Add Parallel Tasks Settings

server/index.go

@@ -18,8 +18,9 @@ type IndexData struct {
 type IndexWorker struct {
 	ID string
 	Worker
-	Status      *types.WorkerStatus
-	QueueEnable bool
+	Status        *types.WorkerStatus
+	QueueEnable   bool
+	ParallelTasks int
 }
 
 func handleIndex(w http.ResponseWriter, r *http.Request) {
@@ -31,6 +32,15 @@ func handleIndex(w http.ResponseWriter, r *http.Request) {
 		WorkersMutex.Lock()
 		defer WorkersMutex.Unlock()
 		for i := range Workers {
+
+			var queueEnable bool
+			var parallelTasks int
+			err := db.QueryRow(r.Context(), "SELECT queue_enable, parallel_tasks FROM workers WHERE id = $1", i).Scan(&queueEnable, &parallelTasks)
+			if err != nil {
+				w.Write([]byte(err.Error()))
+				slog.ErrorContext(r.Context(), "Error Querying Worker", "err", err)
+			}
+
 			if Workers[i].Connected {
 				var status types.WorkerStatus
 				_, err := rpcServer.Call(r.Context(), Workers[i].Conn, "status", nil, &status)
@@ -41,15 +51,19 @@ func handleIndex(w http.ResponseWriter, r *http.Request) {
 				}
 				slog.InfoContext(r.Context(), "Got Worker Status", "id", i, "status", status)
 				data.Workers = append(data.Workers, IndexWorker{
-					ID:     i,
-					Worker: *Workers[i],
-					Status: &status,
+					ID:            i,
+					Worker:        *Workers[i],
+					Status:        &status,
+					QueueEnable:   queueEnable,
+					ParallelTasks: parallelTasks,
 				})
 			} else {
 				data.Workers = append(data.Workers, IndexWorker{
-					ID:     i,
-					Worker: *Workers[i],
-					Status: nil,
+					ID:            i,
+					Worker:        *Workers[i],
+					Status:        nil,
+					QueueEnable:   queueEnable,
+					ParallelTasks: parallelTasks,
 				})
 			}
 		}