✨ Status checking

pkg/internal/directory/validator.go

@@ -1,8 +1,34 @@
 package directory
 
-import "github.com/rs/zerolog/log"
+import (
+	"sync"
+
+	"github.com/rs/zerolog/log"
+)
+
+var statusOfServices = make(map[string]bool)
+var statusLock sync.Mutex
+
+func GetServiceStatus() map[string]bool {
+	out := make(map[string]bool)
+	for k, v := range statusOfServices {
+		out[k] = v
+	}
+
+	services := ListServiceInstance()
+	for _, service := range services {
+		if _, ok := out[service.Type]; !ok {
+			out[service.Type] = false
+		}
+	}
+
+	return out
+}
 
 func ValidateServices() {
+	statusLock.Lock()
+	defer statusLock.Unlock()
+
 	services := ListServiceInstance()
 	if len(services) == 0 {
 		return
@@ -19,9 +45,12 @@ func ValidateServices() {
 		}
 		// Directly use the connect method to skip cache
 		if _, err := ConnectService(service); err != nil {
+			statusOfServices[service.Type] = false
 			_ = RemoveServiceInstance(service.ID)
 			log.Warn().Err(err).Str("id", service.ID).Str("addr", service.GrpcAddr).Msg("Unable connect to service, dropped...")
 			continue
+		} else {
+			statusOfServices[service.Type] = true
 		}
 
 		successCount++

pkg/internal/web/api/index.go

@@ -13,19 +13,19 @@ import (
 
 func MapControllers(app *fiber.App) {
 	app.Get("/check-ip", getClientIP)
+	app.Get("/", func(c *fiber.Ctx) error {
+		return c.JSON(fiber.Map{
+			"api_level": pkg.ApiLevel,
+			"version":   pkg.AppVersion,
+			"status":    true,
+		})
+	})
+	app.Get("/directory/status", getServicesStatus)
+	app.Get("/directory/services", listExistsService)
 
 	// Some built-in public-accessible APIs
 	wellKnown := app.Group("/.well-known").Name("Well Known")
 	{
-		wellKnown.Get("/", func(c *fiber.Ctx) error {
-			return c.JSON(fiber.Map{
-				"api_level": pkg.ApiLevel,
-				"version":   pkg.AppVersion,
-				"status":    true,
-			})
-		})
-		wellKnown.Get("/directory/services", listExistsService)
-
 		wellKnown.Get("/openid-configuration", func(c *fiber.Ctx) error {
 			service := directory.GetServiceInstanceByType(nex.ServiceTypeAuth)
 			if service == nil || service.HttpAddr == nil {
@@ -45,7 +45,7 @@ func MapControllers(app *fiber.App) {
 	// WatchTower administration APIs
 	wt := app.Group("/wt").Name("WatchTower").Use(auth.ValidatorMiddleware)
 	{
-		wt.Post("/maintenance/database", wtRunDbMaintenance)
+		wt.Post("/maintenance/database", watchRunDbMaintenance)
 	}
 
 	// Common websocket gateway

pkg/internal/web/api/status.go

@@ -0,0 +1,10 @@
+package api
+
+import (
+	"git.solsynth.dev/hypernet/nexus/pkg/internal/directory"
+	"github.com/gofiber/fiber/v2"
+)
+
+func getServicesStatus(c *fiber.Ctx) error {
+	return c.JSON(directory.GetServiceStatus())
+}

pkg/internal/web/api/watchtower.go

@@ -6,7 +6,7 @@ import (
 	"github.com/gofiber/fiber/v2"
 )
 
-func wtRunDbMaintenance(c *fiber.Ctx) error {
+func watchRunDbMaintenance(c *fiber.Ctx) error {
 	if err := sec.EnsureGrantedPerm(c, "AdminOperateWatchTower", true); err != nil {
 		return err
 	}

pkg/internal/web/server.go

@@ -23,8 +23,8 @@ func NewServer() *WebApp {
 	app := fiber.New(fiber.Config{
 		DisableStartupMessage: true,
 		EnableIPValidation:    true,
-		ServerHeader:          "Hypernet.Nexus",
+		ServerHeader:          "HyperNet.Nexus",
-		AppName:               "Hypernet.Nexus",
+		AppName:               "HyperNet.Nexus",
 		ProxyHeader:           fiber.HeaderXForwardedFor,
 		JSONEncoder:           json.Marshal,
 		JSONDecoder:           json.Unmarshal,

pkg/main.go

@@ -117,6 +117,7 @@ func main() {
 	// Configure timed tasks
 	quartz := cron.New(cron.WithLogger(cron.VerbosePrintfLogger(&log.Logger)))
 	quartz.AddFunc("@midnight", watchtower.RunDbMaintenance)
+	quartz.AddFunc("@every 5m", directory.ValidateServices)
 	quartz.Start()
 
 	// Messages