From 60942903ec6ec86c98b3b81b699d08939cafd2db Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=C3=81d=C3=A1m=20Kov=C3=A1cs?= Date: Thu, 27 Mar 2025 11:09:56 +0100 Subject: [PATCH] feat: app --- db.go | 160 +++++++++++++++++++++++++++++++++++++++++++++++ go.mod | 17 +++++ go.sum | 23 +++++++ main.go | 13 ++++ server.go | 100 +++++++++++++++++++++++++++++ service-model.go | 6 ++ 6 files changed, 319 insertions(+) create mode 100644 db.go create mode 100644 go.mod create mode 100644 go.sum create mode 100644 main.go create mode 100644 server.go create mode 100644 service-model.go diff --git a/db.go b/db.go new file mode 100644 index 0000000..4bb5966 --- /dev/null +++ b/db.go @@ -0,0 +1,160 @@ +package main + +import ( + "database/sql" + _ "embed" + "errors" + _ "github.com/glebarez/go-sqlite" + "log" + "os" +) + +const dbFileName = "healthr.db" +const driverName = "sqlite" + +type Database interface { + Initialize() error + GetAllStatus() ([]Service, error) + Get(serviceName string) *Service + Update(service Service) error +} + +type SqliteDatabase struct { +} + +//go:embed sql_queries/init-db.sql +var initDbQuery string + +func (sqldb *SqliteDatabase) Initialize() error { + fileInfo, err := os.Stat(dbFileName) + if fileInfo != nil && fileInfo.IsDir() { + return errors.New(dbFileName + " is exists, but is a directory instead of a file") + } + + if err == nil { + return nil + } + + db, err := sql.Open(driverName, dbFileName) + if err != nil { + log.Fatal(err.Error()) + } + defer db.Close() + + db.Exec(initDbQuery) + log.Default().Println("Database initialized!") + + return nil +} + +//go:embed sql_queries/get-service.sql +var getServiceQuery string + +func (sqldb *SqliteDatabase) Get(serviceName string) *Service { + db, err := sql.Open(driverName, dbFileName) + if err != nil { + log.Fatal(err.Error()) + } + defer db.Close() + + rows, err := db.Query(getServiceQuery, serviceName) + if err != nil { + log.Fatal(err.Error()) + } + defer rows.Close() + + hasItem := rows.Next() + if !hasItem { + return nil + } + + var service Service + err = rows.Scan(&service.Name, &service.Healthy) + if err != nil { + log.Fatal(err.Error()) + return nil + } + + return &service +} + +//go:embed sql_queries/get-all.sql +var getAllQuery string + +func (sqldb *SqliteDatabase) GetAllStatus() ([]Service, error) { + db, err := sql.Open(driverName, dbFileName) + if err != nil { + log.Fatal(err) + } + defer db.Close() + + rows, err := db.Query(getAllQuery) + if err != nil { + log.Fatal(err) + } + defer rows.Close() + + services := make([]Service, 0) + + for rows.Next() { + var service Service + if err := rows.Scan(&service.Name, &service.Healthy); err != nil { + log.Default().Print("Error while parsing row", err) + } + + services = append(services, service) + } + + return services, nil +} + +func (sqldb *SqliteDatabase) Update(service Service) error { + sqldb.Get(service.Name) + existingService := sqldb.Get(service.Name) + + if existingService == nil { + return sqldb.insert(service) + } else { + return sqldb.update(service) + } +} + +//go:embed sql_queries/insert-service.sql +var insertQuery string + +func (sqldb *SqliteDatabase) insert(service Service) error { + db, err := sql.Open(driverName, dbFileName) + if err != nil { + log.Fatal(err.Error()) + } + defer db.Close() + + _, err = db.Exec(insertQuery, service.Name, service.Healthy) + + if err != nil { + println("ERROR", err.Error()) + return err + } + + return nil +} + +//go:embed sql_queries/update-service.sql +var updateQuery string + +func (sqldb *SqliteDatabase) update(service Service) error { + db, err := sql.Open(driverName, dbFileName) + if err != nil { + log.Fatal(err.Error()) + } + defer db.Close() + + _, err = db.Exec(updateQuery, service.Name, service.Healthy, service.Name) + + if err != nil { + println("ERROR", err.Error()) + return err + } + + return nil +} diff --git a/go.mod b/go.mod new file mode 100644 index 0000000..0cb3b9f --- /dev/null +++ b/go.mod @@ -0,0 +1,17 @@ +module adix7/healthr + +go 1.23.7 + +require github.com/glebarez/go-sqlite v1.22.0 + +require ( + github.com/dustin/go-humanize v1.0.1 // indirect + github.com/google/uuid v1.5.0 // indirect + github.com/mattn/go-isatty v0.0.20 // indirect + github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec // indirect + golang.org/x/sys v0.15.0 // indirect + modernc.org/libc v1.37.6 // indirect + modernc.org/mathutil v1.6.0 // indirect + modernc.org/memory v1.7.2 // indirect + modernc.org/sqlite v1.28.0 // indirect +) diff --git a/go.sum b/go.sum new file mode 100644 index 0000000..e635754 --- /dev/null +++ b/go.sum @@ -0,0 +1,23 @@ +github.com/dustin/go-humanize v1.0.1 h1:GzkhY7T5VNhEkwH0PVJgjz+fX1rhBrR7pRT3mDkpeCY= +github.com/dustin/go-humanize v1.0.1/go.mod h1:Mu1zIs6XwVuF/gI1OepvI0qD18qycQx+mFykh5fBlto= +github.com/glebarez/go-sqlite v1.22.0 h1:uAcMJhaA6r3LHMTFgP0SifzgXg46yJkgxqyuyec+ruQ= +github.com/glebarez/go-sqlite v1.22.0/go.mod h1:PlBIdHe0+aUEFn+r2/uthrWq4FxbzugL0L8Li6yQJbc= +github.com/google/pprof v0.0.0-20221118152302-e6195bd50e26 h1:Xim43kblpZXfIBQsbuBVKCudVG457BR2GZFIz3uw3hQ= +github.com/google/pprof v0.0.0-20221118152302-e6195bd50e26/go.mod h1:dDKJzRmX4S37WGHujM7tX//fmj1uioxKzKxz3lo4HJo= +github.com/google/uuid v1.5.0 h1:1p67kYwdtXjb0gL0BPiP1Av9wiZPo5A8z2cWkTZ+eyU= +github.com/google/uuid v1.5.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo= +github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY= +github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y= +github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec h1:W09IVJc94icq4NjY3clb7Lk8O1qJ8BdBEF8z0ibU0rE= +github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec/go.mod h1:qqbHyh8v60DhA7CoWK5oRCqLrMHRGoxYCSS9EjAz6Eo= +golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.15.0 h1:h48lPFYpsTvQJZF4EKyI4aLHaev3CxivZmv7yZig9pc= +golang.org/x/sys v0.15.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA= +modernc.org/libc v1.37.6 h1:orZH3c5wmhIQFTXF+Nt+eeauyd+ZIt2BX6ARe+kD+aw= +modernc.org/libc v1.37.6/go.mod h1:YAXkAZ8ktnkCKaN9sw/UDeUVkGYJ/YquGO4FTi5nmHE= +modernc.org/mathutil v1.6.0 h1:fRe9+AmYlaej+64JsEEhoWuAYBkOtQiMEU7n/XgfYi4= +modernc.org/mathutil v1.6.0/go.mod h1:Ui5Q9q1TR2gFm0AQRqQUaBWFLAhQpCwNcuhBOSedWPo= +modernc.org/memory v1.7.2 h1:Klh90S215mmH8c9gO98QxQFsY+W451E8AnzjoE2ee1E= +modernc.org/memory v1.7.2/go.mod h1:NO4NVCQy0N7ln+T9ngWqOQfi7ley4vpwvARR+Hjw95E= +modernc.org/sqlite v1.28.0 h1:Zx+LyDDmXczNnEQdvPuEfcFVA2ZPyaD7UCZDjef3BHQ= +modernc.org/sqlite v1.28.0/go.mod h1:Qxpazz0zH8Z1xCFyi5GSL3FzbtZ3fvbjmywNogldEW0= diff --git a/main.go b/main.go new file mode 100644 index 0000000..8c70fe7 --- /dev/null +++ b/main.go @@ -0,0 +1,13 @@ +package main + +import "log" + +func main() { + var db Database = &SqliteDatabase{} + err := db.Initialize() + + if err != nil { + log.Fatal("There was an error initializing the database %w", err) + } + webServer(db) +} diff --git a/server.go b/server.go new file mode 100644 index 0000000..ad8f566 --- /dev/null +++ b/server.go @@ -0,0 +1,100 @@ +package main + +import ( + "encoding/json" + "errors" + "log" + "net/http" +) + +var database Database + +type UpdateRequest struct { + Name string `json:"name"` + Healthy bool `json:"healthy"` +} + +func health(w http.ResponseWriter, req *http.Request) { + if req.Method == "GET" { + getHealth(w, req) + } else if req.Method == "POST" { + updateHealth(w, req) + } else { + w.WriteHeader(http.StatusBadRequest) + } +} + +func getHealth(w http.ResponseWriter, req *http.Request) { + serviceId := req.URL.Query().Get("service") + + if serviceId == "" { + w.WriteHeader(http.StatusBadRequest) + w.Write([]byte("Service parameter not supplied")) + return + } + + service := database.Get(serviceId) + + if service == nil { + w.WriteHeader(http.StatusNotFound) + w.Write([]byte("Service not found with id " + serviceId)) + return + } + + response, err := json.Marshal(service) + if err != nil { + log.Fatalf("Error marshalling JSON response: %v", err) + } + + w.Header().Set("Content-Type", "application/json") + w.Write(response) +} +func updateHealth(w http.ResponseWriter, req *http.Request) error { + var updateRequest UpdateRequest + if req.Body == nil { + http.Error(w, "Request has no body", http.StatusBadRequest) + return errors.New("request has no body") + } + err := json.NewDecoder(req.Body).Decode(&updateRequest) + if err != nil { + http.Error(w, "Error while parsing request", http.StatusBadRequest) + log.Default().Println(err) + return err + } + + err = database.Update(Service{Name: updateRequest.Name, Healthy: updateRequest.Healthy}) + if err != nil { + http.Error(w, "Error while updating service", 500) + return err + } + + w.WriteHeader(200) + return nil +} + +func status(w http.ResponseWriter, req *http.Request) { + result, err := database.GetAllStatus() + + if err != nil { + http.Error(w, "Error while fetching status", 500) + return + } + + response, err := json.Marshal(result) + if err != nil { + log.Fatalf("Error marshalling JSON response: %v", err) + } + + w.Header().Set("Content-Type", "application/json") + w.Header().Set("Access-Control-Allow-Origin", "*") // URL of the Kea control agent API endpoint + w.Write(response) +} + +func webServer(db Database) { + database = db + + http.HandleFunc("/health", health) + http.HandleFunc("/status", status) + + http.ListenAndServe(":8090", nil) +} diff --git a/service-model.go b/service-model.go new file mode 100644 index 0000000..102a7af --- /dev/null +++ b/service-model.go @@ -0,0 +1,6 @@ +package main; + +type Service struct { + Name string `json:"name"`; + Healthy bool `json:"healthy"`; +}