org.wltea.analyzer.cfg.DefaultConfig Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ik-analyzers Show documentation
bestwu ik-analyzer
The newest version!
package org.wltea.analyzer.cfg;

import lombok.extern.slf4j.Slf4j;

import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.List;
import java.util.Properties;

/**
 * Configuration 默认实现
 * 2012-5-8
 */
@Slf4j
public class DefaultConfig implements Configuration {

	/*
	 * 分词器默认字典路径 
	 */
	private static final String PATH_DIC_MAIN = "org/wltea/analyzer/dic/main2012.dic";
	private static final String PATH_DIC_QUANTIFIER = "org/wltea/analyzer/dic/quantifier.dic";

	/*
	 * 分词器配置文件路径
	 */
	private static final String FILE_NAME = "IKAnalyzer.cfg.xml";
	//配置属性——扩展字典
	private static final String EXT_DICT = "ext_dict";
	//配置属性——扩展停止词典
	private static final String EXT_STOP = "ext_stopwords";

	private final Properties props;
	/*
	 * 是否使用smart方式分词
	 */
	private boolean useSmart;

	/**
	 * 返回单例
	 *
	 * @return Configuration单例
	 */
	public static Configuration getInstance() {
		return new DefaultConfig();
	}

	/*
	 * 初始化配置文件
	 */
	private DefaultConfig() {
		props = new Properties();

		InputStream input = this.getClass().getClassLoader().getResourceAsStream(FILE_NAME);
		if (input != null) {
			try {
				props.loadFromXML(input);
			} catch (IOException e) {
				log.error(e.getMessage(), e);
			}
		}
	}

	/**
	 * 返回useSmart标志位
	 * useSmart =true ，分词器使用智能切分策略， =false则使用细粒度切分
	 *
	 * @return useSmart
	 */
	public boolean useSmart() {
		return useSmart;
	}

	/**
	 * 设置useSmart标志位
	 * useSmart =true ，分词器使用智能切分策略， =false则使用细粒度切分
	 *
	 * @param useSmart useSmart
	 */
	public void setUseSmart(boolean useSmart) {
		this.useSmart = useSmart;
	}

	/**
	 * 获取主词典路径
	 *
	 * @return String 主词典路径
	 */
	public String getMainDictionary() {
		return PATH_DIC_MAIN;
	}

	/**
	 * 获取量词词典路径
	 *
	 * @return String 量词词典路径
	 */
	public String getQuantifierDicionary() {
		return PATH_DIC_QUANTIFIER;
	}

	/**
	 * 获取扩展字典配置路径
	 *
	 * @return 相对类加载器的路径
	 */
	public List getExtDictionarys() {
		List extDictFiles = new ArrayList<>(2);
		String extDictCfg = props.getProperty(EXT_DICT);
		if (extDictCfg != null) {
			//使用;分割多个扩展字典配置
			String[] filePaths = extDictCfg.split(";");
			if (filePaths != null) {
				for (String filePath : filePaths) {
					if (filePath != null && !"".equals(filePath.trim())) {
						extDictFiles.add(filePath.trim());
					}
				}
			}
		}
		return extDictFiles;
	}

	/**
	 * 获取扩展停止词典配置路径
	 *
	 * @return 相对类加载器的路径
	 */
	public List getExtStopWordDictionarys() {
		List extStopWordDictFiles = new ArrayList<>(2);
		String extStopWordDictCfg = props.getProperty(EXT_STOP);
		if (extStopWordDictCfg != null) {
			//使用;分割多个扩展字典配置
			String[] filePaths = extStopWordDictCfg.split(";");
			if (filePaths != null) {
				for (String filePath : filePaths) {
					if (filePath != null && !"".equals(filePath.trim())) {
						extStopWordDictFiles.add(filePath.trim());
					}
				}
			}
		}
		return extStopWordDictFiles;
	}

}