From e159beef7cb5e4dedba602268799b5f64a081681 Mon Sep 17 00:00:00 2001 From: arkiver Date: Thu, 16 Feb 2023 22:55:27 +0100 Subject: [PATCH 1/2] Support stash:* queues registered in stashes set. --- main.go | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/main.go b/main.go index 939d597..cd31618 100644 --- a/main.go +++ b/main.go @@ -86,6 +86,7 @@ func (that *Offloader) RefreshQueues() error { pipe := that.RedisClient.Pipeline() prioritiesCmdRes := pipe.ZRange(that.Context, fmt.Sprintf("%s:priorities", that.Name), 0, -1) filtersCmdRes := pipe.SMembers(that.Context, fmt.Sprintf("%s:filters", that.Name)) + stashesCmdRes := pipe.SMembers(that.Context, fmt.Sprintf("%s:stashes", that.Name)) _, err := pipe.Exec(that.Context) if err != nil { return err @@ -98,6 +99,10 @@ func (that *Offloader) RefreshQueues() error { if err != nil { return err } + stashes, err := stashesCmdRes.Result() + if err != nil { + return err + } setQueueMap := map[string]string{ "todo": "todo", "todo:secondary": "todo:secondary", @@ -112,6 +117,9 @@ func (that *Offloader) RefreshQueues() error { for _, priority := range priorities { setQueueMap[fmt.Sprintf("todo:prio:%s", priority)] = fmt.Sprintf("todo:prio:%s", priority) } + for _, stash := range stashes { + setQueueMap[fmt.Sprintf("stash:%s", stash)] = fmt.Sprintf("stash:%s", stash) + } needQueueMap := map[string]bool{} for setName, queueName := range setQueueMap { needQueueMap[queueName] = true -- 2.17.1 From ff57dbb637c3e4e0f0666065c5f2064abd981142 Mon Sep 17 00:00:00 2001 From: arkiver Date: Sun, 19 Feb 2023 13:59:17 +0100 Subject: [PATCH 2/2] Handle backfeed_retry set. --- main.go | 1 + 1 file changed, 1 insertion(+) diff --git a/main.go b/main.go index cd31618..53bc2b2 100644 --- a/main.go +++ b/main.go @@ -110,6 +110,7 @@ func (that *Offloader) RefreshQueues() error { "todo:backfeed": "todo:backfeed", "done": "done", "unretrievable": "unretrievable", + "backfeed_retry": "backfeed_retry", } for _, filter := range filters { setQueueMap[fmt.Sprintf("filtered:%s", filter)] = "filtered" -- 2.17.1