[Summary view] [Print] [Text view]

   1  <?php
   2  
   3  final class HarbormasterBuildPlanQuery
   4    extends PhabricatorCursorPagedPolicyAwareQuery {
   5  
   6    private $ids;
   7    private $phids;
   8    private $statuses;
   9  
  10    public function withIDs(array $ids) {
  11      $this->ids = $ids;
  12      return $this;
  13    }
  14  
  15    public function withPHIDs(array $phids) {
  16      $this->phids = $phids;
  17      return $this;
  18    }
  19  
  20    public function withStatuses(array $statuses) {
  21      $this->statuses = $statuses;
  22      return $this;
  23    }
  24  
  25    protected function loadPage() {
  26      $table = new HarbormasterBuildPlan();
  27      $conn_r = $table->establishConnection('r');
  28  
  29      $data = queryfx_all(
  30        $conn_r,
  31        'SELECT * FROM %T %Q %Q %Q',
  32        $table->getTableName(),
  33        $this->buildWhereClause($conn_r),
  34        $this->buildOrderClause($conn_r),
  35        $this->buildLimitClause($conn_r));
  36  
  37      return $table->loadAllFromArray($data);
  38    }
  39  
  40    private function buildWhereClause(AphrontDatabaseConnection $conn_r) {
  41      $where = array();
  42  
  43      if ($this->ids) {
  44        $where[] = qsprintf(
  45          $conn_r,
  46          'id IN (%Ld)',
  47          $this->ids);
  48      }
  49  
  50      if ($this->phids) {
  51        $where[] = qsprintf(
  52          $conn_r,
  53          'phid IN (%Ls)',
  54          $this->phids);
  55      }
  56  
  57      if ($this->statuses) {
  58        $where[] = qsprintf(
  59          $conn_r,
  60          'planStatus IN (%Ls)',
  61          $this->statuses);
  62      }
  63  
  64      $where[] = $this->buildPagingClause($conn_r);
  65  
  66      return $this->formatWhereClause($where);
  67    }
  68  
  69    public function getQueryApplicationClass() {
  70      return 'PhabricatorHarbormasterApplication';
  71    }
  72  
  73  }