mirror of
https://github.com/archtechx/tenancy.git
synced 2026-02-05 02:34:03 +00:00
7317d2638a
This PR adds Postgres RLS (trait manager + table manager approach) and permission controlled managers for PostgreSQL. --------- Co-authored-by: lukinovec <lukinovec@gmail.com> Co-authored-by: PHP CS Fixer <phpcsfixer@example.com>
63 lines
2.1 KiB
PHP
63 lines
2.1 KiB
PHP
<?php
|
|
|
|
declare(strict_types=1);
|
|
|
|
namespace Stancl\Tenancy\Bootstrappers;
|
|
|
|
use Illuminate\Contracts\Config\Repository;
|
|
use Illuminate\Database\DatabaseManager;
|
|
use Stancl\Tenancy\Contracts\TenancyBootstrapper;
|
|
use Stancl\Tenancy\Contracts\Tenant;
|
|
use Stancl\Tenancy\Database\DatabaseManager as TenantConnectionManager;
|
|
|
|
/**
|
|
* When initializing tenancy, use the tenant connection with the configured RLS user credentials
|
|
* and set the configured session variable to the current tenant's key.
|
|
*
|
|
* When ending tenancy, reset the session variable (to invalidate the connection)
|
|
* and switch back to the central connection again.
|
|
*
|
|
* This bootstrapper is intended to be used with Postgres RLS.
|
|
*
|
|
* @see \Stancl\Tenancy\Commands\CreateUserWithRLSPolicies
|
|
*/
|
|
class PostgresRLSBootstrapper implements TenancyBootstrapper
|
|
{
|
|
public function __construct(
|
|
protected Repository $config,
|
|
protected DatabaseManager $database,
|
|
protected TenantConnectionManager $tenantConnectionManager,
|
|
) {}
|
|
|
|
public function bootstrap(Tenant $tenant): void
|
|
{
|
|
$this->connectToTenant();
|
|
|
|
$tenantSessionKey = $this->config->get('tenancy.rls.session_variable_name');
|
|
|
|
$this->database->statement("SET {$tenantSessionKey} = '{$tenant->getTenantKey()}'");
|
|
}
|
|
|
|
public function revert(): void
|
|
{
|
|
$this->database->statement("RESET {$this->config->get('tenancy.rls.session_variable_name')}");
|
|
|
|
$this->tenantConnectionManager->reconnectToCentral();
|
|
}
|
|
|
|
protected function connectToTenant(): void
|
|
{
|
|
$centralConnection = $this->config->get('tenancy.database.central_connection');
|
|
|
|
$this->tenantConnectionManager->purgeTenantConnection();
|
|
|
|
$tenantConnection = array_merge($this->config->get('database.connections.' . $centralConnection), [
|
|
'username' => $this->config->get('tenancy.rls.user.username'),
|
|
'password' => $this->config->get('tenancy.rls.user.password'),
|
|
]);
|
|
|
|
$this->config['database.connections.tenant'] = $tenantConnection;
|
|
|
|
$this->tenantConnectionManager->setDefaultConnection('tenant');
|
|
}
|
|
}
|