archtechx/tenancy
1
0
Fork 0
mirror of https://github.com/archtechx/tenancy.git synced 2026-02-05 14:54:03 +00:00
Code Activity

typehint DatabaseBatchRepository

Browse source
This commit is contained in:
Abrar Ahmad 2022-09-22 11:16:14 +05:00
parent 8e919b8101
commit 8bbf92aa9b
1 changed files with 5 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

13
src/Bootstrappers/BatchTenancyBootstrapper.php
View file

@ -4,7 +4,6 @@ declare(strict_types=1);
namespace Stancl\Tenancy\Bootstrappers;
use Illuminate\Bus\BatchRepository;
use Illuminate\Bus\DatabaseBatchRepository;
use Illuminate\Support\Facades\DB;
use Stancl\Tenancy\Contracts\TenancyBootstrapper;
@ -19,24 +18,22 @@ class BatchTenancyBootstrapper implements TenancyBootstrapper
*/
protected $previousConnection = null;
public function __construct(protected BatchRepository $batchRepository)
public function __construct(protected DatabaseBatchRepository $databaseBatchRepository)
{
}
public function bootstrap(Tenant $tenant)
{
if ($this->batchRepository instanceof DatabaseBatchRepository) {
// Update batch repository connection to use the tenant connection
$this->previousConnection = $this->batchRepository->getConnection();
$this->batchRepository->setConnection(DB::connection('tenant'));
}
// Update batch repository connection to use the tenant connection
$this->previousConnection = $this->databaseBatchRepository->getConnection();
$this->databaseBatchRepository->setConnection(DB::connection('tenant'));
}
public function revert()
{
if ($this->previousConnection) {
// Replace batch repository connection with the previously replaced one
$this->batchRepository->setConnection($this->previousConnection);
$this->databaseBatchRepository->setConnection($this->previousConnection);
$this->previousConnection = null;
}
}