[Scheduler] Add AbstractTriggerDecorator

kbond · kbond · commit 3ca7eb3a5583 · 2023-07-28T16:05:45.000-04:00
diff --git a/src/Symfony/Component/Scheduler/CHANGELOG.md b/src/Symfony/Component/Scheduler/CHANGELOG.md
@@ -5,6 +5,7 @@ CHANGELOG
 ---
 
  * Allow setting timezone of next run date in CronExpressionTrigger
+ * Add `AbstractTriggerDecorator`
 
 6.3
 ---
diff --git a/src/Symfony/Component/Scheduler/Tests/Trigger/AbstractDecoratedTriggerTest.php b/src/Symfony/Component/Scheduler/Tests/Trigger/AbstractDecoratedTriggerTest.php
@@ -0,0 +1,40 @@
+<?php
+
+/*
+ * This file is part of the Symfony package.
+ *
+ * (c) Fabien Potencier <fabien@symfony.com>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace Symfony\Component\Scheduler\Tests\Trigger;
+
+use PHPUnit\Framework\TestCase;
+use Symfony\Component\Scheduler\Trigger\ExcludeTimeTrigger;
+use Symfony\Component\Scheduler\Trigger\JitterTrigger;
+use Symfony\Component\Scheduler\Trigger\TriggerInterface;
+
+class AbstractDecoratedTriggerTest extends TestCase
+{
+    public function testCanGetInnerTrigger()
+    {
+        $trigger = new JitterTrigger($inner = $this->createMock(TriggerInterface::class));
+
+        $this->assertSame($inner, $trigger->inner());
+        $this->assertSame([$trigger], iterator_to_array($trigger->decorators()));
+    }
+
+    public function testCanGetNestedInnerTrigger()
+    {
+        $trigger = new ExcludeTimeTrigger(
+            $jitter = new JitterTrigger($inner = $this->createMock(TriggerInterface::class)),
+            new \DateTimeImmutable(),
+            new \DateTimeImmutable(),
+        );
+
+        $this->assertSame($inner, $trigger->inner());
+        $this->assertSame([$trigger, $jitter], iterator_to_array($trigger->decorators()));
+    }
+}
diff --git a/src/Symfony/Component/Scheduler/Trigger/AbstractDecoratedTrigger.php b/src/Symfony/Component/Scheduler/Trigger/AbstractDecoratedTrigger.php
@@ -0,0 +1,49 @@
+<?php
+
+/*
+ * This file is part of the Symfony package.
+ *
+ * (c) Fabien Potencier <fabien@symfony.com>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace Symfony\Component\Scheduler\Trigger;
+
+/**
+ * @author Kevin Bond <kevinbond@gmail.com>
+ */
+abstract class AbstractDecoratedTrigger implements TriggerInterface
+{
+    public function __construct(private TriggerInterface $inner)
+    {
+    }
+
+    final public function inner(): TriggerInterface
+    {
+        $inner = $this->inner;
+
+        while ($inner instanceof self) {
+            $inner = $inner->inner;
+        }
+
+        return $inner;
+    }
+
+    /**
+     * @return \Traversable<self>
+     */
+    final public function decorators(): \Traversable
+    {
+        yield $this;
+
+        $inner = $this->inner;
+
+        while ($inner instanceof self) {
+            yield $inner;
+
+            $inner = $inner->inner;
+        }
+    }
+}
diff --git a/src/Symfony/Component/Scheduler/Trigger/ExcludeTimeTrigger.php b/src/Symfony/Component/Scheduler/Trigger/ExcludeTimeTrigger.php
@@ -14,13 +14,14 @@
 /**
  * @experimental
  */
-final class ExcludeTimeTrigger implements TriggerInterface
+final class ExcludeTimeTrigger extends AbstractDecoratedTrigger
 {
     public function __construct(
         private readonly TriggerInterface $inner,
         private readonly \DateTimeImmutable $from,
         private readonly \DateTimeImmutable $until,
     ) {
+        parent::__construct($inner);
     }
 
     public function __toString(): string
diff --git a/src/Symfony/Component/Scheduler/Trigger/JitterTrigger.php b/src/Symfony/Component/Scheduler/Trigger/JitterTrigger.php
@@ -14,13 +14,14 @@
 /**
  * @author Kevin Bond <kevinbond@gmail.com>
  */
-final class JitterTrigger implements TriggerInterface
+final class JitterTrigger extends AbstractDecoratedTrigger
 {
     /**
      * @param positive-int $maxSeconds
      */
     public function __construct(private readonly TriggerInterface $trigger, private readonly int $maxSeconds = 60)
     {
+        parent::__construct($trigger);
     }
 
     public function __toString(): string
diff --git a/src/Symfony/Component/Scheduler/Trigger/PeriodicalTrigger.php b/src/Symfony/Component/Scheduler/Trigger/PeriodicalTrigger.php
@@ -16,7 +16,7 @@
 /**
  * @experimental
  */
-class PeriodicalTrigger implements TriggerInterface, \Stringable
+class PeriodicalTrigger implements TriggerInterface
 {
     private float $intervalInSeconds = 0.0;
     private \DateTimeImmutable $from;

Original file line number	Diff line number	Diff line change
`@@ -14,13 +14,14 @@`
`14`	`14`	`/**`
`15`	`15`	`* @author Kevin Bond <kevinbond@gmail.com>`
`16`	`16`	`*/`
`17`		`-final class JitterTrigger implements TriggerInterface`
	`17`	`+final class JitterTrigger extends AbstractDecoratedTrigger`
`18`	`18`	`{`
`19`	`19`	`/**`
`20`	`20`	`* @param positive-int $maxSeconds`
`21`	`21`	`*/`
`22`	`22`	`public function __construct(private readonly TriggerInterface $trigger, private readonly int $maxSeconds = 60)`
`23`	`23`	`{`
	`24`	`+ parent::__construct($trigger);`
`24`	`25`	`}`
`25`	`26`
`26`	`27`	`public function __toString(): string`
Original file line number	Diff line number	Diff line change
`@@ -16,7 +16,7 @@`
`16`	`16`	`/**`
`17`	`17`	`* @experimental`
`18`	`18`	`*/`
`19`		`-class PeriodicalTrigger implements TriggerInterface, \Stringable`
	`19`	`+class PeriodicalTrigger implements TriggerInterface`
`20`	`20`	`{`
`21`	`21`	`private float $intervalInSeconds = 0.0;`
`22`	`22`	`private \DateTimeImmutable $from;`