Diffusion Phabricator (feature/testing) Browse

testing

DrydockLease.php
View Options

			<?php

			final class DrydockLease extends DrydockDAO {

			protected $resourceID;
			protected $resourceType;
			protected $until;
			protected $ownerPHID;
			protected $attributes = array();
			protected $status = DrydockLeaseStatus::STATUS_PENDING;
			protected $taskID;

			private $resource = self::ATTACHABLE;
			private $releaseOnDestruction;

			/**
			* Flag this lease to be released when its destructor is called. This is
			* mostly useful if you have a script which acquires, uses, and then releases
			* a lease, as you don't need to explicitly handle exceptions to properly
			* release the lease.
			*/
			public function releaseOnDestruction() {
			$this->releaseOnDestruction = true;
			return $this;
			}

			public function __destruct() {
			if ($this->releaseOnDestruction) {
			if ($this->isActive()) {
			$this->release();
			}
			}
			}

			public function getLeaseName() {
			return pht('Lease %d', $this->getID());
			}

			public function getConfiguration() {
			return array(
			self::CONFIG_AUX_PHID => true,
			self::CONFIG_SERIALIZATION => array(
			'attributes' => self::SERIALIZATION_JSON,
			),
			) + parent::getConfiguration();
			}

			public function setAttribute($key, $value) {
			$this->attributes[$key] = $value;
			return $this;
			}

			public function getAttribute($key, $default = null) {
			return idx($this->attributes, $key, $default);
			}

			public function generatePHID() {
			return PhabricatorPHID::generateNewPHID(
			PhabricatorPHIDConstants::PHID_TYPE_DRYL);
			}

			public function getInterface($type) {
			return $this->getResource()->getInterface($this, $type);
			}

			public function getResource() {
			return $this->assertAttached($this->resource);
			}

			public function attachResource(DrydockResource $resource) {
			$this->resource = $resource;
			return $this;
			}

			public function hasAttachedResource() {
			return ($this->resource !== null);
			}

			public function loadResource() {
			return id(new DrydockResource())->loadOneWhere(
			'id = %d',
			$this->getResourceID());
			}

			public function queueForActivation() {
			if ($this->getID()) {
			throw new Exception(
			"Only new leases may be queued for activation!");
			}

			$this->setStatus(DrydockLeaseStatus::STATUS_PENDING);
			$this->save();

			// NOTE: Prevent a race where some eager worker quickly grabs the task
			// before we can save the Task ID.

			$this->openTransaction();
			$this->beginReadLocking();

			$this->reload();

			$task = PhabricatorWorker::scheduleTask(
			'DrydockAllocatorWorker',
			$this->getID());

			$this->setTaskID($task->getID());
			$this->save();

			$this->endReadLocking();
			$this->saveTransaction();

			return $this;
			}

			public function release() {
			$this->assertActive();
			$this->setStatus(DrydockLeaseStatus::STATUS_RELEASED);
			$this->save();

			$this->resource = null;

			return $this;
			}

			public function isActive() {
			switch ($this->status) {
			case DrydockLeaseStatus::STATUS_ACTIVE:
			case DrydockLeaseStatus::STATUS_ACQUIRING:
			return true;
			}
			return false;
			}

			private function assertActive() {
			if (!$this->isActive()) {
			throw new Exception(
			"Lease is not active! You can not interact with resources through ".
			"an inactive lease.");
			}
			}

			public static function waitForLeases(array $leases) {
			assert_instances_of($leases, 'DrydockLease');

			$task_ids = array_filter(mpull($leases, 'getTaskID'));

			PhabricatorWorker::waitForTasks($task_ids);

			$unresolved = $leases;
			while (true) {
			foreach ($unresolved as $key => $lease) {
			$lease->reload();
			switch ($lease->getStatus()) {
			case DrydockLeaseStatus::STATUS_ACTIVE:
			unset($unresolved[$key]);
			break;
			case DrydockLeaseStatus::STATUS_RELEASED:
			throw new Exception("Lease has already been released!");
			case DrydockLeaseStatus::STATUS_EXPIRED:
			throw new Exception("Lease has already expired!");
			case DrydockLeaseStatus::STATUS_BROKEN:
			throw new Exception("Lease has been broken!");
			case DrydockLeaseStatus::STATUS_PENDING:
			case DrydockLeaseStatus::STATUS_ACQUIRING:
			break;
			}
			}

			if ($unresolved) {
			sleep(1);
			} else {
			break;
			}
			}

			foreach ($leases as $lease) {
			$lease->attachResource($lease->loadResource());
			}
			}

			public function waitUntilActive() {
			if (!$this->getID()) {
			$this->queueForActivation();
			}

			self::waitForLeases(array($this));
			return $this;
			}

			}

phab/src/applications/drydock/storage/DrydockLease.php1f53017f1f46feature/testing

DrydockLease.phpView Options

phab/src/applications/drydock/storage/DrydockLease.php
1f53017f1f46feature/testing

DrydockLease.php
View Options