Differential D17384 Diff 42244 src/applications/search/management/PhabricatorSearchManagementInitWorkflow.php

Changeset View

Standalone View

src/applications/search/management/PhabricatorSearchManagementInitWorkflow.php

	<?php			<?php

	final class PhabricatorSearchManagementInitWorkflow			final class PhabricatorSearchManagementInitWorkflow
	extends PhabricatorSearchManagementWorkflow {			extends PhabricatorSearchManagementWorkflow {

	protected function didConstruct() {			protected function didConstruct() {
	$this			$this
	->setName('init')			->setName('init')
	->setSynopsis(pht('Initialize or repair an index.'))			->setSynopsis(pht('Initialize or repair an index.'))
	->setExamples('init');			->setExamples('init');
	}			}

	public function execute(PhutilArgumentParser $args) {			public function execute(PhutilArgumentParser $args) {
	$console = PhutilConsole::getConsole();			$console = PhutilConsole::getConsole();

	$engine = PhabricatorFulltextStorageEngine::loadEngine();

	$work_done = false;			$work_done = false;
				foreach (PhabricatorSearchService::getAllServices() as $service) {
				$console->writeOut(
				"%s\n",
				pht('Initializing search service "%s"', $service->getDisplayName()));

				try {
				$host = $service->getAnyHostForRole('write');
				} catch (PhabricatorClusterNoHostForRoleException $e) {
				// If there are no writable hosts for a given cluster, skip it
				$console->writeOut("%s\n", $e->getMessage());
				epriestleyUnsubmitted Done Inline Actions In this case (CLI script with noncritical exception), it's OK to just `phlog($e);`. epriestley: In this case (CLI script with noncritical exception), it's OK to just `phlog($e);`.
				continue;
				}

				$engine = $host->getEngine();

	if (!$engine->indexExists()) {			if (!$engine->indexExists()) {
	$console->writeOut(			$console->writeOut(
	'%s',			'%s',
	pht('Index does not exist, creating...'));			pht('Index does not exist, creating...'));
	$engine->initIndex();			$engine->initIndex();
	$console->writeOut(			$console->writeOut(
	"%s\n",			"%s\n",
	pht('done.'));			pht('done.'));
	$work_done = true;			$work_done = true;
	} else if (!$engine->indexIsSane()) {			} else if (!$engine->indexIsSane()) {
	$console->writeOut(			$console->writeOut(
	'%s',			'%s',
	pht('Index exists but is incorrect, fixing...'));			pht('Index exists but is incorrect, fixing...'));
	$engine->initIndex();			$engine->initIndex();
	$console->writeOut(			$console->writeOut(
	"%s\n",			"%s\n",
	pht('done.'));			pht('done.'));
	$work_done = true;			$work_done = true;
	}			}
				}

	if ($work_done) {			if ($work_done) {
	$console->writeOut(			$console->writeOut(
	"%s\n",			"%s\n",
	pht(			pht(
	'Index maintenance complete. Run `%s` to reindex documents',			'Index maintenance complete. Run `%s` to reindex documents',
	'./bin/search index'));			'./bin/search index'));
	} else {			} else {
	$console->writeOut(			$console->writeOut(
	"%s\n",			"%s\n",
	pht('Nothing to do.'));			pht('Nothing to do.'));
	}			}
	}			}
	}			}