db select query result returneed

Author: mikebionic

app.go

@@ -47,6 +47,22 @@ func (a *App) initializeRoutes() {
 
 }
 
+////////
+
+func respondWithJSON(w http.ResponseWriter, code int, payload interface{}) {
+	response, _ := json.Marshal(payload)
+
+	w.Header().Set("Content-Type", "application/json")
+	w.WriteHeader(code)
+	w.Write(response)
+}
+
+func respondWithError(w http.ResponseWriter, code int, message string) {
+	respondWithJSON(w, code, map[string]string{"error": message})
+}
+
+////////
+
 type QueryRequest struct {
 	QueryString string `json:"query_string"`
 }
@@ -59,30 +75,28 @@ func (a *App) apiMakeDbRequest(w http.ResponseWriter, r *http.Request) {
 	json.Unmarshal(reqBody, &queryrequest)
 	fmt.Println(queryrequest)
 
-	response, err := do_db_query(queryrequest.QueryString)
+	response, err := do_db_query(a.DB, queryrequest.QueryString)
 	if err != nil {
-		w.WriteHeader(http.StatusInternalServerError)
-		w.Write([]byte(`{"message": "error"}`))
+		respondWithError(w, http.StatusInternalServerError, err.Error())
+		return
 	}
+	respondWithJSON(w, http.StatusOK, response)
 
 	fmt.Println(response)
-	type RespObject struct {
-		Data []interface{}
-	}
-	respObject := RespObject{response}
-	data, err := json.Marshal(respObject)
-	if err != nil {
-		fmt.Println(err)
-	}
-	fmt.Println("", data)
-
-	w.Header().Set("Content-Type", "application/json")
-	w.WriteHeader(http.StatusOK)
-	w.Write(data)
-	// var res map[string]interface{}
-	// json.Marshal(&response, &res)
+	// type RespObject struct {
+	// 	Data map[string]interface{}
+	// }
+	// respObject := RespObject{response}
+	// data, err := json.Marshal(respObject)
+	// if err != nil {
+	// 	fmt.Println(err)
+	// }
+	// fmt.Println("", data)
+
+	// w.Header().Set("Content-Type", "application/json")
+	// w.WriteHeader(http.StatusOK)
+	// // w.Write(data)
 
-	// w.Write([]byte(`{"message": "post called"}`))
 }
 
 func (a *App) getRequest(w http.ResponseWriter, r *http.Request) {

do_db_query.go

@@ -4,51 +4,41 @@ import (
 	"database/sql"
 	"errors"
 	"fmt"
-	"log"
 )
 
-func do_db_query(query_string string) ([]interface{}, error) {
-	db_type := "postgres"
-	db_user := "postgres"
-	db_password := "123456"
-	db_host := "localhost"
-	db_database := "dbSapHasap"
+type query_response map[string]interface{}
 
-	connStr := fmt.Sprintf("%s://%s:%s@%s/%s?sslmode=disable", db_type, db_user, db_password, db_host, db_database)
+func do_db_query(db *sql.DB, query_string string) ([]query_response, error) {
 
-	db, err := sql.Open("postgres", connStr)
+	rows, err := db.Query(query_string)
 	if err != nil {
-		log.Fatal(err)
-	}
-
-	query_request := query_string
-
-	rows, err := db.Query(query_request)
-	if err != nil {
-		log.Fatal(err)
+		return nil, err
 	}
 	defer rows.Close()
 
-	cols, _ := rows.Columns()
-
-	var query_results []interface{}
-	var error error
+	query_results := []query_response{}
 
 	for rows.Next() {
 
+		r := make(query_response)
+		cols, _ := rows.Columns()
+
+		fmt.Println(cols, "\n", len(cols))
 		values := make([]interface{}, len(cols))
 		valuePointers := make([]interface{}, len(cols))
 
 		for i := range cols {
 			valuePointers[i] = &values[i]
 		}
 
+		fmt.Println("\n", valuePointers)
+
 		if err := rows.Scan(valuePointers...); err != nil {
-			error = err
+			return nil, err
 		}
 
 		for i, col := range cols {
-			query_result_dict := map[interface{}]interface{}{}
+
 			var query_value interface{}
 
 			val := values[i]
@@ -59,22 +49,23 @@ func do_db_query(query_string string) ([]interface{}, error) {
 			} else {
 				query_value = val
 			}
-			query_result_dict[col] = query_value
+			r[col] = query_value
 
-			fmt.Println(query_result_dict)
-			query_results = append(query_results, query_result_dict)
 		}
+		fmt.Println(r)
+		query_results = append(query_results, r)
 
-		fmt.Println("resutls---------", "")
-		// fmt.Println("")
-		fmt.Println(query_results)
+		// fmt.Println("\n resutls---------")
+		// fmt.Println(query_results)
+		// fmt.Println("\n", "fuck", "you", "+++++++++++++++++")
 
 	}
 
 	if len(query_results) < 1 {
-		error = errors.New("empty name")
+		err = errors.New("ot found")
+		return nil, err
 	}
 
-	defer db.Close()
-	return query_results, error
+	// defer db.Close()
+	return query_results, nil
 }