From 6bb6071ec1dce0cce59df0cb1c38168918061041 Mon Sep 17 00:00:00 2001 From: nsfisis Date: Wed, 7 Aug 2024 22:14:51 +0900 Subject: refactor(backend): move ownership of channel to send task results --- backend/taskqueue/worker_server.go | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) (limited to 'backend/taskqueue/worker_server.go') diff --git a/backend/taskqueue/worker_server.go b/backend/taskqueue/worker_server.go index 485d6d3..67803e2 100644 --- a/backend/taskqueue/worker_server.go +++ b/backend/taskqueue/worker_server.go @@ -9,10 +9,10 @@ import ( type WorkerServer struct { server *asynq.Server queries *db.Queries - c chan string + results chan TaskExecResult } -func NewWorkerServer(redisAddr string, queries *db.Queries, c chan string) *WorkerServer { +func NewWorkerServer(redisAddr string, queries *db.Queries) *WorkerServer { return &WorkerServer{ server: asynq.NewServer( asynq.RedisClientOpt{ @@ -21,13 +21,17 @@ func NewWorkerServer(redisAddr string, queries *db.Queries, c chan string) *Work asynq.Config{}, ), queries: queries, - c: c, + results: make(chan TaskExecResult), } } func (s *WorkerServer) Run() error { mux := asynq.NewServeMux() - mux.Handle(TaskTypeExec, NewExecProcessor(s.queries, s.c)) + mux.Handle(TaskTypeExec, NewExecProcessor(s.queries, s.results)) return s.server.Run(mux) } + +func (s *WorkerServer) Results() chan TaskExecResult { + return s.results +} -- cgit v1.2.3-70-g09d2