add subscription setup cli commands & skip setup in subscription run

patchlevel · Jul 31, 2024 · 55e10ec · 55e10ec
1 parent fe23cbc
commit 55e10ec
Show file tree

Hide file tree

Showing 6 changed files with 150 additions and 1 deletion.
diff --git a/src/Console/Command/SchemaSubscriptionSetupCommand.php b/src/Console/Command/SchemaSubscriptionSetupCommand.php
@@ -0,0 +1,47 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Patchlevel\EventSourcing\Console\Command;
+
+use Patchlevel\EventSourcing\Console\OutputStyle;
+use Patchlevel\EventSourcing\Store\Store;
+use Patchlevel\EventSourcing\Store\SubscriptionStore;
+use Symfony\Component\Console\Attribute\AsCommand;
+use Symfony\Component\Console\Command\Command;
+use Symfony\Component\Console\Input\InputInterface;
+use Symfony\Component\Console\Output\OutputInterface;
+
+#[AsCommand(
+    'event-sourcing:schema:subscription-setup',
+    'setup subscription (pub/sub) for store',
+)]
+final class SchemaSubscriptionSetupCommand extends Command
+{
+    public function __construct(
+        private readonly Store $store,
+    ) {
+        parent::__construct();
+    }
+
+    protected function execute(InputInterface $input, OutputInterface $output): int
+    {
+        $io = new OutputStyle($input, $output);
+
+        if (!$this->store instanceof SubscriptionStore) {
+            $io->error('store does not support subscriptions');
+
+            return 1;
+        }
+
+        if (!$this->store->supportSubscription()) {
+            $io->error('store does not support subscriptions');
+
+            return 1;
+        }
+
+        $this->store->setupSubscription();
+
+        return 0;
+    }
+}
diff --git a/src/Console/Command/SchemaSubscriptionTeardownCommand.php b/src/Console/Command/SchemaSubscriptionTeardownCommand.php
@@ -0,0 +1,55 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Patchlevel\EventSourcing\Console\Command;
+
+use Patchlevel\EventSourcing\Console\OutputStyle;
+use Patchlevel\EventSourcing\Store\Store;
+use Patchlevel\EventSourcing\Store\SubscriptionStore;
+use Symfony\Component\Console\Attribute\AsCommand;
+use Symfony\Component\Console\Command\Command;
+use Symfony\Component\Console\Input\InputInterface;
+use Symfony\Component\Console\Output\OutputInterface;
+
+use function method_exists;
+
+#[AsCommand(
+    'event-sourcing:schema:subscription-teardown',
+    'teardown subscription (pub/sub) for store',
+)]
+final class SchemaSubscriptionTeardownCommand extends Command
+{
+    public function __construct(
+        private readonly Store $store,
+    ) {
+        parent::__construct();
+    }
+
+    protected function execute(InputInterface $input, OutputInterface $output): int
+    {
+        $io = new OutputStyle($input, $output);
+
+        if (!$this->store instanceof SubscriptionStore) {
+            $io->error('store does not support subscriptions');
+
+            return 1;
+        }
+
+        if (!$this->store->supportSubscription()) {
+            $io->error('store does not support subscriptions');
+
+            return 1;
+        }
+
+        if (method_exists($this->store, 'teardownSubscription')) {
+            $this->store->teardownSubscription();
+
+            return 0;
+        }
+
+        $io->error('store does not support teardownSubscription');
+
+        return 1;
+    }
+}
diff --git a/src/Console/Command/SubscriptionRunCommand.php b/src/Console/Command/SubscriptionRunCommand.php
@@ -70,6 +70,12 @@ protected function configure(): void
                 null,
                 InputOption::VALUE_NONE,
                 'rebuild (remove & boot) subscriptions before run',
+            )
+            ->addOption(
+                'skip-subscription-setup',
+                null,
+                InputOption::VALUE_NONE,
+                'skip subscription setup',
             );
     }
 
@@ -81,11 +87,12 @@ protected function execute(InputInterface $input, OutputInterface $output): int
         $timeLimit = InputHelper::nullablePositiveInt($input->getOption('time-limit'));
         $sleep = InputHelper::positiveIntOrZero($input->getOption('sleep'));
         $rebuild = InputHelper::bool($input->getOption('rebuild'));
+        $skipSubscriptionSetup = InputHelper::bool($input->getOption('skip-subscription-setup'));
 
         $criteria = $this->subscriptionEngineCriteria($input);
         $criteria = $this->resolveCriteriaIntoCriteriaWithOnlyIds($criteria);
 
-        if ($this->store instanceof SubscriptionStore) {
+        if ($this->store instanceof SubscriptionStore && !$skipSubscriptionSetup) {
             $this->store->setupSubscription();
         }
 

diff --git a/src/Store/DoctrineDbalStore.php b/src/Store/DoctrineDbalStore.php
@@ -410,6 +410,25 @@ public function setupSubscription(): void
         ));
     }
 
+    public function teardownSubscription(): void
+    {
+        if (!$this->supportSubscription()) {
+            return;
+        }
+
+        $functionName = $this->createTriggerFunctionName();
+
+        $this->connection->executeStatement(sprintf(
+            'DROP FUNCTION IF EXISTS %s() CASCADE;',
+            $functionName,
+        ));
+
+        $this->connection->executeStatement(sprintf(
+            'DROP TRIGGER IF EXISTS notify_trigger ON %s;',
+            $this->config['table_name'],
+        ));
+    }
+
     private function createTriggerFunctionName(): string
     {
         $tableConfig = explode('.', $this->config['table_name']);

diff --git a/src/Store/StreamDoctrineDbalStore.php b/src/Store/StreamDoctrineDbalStore.php
@@ -442,6 +442,25 @@ public function setupSubscription(): void
         ));
     }
 
+    public function teardownSubscription(): void
+    {
+        if (!$this->supportSubscription()) {
+            return;
+        }
+
+        $functionName = $this->createTriggerFunctionName();
+
+        $this->connection->executeStatement(sprintf(
+            'DROP FUNCTION IF EXISTS %s() CASCADE;',
+            $functionName,
+        ));
+
+        $this->connection->executeStatement(sprintf(
+            'DROP TRIGGER IF EXISTS notify_trigger ON %s;',
+            $this->config['table_name'],
+        ));
+    }
+
     private function createTriggerFunctionName(): string
     {
         $tableConfig = explode('.', $this->config['table_name']);

diff --git a/src/Store/SubscriptionStore.php b/src/Store/SubscriptionStore.php
@@ -11,4 +11,6 @@ public function supportSubscription(): bool;
     public function setupSubscription(): void;
 
     public function wait(int $timeoutMilliseconds): void;
+
+    // public function teardownSubscription(): void;
 }