From e97b271bd7ed6cc8ed3db2cdc9954d82d7dcb7d8 Mon Sep 17 00:00:00 2001
From: trafficlunar <trafficlunar@proton.me>
Date: Fri, 29 Nov 2024 23:28:13 +0000
Subject: [PATCH] fix: create worker to add data when computer is offline

---
 internal/worker/computer.go | 21 +++++++++++++++++++++
 internal/worker/worker.go   |  1 +
 2 files changed, 22 insertions(+)
 create mode 100644 internal/worker/computer.go

diff --git a/internal/worker/computer.go b/internal/worker/computer.go
new file mode 100644
index 0000000..ff54b0b
--- /dev/null
+++ b/internal/worker/computer.go
@@ -0,0 +1,21 @@
+package worker
+
+import (
+	"backend/internal/model"
+	"backend/internal/service"
+	"log/slog"
+	"time"
+)
+
+func StartComputerWorker() {
+	slog.Info("Starting computer worker...")
+
+	for range time.Tick(1 * time.Minute) {
+		if !service.ComputerData.Online {
+			service.AddComputerData(model.ComputerWebSocketMessage{
+				Cpu: 0,
+				Ram: 0,
+			})
+		}
+	}
+}
diff --git a/internal/worker/worker.go b/internal/worker/worker.go
index ade4e6e..9908221 100644
--- a/internal/worker/worker.go
+++ b/internal/worker/worker.go
@@ -4,6 +4,7 @@ import "os"
 
 func StartWorkers() {
 	go StartLastFMWorker()
+	go StartComputerWorker()
 	if os.Getenv("UPTIME_KUMA_ENABLED") == "true" {
 		go StartStatusWorker()
 	}