31 lines
783 B
Go
31 lines
783 B
Go
|
package task
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/WHUPRJ/woj-server/internal/e"
|
||
|
|
"github.com/hibiken/asynq"
|
||
|
|
"go.uber.org/zap"
|
||
|
|
)
|
||
|
|
|
||
|
|
func (s *service) submit(typename string, payload []byte) (*asynq.TaskInfo, e.Status) {
|
||
|
|
task := asynq.NewTask(typename, payload)
|
||
|
|
|
||
|
|
info, err := s.queue.Enqueue(task)
|
||
|
|
if err != nil {
|
||
|
|
s.log.Warn("failed to enqueue task", zap.Error(err), zap.Any("task", task))
|
||
|
|
return nil, e.TaskEnqueueFailed
|
||
|
|
}
|
||
|
|
|
||
|
|
s.log.Debug("Successfully enqueued task", zap.Any("info", info))
|
||
|
|
|
||
|
|
return info, e.Success
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *service) GetTaskInfo(id string) (*asynq.TaskInfo, e.Status) {
|
||
|
|
task, err := s.inspector.GetTaskInfo("", id)
|
||
|
|
if err != nil {
|
||
|
|
s.log.Debug("get task info failed", zap.Error(err), zap.String("id", id))
|
||
|
|
return nil, e.TaskGetInfoFailed
|
||
|
|
}
|
||
|
|
return task, e.Success
|
||
|
|
}
|