diff options
| author | nsfisis <nsfisis@gmail.com> | 2024-08-07 22:14:51 +0900 |
|---|---|---|
| committer | nsfisis <nsfisis@gmail.com> | 2024-08-08 04:02:37 +0900 |
| commit | 6bb6071ec1dce0cce59df0cb1c38168918061041 (patch) | |
| tree | feca06471f51ac5a1cdaaed4a0333ee9c0cf38e3 /backend/taskqueue/worker_server.go | |
| parent | 401a28944fc0408811aedadd1c3104e2e2d4d7fe (diff) | |
| download | iosdc-japan-2024-albatross-6bb6071ec1dce0cce59df0cb1c38168918061041.tar.gz iosdc-japan-2024-albatross-6bb6071ec1dce0cce59df0cb1c38168918061041.tar.zst iosdc-japan-2024-albatross-6bb6071ec1dce0cce59df0cb1c38168918061041.zip | |
refactor(backend): move ownership of channel to send task results
Diffstat (limited to 'backend/taskqueue/worker_server.go')
| -rw-r--r-- | backend/taskqueue/worker_server.go | 12 |
1 files changed, 8 insertions, 4 deletions
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 +} |