go-pkg/asynq
2
0
Fork 0
mirror of https://github.com/hibiken/asynq.git synced 2025-04-22 16:50:18 +08:00
Code Issues Projects Releases Wiki Activity

Update all variants of RDB.RunAll methods

Browse Source
This commit is contained in:
Ken Hibino 2021-02-26 15:32:25 -08:00
parent 0f09f936a9
commit ef0c390642
2 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
internal/rdb/inspect.go
View File

@ -513,12 +513,12 @@ func (r *RDB) removeAndRun(zset, qkey, id string) (int64, error) {
} }
var removeAndRunAllCmd = redis.NewScript(` var removeAndRunAllCmd = redis.NewScript(`
local msgs = redis.call("ZRANGE", KEYS[1], 0, -1) local ids = redis.call("ZRANGE", KEYS[1], 0, -1)
for _, msg in ipairs(msgs) do for _, id in ipairs(ids) do
redis.call("LPUSH", KEYS[2], msg) redis.call("LPUSH", KEYS[2], id)
redis.call("ZREM", KEYS[1], msg) redis.call("ZREM", KEYS[1], id)
end end
return table.getn(msgs)`) return table.getn(ids)`)
func (r *RDB) removeAndRunAll(zset, qkey string) (int64, error) { func (r *RDB) removeAndRunAll(zset, qkey string) (int64, error) {
res, err := removeAndRunAllCmd.Run(r.client, []string{zset, qkey}).Result() res, err := removeAndRunAllCmd.Run(r.client, []string{zset, qkey}).Result()

2
internal/rdb/inspect_test.go
View File

@ -1511,7 +1511,7 @@ func TestRunAllRetryTasks(t *testing.T) {
} }
} }
func TestRunAllDeadTasks(t *testing.T) { func TestRunAllArchivedTasks(t *testing.T) {
r := setup(t) r := setup(t)
defer r.Close() defer r.Close()
t1 := h.NewTaskMessage("send_email", nil) t1 := h.NewTaskMessage("send_email", nil)