fix: change SupervisorStatusType to enum

Author: GaminRick7

src/api.go

@@ -225,7 +225,6 @@ func (a *App) getSupervisorStatusByID(w http.ResponseWriter, r *http.Request) {
 }
 
 func (a *App) getAllSupervisors(w http.ResponseWriter, r *http.Request) {
-	// check if we want only active supervisors
 	activeOnly := r.URL.Query().Get("active") == "true"
 
 	var supervisors []SupervisorStatus

src/status.go

@@ -30,21 +30,21 @@ func addDummySupervisors(statusRegistry *StatusRegistry, log *slog.Logger) {
 		{
 			ConsumerID: "worker_amd_001",
 			GPUType:    "AMD",
-			Status:     "active",
+			Status:     SupervisorStateActive,
 			LastSeen:   now,                     // now
 			StartedAt:  now.Add(-2 * time.Hour), // 2hours ago
 		},
 		{
 			ConsumerID: "worker_nvidia_002",
 			GPUType:    "NVIDIA",
-			Status:     "active",
+			Status:     SupervisorStateActive,
 			LastSeen:   now.Add(-30 * time.Second), // 30 seconds ago
 			StartedAt:  now.Add(-1 * time.Hour),    // 1 hour ago
 		},
 		{
 			ConsumerID: "worker_tt_003",
 			GPUType:    "TT",
-			Status:     "inactive",
+			Status:     SupervisorStateInactive,
 			LastSeen:   now.Add(-5 * time.Minute), // seen 5 minutes ago
 			StartedAt:  now.Add(-3 * time.Hour),   // 3 hours ago
 		},
@@ -103,7 +103,7 @@ func (sr *StatusRegistry) GetActiveSupervisors() ([]SupervisorStatus, error) {
 
 	var activeSupervisors []SupervisorStatus
 	for _, supervisor := range allSupervisors {
-		if supervisor.Status == "active" {
+		if supervisor.Status == SupervisorStateActive {
 			activeSupervisors = append(activeSupervisors, supervisor)
 		}
 	}

src/util.go

@@ -7,10 +7,11 @@ import (
 )
 
 const (
-	StreamName    = "jobs:stream"
-	ConsumerGroup = "workers"
-	MaxRetries    = 3
-	RetryDelay    = 5 * time.Second
+	StreamName          = "jobs:stream"
+	ConsumerGroup       = "workers"
+	SupervisorStatusKey = "supervisors:status"
+	MaxRetries          = 3
+	RetryDelay          = 5 * time.Second
 )
 
 type Job struct {
@@ -22,12 +23,20 @@ type Job struct {
 	RequiredGPU string                 `json:"gpu"`
 }
 
+type SupervisorState string
+
+const (
+	SupervisorStateActive   SupervisorState = "active"
+	SupervisorStateInactive SupervisorState = "inactive"
+	SupervisorStateFailed   SupervisorState = "failed"
+)
+
 type SupervisorStatus struct {
-	ConsumerID string    `json:"consumer_id"`
-	GPUType    string    `json:"gpu_type"`
-	Status     string    `json:"status"` // "active", "inactive", "failed"
-	LastSeen   time.Time `json:"last_seen"`
-	StartedAt  time.Time `json:"started_at"`
+	ConsumerID string          `json:"consumer_id"`
+	GPUType    string          `json:"gpu_type"`
+	Status     SupervisorState `json:"status"`
+	LastSeen   time.Time       `json:"last_seen"`
+	StartedAt  time.Time       `json:"started_at"`
 }
 
 func generateJobID() string {