phorge / src / infrastructure / storage / connection / AphrontDatabaseConnection.php

<?php

/**
 * @task  xaction Transaction Management
 */
abstract class AphrontDatabaseConnection
  extends Phobject
  implements PhutilQsprintfInterface {

  private $transactionState;
  private $readOnly;
  private $queryTimeout;
  private $locks = array();
  private $lastActiveEpoch;
  private $persistent;

  abstract public function getInsertID();
  abstract public function getAffectedRows();
  abstract public function selectAllResults();
  abstract public function executeQuery(PhutilQueryString $query);
  abstract public function executeRawQueries(array $raw_queries);
  abstract public function close();
  abstract public function openConnection();

  public function __destruct() {
    // NOTE: This does not actually close persistent connections: PHP maintains
    // them in the connection pool.
    $this->close();
  }

  final public function setLastActiveEpoch($epoch) {
    $this->lastActiveEpoch = $epoch;
    return $this;
  }

  final public function getLastActiveEpoch() {
    return $this->lastActiveEpoch;
  }

  final public function setPersistent($persistent) {
    $this->persistent = $persistent;
    return $this;
  }

  final public function getPersistent() {
    return $this->persistent;
  }

  public function queryData($pattern/* , $arg, $arg, ... */) {
    $args = func_get_args();
    array_unshift($args, $this);
    return call_user_func_array('queryfx_all', $args);
  }

  public function query($pattern/* , $arg, $arg, ... */) {
    $args = func_get_args();
    array_unshift($args, $this);
    return call_user_func_array('queryfx', $args);
  }

  public function setReadOnly($read_only) {
    $this->readOnly = $read_only;
    return $this;
  }

  public function getReadOnly() {
    return $this->readOnly;
  }

  public function setQueryTimeout($query_timeout) {
    $this->queryTimeout = $query_timeout;
    return $this;
  }

  public function getQueryTimeout() {
    return $this->queryTimeout;
  }

  public function asyncQuery($raw_query) {
    throw new Exception(pht('Async queries are not supported.'));
  }

  public static function resolveAsyncQueries(array $conns, array $asyncs) {
    throw new Exception(pht('Async queries are not supported.'));
  }

  /**
   * Is this connection idle and safe to close?
   *
   * A connection is "idle" if it can be safely closed without loss of state.
   * Connections inside a transaction or holding locks are not idle, even
   * though they may not actively be executing queries.
   *
   * @return bool True if the connection is idle and can be safely closed.
   */
  public function isIdle() {
    if ($this->isInsideTransaction()) {
      return false;
    }

    if ($this->isHoldingAnyLock()) {
      return false;
    }

    return true;
  }


/* -(  Global Locks  )------------------------------------------------------- */


  public function rememberLock($lock) {
    if (isset($this->locks[$lock])) {
      throw new Exception(
        pht(
          'Trying to remember lock "%s", but this lock has already been '.
          'remembered.',
          $lock));
    }

    $this->locks[$lock] = true;
    return $this;
  }


  public function forgetLock($lock) {
    if (empty($this->locks[$lock])) {
      throw new Exception(
        pht(
          'Trying to forget lock "%s", but this connection does not remember '.
          'that lock.',
          $lock));
    }

    unset($this->locks[$lock]);
    return $this;
  }


  public function forgetAllLocks() {
    $this->locks = array();
    return $this;
  }


  public function isHoldingAnyLock() {
    return (bool)$this->locks;
  }


/* -(  Transaction Management  )--------------------------------------------- */


  /**
   * Begin a transaction, or set a savepoint if the connection is already
   * transactional.
   *
   * @return $this
   * @task xaction
   */
  public function openTransaction() {
    $state = $this->getTransactionState();
    $point = $state->getSavepointName();
    $depth = $state->getDepth();

    $new_transaction = ($depth == 0);
    if ($new_transaction) {
      $this->query('START TRANSACTION');
    } else {
      $this->query('SAVEPOINT '.$point);
    }

    $state->increaseDepth();

    return $this;
  }


  /**
   * Commit a transaction, or stage a savepoint for commit once the entire
   * transaction completes if inside a transaction stack.
   *
   * @return $this
   * @task xaction
   */
  public function saveTransaction() {
    $state = $this->getTransactionState();
    $depth = $state->decreaseDepth();

    if ($depth == 0) {
      $this->query('COMMIT');
    }

    return $this;
  }


  /**
   * Rollback a transaction, or unstage the last savepoint if inside a
   * transaction stack.
   *
   * @return $this
   */
  public function killTransaction() {
    $state = $this->getTransactionState();
    $depth = $state->decreaseDepth();

    if ($depth == 0) {
      $this->query('ROLLBACK');
    } else {
      $this->query('ROLLBACK TO SAVEPOINT '.$state->getSavepointName());
    }

    return $this;
  }


  /**
   * Returns true if the connection is transactional.
   *
   * @return bool True if the connection is currently transactional.
   * @task xaction
   */
  public function isInsideTransaction() {
    $state = $this->getTransactionState();
    return ($state->getDepth() > 0);
  }


  /**
   * Get the current @{class:AphrontDatabaseTransactionState} object, or create
   * one if none exists.
   *
   * @return AphrontDatabaseTransactionState Current transaction state.
   * @task xaction
   */
  protected function getTransactionState() {
    if (!$this->transactionState) {
      $this->transactionState = new AphrontDatabaseTransactionState();
    }
    return $this->transactionState;
  }


  /**
   * @task xaction
   */
  public function beginReadLocking() {
    $this->getTransactionState()->beginReadLocking();
    return $this;
  }


  /**
   * @task xaction
   */
  public function endReadLocking() {
    $this->getTransactionState()->endReadLocking();
    return $this;
  }


  /**
   * @task xaction
   */
  public function isReadLocking() {
    return $this->getTransactionState()->isReadLocking();
  }


  /**
   * @task xaction
   */
  public function beginWriteLocking() {
    $this->getTransactionState()->beginWriteLocking();
    return $this;
  }


  /**
   * @task xaction
   */
  public function endWriteLocking() {
    $this->getTransactionState()->endWriteLocking();
    return $this;
  }


  /**
   * @task xaction
   */
  public function isWriteLocking() {
    return $this->getTransactionState()->isWriteLocking();
  }

}