org.codelibs.opensearch.extension.ExtensionPlugin Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of opensearch-analysis-extension Show documentation
This plugin provides an analysis library for Japanese.
The newest version!
package org.codelibs.opensearch.extension;

import java.util.HashMap;
import java.util.Map;

import org.codelibs.opensearch.extension.analysis.AlphaNumWordFilterFactory;
import org.codelibs.opensearch.extension.analysis.CharTypeFilterFactory;
import org.codelibs.opensearch.extension.analysis.DisableGraphFilterFactory;
import org.codelibs.opensearch.extension.analysis.FlexiblePorterStemFilterFactory;
import org.codelibs.opensearch.extension.analysis.IterationMarkCharFilterFactory;
import org.codelibs.opensearch.extension.analysis.KanjiNumberFilterFactory;
import org.codelibs.opensearch.extension.analysis.NGramSynonymTokenizerFactory;
import org.codelibs.opensearch.extension.analysis.NumberConcatenationFilterFactory;
import org.codelibs.opensearch.extension.analysis.PatternConcatenationFilterFactory;
import org.codelibs.opensearch.extension.analysis.PosConcatenationFilterFactory;
import org.codelibs.opensearch.extension.analysis.ProlongedSoundMarkCharFilterFactory;
import org.codelibs.opensearch.extension.analysis.ReloadableKeywordMarkerFilterFactory;
import org.codelibs.opensearch.extension.analysis.ReloadableKuromojiTokenizerFactory;
import org.codelibs.opensearch.extension.analysis.ReloadableStopFilterFactory;
import org.codelibs.opensearch.extension.analysis.StopTokenPrefixFilterFactory;
import org.codelibs.opensearch.extension.analysis.StopTokenSuffixFilterFactory;
import org.codelibs.opensearch.extension.kuromoji.index.analysis.JapaneseStopTokenFilterFactory;
import org.codelibs.opensearch.extension.kuromoji.index.analysis.KuromojiBaseFormFilterFactory;
import org.codelibs.opensearch.extension.kuromoji.index.analysis.KuromojiIterationMarkCharFilterFactory;
import org.codelibs.opensearch.extension.kuromoji.index.analysis.KuromojiKatakanaStemmerFactory;
import org.codelibs.opensearch.extension.kuromoji.index.analysis.KuromojiNumberFilterFactory;
import org.codelibs.opensearch.extension.kuromoji.index.analysis.KuromojiPartOfSpeechFilterFactory;
import org.codelibs.opensearch.extension.kuromoji.index.analysis.KuromojiReadingFormFilterFactory;
import org.opensearch.index.analysis.CharFilterFactory;
import org.opensearch.index.analysis.TokenFilterFactory;
import org.opensearch.index.analysis.TokenizerFactory;
import org.opensearch.indices.analysis.AnalysisModule.AnalysisProvider;
import org.opensearch.plugins.AnalysisPlugin;
import org.opensearch.plugins.Plugin;

public class ExtensionPlugin extends Plugin implements AnalysisPlugin {

    @Override
    public Map> getCharFilters() {
        final Map> extra = new HashMap<>();
        extra.put("iteration_mark", IterationMarkCharFilterFactory::new);
        extra.put("prolonged_sound_mark", ProlongedSoundMarkCharFilterFactory::new);
        extra.put("reloadable_kuromoji_iteration_mark", KuromojiIterationMarkCharFilterFactory::new);
        return extra;
    }

    @Override
    public Map> getTokenFilters() {
        final Map> extra = new HashMap<>();
        extra.put("reloadable_kuromoji_baseform", KuromojiBaseFormFilterFactory::new);
        extra.put("reloadable_kuromoji_part_of_speech", KuromojiPartOfSpeechFilterFactory::new);
        extra.put("reloadable_kuromoji_readingform", KuromojiReadingFormFilterFactory::new);
        extra.put("reloadable_kuromoji_stemmer", KuromojiKatakanaStemmerFactory::new);
        extra.put("reloadable_kuromoji_number", KuromojiNumberFilterFactory::new);
        extra.put("reloadable_ja_stop", JapaneseStopTokenFilterFactory::new);
        extra.put("kanji_number", KanjiNumberFilterFactory::new);
        extra.put("kuromoji_pos_concat", PosConcatenationFilterFactory::new);
        extra.put("char_type", CharTypeFilterFactory::new);
        extra.put("number_concat", NumberConcatenationFilterFactory::new);
        extra.put("pattern_concat", PatternConcatenationFilterFactory::new);
        extra.put("stop_prefix", StopTokenPrefixFilterFactory::new);
        extra.put("stop_suffix", StopTokenSuffixFilterFactory::new);
        extra.put("reloadable_keyword_marker", ReloadableKeywordMarkerFilterFactory::new);
        extra.put("reloadable_stop", ReloadableStopFilterFactory::new);
        extra.put("flexible_porter_stem", FlexiblePorterStemFilterFactory::new);
        extra.put("alphanum_word", AlphaNumWordFilterFactory::new);
        extra.put("disable_graph", DisableGraphFilterFactory::new);
        return extra;
    }

    @Override
    public Map> getTokenizers() {
        final Map> extra = new HashMap<>();
        extra.put("reloadable_kuromoji_tokenizer", ReloadableKuromojiTokenizerFactory::new);
        extra.put("reloadable_kuromoji", ReloadableKuromojiTokenizerFactory::new);
        extra.put("ngram_synonym", NGramSynonymTokenizerFactory::new);
        return extra;
    }

}