From f78171aa02ea070d8bd7abd09073e2b3e7fb9a99 Mon Sep 17 00:00:00 2001 From: mudler Date: Mon, 28 Oct 2024 18:29:15 +0100 Subject: [PATCH] chore(refactor): move scraper type to a constant Signed-off-by: mudler --- internal/jobs/webscraper.go | 2 ++ internal/jobserver/worker.go | 4 ++-- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/internal/jobs/webscraper.go b/internal/jobs/webscraper.go index e50c5db..b9aee91 100644 --- a/internal/jobs/webscraper.go +++ b/internal/jobs/webscraper.go @@ -4,6 +4,8 @@ import ( "github.com/masa-finance/tee-worker/api/types" ) +const WebScraperType = "web-scraper" + type WebScraper struct { } diff --git a/internal/jobserver/worker.go b/internal/jobserver/worker.go index 62ebd95..6f29ab4 100644 --- a/internal/jobserver/worker.go +++ b/internal/jobserver/worker.go @@ -29,8 +29,8 @@ type worker interface { func (js *JobServer) doWork(j types.Job) error { var w worker switch j.Type { - case "web-scraper": - w = &jobs.WebScraper{} + case jobs.WebScraperType: + w = jobs.NewWebScraper() default: js.Lock() js.results[j.UUID] = types.JobResult{