mirror of
https://we.phorge.it/source/phorge.git
synced 2024-12-03 20:22:46 +01:00
f5580c7a08
Summary: Ref T4100. Ref T5595. To support a unified "Projects:" query across all applications, a future diff is going to add a set of "Edge Logic" capabilities to `PolicyAwareQuery` which write the required SELECT, JOIN, WHERE, HAVING and GROUP clauses for you. With the addition of "Edge Logic", we'll have three systems which may need to build components of query claues: ordering/paging, customfields/applicationsearch, and edge logic. For most clauses, queries don't currently call into the parent explicitly to get default components. I want to move more query construction logic up the class tree so it can be shared. For most methods, this isn't a problem, but many subclasses define a `buildWhereClause()`. Make all such definitions protected and consistent. This causes no behavioral changes. Test Plan: Ran `arc unit --everything`, which does a pretty through job of verifying this statically. Reviewers: btrahan Reviewed By: btrahan Subscribers: yelirekim, hach-que, epriestley Maniphest Tasks: T4100, T5595 Differential Revision: https://secure.phabricator.com/D12453
124 lines
3 KiB
PHP
124 lines
3 KiB
PHP
<?php
|
|
|
|
final class HarbormasterBuildArtifactQuery
|
|
extends PhabricatorCursorPagedPolicyAwareQuery {
|
|
|
|
private $ids;
|
|
private $buildTargetPHIDs;
|
|
private $artifactTypes;
|
|
private $artifactKeys;
|
|
private $keyBuildPHID;
|
|
private $keyBuildGeneration;
|
|
|
|
public function withIDs(array $ids) {
|
|
$this->ids = $ids;
|
|
return $this;
|
|
}
|
|
|
|
public function withBuildTargetPHIDs(array $build_target_phids) {
|
|
$this->buildTargetPHIDs = $build_target_phids;
|
|
return $this;
|
|
}
|
|
|
|
public function withArtifactTypes(array $artifact_types) {
|
|
$this->artifactTypes = $artifact_types;
|
|
return $this;
|
|
}
|
|
|
|
public function withArtifactKeys(
|
|
$build_phid,
|
|
$build_gen,
|
|
array $artifact_keys) {
|
|
$this->keyBuildPHID = $build_phid;
|
|
$this->keyBuildGeneration = $build_gen;
|
|
$this->artifactKeys = $artifact_keys;
|
|
return $this;
|
|
}
|
|
|
|
protected function loadPage() {
|
|
$table = new HarbormasterBuildArtifact();
|
|
$conn_r = $table->establishConnection('r');
|
|
|
|
$data = queryfx_all(
|
|
$conn_r,
|
|
'SELECT * FROM %T %Q %Q %Q',
|
|
$table->getTableName(),
|
|
$this->buildWhereClause($conn_r),
|
|
$this->buildOrderClause($conn_r),
|
|
$this->buildLimitClause($conn_r));
|
|
|
|
return $table->loadAllFromArray($data);
|
|
}
|
|
|
|
protected function willFilterPage(array $page) {
|
|
$build_targets = array();
|
|
|
|
$build_target_phids = array_filter(mpull($page, 'getBuildTargetPHID'));
|
|
if ($build_target_phids) {
|
|
$build_targets = id(new HarbormasterBuildTargetQuery())
|
|
->setViewer($this->getViewer())
|
|
->withPHIDs($build_target_phids)
|
|
->setParentQuery($this)
|
|
->execute();
|
|
$build_targets = mpull($build_targets, null, 'getPHID');
|
|
}
|
|
|
|
foreach ($page as $key => $build_log) {
|
|
$build_target_phid = $build_log->getBuildTargetPHID();
|
|
if (empty($build_targets[$build_target_phid])) {
|
|
unset($page[$key]);
|
|
continue;
|
|
}
|
|
$build_log->attachBuildTarget($build_targets[$build_target_phid]);
|
|
}
|
|
|
|
return $page;
|
|
}
|
|
|
|
protected function buildWhereClause(AphrontDatabaseConnection $conn_r) {
|
|
$where = array();
|
|
|
|
if ($this->ids) {
|
|
$where[] = qsprintf(
|
|
$conn_r,
|
|
'id IN (%Ld)',
|
|
$this->ids);
|
|
}
|
|
|
|
if ($this->buildTargetPHIDs) {
|
|
$where[] = qsprintf(
|
|
$conn_r,
|
|
'buildTargetPHID IN (%Ls)',
|
|
$this->buildTargetPHIDs);
|
|
}
|
|
|
|
if ($this->artifactTypes) {
|
|
$where[] = qsprintf(
|
|
$conn_r,
|
|
'artifactType in (%Ls)',
|
|
$this->artifactTypes);
|
|
}
|
|
|
|
if ($this->artifactKeys) {
|
|
$indexes = array();
|
|
foreach ($this->artifactKeys as $key) {
|
|
$indexes[] = PhabricatorHash::digestForIndex(
|
|
$this->keyBuildPHID.$this->keyBuildGeneration.$key);
|
|
}
|
|
|
|
$where[] = qsprintf(
|
|
$conn_r,
|
|
'artifactIndex IN (%Ls)',
|
|
$indexes);
|
|
}
|
|
|
|
$where[] = $this->buildPagingClause($conn_r);
|
|
|
|
return $this->formatWhereClause($where);
|
|
}
|
|
|
|
public function getQueryApplicationClass() {
|
|
return 'PhabricatorHarbormasterApplication';
|
|
}
|
|
|
|
}
|