[Summary view] [Print] [Text view]

   1  <?php
   2  
   3  final class FundBackerQuery
   4    extends PhabricatorCursorPagedPolicyAwareQuery {
   5  
   6    private $ids;
   7    private $phids;
   8    private $statuses;
   9  
  10    private $initiativePHIDs;
  11    private $backerPHIDs;
  12  
  13    public function withIDs(array $ids) {
  14      $this->ids = $ids;
  15      return $this;
  16    }
  17  
  18    public function withPHIDs(array $phids) {
  19      $this->phids = $phids;
  20      return $this;
  21    }
  22  
  23    public function withStatuses(array $statuses) {
  24      $this->statuses = $statuses;
  25      return $this;
  26    }
  27  
  28    public function withInitiativePHIDs(array $phids) {
  29      $this->initiativePHIDs = $phids;
  30      return $this;
  31    }
  32  
  33    public function withBackerPHIDs(array $phids) {
  34      $this->backerPHIDs = $phids;
  35      return $this;
  36    }
  37  
  38    protected function loadPage() {
  39      $table = new FundBacker();
  40      $conn_r = $table->establishConnection('r');
  41  
  42      $rows = queryfx_all(
  43        $conn_r,
  44        'SELECT * FROM %T %Q %Q %Q',
  45        $table->getTableName(),
  46        $this->buildWhereClause($conn_r),
  47        $this->buildOrderClause($conn_r),
  48        $this->buildLimitClause($conn_r));
  49  
  50      return $table->loadAllFromArray($rows);
  51    }
  52  
  53    protected function willFilterPage(array $backers) {
  54      $initiative_phids = mpull($backers, 'getInitiativePHID');
  55      $initiatives = id(new PhabricatorObjectQuery())
  56        ->setParentQuery($this)
  57        ->setViewer($this->getViewer())
  58        ->withPHIDs($initiative_phids)
  59        ->execute();
  60      $initiatives = mpull($initiatives, null, 'getPHID');
  61  
  62      foreach ($backers as $backer) {
  63        $initiative_phid = $backer->getInitiativePHID();
  64        $initiative = idx($initiatives, $initiative_phid);
  65        $backer->attachInitiative($initiative);
  66      }
  67  
  68      return $backers;
  69    }
  70  
  71    private function buildWhereClause(AphrontDatabaseConnection $conn_r) {
  72      $where = array();
  73  
  74      $where[] = $this->buildPagingClause($conn_r);
  75  
  76      if ($this->ids !== null) {
  77        $where[] = qsprintf(
  78          $conn_r,
  79          'id IN (%Ld)',
  80          $this->ids);
  81      }
  82  
  83      if ($this->phids !== null) {
  84        $where[] = qsprintf(
  85          $conn_r,
  86          'phid IN (%Ls)',
  87          $this->phids);
  88      }
  89  
  90      if ($this->initiativePHIDs !== null) {
  91        $where[] = qsprintf(
  92          $conn_r,
  93          'initiativePHID IN (%Ls)',
  94          $this->initiativePHIDs);
  95      }
  96  
  97      if ($this->backerPHIDs !== null) {
  98        $where[] = qsprintf(
  99          $conn_r,
 100          'backerPHID IN (%Ls)',
 101          $this->backerPHIDs);
 102      }
 103  
 104      if ($this->statuses !== null) {
 105        $where[] = qsprintf(
 106          $conn_r,
 107          'status IN (%Ls)',
 108          $this->statuses);
 109      }
 110  
 111      return $this->formatWhereClause($where);
 112    }
 113  
 114    public function getQueryApplicationClass() {
 115      return 'PhabricatorFundApplication';
 116    }
 117  
 118  }