added health route for server stats

1 files changed, 36 insertions, 0 deletions

diff --git a/controllers/health.go b/controllers/health.go
new file mode 100644
index 0000000..3f12bce
--- /dev/null
+++ b/controllers/health.go
@@ -0,0 +1,36 @@
+package controllers
+
+import (
+	"metachan/database"
+	"metachan/tasks"
+	"metachan/types"
+	"metachan/utils/stats"
+
+	"github.com/gofiber/fiber/v2"
+)
+
+func HealthStatus(c *fiber.Ctx) error {
+	// Check if the database is connected
+	databaseStatus := database.DatabaseConnectionStatus()
+
+	// Get the memory stats
+	memoryStats := stats.GetMemoryStats()
+
+	// Get the task statuses
+	taskStatuses := tasks.GlobalTaskManager.GetAllTaskStatuses()
+
+	statusString := map[bool]string{
+		true:  "healthy",
+		false: "unhealthy",
+	}[databaseStatus]
+	// Create the health status response
+	healthStatus := types.HealthStatus{
+		Status:    statusString,
+		Timestamp: stats.GetCurrentTimestamp(),
+		Uptime:    stats.GetUptime(),
+		Memory:    memoryStats,
+		Database:  types.DatabaseStatus{Connected: databaseStatus, LastChecked: stats.GetCurrentTimestamp()},
+		Tasks:     taskStatuses,
+	}
+	return c.JSON(healthStatus)
+}