archtechx/tenancy
1
0
Fork 0
mirror of https://github.com/archtechx/tenancy.git synced 2026-02-05 20:34: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

11
src/Bootstrappers/BatchTenancyBootstrapper.php
View file

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