Skip to content

Commit

Permalink
Fix #1229
Browse files Browse the repository at this point in the history
  • Loading branch information
stancl committed Sep 10, 2024
1 parent 8f9c7ef commit 99eabfb
Showing 1 changed file with 12 additions and 33 deletions.
45 changes: 12 additions & 33 deletions src/Bootstrappers/QueueTenancyBootstrapper.php
Original file line number Diff line number Diff line change
Expand Up @@ -31,6 +31,8 @@ class QueueTenancyBootstrapper implements TenancyBootstrapper
* This is useful when you're changing the tenant's state (e.g. properties in the `data` column) and want the next job to initialize tenancy again
* with the new data. Features like the Tenant Config are only executed when tenancy is initialized, so the re-initialization is needed in some cases.
*
* @deprecated This now has no effect, tenancy is always ended between queued jobs.
*
* @var bool
*/
public static $forceRefresh = false;
Expand All @@ -42,7 +44,7 @@ class QueueTenancyBootstrapper implements TenancyBootstrapper
*/
public static function __constructStatic(Application $app)
{
static::setUpJobListener($app->make(Dispatcher::class), $app->runningUnitTests());
static::setUpJobListener($app->make(Dispatcher::class));
}

public function __construct(Repository $config, QueueManager $queue)
Expand All @@ -53,7 +55,7 @@ public function __construct(Repository $config, QueueManager $queue)
$this->setUpPayloadGenerator();
}

protected static function setUpJobListener($dispatcher, $runningTests)
protected static function setUpJobListener($dispatcher)
{
$previousTenant = null;

Expand All @@ -69,11 +71,8 @@ protected static function setUpJobListener($dispatcher, $runningTests)
static::initializeTenancyForQueue($event->payload()['tenant_id'] ?? null);
});

// If we're running tests, we make sure to clean up after any artisan('queue:work') calls
$revertToPreviousState = function ($event) use (&$previousTenant, $runningTests) {
if ($runningTests) {
static::revertToPreviousState($event, $previousTenant);
}
$revertToPreviousState = function ($event) use (&$previousTenant) {
static::revertToPreviousState($event, $previousTenant);
};

$dispatcher->listen(JobProcessed::class, $revertToPreviousState); // artisan('queue:work') which succeeds
Expand All @@ -83,55 +82,35 @@ protected static function setUpJobListener($dispatcher, $runningTests)
protected static function initializeTenancyForQueue($tenantId)
{
if (! $tenantId) {
// The job is not tenant-aware
if (tenancy()->initialized) {
// Tenancy was initialized, so we revert back to the central context
tenancy()->end();
}

return;
}

if (static::$forceRefresh) {
// Re-initialize tenancy between all jobs
// The job is not tenant-aware, so we make sure tenancy isn't initialized.
if (tenancy()->initialized) {
tenancy()->end();
}

tenancy()->initialize(tenancy()->find($tenantId));

return;
}

if (tenancy()->initialized) {
// Tenancy is already initialized
if (tenant()->getTenantKey() === $tenantId) {
// It's initialized for the same tenant (e.g. dispatchNow was used, or the previous job also ran for this tenant)
return;
}
}

// Tenancy was either not initialized, or initialized for a different tenant.
// Therefore, we initialize it for the correct tenant.
tenancy()->initialize(tenancy()->find($tenantId));
}

protected static function revertToPreviousState($event, &$previousTenant)
{
$tenantId = $event->job->payload()['tenant_id'] ?? null;

// The job was not tenant-aware
if (! $tenantId) {
// The job was not tenant-aware, so there's nothing to revert
return;
}

// Revert back to the previous tenant
if (tenant() && $previousTenant && $previousTenant->isNot(tenant())) {
// Revert back to the previous tenant (since Tenancy v3.8.5 this should should *likely* not happen)
tenancy()->initialize($previousTenant);
return;
}

// End tenancy
if (tenant() && (! $previousTenant)) {
// No previous tenant = previous context was central
// NOTE: Since Tenancy v3.8.5 this should *likely* not happen
tenancy()->end();
}
}
Expand Down

0 comments on commit 99eabfb

Please sign in to comment.