/* * Hibernate Search, full-text search for your domain model * * License: GNU Lesser General Public License (LGPL), version 2.1 or later * See the lgpl.txt file in the root directory or <http://www.gnu.org/licenses/lgpl-2.1.html>. */ package org.hibernate.search.query.dsl.impl; import java.util.List; import org.hibernate.search.engine.metadata.impl.FacetMetadata; import org.hibernate.search.query.facet.Facet; /** * @author Hardy Ferentschik */ public class RangeFacetRequest<T> extends FacetingRequestImpl { private final List<FacetRange<T>> facetRangeList; RangeFacetRequest(String name, String fieldName, List<FacetRange<T>> facetRanges) { super( name, fieldName ); if ( facetRanges == null || facetRanges.isEmpty() ) { throw new IllegalArgumentException( "At least one facet range must be specified" ); } this.facetRangeList = facetRanges; } public List<FacetRange<T>> getFacetRangeList() { return facetRangeList; } @Override public Class<?> getFacetValueType() { // safe since we have at least one facet range set Object o = facetRangeList.get( 0 ).getMin(); if ( o == null ) { o = facetRangeList.get( 0 ).getMax(); } return o.getClass(); } @Override public Facet createFacet(FacetMetadata facetMetadata, String value, int count) { int facetIndex = findFacetRangeIndex( value ); FacetRange<T> range = facetRangeList.get( facetIndex ); return new RangeFacetImpl<>( getFacetingName(), getFieldName(), facetMetadata.getSourceField().getAbsoluteName(), range, count, facetIndex ); } @Override public String toString() { return "RangeFacetRequest{" + "facetRangeList=" + facetRangeList + "} " + super.toString(); } private int findFacetRangeIndex(String value) { int index = 0; for ( FacetRange<T> facetRange : facetRangeList ) { if ( facetRange.getRangeString().equals( value ) ) { return index; } index++; } return -1; } }