org.codelibs.elasticsearch.kuromoji.neologd.KuromojiNeologdPlugin Maven / Gradle / Ivy

Go to download
package org.codelibs.elasticsearch.kuromoji.neologd;

import static java.util.Collections.singletonMap;

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

import org.apache.lucene.analysis.Analyzer;
import org.codelibs.elasticsearch.kuromoji.neologd.index.analysis.KuromojiAnalyzerProvider;
import org.codelibs.elasticsearch.kuromoji.neologd.index.analysis.KuromojiBaseFormFilterFactory;
import org.codelibs.elasticsearch.kuromoji.neologd.index.analysis.KuromojiIterationMarkCharFilterFactory;
import org.codelibs.elasticsearch.kuromoji.neologd.index.analysis.KuromojiKatakanaStemmerFactory;
import org.codelibs.elasticsearch.kuromoji.neologd.index.analysis.KuromojiPartOfSpeechFilterFactory;
import org.codelibs.elasticsearch.kuromoji.neologd.index.analysis.KuromojiReadingFormFilterFactory;
import org.codelibs.elasticsearch.kuromoji.neologd.index.analysis.KuromojiTokenizerFactory;
import org.codelibs.elasticsearch.kuromoji.neologd.index.analysis.PosConcatenationFilterFactory;
import org.codelibs.elasticsearch.kuromoji.neologd.index.analysis.ReloadableKuromojiTokenizerFactory;
import org.elasticsearch.index.analysis.AnalyzerProvider;
import org.elasticsearch.index.analysis.CharFilterFactory;
import org.elasticsearch.index.analysis.TokenFilterFactory;
import org.elasticsearch.index.analysis.TokenizerFactory;
import org.elasticsearch.indices.analysis.AnalysisModule.AnalysisProvider;
import org.elasticsearch.plugins.AnalysisPlugin;
import org.elasticsearch.plugins.Plugin;

public class KuromojiNeologdPlugin extends Plugin implements AnalysisPlugin {

    @Override
    public Map> getCharFilters() {
        return singletonMap("kuromoji_neologd_iteration_mark", KuromojiIterationMarkCharFilterFactory::new);
    }

    @Override
    public Map> getTokenFilters() {
        Map> extra = new HashMap<>();
        extra.put("kuromoji_neologd_baseform", KuromojiBaseFormFilterFactory::new);
        extra.put("kuromoji_neologd_part_of_speech", KuromojiPartOfSpeechFilterFactory::new);
        extra.put("kuromoji_neologd_readingform", KuromojiReadingFormFilterFactory::new);
        extra.put("kuromoji_neologd_stemmer", KuromojiKatakanaStemmerFactory::new);
        extra.put("kuromoji_neologd_pos_concat", PosConcatenationFilterFactory::new);
        return extra;
    }

    @Override
    public Map> getTokenizers() {
        Map> extra = new HashMap<>();
        extra.put("kuromoji_neologd_tokenizer", KuromojiTokenizerFactory::new);
        extra.put("reloadable_kuromoji_neologd_tokenizer", ReloadableKuromojiTokenizerFactory::new);
        extra.put("reloadable_kuromoji_neologd", ReloadableKuromojiTokenizerFactory::new);
        return extra;
    }

    @Override
    public Map>> getAnalyzers() {
        return singletonMap("kuromoji_neologd", KuromojiAnalyzerProvider::new);
    }
}