org.hibernate.search.cfg.SearchMapping Maven / Gradle / Ivy

Go to download
/*
 * 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.cfg;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;

import org.apache.lucene.analysis.util.TokenizerFactory;

/**
 * @author Emmanuel Bernard
 */
public class SearchMapping {
	private Set> analyzerDefs = new HashSet>();
	private Set> fullTextFilterDefs = new HashSet>();
	private Map, EntityDescriptor> entities = new HashMap, EntityDescriptor>();

	public Set> getAnalyzerDefs() {
		return analyzerDefs;
	}

	public Set> getFullTextFilterDefs() {
		return fullTextFilterDefs;
	}

	public EntityDescriptor getEntityDescriptor(Class entityType) {
		return entities.get( entityType );
	}

	public Set> getMappedEntities() {
		return entities.keySet();
	}

	public AnalyzerDefMapping analyzerDef(String name, Class tokenizerFactory) {
		return new AnalyzerDefMapping(name, tokenizerFactory, this);
	}

	public EntityMapping entity(Class entityType) {
		return new EntityMapping(entityType, this);
	}

	public FullTextFilterDefMapping fullTextFilterDef(String name, Class impl) {
		return new FullTextFilterDefMapping(this, name, impl );
	}

	/**
	 * eg @Containing(things={@Thing(...), @Thing(...)}
	 * Map addedThing = addElementToAnnotationArray(containing, "things");
	 */

	static Map addElementToAnnotationArray(Map containingAnnotation,
													String attributeName) {
		@SuppressWarnings("unchecked") List> array = (List>) containingAnnotation.get( attributeName );
		if ( array == null ) {
			array = new ArrayList>();
			containingAnnotation.put( attributeName, array );
		}
		Map param = new HashMap();
		array.add( param );
		return param;
	}

	void addAnalyzerDef(Map analyzerDef) {
		analyzerDefs.add( analyzerDef );
	}

	EntityDescriptor getEntity(Class entityType) {
		EntityDescriptor entityDescriptor = entities.get( entityType );
		if ( entityDescriptor == null ) {
			entityDescriptor = new EntityDescriptor( );
			entities.put( entityType, entityDescriptor );
		}
		return entityDescriptor;
	}

	void addFulltextFilterDef(Map fullTextFilterDef) {
		fullTextFilterDefs.add( fullTextFilterDef );
	}

}