org.wikimedia.search.extra.analysis.ukrainian.ExtraAnalysisUkrainianPlugin Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of opensearch-extra-analysis-ukrainian Show documentation

Ukrainian analyzer.

The newest version!

package org.wikimedia.search.extra.analysis.ukrainian;

import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.opensearch.index.analysis.PreConfiguredTokenFilter;
import org.opensearch.index.analysis.TokenFilterFactory;
import org.opensearch.indices.analysis.AnalysisModule.AnalysisProvider;
import org.opensearch.plugins.AnalysisPlugin;
import org.opensearch.plugins.Plugin;

/**
 * Setup the OpenSearch plugin.
 */
public class ExtraAnalysisUkrainianPlugin extends Plugin implements AnalysisPlugin {

    @Override
    public Map> getTokenFilters() {
        Map> map = new HashMap<>();
        map.put("ukrainian_stop", UkrainianStopFilterFactory::new);
        map.put("ukrainian_stemmer", UkrainianStemmerFilterFactory::new);
        return Collections.unmodifiableMap(map);
    }

    @Override
    public List getPreConfiguredTokenFilters() {
        List list = new ArrayList<>();
        list.add(PreConfiguredTokenFilter.singleton("ukrainian_stop", true, in -> new UkrainianStopFilter(in, UkrainianStopFilterFactory.UK_STOP)));
        list.add(PreConfiguredTokenFilter.singleton("ukrainian_stemmer", true, in -> new UkrainianStemmerFilter(in, UkrainianStemmerFilterFactory.UK_DICT)));
        return Collections.unmodifiableList(list);
    }

}