Skynet/misc_pterodactyl-panel
7
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat(database-hosts): allow linking of multiple nodes

Browse source
This commit is contained in:
Matthew Penner 2021-01-30 12:46:27 -07:00
parent 72983e8385
commit 2b5cc99abd
4 changed files with 73 additions and 33 deletions
Download patch file Download diff file Expand all files Collapse all files

26
app/Services/Databases/DeployServerDatabaseService.php
View file

@ -10,10 +10,7 @@ use Pterodactyl\Exceptions\Service\Database\NoSuitableDatabaseHostException;
class DeployServerDatabaseService
{
/**
* @var \Pterodactyl\Services\Databases\DatabaseManagementService
*/
private $managementService;
private DatabaseManagementService $managementService;
/**
* ServerDatabaseCreationService constructor.
@ -35,21 +32,22 @@ class DeployServerDatabaseService
Assert::notEmpty($data['database'] ?? null);
Assert::notEmpty($data['remote'] ?? null);
$hosts = DatabaseHost::query()->get()->toBase();
if ($hosts->isEmpty()) {
throw new NoSuitableDatabaseHostException();
} else {
$nodeHosts = $hosts->where('node_id', $server->node_id)->toBase();
if ($nodeHosts->isEmpty() && !config('pterodactyl.client_features.databases.allow_random')) {
$databaseHostId = $server->node->database_host_id;
if (is_null($databaseHostId)) {
if (!config('pterodactyl.client_features.databases.allow_random')) {
throw new NoSuitableDatabaseHostException();
}
$hosts = DatabaseHost::query()->get()->toBase();
if ($hosts->isEmpty()) {
throw new NoSuitableDatabaseHostException();
}
$databaseHostId = $hosts->random()->id;
}
return $this->managementService->create($server, [
'database_host_id' => $nodeHosts->isEmpty()
? $hosts->random()->id
: $nodeHosts->random()->id,
'database_host_id' => $databaseHostId,
'database' => DatabaseManagementService::generateUniqueDatabaseName($data['database'], $server->id),
'remote' => $data['remote'],
]);