added health route for server stats

1 files changed, 37 insertions, 0 deletions

diff --git a/database/status.go b/database/status.go
new file mode 100644
index 0000000..14f24ed
--- /dev/null
+++ b/database/status.go
@@ -0,0 +1,37 @@
+package database
+
+import (
+	"context"
+	"fmt"
+	"metachan/types"
+	"metachan/utils/logger"
+	"time"
+)
+
+func DatabaseConnectionStatus() bool {
+	if DB == nil {
+		return false
+	}
+
+	sqlDB, err := DB.DB()
+	if err != nil {
+		logger.Log(fmt.Sprintf("Unable to get SQL DB: %v", err), types.LogOptions{
+			Prefix: "Database",
+			Level:  types.Error,
+		})
+		return false
+	}
+
+	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+	defer cancel()
+
+	err = sqlDB.PingContext(ctx)
+	if err != nil {
+		logger.Log(fmt.Sprintf("Database connection error: %v", err), types.LogOptions{
+			Prefix: "Database",
+			Level:  types.Error,
+		})
+		return false
+	}
+	return true
+}