113 lines
3.2 KiB
PHP
113 lines
3.2 KiB
PHP
<?php
|
|
|
|
namespace Pterodactyl\Repositories\Eloquent;
|
|
|
|
use Pterodactyl\Models\Allocation;
|
|
use Illuminate\Database\Eloquent\Builder;
|
|
use Pterodactyl\Contracts\Repository\AllocationRepositoryInterface;
|
|
|
|
class AllocationRepository extends EloquentRepository implements AllocationRepositoryInterface
|
|
{
|
|
/**
|
|
* Return the model backing this repository.
|
|
*
|
|
* @return string
|
|
*/
|
|
public function model()
|
|
{
|
|
return Allocation::class;
|
|
}
|
|
|
|
/**
|
|
* Return all of the allocations that exist for a node that are not currently
|
|
* allocated.
|
|
*
|
|
* @param int $node
|
|
* @return array
|
|
*/
|
|
public function getUnassignedAllocationIds(int $node): array
|
|
{
|
|
return Allocation::query()->select('id')
|
|
->whereNull('server_id')
|
|
->where('node_id', $node)
|
|
->get()
|
|
->pluck('id')
|
|
->toArray();
|
|
}
|
|
|
|
/**
|
|
* Return a concatenated result set of node ips that already have at least one
|
|
* server assigned to that IP. This allows for filtering out sets for
|
|
* dedicated allocation IPs.
|
|
*
|
|
* If an array of nodes is passed the results will be limited to allocations
|
|
* in those nodes.
|
|
*
|
|
* @param array $nodes
|
|
* @return array
|
|
*/
|
|
protected function getDiscardableDedicatedAllocations(array $nodes = []): array
|
|
{
|
|
$query = Allocation::query()->selectRaw('CONCAT_WS("-", node_id, ip) as result');
|
|
|
|
if (! empty($nodes)) {
|
|
$query->whereIn('node_id', $nodes);
|
|
}
|
|
|
|
return $query->whereNotNull('server_id')
|
|
->groupByRaw('CONCAT(node_id, ip)')
|
|
->get()
|
|
->pluck('result')
|
|
->toArray();
|
|
}
|
|
|
|
/**
|
|
* Return a single allocation from those meeting the requirements.
|
|
*
|
|
* @param array $nodes
|
|
* @param array $ports
|
|
* @param bool $dedicated
|
|
* @return \Pterodactyl\Models\Allocation|null
|
|
*/
|
|
public function getRandomAllocation(array $nodes, array $ports, bool $dedicated = false)
|
|
{
|
|
$query = Allocation::query()->whereNull('server_id');
|
|
|
|
if (! empty($nodes)) {
|
|
$query->whereIn('node_id', $nodes);
|
|
}
|
|
|
|
if (! empty($ports)) {
|
|
$query->where(function (Builder $inner) use ($ports) {
|
|
$whereIn = [];
|
|
foreach ($ports as $port) {
|
|
if (is_array($port)) {
|
|
$inner->orWhereBetween('port', $port);
|
|
continue;
|
|
}
|
|
|
|
$whereIn[] = $port;
|
|
}
|
|
|
|
if (! empty($whereIn)) {
|
|
$inner->orWhereIn('port', $whereIn);
|
|
}
|
|
});
|
|
}
|
|
|
|
// If this allocation should not be shared with any other servers get
|
|
// the data and modify the query as necessary,
|
|
if ($dedicated) {
|
|
$discard = $this->getDiscardableDedicatedAllocations($nodes);
|
|
|
|
if (! empty($discard)) {
|
|
$query->whereNotIn(
|
|
$this->getBuilder()->raw('CONCAT_WS("-", node_id, ip)'), $discard
|
|
);
|
|
}
|
|
}
|
|
|
|
return $query->inRandomOrder()->first();
|
|
}
|
|
}
|