refactored api to handle post request with body processing

Author: mikebionic

do_db_query.go

@@ -2,11 +2,12 @@ package main
 
 import (
 	"database/sql"
+	"errors"
 	"fmt"
 	"log"
 )
 
-func do_db_query(query_string string) {
+func do_db_query(query_string string) ([]interface{}, error) {
 	db_type := "postgres"
 	db_user := "postgres"
 	db_password := "123456"
@@ -31,6 +32,7 @@ func do_db_query(query_string string) {
 	cols, _ := rows.Columns()
 
 	var query_results []interface{}
+	var error error
 
 	for rows.Next() {
 
@@ -42,33 +44,37 @@ func do_db_query(query_string string) {
 		}
 
 		if err := rows.Scan(valuePointers...); err != nil {
-			log.Fatal(err)
+			error = err
 		}
 
 		for i, col := range cols {
 			query_result_dict := map[interface{}]interface{}{}
+			var query_value interface{}
 
 			val := values[i]
 			b, ok := val.([]byte)
 
-			var v interface{}
-
 			if ok {
-				v = string(b)
+				query_value = string(b)
 			} else {
-				v = val
+				query_value = val
 			}
-			query_result_dict[col] = v
+			query_result_dict[col] = query_value
 
 			fmt.Println(query_result_dict)
 			query_results = append(query_results, query_result_dict)
 		}
 
-		fmt.Println("resutls---------")
+		fmt.Println("resutls---------", "")
+		// fmt.Println("")
 		fmt.Println(query_results)
 
 	}
 
+	if len(query_results) < 1 {
+		error = errors.New("empty name")
+	}
+
 	defer db.Close()
-	// return nil
+	return query_results, error
 }

main.go

@@ -1,7 +1,9 @@
 package main
 
 import (
+	"encoding/json"
 	"fmt"
+	"io/ioutil"
 	"log"
 	"net/http"
 	"strconv"
@@ -18,6 +20,10 @@ import (
 
 // }
 
+type QueryRequest struct {
+	QueryString string `json:"query_string"`
+}
+
 func notFound(w http.ResponseWriter, r *http.Request) {
 	w.Header().Set("Content-Type", "application/json")
 	w.WriteHeader(http.StatusNotFound)
@@ -26,6 +32,20 @@ func notFound(w http.ResponseWriter, r *http.Request) {
 
 func post(w http.ResponseWriter, r *http.Request) {
 	w.Header().Set("Content-Type", "application/json")
+	reqBody, _ := ioutil.ReadAll(r.Body)
+
+	var queryrequest QueryRequest
+	json.Unmarshal(reqBody, &queryrequest)
+	fmt.Println(queryrequest)
+
+	response, err := do_db_query(queryrequest.QueryString)
+	if err != nil {
+		w.WriteHeader(http.StatusInternalServerError)
+		w.Write([]byte(`{"message": "error"}`))
+	}
+
+	fmt.Println(response)
+
 	w.WriteHeader(http.StatusCreated)
 	w.Write([]byte(`{"message": "post called"}`))
 }
@@ -68,7 +88,7 @@ func withParams(w http.ResponseWriter, r *http.Request) {
 	w.Write([]byte(fmt.Sprintf(`{"userID": %d, "commentID": %d, "location": "%s" }`, userID, commentID, location)))
 }
 
-func main() {
+func handleRequests() {
 	r := mux.NewRouter()
 	api := r.PathPrefix("/api/v1").Subrouter()
 	api.HandleFunc("", get).Methods(http.MethodGet)
@@ -78,3 +98,7 @@ func main() {
 
 	log.Fatal(http.ListenAndServe(":8080", r))
 }
+
+func main() {
+	handleRequests()
+}