Merge pull request #471 from raymadrona/main

Bind or forget current tenant
spatie · Jun 30, 2023 · d5aaa4c · d5aaa4c
2 parents cab0ca6 + 292fbcc
commit d5aaa4c
Show file tree

Hide file tree

Showing 3 changed files with 37 additions and 21 deletions.
diff --git a/src/Actions/MakeQueueTenantAwareAction.php b/src/Actions/MakeQueueTenantAwareAction.php
@@ -5,6 +5,7 @@
 use Illuminate\Queue\Events\JobProcessing;
 use Illuminate\Queue\Events\JobRetryRequested;
 use Illuminate\Support\Arr;
+use Spatie\Multitenancy\Concerns\BindAsCurrentTenant;
 use Spatie\Multitenancy\Exceptions\CurrentTenantCouldNotBeDeterminedInTenantAwareJob;
 use Spatie\Multitenancy\Jobs\NotTenantAware;
 use Spatie\Multitenancy\Jobs\TenantAware;
@@ -14,6 +15,7 @@
 class MakeQueueTenantAwareAction
 {
     use UsesTenantModel;
+    use BindAsCurrentTenant;
 
     public function execute(): void
     {
@@ -41,11 +43,7 @@ protected function listenForJobsBeingQueued(): static
     protected function listenForJobsBeingProcessed(): static
     {
         app('events')->listen(JobProcessing::class, function (JobProcessing $event) {
-            $this->getTenantModel()::forgetCurrent();
-
-            if (array_key_exists('tenantId', $event->job->payload())) {
-                $this->findTenant($event)->makeCurrent();
-            }
+            $this->bindOrForgetCurrentTenant($event);
         });
 
         return $this;
@@ -54,11 +52,7 @@ protected function listenForJobsBeingProcessed(): static
     protected function listenForJobsRetryRequested(): static
     {
         app('events')->listen(JobRetryRequested::class, function (JobRetryRequested $event) {
-            $this->getTenantModel()::forgetCurrent();
-
-            if (array_key_exists('tenantId', $event->payload())) {
-                $this->findTenant($event)->makeCurrent();
-            }
+            $this->bindOrForgetCurrentTenant($event);
         });
 
         return $this;
@@ -131,4 +125,15 @@ protected function getJobFromQueueable(object $queueable)
 
         return $queueable->$job;
     }
+
+    protected function bindOrForgetCurrentTenant(JobProcessing|JobRetryRequested $event): void
+    {
+        if (array_key_exists('tenantId', $this->getEventPayload($event))) {
+            $this->bindAsCurrentTenant($this->findTenant($event)->makeCurrent());
+
+            return;
+        }
+
+        $this->getTenantModel()::forgetCurrent();
+    }
 }
diff --git a/src/Actions/MakeTenantCurrentAction.php b/src/Actions/MakeTenantCurrentAction.php
@@ -2,6 +2,7 @@
 
 namespace Spatie\Multitenancy\Actions;
 
+use Spatie\Multitenancy\Concerns\BindAsCurrentTenant;
 use Spatie\Multitenancy\Events\MadeTenantCurrentEvent;
 use Spatie\Multitenancy\Events\MakingTenantCurrentEvent;
 use Spatie\Multitenancy\Models\Tenant;
@@ -10,6 +11,8 @@
 
 class MakeTenantCurrentAction
 {
+    use BindAsCurrentTenant;
+
     public function __construct(
         protected TasksCollection $tasksCollection
     ) {
@@ -34,15 +37,4 @@ protected function performTasksToMakeTenantCurrent(Tenant $tenant): self
 
         return $this;
     }
-
-    protected function bindAsCurrentTenant(Tenant $tenant): self
-    {
-        $containerKey = config('multitenancy.current_tenant_container_key');
-
-        app()->forgetInstance($containerKey);
-
-        app()->instance($containerKey, $tenant);
-
-        return $this;
-    }
 }
diff --git a/src/Concerns/BindAsCurrentTenant.php b/src/Concerns/BindAsCurrentTenant.php
@@ -0,0 +1,19 @@
+<?php
+
+namespace Spatie\Multitenancy\Concerns;
+
+use Spatie\Multitenancy\Models\Tenant;
+
+trait BindAsCurrentTenant
+{
+    protected function bindAsCurrentTenant(Tenant $tenant): self
+    {
+        $containerKey = config('multitenancy.current_tenant_container_key');
+
+        app()->forgetInstance($containerKey);
+
+        app()->instance($containerKey, $tenant);
+
+        return $this;
+    }
+}