2
0
mirror of https://github.com/hibiken/asynq.git synced 2024-09-20 11:05:58 +08:00

client.Enqueue - prevent empty task's typename

This commit is contained in:
Luqqk 2021-08-01 16:24:45 +02:00 committed by Ken Hibino
parent 73f930313c
commit 4bce28d677
2 changed files with 11 additions and 0 deletions

View File

@ -178,6 +178,9 @@ func (d processInOption) Value() interface{} { return time.Duration(d) }
// ErrDuplicateTask error only applies to tasks enqueued with a Unique option.
var ErrDuplicateTask = errors.New("task already exists")
// ErrEmptyTypeTask indicates that task's typename is not specified.
var ErrEmptyTypeTask = errors.New("task typename not specified")
type option struct {
retry int
queue string
@ -266,6 +269,9 @@ func (c *Client) Close() error {
//
// If no ProcessAt or ProcessIn options are provided, the task will be pending immediately.
func (c *Client) Enqueue(task *Task, opts ...Option) (*TaskInfo, error) {
if task.Type() == "" {
return nil, fmt.Errorf("%w", ErrEmptyTypeTask)
}
c.mu.Lock()
if defaults, ok := c.opts[task.Type()]; ok {
opts = append(defaults, opts...)

View File

@ -585,6 +585,11 @@ func TestClientEnqueueError(t *testing.T) {
Queue(""),
},
},
{
desc: "With empty task typename",
task: NewTask("", h.JSON(map[string]interface{}{})),
opts: []Option{},
},
}
for _, tc := range tests {