All Downloads are FREE. Search and download functionalities are using the official Maven repository.

org.codelibs.elasticsearch.fess.FessAnalysisPlugin Maven / Gradle / Ivy

The newest version!
/*
 * Copyright 2012-2022 CodeLibs Project and the Others.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND,
 * either express or implied. See the License for the specific language
 * governing permissions and limitations under the License.
 */
package org.codelibs.elasticsearch.fess;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.codelibs.elasticsearch.fess.action.AnalysisAction;
import org.codelibs.elasticsearch.fess.action.TransportAnalysisAction;
import org.codelibs.elasticsearch.fess.index.analysis.chinese.ChineseNoOpTokenFilterFactory;
import org.codelibs.elasticsearch.fess.index.analysis.chinese.ChineseStopTokenFilterFactory;
import org.codelibs.elasticsearch.fess.index.analysis.chinese.ChineseTokenizerFactory;
import org.codelibs.elasticsearch.fess.index.analysis.chinese.TraditionalChineseConvertCharFilterFactory;
import org.codelibs.elasticsearch.fess.index.analysis.japanese.JapaneseBaseFormFilterFactory;
import org.codelibs.elasticsearch.fess.index.analysis.japanese.JapaneseIterationMarkCharFilterFactory;
import org.codelibs.elasticsearch.fess.index.analysis.japanese.JapaneseKatakanaStemmerFactory;
import org.codelibs.elasticsearch.fess.index.analysis.japanese.JapanesePartOfSpeechFilterFactory;
import org.codelibs.elasticsearch.fess.index.analysis.japanese.JapaneseReadingFormFilterFactory;
import org.codelibs.elasticsearch.fess.index.analysis.japanese.JapaneseTokenizerFactory;
import org.codelibs.elasticsearch.fess.index.analysis.japanese.ReloadableJapaneseTokenizerFactory;
import org.codelibs.elasticsearch.fess.index.analysis.korean.KoreanNumberFilterFactory;
import org.codelibs.elasticsearch.fess.index.analysis.korean.KoreanPartOfSpeechStopFilterFactory;
import org.codelibs.elasticsearch.fess.index.analysis.korean.KoreanReadingFormFilterFactory;
import org.codelibs.elasticsearch.fess.index.analysis.korean.KoreanTokenizerFactory;
import org.codelibs.elasticsearch.fess.index.analysis.vietnamese.VietnameseTokenizerFactory;
import org.codelibs.elasticsearch.fess.service.FessAnalysisService;
import org.elasticsearch.action.ActionRequest;
import org.elasticsearch.action.ActionResponse;
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.ActionPlugin;
import org.elasticsearch.plugins.AnalysisPlugin;
import org.elasticsearch.plugins.Plugin;

@SuppressWarnings("deprecation")
public class FessAnalysisPlugin extends Plugin implements ActionPlugin, AnalysisPlugin {

    private FessAnalysisService service;

    @Override
    public List> getActions() {
        return Arrays.asList(//
                new ActionHandler<>(AnalysisAction.INSTANCE, TransportAnalysisAction.class));
    }

    @Override
    public Collection createComponents(PluginServices services) {
        final Collection components = new ArrayList<>();
        service = new FessAnalysisService();
        components.add(service);
        return components;
    }

    @Override
    public Map> getCharFilters() {
        final Map> extra = new HashMap<>();
        extra.put("fess_japanese_iteration_mark", (indexSettings, env, name,
                settings) -> new JapaneseIterationMarkCharFilterFactory(indexSettings, env, name, settings, service));
        extra.put("fess_traditional_chinese_convert", (indexSettings, env, name,
                settings) -> new TraditionalChineseConvertCharFilterFactory(indexSettings, env, name, settings, service));
        return extra;
    }

    @Override
    public Map> getTokenFilters() {
        final Map> extra = new HashMap<>();
        extra.put("fess_japanese_baseform",
                (indexSettings, env, name, settings) -> new JapaneseBaseFormFilterFactory(indexSettings, env, name, settings, service));
        extra.put("fess_japanese_part_of_speech",
                (indexSettings, env, name, settings) -> new JapanesePartOfSpeechFilterFactory(indexSettings, env, name, settings, service));
        extra.put("fess_japanese_readingform",
                (indexSettings, env, name, settings) -> new JapaneseReadingFormFilterFactory(indexSettings, env, name, settings, service));
        extra.put("fess_japanese_stemmer",
                (indexSettings, env, name, settings) -> new JapaneseKatakanaStemmerFactory(indexSettings, env, name, settings, service));
        extra.put("fess_simplified_chinese_stop",
                (indexSettings, env, name, settings) -> new ChineseStopTokenFilterFactory(indexSettings, env, name, settings, service));
        extra.put("fess_simplified_chinese_noop",
                (indexSettings, env, name, settings) -> new ChineseNoOpTokenFilterFactory(indexSettings, env, name, settings, service));
        extra.put("fess_korean_number",
                (indexSettings, env, name, settings) -> new KoreanNumberFilterFactory(indexSettings, env, name, settings, service));
        extra.put("fess_korean_part_of_speech", (indexSettings, env, name,
                settings) -> new KoreanPartOfSpeechStopFilterFactory(indexSettings, env, name, settings, service));
        extra.put("fess_korean_readingform",
                (indexSettings, env, name, settings) -> new KoreanReadingFormFilterFactory(indexSettings, env, name, settings, service));
        return extra;
    }

    @Override
    public Map> getTokenizers() {
        final Map> extra = new HashMap<>();
        extra.put("fess_japanese_tokenizer",
                (indexSettings, env, name, settings) -> new JapaneseTokenizerFactory(indexSettings, env, name, settings, service));
        extra.put("fess_japanese_reloadable_tokenizer", (indexSettings, env, name,
                settings) -> new ReloadableJapaneseTokenizerFactory(indexSettings, env, name, settings, service));
        extra.put("fess_korean_tokenizer",
                (indexSettings, env, name, settings) -> new KoreanTokenizerFactory(indexSettings, env, name, settings, service));
        extra.put("fess_vietnamese_tokenizer",
                (indexSettings, env, name, settings) -> new VietnameseTokenizerFactory(indexSettings, env, name, settings, service));
        extra.put("fess_simplified_chinese_tokenizer",
                (indexSettings, env, name, settings) -> new ChineseTokenizerFactory(indexSettings, env, name, settings, service));
        return extra;
    }

}