feat: app

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
+}

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
+)

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=

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)
+}

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)
+}

service-model.go

@@ -0,0 +1,6 @@
+package main;
+
+type Service struct {
+    Name string `json:"name"`;
+    Healthy bool `json:"healthy"`;
+}