Timeout

Author: dantleech

lib/Extension/Core/CoreExtension.php

@@ -14,6 +14,7 @@
 use PhpBench\Pipeline\Extension\Core\Stage\Aggregator\DescribeAggregator;
 use PhpBench\Pipeline\Extension\Core\Stage\Parameter\CounterParameter;
 use PhpBench\Pipeline\Extension\Core\Stage\Valve\DelayValve;
+use PhpBench\Pipeline\Extension\Core\Stage\Valve\TimeoutValve;
 
 class CoreExtension implements PipelineExtension
 {
@@ -35,6 +36,7 @@ public function __construct()
             'sampler/curl' => new CurlSampler(),
             'valve/delay' => new DelayValve(),
             'valve/take' => new TakeValve(),
+            'valve/timeout' => new TimeoutValve(),
         ];
     }
 

lib/Extension/Core/Stage/Valve/TimeoutValve.php

@@ -0,0 +1,29 @@
+<?php
+
+namespace PhpBench\Pipeline\Extension\Core\Stage\Valve;
+
+use PhpBench\Pipeline\Core\Stage;
+use Generator;
+use PhpBench\Pipeline\Core\Schema;
+
+class TimeoutValve implements Stage
+{
+    public function __invoke(): Generator
+    {
+        list($config, $data) = yield;
+
+        $runningTime = 0;
+        $start = microtime(true);
+        while ($runningTime < $config['time']) {
+            list($config, $data) = yield $data;
+            $runningTime = (microtime(true) - $start) * 1E6;
+        }
+    }
+
+    public function configure(Schema $schema)
+    {
+        $schema->setDefaults([
+            'time' => 1E6,
+        ]);
+    }
+}

tests/Unit/Extension/Core/Stage/Valve/TimeoutValveTest.php

@@ -0,0 +1,21 @@
+<?php
+
+namespace PhpBench\Pipeline\Tests\Unit\Extension\Core\Stage\Valve;
+
+use PHPUnit\Framework\TestCase;
+use PhpBench\Pipeline\Tests\Unit\Extension\Core\CoreTestCase;
+
+class TimeoutValveTest extends CoreTestCase
+{
+    public function testTerminatesAfterAGivenNumberOfMicroseconds()
+    {
+        $start = microtime(true);
+        $this->pipeline()
+            ->stage('valve/timeout', [ 'time' => 100000 ])
+            ->run();
+
+        $time = (microtime(true) - $start) * 1E6;
+
+        $this->assertGreaterThanOrEqual(100000, $time);
+    }
+}