deblan/FOSElasticaBundle
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

Add explain capabilities to the search command

Browse source
This commit is contained in:
ornicar 2011-04-25 12:31:43 -07:00
parent 42688c637e
commit d1407950b5
1 changed files with 11 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
Command/SearchCommand.php
View file

@ -29,6 +29,7 @@ class SearchCommand extends Command
->addOption('index', null, InputOption::VALUE_NONE, 'The index to search in')
->addOption('limit', null, InputOption::VALUE_REQUIRED, 'The maximum number of documents to return', 20)
->addOption('show-source', null, InputOption::VALUE_NONE, 'Show the documents sources')
->addOption('explain', null, InputOption::VALUE_NONE, 'Enables explanation for each hit on how its score was computed.')
->setName('foq:elastica:search')
->setDescription('Searches documents in a given type and index');
}
@ -42,22 +43,27 @@ class SearchCommand extends Command
$type = $index->getType($input->getArgument('type'));
$query = Elastica_Query::create($input->getArgument('query'));
$query->setLimit($input->getOption('limit'));
if ($input->getOption('explain')) {
$query->setExplain(true);
}
$resultSet = $type->search($query);
$output->writeLn(sprintf('Found %d results', $type->count($query)));
foreach ($resultSet->getResults() as $result) {
$output->writeLn($this->formatResult($result, $input->getOption('show-source')));
$output->writeLn($this->formatResult($result, $input->getOption('show-source'), $input->getOption('explain')));
}
}
protected function formatResult(Elastica_Result $result, $showSource)
protected function formatResult(Elastica_Result $result, $showSource, $explain)
{
$source = $result->getSource();
$string = sprintf('[%0.2f] %s', $result->getScore(), var_export(reset($source), true));
if ($showSource) {
$string = sprintf('[%0.2f] %s %s', $result->getScore(), var_export(reset($source), true), json_encode($source));
} else {
$string = sprintf('[%0.2f] %s', $result->getScore(), var_export(reset($source), true));
$string = sprintf('%s %s', $string, json_encode($source));
}
if ($explain) {
$string = sprintf('%s %s', $string, json_encode($result->getExplanation()));
}
return $string;