better-testing

PhabricatorWorkerArchiveTask.php
View Options

			<?php

			final class PhabricatorWorkerArchiveTask extends PhabricatorWorkerTask {

			const RESULT_SUCCESS = 0;
			const RESULT_FAILURE = 1;
			const RESULT_CANCELLED = 2;

			protected $duration;
			protected $result;

			public function save() {
			if ($this->getID() === null) {
			throw new Exception(
			"Trying to archive a task with no ID.");
			}

			$other = new PhabricatorWorkerActiveTask();
			$conn_w = $this->establishConnection('w');

			$this->openTransaction();
			queryfx(
			$conn_w,
			'DELETE FROM %T WHERE id = %d',
			$other->getTableName(),
			$this->getID());
			$result = parent::insert();
			$this->saveTransaction();

			return $result;
			}

			public function delete() {
			$this->openTransaction();
			if ($this->getDataID()) {
			$conn_w = $this->establishConnection('w');
			$data_table = new PhabricatorWorkerTaskData();

			queryfx(
			$conn_w,
			'DELETE FROM %T WHERE id = %d',
			$data_table->getTableName(),
			$this->getDataID());
			}

			$result = parent::delete();
			$this->saveTransaction();
			return $result;
			}

			public function unarchiveTask() {
			$this->openTransaction();
			$active = id(new PhabricatorWorkerActiveTask())
			->setID($this->getID())
			->setTaskClass($this->getTaskClass())
			->setLeaseOwner(null)
			->setLeaseExpires(0)
			->setFailureCount(0)
			->setDataID($this->getDataID())
			->insert();

			$this->setDataID(null);
			$this->delete();
			$this->saveTransaction();

			return $active;
			}

			}