Monitor specific queue for long wait if configured

src/Listeners/MonitorWaitTimes.php

@@ -46,10 +46,20 @@ public function handle()
             return;
         }
 
-        // Here we will calculate the wait time in seconds for each of the queues that
-        // the application is working. Then, we will filter the results to find the
-        // queues with the longest wait times and raise events for each of these.
-        $results = app(WaitTimeCalculator::class)->calculate();
+        // Here we will calculate the wait time in seconds for each of the queues
+        // the application is working or monitoring. Then we will compare the
+        // results against the configured times for any long wait times.
+        $waitTime = app(WaitTimeCalculator::class);
+        $results = $waitTime->calculate();
+
+        $results = collect(config('horizon.waits'))
+            ->diffKeys($results)
+            ->filter()
+            ->map(function ($wait, $queue) use ($waitTime) {
+                return $waitTime->calculateFor($queue);
+            })
+            ->merge($results)
+            ->all();
 
         $long = collect($results)->filter(function ($wait, $queue) {
             return config("horizon.waits.{$queue}") !== 0

tests/Feature/MonitorWaitTimesTest.php

@@ -15,13 +15,17 @@ class MonitorWaitTimesTest extends IntegrationTest
 {
     public function test_queues_with_long_waits_are_found()
     {
+        config(['horizon.waits' => []]);
+
         Event::fake();
 
         $calc = Mockery::mock(WaitTimeCalculator::class);
-        $calc->shouldReceive('calculate')->andReturn([
-            'redis:test-queue' => 10,
-            'redis:test-queue-2' => 80,
-        ]);
+        $calc->expects('calculate')
+            ->withNoArgs()
+            ->andReturn([
+                'redis:test-queue' => 10,
+                'redis:test-queue-2' => 80,
+            ]);
         $this->app->instance(WaitTimeCalculator::class, $calc);
 
         $listener = new MonitorWaitTimes(resolve(MetricsRepository::class));
@@ -34,23 +38,53 @@ public function test_queues_with_long_waits_are_found()
         });
     }
 
-    public function test_queue_ignores_long_waits()
+    public function test_ignores_queue_with_long_wait()
     {
         config(['horizon.waits' => ['redis:ignore-queue' => 0]]);
 
         Event::fake();
 
         $calc = Mockery::mock(WaitTimeCalculator::class);
-        $calc->expects('calculate')->andReturn([
-            'redis:ignore-queue' => 10,
-        ]);
+        $calc->expects('calculate')
+            ->withNoArgs()
+            ->andReturn([
+                'redis:ignore-queue' => 10,
+            ]);
         $this->app->instance(WaitTimeCalculator::class, $calc);
 
         $listener = new MonitorWaitTimes(resolve(MetricsRepository::class));
-        $listener->lastMonitoredAt = CarbonImmutable::now()->subDays(1);
+        $listener->lastMonitoredAt = CarbonImmutable::now()->subDay();
 
         $listener->handle();
 
         Event::assertNotDispatched(LongWaitDetected::class);
     }
+
+    public function test_monitors_specific_queue_with_long_wait()
+    {
+        config(['horizon.waits' => ['redis:default' => 0, 'redis:shared-1' => 10]]);
+
+        Event::fake();
+
+        $calc = Mockery::mock(WaitTimeCalculator::class);
+        $calc->expects('calculate')
+            ->withNoArgs()
+            ->andReturn([
+                'redis:default' => 90,
+                'redis:shared-1,shared-2' => 25,
+            ]);
+        $calc->expects('calculateFor')
+            ->with('redis:shared-1')
+            ->andReturn(15);
+        $this->app->instance(WaitTimeCalculator::class, $calc);
+
+        $listener = new MonitorWaitTimes(resolve(MetricsRepository::class));
+        $listener->lastMonitoredAt = CarbonImmutable::now()->subDay();
+
+        $listener->handle();
+
+        Event::assertDispatched(LongWaitDetected::class, function ($event) {
+            return $event->connection == 'redis' && $event->queue == 'shared-1';
+        });
+    }
 }