Add Worker Cleanup

server/worker.go

@@ -101,3 +101,26 @@ func readMessage(ctx context.Context, c *websocket.Conn) error {
 	slog.InfoContext(ctx, "Got Websocket Message", "type", t.String(), "data", data)
 	return c.Write(ctx, t, data)
 }
+
+func cleanupDeadWorkers(stop chan bool) {
+	ticker := time.NewTicker(time.Second)
+	for {
+		select {
+		case <-ticker.C:
+			func() {
+				WorkersMutex.Lock()
+				defer WorkersMutex.Unlock()
+
+				for uuid, w := range Workers {
+					if !w.Connected && w.ConnectionChanged.Add(time.Minute*5).Before(time.Now()) {
+						slog.Warn("Removing Dead Worker", "uuid", uuid, "name", w.Name)
+						delete(Workers, uuid)
+						// TODO Free any Jobs that the Worker had
+					}
+				}
+			}()
+		case <-stop:
+			return
+		}
+	}
+}