ADIX7/healthr-go
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: app

Browse Source
This commit is contained in:
Ádám Kovács
2025-03-27 11:09:56 +01:00
parent d02d210aff
commit 60942903ec
6 changed files with 319 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

160
db.go Normal file
View File

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

17
go.mod Normal file
View File

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

23
go.sum Normal file
View File

@@ -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=

13
main.go Normal file
View File

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

100
server.go Normal file
View File

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

6
service-model.go Normal file
View File

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