From 1f52d5504b958ac0a1c870f669beec34d94b3cda Mon Sep 17 00:00:00 2001 From: Eric Davis Date: Wed, 20 Nov 2024 09:16:15 -0800 Subject: [PATCH] feat: check if context is done before grabbing next item --- cmd/bsky-users/main.go | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/cmd/bsky-users/main.go b/cmd/bsky-users/main.go index 385f698..34dc791 100644 --- a/cmd/bsky-users/main.go +++ b/cmd/bsky-users/main.go @@ -99,7 +99,14 @@ func handler(ctx context.Context, queue *Queue, dbCnx *sql.DB) { eventCount int ) +queueLoop: for { + select { + case <-ctx.Done(): + break queueLoop + default: + } + event, ok := queue.Dequeue() if !ok { time.Sleep(100 * time.Millisecond) -- 2.51.2