Homec4science
Diffusion Phabricator (feature/better-testing) Browse

phab/src/applications/drydock/allocator/resource/DrydockAllocator.php
d1ee08b2df26feature/better-testing

Last Change
Raw File

DrydockAllocator.php
View Options

<?php
/*
* Copyright 2012 Facebook, Inc.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
final class DrydockAllocator {
private $resourceType;
private $lease;
public function setResourceType($resource_type) {
$this->resourceType = $resource_type;
return $this;
}
public function getResourceType() {
return $this->resourceType;
}
public function getPendingLease() {
if (!$this->lease) {
$lease = new DrydockLease();
$lease->setStatus(DrydockLeaseStatus::STATUS_PENDING);
$lease->save();
$this->lease = $lease;
}
return $lease;
}
public function allocate() {
$type = $this->getResourceType();
$candidates = id(new DrydockResource())->loadAllWhere(
'type = %s AND status = %s',
$type,
DrydockResourceStatus::STATUS_OPEN);
if ($candidates) {
shuffle($candidates);
$resource = head($candidates);
} else {
$blueprints = DrydockBlueprint::getAllBlueprintsForResource($type);
foreach ($blueprints as $key => $blueprint) {
if (!$blueprint->canAllocateResources()) {
unset($blueprints[$key]);
continue;
}
}
if (!$blueprints) {
throw new Exception(
"There are no valid existing '{$type}' resources, and no valid ".
"blueprints to build new ones.");
}
// TODO: Rank intelligently.
shuffle($blueprints);
$blueprint = head($blueprints);
$resource = $blueprint->allocateResource();
}
$lease = $this->getPendingLease();
$lease->setResourceID($resource->getID());
$lease->setStatus(DrydockLeaseStatus::STATUS_ACTIVE);
$lease->save();
$lease->attachResource($resource);
return $lease;
}
}
c4science · Help