PhabricatorRepositoryManagementParentsWorkflow.php
View Options

	<?php

	final class PhabricatorRepositoryManagementParentsWorkflow
	extends PhabricatorRepositoryManagementWorkflow {

	protected function didConstruct() {
	$this
	->setName('parents')
	->setExamples('parents [options] [__repository__] ...')
	->setSynopsis(
	pht(
	'Build parent caches in repositories that are missing the data, '.
	'or rebuild them in a specific __repository__.'))
	->setArguments(
	array(
	array(
	'name' => 'repos',
	'wildcard' => true,
	),
	));
	}

	public function execute(PhutilArgumentParser $args) {
	$repos = $this->loadRepositories($args, 'repos');
	if (!$repos) {
	$repos = id(new PhabricatorRepositoryQuery())
	->setViewer($this->getViewer())
	->execute();
	}

	$console = PhutilConsole::getConsole();
	foreach ($repos as $repo) {
	$monogram = $repo->getMonogram();
	if ($repo->isSVN()) {
	$console->writeOut(
	"%s\n",
	pht(
	'Skipping "%s": Subversion repositories do not require this '.
	'cache to be built.',
	$monogram));
	continue;
	}
	$this->rebuildRepository($repo);
	}

	return 0;
	}

	private function rebuildRepository(PhabricatorRepository $repo) {
	$console = PhutilConsole::getConsole();
	$console->writeOut("%s\n", pht('Rebuilding "%s"...', $repo->getMonogram()));

	$refs = id(new PhabricatorRepositoryRefCursorQuery())
	->setViewer($this->getViewer())
	->withRefTypes(array(PhabricatorRepositoryRefCursor::TYPE_BRANCH))
	->withRepositoryPHIDs(array($repo->getPHID()))
	->needPositions(true)
	->execute();

	$graph = array();
	foreach ($refs as $ref) {
	if (!$repo->shouldTrackBranch($ref->getRefName())) {
	continue;
	}

	$console->writeOut(
	"%s\n",
	pht('Rebuilding branch "%s"...', $ref->getRefName()));

	foreach ($ref->getPositionIdentifiers() as $commit) {
	if ($repo->isGit()) {
	$stream = new PhabricatorGitGraphStream($repo, $commit);
	} else {
	$stream = new PhabricatorMercurialGraphStream($repo, $commit);
	}

	$discover = array($commit);
	while ($discover) {
	$target = array_pop($discover);
	if (isset($graph[$target])) {
	continue;
	}
	$graph[$target] = $stream->getParents($target);
	foreach ($graph[$target] as $parent) {
	$discover[] = $parent;
	}
	}
	}
	}

	$console->writeOut(
	"%s\n",
	pht(
	'Found %s total commit(s); updating...',
	phutil_count($graph)));

	$commit_table = id(new PhabricatorRepositoryCommit());
	$commit_table_name = $commit_table->getTableName();
	$conn_w = $commit_table->establishConnection('w');

	$bar = id(new PhutilConsoleProgressBar())
	->setTotal(count($graph));

	$need = array();
	foreach ($graph as $child => $parents) {
	foreach ($parents as $parent) {
	$need[$parent] = $parent;
	}
	$need[$child] = $child;
	}

	$map = array();
	foreach (array_chunk($need, 2048) as $chunk) {
	$rows = queryfx_all(
	$conn_w,
	'SELECT id, commitIdentifier FROM %T
	WHERE commitIdentifier IN (%Ls) AND repositoryID = %d',
	$commit_table_name,
	$chunk,
	$repo->getID());
	foreach ($rows as $row) {
	$map[$row['commitIdentifier']] = $row['id'];
	}
	}

	$insert_sql = array();
	$delete_sql = array();

	foreach ($graph as $child => $parents) {
	$names = $parents;
	$names[] = $child;

	foreach ($names as $name) {
	if (empty($map[$name])) {
	throw new Exception(pht('Unknown commit "%s"!', $name));
	}
	}

	if (!$parents) {
	// Write an explicit 0 to indicate "no parents" instead of "no data".
	$insert_sql[] = qsprintf(
	$conn_w,
	'(%d, 0)',
	$map[$child]);
	} else {
	foreach ($parents as $parent) {
	$insert_sql[] = qsprintf(
	$conn_w,
	'(%d, %d)',
	$map[$child],
	$map[$parent]);
	}
	}

	$delete_sql[] = $map[$child];

	$bar->update(1);
	}

	$commit_table->openTransaction();
	foreach (PhabricatorLiskDAO::chunkSQL($delete_sql) as $chunk) {
	queryfx(
	$conn_w,
	'DELETE FROM %T WHERE childCommitID IN (%LQ)',
	PhabricatorRepository::TABLE_PARENTS,
	$chunk);
	}

	foreach (PhabricatorLiskDAO::chunkSQL($insert_sql) as $chunk) {
	queryfx(
	$conn_w,
	'INSERT INTO %T (childCommitID, parentCommitID) VALUES %LQ',
	PhabricatorRepository::TABLE_PARENTS,
	$chunk);
	}
	$commit_table->saveTransaction();

	$bar->done();
	}

	}

File Metadata

Mime Type: text/x-php
Expires: Thu, Mar 27, 01:02 (1 w, 3 d ago)
Storage Engine: blob
Storage Format: Raw Data
Storage Handle: 1144085
Default Alt Text: PhabricatorRepositoryManagementParentsWorkflow.php (4 KB)

PhabricatorRepositoryManagementParentsWorkflow.php
No OneTemporary
Actions

PhabricatorRepositoryManagementParentsWorkflow.php
View Options

File Metadata

Event Timeline

PhabricatorRepositoryManagementParentsWorkflow.phpNo OneTemporaryActions

PhabricatorRepositoryManagementParentsWorkflow.phpView Options

File Metadata

Event Timeline

PhabricatorRepositoryManagementParentsWorkflow.php
No OneTemporary
Actions

PhabricatorRepositoryManagementParentsWorkflow.php
View Options