Changeset View
Changeset View
Standalone View
Standalone View
src/applications/auth/query/PhabricatorAuthSessionQuery.php
- This file was added.
<?php | |||||
final class PhabricatorAuthSessionQuery | |||||
extends PhabricatorCursorPagedPolicyAwareQuery { | |||||
private $identityPHIDs; | |||||
public function withIdentityPHIDs(array $identity_phids) { | |||||
$this->identityPHIDs = $identity_phids; | |||||
return $this; | |||||
} | |||||
protected function loadPage() { | |||||
$table = new PhabricatorAuthSession(); | |||||
$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 $sessions) { | |||||
$identity_phids = mpull($sessions, 'getUserPHID'); | |||||
$identity_objects = id(new PhabricatorObjectQuery()) | |||||
->setViewer($this->getViewer()) | |||||
->setParentQuery($this) | |||||
->withPHIDs($identity_phids) | |||||
->execute(); | |||||
$identity_objects = mpull($identity_objects, null, 'getPHID'); | |||||
foreach ($sessions as $key => $session) { | |||||
$identity_object = idx($identity_objects, $session->getUserPHID()); | |||||
if (!$identity_object) { | |||||
unset($sessions[$key]); | |||||
} else { | |||||
$session->attachIdentityObject($identity_object); | |||||
} | |||||
} | |||||
return $sessions; | |||||
} | |||||
protected function buildWhereClause(AphrontDatabaseConnection $conn_r) { | |||||
$where = array(); | |||||
if ($this->identityPHIDs) { | |||||
$where[] = qsprintf( | |||||
$conn_r, | |||||
'userPHID IN (%Ls)', | |||||
$this->identityPHIDs); | |||||
} | |||||
$where[] = $this->buildPagingClause($conn_r); | |||||
return $this->formatWhereClause($where); | |||||
} | |||||
public function getPagingColumn() { | |||||
return 'sessionKey'; | |||||
} | |||||
public function getQueryApplicationClass() { | |||||
return 'PhabricatorApplicationAuth'; | |||||
} | |||||
} |