update method done, putted a url flag executeOnly for other than select statements

Author: mikebionic

app.go

@@ -68,35 +68,44 @@ type QueryRequest struct {
 }
 
 func (a *App) apiMakeDbRequest(w http.ResponseWriter, r *http.Request) {
-	w.Header().Set("Content-Type", "application/json")
-	reqBody, _ := ioutil.ReadAll(r.Body)
+
+	executeOnly := r.URL.Query().Get("executeOnly")
+	executeOnly_state := 0
+	var err error
+
+	if len(executeOnly) > 0 {
+		executeOnly_state, err = strconv.Atoi(executeOnly)
+		if err != nil {
+			respondWithError(w, http.StatusBadRequest, err.Error())
+		}
+	}
+
+	reqBody, err := ioutil.ReadAll(r.Body)
+	if err != nil {
+		respondWithError(w, http.StatusInternalServerError, err.Error())
+	}
 
 	var queryrequest QueryRequest
 	json.Unmarshal(reqBody, &queryrequest)
 	fmt.Println(queryrequest)
 
-	response, err := do_db_query(a.DB, queryrequest.QueryString)
+	var response interface{}
+	switch executeOnly_state {
+	case 0:
+		response, err = do_db_select_query(a.DB, queryrequest.QueryString)
+
+	default:
+		err = do_db_query_exec(a.DB, queryrequest.QueryString)
+		response = map[string]interface{}{"status": 1}
+
+	}
+
 	if err != nil {
 		respondWithError(w, http.StatusInternalServerError, err.Error())
 		return
 	}
 	respondWithJSON(w, http.StatusOK, response)
 
-	fmt.Println(response)
-	// 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)
-
 }
 
 func (a *App) getRequest(w http.ResponseWriter, r *http.Request) {

do_db_query.go

@@ -3,12 +3,11 @@ package main
 import (
 	"database/sql"
 	"errors"
-	"fmt"
 )
 
 type query_response map[string]interface{}
 
-func do_db_query(db *sql.DB, query_string string) ([]query_response, error) {
+func do_db_select_query(db *sql.DB, query_string string) ([]query_response, error) {
 
 	rows, err := db.Query(query_string)
 	if err != nil {
@@ -23,16 +22,13 @@ func do_db_query(db *sql.DB, query_string string) ([]query_response, error) {
 		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 {
 			return nil, err
 		}
@@ -52,13 +48,7 @@ func do_db_query(db *sql.DB, query_string string) ([]query_response, error) {
 			r[col] = query_value
 
 		}
-		fmt.Println(r)
 		query_results = append(query_results, r)
-
-		// fmt.Println("\n resutls---------")
-		// fmt.Println(query_results)
-		// fmt.Println("\n", "fuck", "you", "+++++++++++++++++")
-
 	}
 
 	if len(query_results) < 1 {
@@ -69,3 +59,13 @@ func do_db_query(db *sql.DB, query_string string) ([]query_response, error) {
 	// defer db.Close()
 	return query_results, nil
 }
+
+func do_db_query_exec(db *sql.DB, query_string string) error {
+
+	_, err := db.Query(query_string)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}