public function testPostCreateRequest()
{
// create a very long query
$fq = '';
for ($i = 1; $i <= 1000; $i++) {
$fq .= ' OR price:' . $i;
}
$fq = substr($fq, 4);
$this->query->createFilterQuery('fq')->setQuery($fq);
$requestOutput = $this->client->createRequest($this->query);
$requestInput = clone $requestOutput;
$event = new PostCreateRequestEvent($this->query, $requestOutput);
$this->plugin->postCreateRequest($event);
$this->assertEquals(Request::METHOD_GET, $requestInput->getMethod());
$this->assertEquals(Request::METHOD_POST, $requestOutput->getMethod());
$this->assertEquals($requestInput->getQueryString(), $requestOutput->getRawData());
$this->assertEquals('', $requestOutput->getQueryString());
}
/**
* Adds filter queries configured in TypoScript to $query.
*
* @param \Solarium\QueryType\Select\Query\Query $query
* @return $this
*/
protected function addTypoScriptFilters()
{
if (!empty($this->settings['additionalFilters'])) {
foreach ($this->settings['additionalFilters'] as $key => $filterQuery) {
$this->query->createFilterQuery('additionalFilter-' . $key)->setQuery($filterQuery);
}
}
return $this;
}
public function testCreateFilterQuery()
{
$options = array('optionA' => 1, 'optionB' => 2);
$fq = $this->query->createFilterQuery($options);
// check class
$this->assertThat($fq, $this->isInstanceOf('Solarium\\QueryType\\Select\\Query\\FilterQuery'));
// check option forwarding
$fqOptions = $fq->getOptions();
$this->assertEquals($options['optionB'], $fqOptions['optionB']);
}
protected function createFilterQuery(Query $query)
{
if ($this->expression === NULL) {
return NULL;
}
$this->expression = $this->doInverse($this->expression);
if (($queryString = $this->expression->build()) !== NULL) {
$key = $this->key ?: md5($queryString);
$fq = $query->createFilterQuery($key)->setQuery($queryString);
if ($this->expression instanceof FieldExpression) {
$fq->addTag($this->expression->getField());
}
if ($this->key !== NULL) {
$fq->addTag($this->key);
}
} else {
$fq = NULL;
}
$this->expression = $this->doInverse($this->expression);
//return to original state
return $fq;
}
/**
* @param \Solarium\QueryType\Select\Query\Query $select
*/
public function addKritsToSolr(&$select)
{
foreach ($this->krits as $krit) {
switch ($krit["typ"]) {
case "betreff":
$helper = $select->getHelper();
$select->createFilterQuery("betreff")->setQuery("antrag_betreff:" . $helper->escapeTerm($krit["suchbegriff"]));
break;
case "antrag_typ":
$select->createFilterQuery("antrag_typ")->setQuery("antrag_typ:" . $krit["suchbegriff"]);
break;
case "antrag_wahlperiode":
$select->createFilterQuery("antrag_wahlperiode")->setQuery("antrag_wahlperiode:" . $krit["suchbegriff"]);
break;
case "volltext":
/** @var Solarium\QueryType\Select\Query\Component\DisMax $dismax */
$dismax = $select->getDisMax();
$dismax->setQueryParser('edismax');
$dismax->setQueryFields("text text_ocr");
$select->setQuery($krit["suchbegriff"]);
break;
case "ba":
$select->createFilterQuery("dokument_bas")->setQuery("dokument_bas:" . $krit["ba_nr"]);
break;
case "geo":
$helper = $select->getHelper();
$select->createFilterQuery("geo")->setQuery($helper->geofilt("geo", $krit["lat"], $krit["lng"], $krit["radius"] / 1000));
break;
case "referat":
$helper = $select->getHelper();
$select->createFilterQuery("referat")->setQuery("referat_id:" . $helper->escapeTerm($krit["referat_id"]));
break;
case "antrag_nr":
/** @var Solarium\QueryType\Select\Query\Component\DisMax $dismax */
$dismax = $select->getDisMax();
$dismax->setQueryParser('edismax');
$dismax->setQueryFields("antrag_nr");
$select->setQuery("*" . $krit["suchbegriff"] . "*");
break;
}
}
}
/**
* Adds spatial features to the search query.
*/
public function setSpatial(Query $solarium_query, QueryInterface $query, $spatial_options = array(), $field_names)
{
foreach ($spatial_options as $i => $spatial) {
// reset radius for each option
unset($radius);
if (empty($spatial['field']) || empty($spatial['lat']) || empty($spatial['lon'])) {
continue;
}
$field = $field_names[$spatial['field']];
$escaped_field = SearchApiSolrUtility::escapeFieldName($field);
$point = (double) $spatial['lat'] . ',' . (double) $spatial['lon'];
// Prepare the filter settings.
if (isset($spatial['radius'])) {
$radius = (double) $spatial['radius'];
}
$spatial_method = 'geofilt';
if (isset($spatial['method']) && in_array($spatial['method'], array('geofilt', 'bbox'))) {
$spatial_method = $spatial['method'];
}
$filter_queries = $solarium_query->getFilterQueries();
// Change the fq facet ranges to the correct fq.
foreach ($filter_queries as $key => $filter_query) {
// If the fq consists only of a filter on this field, replace it with
// a range.
$preg_field = preg_quote($escaped_field, '/');
if (preg_match('/^' . $preg_field . ':\\["?(\\*|\\d+(?:\\.\\d+)?)"? TO "?(\\*|\\d+(?:\\.\\d+)?)"?\\]$/', $filter_query, $matches)) {
unset($filter_queries[$key]);
if ($matches[1] && is_numeric($matches[1])) {
$min_radius = isset($min_radius) ? max($min_radius, $matches[1]) : $matches[1];
}
if (is_numeric($matches[2])) {
// Make the radius tighter accordingly.
$radius = isset($radius) ? min($radius, $matches[2]) : $matches[2];
}
}
}
// If either a radius was given in the option, or a filter was
// encountered, set a filter for the lowest value. If a lower boundary
// was set (too), we can only set a filter for that if the field name
// doesn't contains any colons.
if (isset($min_radius) && strpos($field, ':') === FALSE) {
$upper = isset($radius) ? " u={$radius}" : '';
$solarium_query->createFilterQuery($field)->setQuery("{!frange l={$min_radius}{$upper}}geodist({$field},{$point})");
} elseif (isset($radius)) {
$solarium_query->createFilterQuery($field)->setQuery("{!{$spatial_method} pt={$point} sfield={$field} d={$radius}}");
}
// @todo: Check if this object returns the correct value
$sorts = $solarium_query->getSorts();
// Change sort on the field, if set (and not already changed).
if (isset($sorts[$spatial['field']]) && substr($sorts[$spatial['field']], 0, strlen($field)) === $field) {
$sorts[$spatial['field']] = str_replace($field, "geodist({$field},{$point})", $sorts[$spatial['field']]);
}
// Change the facet parameters for spatial fields to return distance
// facets.
$facets = $solarium_query->getFacetSet();
// @todo: Fix this so it takes it from the solarium query
if (!empty($facets)) {
if (!empty($facet_params['facet.field'])) {
$facet_params['facet.field'] = array_diff($facet_params['facet.field'], array($field));
}
foreach ($facets as $delta => $facet) {
if ($facet['field'] != $spatial['field']) {
continue;
}
$steps = $facet['limit'] > 0 ? $facet['limit'] : 5;
$step = (isset($radius) ? $radius : 100) / $steps;
for ($k = $steps - 1; $k > 0; --$k) {
$distance = $step * $k;
$key = "spatial-{$delta}-{$distance}";
$facet_params['facet.query'][] = "{!{$spatial_method} pt={$point} sfield={$field} d={$distance} key={$key}}";
}
foreach (array('limit', 'mincount', 'missing') as $setting) {
unset($facet_params["f.{$field}.facet.{$setting}"]);
}
}
}
}
// Normal sorting on location fields isn't possible.
foreach (array_keys($solarium_query->getSorts()) as $sort) {
if (substr($sort, 0, 3) === 'loc') {
$solarium_query->removeSort($sort);
}
}
}
/**
* Helper method for creating the facet field parameters.
*/
protected function setFacets(array $facets, array $field_names, Query $solarium_query)
{
$fq = array();
if (!$facets) {
return;
}
$facet_set = $solarium_query->getFacetSet();
$facet_set->setSort('count');
$facet_set->setLimit(10);
$facet_set->setMinCount(1);
$facet_set->setMissing(FALSE);
$taggedFields = array();
foreach ($facets as $info) {
if (empty($field_names[$info['field']])) {
continue;
}
// String fields have their own corresponding facet fields.
$field = $field_names[$info['field']];
// Check for the "or" operator.
if (isset($info['operator']) && $info['operator'] === 'or') {
// Remember that filters for this field should be tagged.
$escaped = SearchApiSolrUtility::escapeFieldName($field_names[$info['field']]);
$taggedFields[$escaped] = "{!tag={$escaped}}";
// Add the facet field.
$facet_field = $facet_set->createFacetField($field)->setField("{!ex={$escaped}}{$field}");
} else {
// Add the facet field.
$facet_field = $facet_set->createFacetField($field)->setField($field);
}
// Set limit, unless it's the default.
if ($info['limit'] != 10) {
$limit = $info['limit'] ? $info['limit'] : -1;
$facet_field->setLimit($limit);
}
// Set mincount, unless it's the default.
if ($info['min_count'] != 1) {
$facet_field->setMinCount($info['min_count']);
}
// Set missing, if specified.
if ($info['missing']) {
$facet_field->setMissing(TRUE);
}
}
// Tag filters of fields with "OR" facets.
foreach ($taggedFields as $field => $tag) {
$regex = '#(?<![^( ])' . preg_quote($field, '#') . ':#';
foreach ($fq as $i => $conditions) {
// Solr can't handle two tags on the same filter, so we don't add two.
// Another option here would even be to remove the other tag, too,
// since we can be pretty sure that this filter does not originate from
// a facet – however, wrong results would still be possible, and this is
// definitely an edge case, so don't bother.
if (preg_match($regex, $conditions) && substr($conditions, 0, 6) != '{!tag=') {
$fq[$i] = $tag . $conditions;
}
}
}
foreach ($fq as $key => $conditions_query) {
$solarium_query->createFilterQuery('facets_' . $key)->setQuery($conditions_query);
}
}
/**
* Adds filter queries configured in TypoScript to $query.
*
* @param \Solarium\QueryType\Select\Query\Query $query
*/
private function addTypoScriptFilters($query)
{
if (!empty($this->templateVariableContainer->get('settings')['additionalFilters'])) {
foreach ($this->templateVariableContainer->get('settings')['additionalFilters'] as $key => $filterQuery) {
$query->createFilterQuery('additionalFilter-' . $key)->setQuery($filterQuery);
}
}
}
