107 lines
2.3 KiB
Go
107 lines
2.3 KiB
Go
package http
|
|
|
|
import (
|
|
"encoding/json"
|
|
"gopkg.in/guregu/null.v4"
|
|
db "infra-dashboard/Database"
|
|
"io/ioutil"
|
|
"log"
|
|
"net/http"
|
|
"strconv"
|
|
"time"
|
|
|
|
"github.com/gorilla/mux"
|
|
)
|
|
|
|
func GetServersList(w http.ResponseWriter, r *http.Request) {
|
|
var list []db.Server
|
|
var err error
|
|
|
|
t := time.Now()
|
|
status := 200
|
|
|
|
dbConn := db.GetDatabaseConnection()
|
|
defer dbConn.Close()
|
|
list, err = db.GetServersList(dbConn)
|
|
if err != nil {
|
|
log.Println("Error getting OS list")
|
|
status = 500
|
|
}
|
|
logRequest(t, r, status)
|
|
w.Header().Set("Content-Type", "application/json")
|
|
json.NewEncoder(w).Encode(list)
|
|
}
|
|
|
|
func GetServersbyID(w http.ResponseWriter, r *http.Request) {
|
|
var server db.Server
|
|
var err error
|
|
|
|
t := time.Now()
|
|
status := 200
|
|
|
|
params := mux.Vars(r)
|
|
id, err := strconv.ParseInt(params["id"], 10, 32)
|
|
if err != nil {
|
|
log.Println("Error converting ID", err)
|
|
status = 500
|
|
}
|
|
dbConn := db.GetDatabaseConnection()
|
|
defer dbConn.Close()
|
|
server = db.GetServersbyID(dbConn, int64(id))
|
|
logRequest(t, r, status)
|
|
w.Header().Set("Content-Type", "application/json")
|
|
json.NewEncoder(w).Encode(server)
|
|
}
|
|
|
|
func GetServersbyOS(w http.ResponseWriter, r *http.Request) {
|
|
var servers []db.Server
|
|
var err error
|
|
|
|
t := time.Now()
|
|
status := 200
|
|
|
|
params := mux.Vars(r)
|
|
osId, err := strconv.ParseInt(params["os_id"], 10, 32)
|
|
if err != nil {
|
|
log.Println("Error converting ID", err)
|
|
status = 500
|
|
}
|
|
dbConn := db.GetDatabaseConnection()
|
|
defer dbConn.Close()
|
|
servers, err = db.GetServersbyOS(dbConn, int64(osId))
|
|
if err != nil {
|
|
log.Println("Error getting servers by OS", err)
|
|
status = 500
|
|
}
|
|
logRequest(t, r, status)
|
|
w.Header().Set("Content-Type", "application/json")
|
|
json.NewEncoder(w).Encode(servers)
|
|
}
|
|
|
|
func CreateServer(w http.ResponseWriter, r *http.Request) {
|
|
var server db.Server
|
|
var err error
|
|
|
|
t := time.Now()
|
|
status := 204
|
|
|
|
body, err := ioutil.ReadAll(r.Body)
|
|
if err != nil {
|
|
log.Println(err.Error(), "Error reading request body")
|
|
}
|
|
|
|
params := make(map[string]null.String)
|
|
json.Unmarshal(body, ¶ms)
|
|
server.Hostname = params["hostname"]
|
|
server.OsId = params["os_id"]
|
|
dbConn := db.GetDatabaseConnection()
|
|
defer dbConn.Close()
|
|
err = db.CreateServer(server, dbConn)
|
|
if err != nil {
|
|
log.Println(err.Error(), "Error creating server")
|
|
status = 500
|
|
}
|
|
logRequest(t, r, status)
|
|
w.WriteHeader(status)
|
|
}
|