49b7051cea
This allows mux to compare and match the requests based on the HTTP method as well. If we compare the methods in the middleware chain, then mux will try to redirect the request to the first match it finds.
90 lines
2.2 KiB
Go
90 lines
2.2 KiB
Go
package api
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"log"
|
|
"net/http"
|
|
"strconv"
|
|
|
|
"github.com/gorilla/mux"
|
|
|
|
"gitlab.com/mobtools/internal-tools-server/models"
|
|
"gitlab.com/mobtools/internal-tools-server/services"
|
|
)
|
|
|
|
/*
|
|
This file contains the APIs for the client to invoke in order to fetch data or perform an action
|
|
on the middleware server
|
|
*/
|
|
|
|
// PostQuery executes a custom sql query on the client database
|
|
func PostQuery(w http.ResponseWriter, r *http.Request) {
|
|
queryBody := models.ExecQuery{}
|
|
err := json.NewDecoder(r.Body).Decode(&queryBody)
|
|
if err != nil {
|
|
HandleAPIError(w, r, err)
|
|
return
|
|
}
|
|
|
|
var mapArray []map[string]interface{}
|
|
mapArray, err = services.ExecuteQuery(queryBody)
|
|
if err != nil {
|
|
HandleAPIError(w, r, err)
|
|
return
|
|
}
|
|
|
|
// Write content-type, statuscode, payload
|
|
mapJSON, _ := json.Marshal(mapArray)
|
|
w.Header().Set("Content-Type", "application/json")
|
|
w.WriteHeader(200)
|
|
fmt.Fprintf(w, "%s", mapJSON)
|
|
}
|
|
|
|
// CreateQuery creates a new query for the user in the table
|
|
func CreateQuery(w http.ResponseWriter, r *http.Request) {
|
|
queryBody := models.Query{}
|
|
err := json.NewDecoder(r.Body).Decode(&queryBody)
|
|
if err != nil {
|
|
HandleAPIError(w, r, err)
|
|
return
|
|
}
|
|
|
|
queryBody, err = services.CreateQuery(queryBody)
|
|
if err != nil {
|
|
HandleAPIError(w, r, err)
|
|
return
|
|
}
|
|
|
|
// Write content-type, statuscode, payload
|
|
queryJSON, _ := json.Marshal(queryBody)
|
|
w.Header().Set("Content-Type", "application/json")
|
|
w.WriteHeader(200)
|
|
fmt.Fprintf(w, "%s", queryJSON)
|
|
}
|
|
|
|
// UpdateQuery updates a given query in the database for a given account
|
|
func UpdateQuery(w http.ResponseWriter, r *http.Request) {
|
|
queryBody := models.Query{}
|
|
err := json.NewDecoder(r.Body).Decode(&queryBody)
|
|
if err != nil {
|
|
log.Printf("Got error when decoding the queryBody. %s", err.Error())
|
|
HandleAPIError(w, r, err)
|
|
return
|
|
}
|
|
|
|
queryBody.ID, _ = strconv.ParseInt(mux.Vars(r)["id"], 10, 64)
|
|
log.Printf("Got query.ID as %d", queryBody.ID)
|
|
queryBody, err = services.UpdateQuery(queryBody)
|
|
if err != nil {
|
|
HandleAPIError(w, r, err)
|
|
return
|
|
}
|
|
|
|
// Write content-type, statuscode, payload
|
|
queryJSON, _ := json.Marshal(queryBody)
|
|
w.Header().Set("Content-Type", "application/json")
|
|
w.WriteHeader(200)
|
|
fmt.Fprintf(w, "%s", queryJSON)
|
|
}
|