Merge pull request #2 from masa-finance/chore/refactor_scraper

chore(refactor): move scraper type to a constant
masa-finance · Oct 28, 2024 · c31771b · c31771b
2 parents 5dc9a18 + f78171a
commit c31771b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git 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
@@ -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{
-Original file line number
+Diff line change
@@ Expand Up / @@ -4,6 +4,8 @@ import ( @@
     	"github.com/masa-finance/tee-worker/api/types"
     )
+    const WebScraperType = "web-scraper"
     type WebScraper struct {
     }
@@ Expand Down @@