RunnableQueue - cancelAll()

ondrejmirtes · ondrejmirtes · commit 51e6cc4b1c6f · 2020-09-03T23:32:39.000+02:00
diff --git a/src/Process/Runnable/RunnableQueue.php b/src/Process/Runnable/RunnableQueue.php
@@ -123,4 +123,16 @@ private function drainQueue(): void
 		});
 	}
 
+	public function cancelAll(): void
+	{
+		foreach ($this->queue as [$runnable, $size, $deferred]) {
+			$deferred->promise()->cancel(); // @phpstan-ignore-line
+		}
+
+		foreach ($this->running as $running) { // phpcs:ignore
+			[, $deferred] = $this->running->getInfo();
+			$deferred->promise()->cancel(); // @phpstan-ignore-line
+		}
+	}
+
 }
diff --git a/tests/PHPStan/Process/Runnable/RunnableQueueTest.php b/tests/PHPStan/Process/Runnable/RunnableQueueTest.php
@@ -122,4 +122,43 @@ public function testCancel(): void
 		], $logger->getMessages());
 	}
 
+	public function testCancelAll(): void
+	{
+		$logger = new RunnableQueueLoggerStub();
+		$queue = new RunnableQueue($logger, 4);
+		$one = new RunnableStub('1');
+		$two = new RunnableStub('2');
+		$three = new RunnableStub('3');
+		$queue->queue($one, 3);
+		$queue->queue($two, 2);
+		$queue->queue($three, 3);
+
+		$this->assertSame(5, $queue->getQueueSize());
+		$this->assertSame(3, $queue->getRunningSize());
+
+		$queue->cancelAll();
+		$this->assertSame(0, $queue->getQueueSize());
+		$this->assertSame(0, $queue->getRunningSize());
+
+		$this->assertSame([
+			0 => 'Queue not full - looking at first item in the queue',
+			1 => 'Removing top item from queue - new size is 3',
+			2 => 'Running process 1',
+			3 => 'Queue not full - looking at first item in the queue',
+			4 => 'Canot remote first item from the queue - it has size 2, current queue size is 3, new size would be 5',
+			5 => 'Queue not full - looking at first item in the queue',
+			6 => 'Canot remote first item from the queue - it has size 2, current queue size is 3, new size would be 5',
+			7 => 'Process 1 finished unsuccessfully: Runnable 1 canceled',
+			8 => 'Queue not full - looking at first item in the queue',
+			9 => 'Removing top item from queue - new size is 2',
+			10 => 'Running process 2',
+			11 => 'Process 2 finished unsuccessfully: Runnable 2 canceled',
+			12 => 'Queue not full - looking at first item in the queue',
+			13 => 'Removing top item from queue - new size is 3',
+			14 => 'Running process 3',
+			15 => 'Process 3 finished unsuccessfully: Runnable 3 canceled',
+			16 => 'Queue empty',
+		], $logger->getMessages());
+	}
+
 }
