/*
* Copyright 2012 -2014 the original author or authors.
*
* 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.springframework.data.solr.repository.query;
import org.springframework.data.mapping.context.MappingContext;
import org.springframework.data.repository.query.parser.PartTree;
import org.springframework.data.solr.core.SolrOperations;
import org.springframework.data.solr.core.mapping.SolrPersistentProperty;
import org.springframework.data.solr.core.query.Query;
/**
* Solr specific implementation of a query derived from method name
*
* @author Christoph Strobl
*/
public class PartTreeSolrQuery extends AbstractSolrQuery {
private final PartTree tree;
private final MappingContext<?, SolrPersistentProperty> mappingContext;
public PartTreeSolrQuery(SolrQueryMethod method, SolrOperations solrOperations) {
this(null, method, solrOperations);
}
public PartTreeSolrQuery(String collection, SolrQueryMethod method, SolrOperations solrOperations) {
super(collection, solrOperations, method);
this.tree = new PartTree(method.getName(), method.getEntityInformation().getJavaType());
this.mappingContext = solrOperations.getConverter().getMappingContext();
}
public PartTree getTree() {
return tree;
}
@Override
protected Query createQuery(SolrParameterAccessor parameterAccessor) {
Query query = new SolrQueryCreator(tree, parameterAccessor, mappingContext).createQuery();
appendProjection(query);
return query;
}
/**
* @see PartTree#isCountProjection()
* @since 1.2
*/
public boolean isCountQuery() {
return tree.isCountProjection();
}
/**
* @see PartTree#isDelete()
* @since 1.2
*/
@Override
public boolean isDeleteQuery() {
return tree.isDelete();
}
/**
* @see PartTree#isLimiting()
* @since 1.3
*/
@Override
public boolean isLimiting() {
return tree.isLimiting();
}
@Override
public int getLimit() {
if (isLimiting()) {
return this.tree.getMaxResults();
}
return super.getLimit();
}
}