/*
* Copyright (c) 2011, Rickard Öberg. All Rights Reserved.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*
*/
package org.qi4j.index.solr.internal;
import java.lang.reflect.Member;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import org.apache.solr.client.solrj.SolrServer;
import org.apache.solr.client.solrj.SolrServerException;
import org.apache.solr.client.solrj.response.QueryResponse;
import org.apache.solr.common.SolrDocument;
import org.apache.solr.common.SolrDocumentList;
import org.apache.solr.common.params.SolrParams;
import org.apache.solr.common.util.NamedList;
import org.qi4j.api.common.Optional;
import org.qi4j.api.composite.Composite;
import org.qi4j.api.entity.EntityReference;
import org.qi4j.api.injection.scope.Service;
import org.qi4j.api.query.grammar.OrderBy;
import org.qi4j.api.query.grammar.QuerySpecification;
import org.qi4j.functional.Iterables;
import org.qi4j.functional.Specification;
import org.qi4j.index.solr.EmbeddedSolrService;
import org.qi4j.index.solr.SolrSearch;
import org.qi4j.spi.query.EntityFinder;
import org.qi4j.spi.query.EntityFinderException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
/**
* JAVADOC
*/
public class SolrEntityQueryMixin
implements EntityFinder, SolrSearch
{
@Service
private EmbeddedSolrService solr;
private Logger logger = LoggerFactory.getLogger( SolrEntityQueryMixin.class );
@Override
public Iterable<EntityReference> findEntities( Class<?> resultType, @Optional Specification<Composite> whereClause, @Optional OrderBy[] orderBySegments, @Optional Integer firstResult, @Optional Integer maxResults, Map<String, Object> variables ) throws EntityFinderException
{
try
{
QuerySpecification expr = (QuerySpecification) whereClause;
SolrServer server = solr.solrServer();
NamedList<Object> list = new NamedList<Object>();
list.add( "q", expr.query() );
list.add( "rows", maxResults != 0 ? maxResults : 10000 );
list.add( "start", firstResult );
if( orderBySegments != null && orderBySegments.length > 0 )
{
for( OrderBy orderBySegment : orderBySegments )
{
String propName = ((Member)orderBySegment.property().accessor()).getName() + "_for_sort";
String order = orderBySegment.order() == OrderBy.Order.ASCENDING ? "asc" : "desc";
list.add( "sort", propName + " " + order );
}
}
SolrParams solrParams = SolrParams.toSolrParams( list );
logger.debug( "Search:" + list.toString() );
QueryResponse query = server.query( solrParams );
SolrDocumentList results = query.getResults();
List<EntityReference> references = new ArrayList<EntityReference>( results.size() );
for( SolrDocument result : results )
{
references.add( EntityReference.parseEntityReference( result.getFirstValue( "id" ).toString() ) );
}
return references;
} catch( SolrServerException e )
{
throw new EntityFinderException( e );
}
}
@Override
public EntityReference findEntity( Class<?> resultType, @Optional Specification<Composite> whereClause, Map<String, Object> variables ) throws EntityFinderException
{
Iterator<EntityReference> iter = findEntities( resultType, whereClause, null, 0, 1, variables ).iterator();
if( iter.hasNext() )
return iter.next();
else
return null;
}
@Override
public long countEntities( Class<?> resultType, @Optional Specification<Composite> whereClause, Map<String, Object> variables ) throws EntityFinderException
{
return Iterables.count( findEntities( resultType, whereClause, null, 0, 1, variables ) );
}
@Override
public SolrDocumentList search( String queryString ) throws SolrServerException
{
SolrServer server = solr.solrServer();
NamedList<Object> list = new NamedList<Object>();
list.add( "q", queryString );
QueryResponse query = server.query( SolrParams.toSolrParams( list ) );
return query.getResults();
}
}