org.hibernate.search.query.collector.impl.FacetCounter Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of infinispan-embedded-query

Infinispan Embedded Query All-in-One module

There is a newer version: 9.1.7.Final

/*
 * 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 .
 */
package org.hibernate.search.query.collector.impl;

import java.util.List;
import java.util.Map;

import org.hibernate.search.exception.AssertionFailure;
import org.hibernate.search.query.dsl.impl.FacetRange;
import org.hibernate.search.query.dsl.impl.RangeFacetRequest;

import static org.hibernate.search.util.impl.CollectionHelper.newHashMap;

/**
 * @author Hardy Ferentschik
 */
public abstract class FacetCounter {
	private Map counts = newHashMap();

	Map getCounts() {
		return counts;
	}

	void initCount(String value) {
		if ( !counts.containsKey( value ) ) {
			counts.put( value, new IntegerWrapper() );
		}
	}

	void incrementCount(String value) {
		IntegerWrapper integerWrapper = counts.get( value );
		if ( integerWrapper == null ) {
			integerWrapper = new IntegerWrapper();
			counts.put( value, integerWrapper );
		}
		integerWrapper.incrementCount();
	}

	abstract void countValue(Object value);

	public static class SimpleFacetCounter extends FacetCounter {
		@Override
		void countValue(Object value) {
			if ( !( value instanceof String[] ) ) {
				throw new AssertionFailure( "Unexpected field value type " + value.getClass() );
			}
			String[] values = (String[]) value;
			for ( String stringValue : values ) {
				incrementCount( stringValue );
			}
		}
	}

	public static class RangeFacetCounter extends FacetCounter {
		private final List> ranges;
		private final Class fieldCacheType;

		RangeFacetCounter(RangeFacetRequest request) {
			this.fieldCacheType = request.getFieldCacheType();
			this.ranges = request.getFacetRangeList();
			for ( FacetRange range : ranges ) {
				initCount( range.getRangeString() );
			}
		}

		@Override
		@SuppressWarnings("unchecked")
		void countValue(Object value) {
			for ( FacetRange range : ranges ) {
				if ( String[].class.equals( fieldCacheType ) ) {
					String[] stringValues = (String[]) value;
					for ( String stringValue : stringValues ) {
						countIfInRange( (T) stringValue, range );
					}
				}
				else {
					countIfInRange( (T) value, range );
				}
			}
		}

		private void countIfInRange(T value, FacetRange range) {
			if ( range.isInRange( value ) ) {
				incrementCount( range.getRangeString() );
			}
		}
	}
}