diff --git a/server/parallel_tasks.go b/server/parallel_tasks.go index d76b3b7..2ec6798 100644 --- a/server/parallel_tasks.go +++ b/server/parallel_tasks.go @@ -23,10 +23,18 @@ func HandleSetParallelTasks(w http.ResponseWriter, r *http.Request) { slog.Info("Got set Parallel Tasks", "parallel_tasks", parallelTasks, "worker", worker) - _, err = db.Exec(r.Context(), "UPDATE workers SET parallel_tasks = $1", parallelTasks) - if err != nil { - http.Error(w, fmt.Sprintf("Setting Worker Parallel Tasks: %v", err), http.StatusInternalServerError) - return + if worker == "all" { + _, err = db.Exec(r.Context(), "UPDATE workers SET parallel_tasks = $1", parallelTasks) + if err != nil { + http.Error(w, fmt.Sprintf("Setting Worker Parallel Tasks: %v", err), http.StatusInternalServerError) + return + } + } else { + _, err = db.Exec(r.Context(), "UPDATE workers SET parallel_tasks = $1 where id = $2", parallelTasks, worker) + if err != nil { + http.Error(w, fmt.Sprintf("Setting Worker Parallel Tasks: %v", err), http.StatusInternalServerError) + return + } } http.Redirect(w, r, r.Header.Get("Referer"), http.StatusFound)