mirror of
https://we.phorge.it/source/phorge.git
synced 2024-11-15 03:12:41 +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
114 lines
2.6 KiB
PHP
114 lines
2.6 KiB
PHP
<?php
|
|
|
|
final class HeraldTranscriptQuery
|
|
extends PhabricatorCursorPagedPolicyAwareQuery {
|
|
|
|
private $ids;
|
|
private $objectPHIDs;
|
|
private $needPartialRecords;
|
|
|
|
public function withIDs(array $ids) {
|
|
$this->ids = $ids;
|
|
return $this;
|
|
}
|
|
|
|
public function withObjectPHIDs(array $phids) {
|
|
$this->objectPHIDs = $phids;
|
|
return $this;
|
|
}
|
|
|
|
public function needPartialRecords($need_partial) {
|
|
$this->needPartialRecords = $need_partial;
|
|
return $this;
|
|
}
|
|
|
|
protected function loadPage() {
|
|
$transcript = new HeraldTranscript();
|
|
$conn_r = $transcript->establishConnection('r');
|
|
|
|
// NOTE: Transcripts include a potentially enormous amount of serialized
|
|
// data, so we're loading only some of the fields here if the caller asked
|
|
// for partial records.
|
|
|
|
if ($this->needPartialRecords) {
|
|
$fields = implode(
|
|
', ',
|
|
array(
|
|
'id',
|
|
'phid',
|
|
'objectPHID',
|
|
'time',
|
|
'duration',
|
|
'dryRun',
|
|
'host',
|
|
));
|
|
} else {
|
|
$fields = '*';
|
|
}
|
|
|
|
$rows = queryfx_all(
|
|
$conn_r,
|
|
'SELECT %Q FROM %T t %Q %Q %Q',
|
|
$fields,
|
|
$transcript->getTableName(),
|
|
$this->buildWhereClause($conn_r),
|
|
$this->buildOrderClause($conn_r),
|
|
$this->buildLimitClause($conn_r));
|
|
|
|
$transcripts = $transcript->loadAllFromArray($rows);
|
|
|
|
if ($this->needPartialRecords) {
|
|
// Make sure nothing tries to write these; they aren't complete.
|
|
foreach ($transcripts as $transcript) {
|
|
$transcript->makeEphemeral();
|
|
}
|
|
}
|
|
|
|
return $transcripts;
|
|
}
|
|
|
|
protected function willFilterPage(array $transcripts) {
|
|
$phids = mpull($transcripts, 'getObjectPHID');
|
|
|
|
$objects = id(new PhabricatorObjectQuery())
|
|
->setViewer($this->getViewer())
|
|
->withPHIDs($phids)
|
|
->execute();
|
|
|
|
foreach ($transcripts as $key => $transcript) {
|
|
if (empty($objects[$transcript->getObjectPHID()])) {
|
|
$this->didRejectResult($transcript);
|
|
unset($transcripts[$key]);
|
|
}
|
|
}
|
|
|
|
return $transcripts;
|
|
}
|
|
|
|
protected function buildWhereClause(AphrontDatabaseConnection $conn_r) {
|
|
$where = array();
|
|
|
|
if ($this->ids) {
|
|
$where[] = qsprintf(
|
|
$conn_r,
|
|
'id IN (%Ld)',
|
|
$this->ids);
|
|
}
|
|
|
|
if ($this->objectPHIDs) {
|
|
$where[] = qsprintf(
|
|
$conn_r,
|
|
'objectPHID in (%Ls)',
|
|
$this->objectPHIDs);
|
|
}
|
|
|
|
$where[] = $this->buildPagingClause($conn_r);
|
|
|
|
return $this->formatWhereClause($where);
|
|
}
|
|
|
|
public function getQueryApplicationClass() {
|
|
return 'PhabricatorHeraldApplication';
|
|
}
|
|
|
|
}
|