Paul Pan
d42ee0ce54
1. merge woj-runner scripts into woj-server 2. add woj-runner app 3. refactor submission status problem ... 4. jwt middleware update Co-authored-by: cxy004 <cxy004@qq.com> Co-authored-by: wzt <w.zhongtao@qq.com>
31 lines
834 B
Go
31 lines
834 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, queue string) (*asynq.TaskInfo, e.Status) {
|
|
task := asynq.NewTask(typename, payload)
|
|
|
|
info, err := s.queue.Enqueue(task, asynq.Queue(queue))
|
|
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, queue string) (*asynq.TaskInfo, e.Status) {
|
|
task, err := s.inspector.GetTaskInfo(queue, 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
|
|
}
|