org.hibernate.search.analyzer.definition.impl.LuceneTokenFilterDefinitionContextImpl Maven / Gradle / Ivy

/*
 * 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.analyzer.definition.impl;

import org.apache.lucene.analysis.util.CharFilterFactory;
import org.apache.lucene.analysis.util.TokenFilterFactory;
import org.hibernate.annotations.common.annotationfactory.AnnotationDescriptor;
import org.hibernate.annotations.common.annotationfactory.AnnotationFactory;
import org.hibernate.search.analyzer.definition.LuceneAnalyzerDefinitionContext;
import org.hibernate.search.analyzer.definition.LuceneCharFilterDefinitionContext;
import org.hibernate.search.analyzer.definition.LuceneCompositeAnalysisDefinitionContext;
import org.hibernate.search.analyzer.definition.LuceneNormalizerDefinitionContext;
import org.hibernate.search.analyzer.definition.LuceneTokenFilterDefinitionContext;
import org.hibernate.search.annotations.TokenFilterDef;


/**
 * @author Yoann Rodiere
 */
public class LuceneTokenFilterDefinitionContextImpl
		implements LuceneTokenFilterDefinitionContext, LuceneAnalysisDefinitionBuilder {

	private final LuceneCompositeAnalysisDefinitionContext parentContext;

	private final Class factoryClass;

	private final ParametersBuilder params = new ParametersBuilder();

	public LuceneTokenFilterDefinitionContextImpl(
			LuceneCompositeAnalysisDefinitionContext parentContext,
			Class factoryClass) {
		this.parentContext = parentContext;
		this.factoryClass = factoryClass;
	}

	@Override
	public LuceneTokenFilterDefinitionContext param(String name, String value) {
		params.put( name, value );
		return this;
	}

	@Override
	public LuceneAnalyzerDefinitionContext analyzer(String name) {
		return parentContext.analyzer( name );
	}

	@Override
	public LuceneNormalizerDefinitionContext normalizer(String name) {
		return parentContext.normalizer( name );
	}

	@Override
	public LuceneCharFilterDefinitionContext charFilter(Class factory) {
		return parentContext.charFilter( factory );
	}

	@Override
	public LuceneTokenFilterDefinitionContext tokenFilter(Class factory) {
		return parentContext.tokenFilter( factory );
	}

	@Override
	public TokenFilterDef build() {
		AnnotationDescriptor descriptor = new AnnotationDescriptor( TokenFilterDef.class );
		descriptor.setValue( "factory", factoryClass );
		descriptor.setValue( "params", params.build() );
		return AnnotationFactory.create( descriptor );
	}

}