com.xiaoleilu.hutool.setting.profile.Profile Maven / Gradle / Ivy

Go to download
package com.xiaoleilu.hutool.setting.profile;

import java.io.Serializable;
import java.nio.charset.Charset;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

import com.xiaoleilu.hutool.setting.Setting;
import com.xiaoleilu.hutool.util.StrUtil;

/**
 * Profile可以让我们定义一系列的配置信息，然后指定其激活条件。

 * 此类中我们规范一套规则如下：

 * 默认的，我们读取${classpath}/default下的配置文件(*.setting文件)，当调用setProfile方法时，指定一个profile，即可读取其目录下的配置文件。

 * 比如我们定义几个profile：test，develop，production，分别代表测试环境、开发环境和线上环境，我希望读取数据库配置文件db.setting，那么：
 * 
 * 	test =》 ${classpath}/test/db.setting
 * 	develop =》 ${classpath}/develop/db.setting
 * 	production =》 ${classpath}/production/db.setting
 * 
 * 
 * @author Looly
 *
 */
public class Profile implements Serializable{
	private static final long serialVersionUID = -4189955219454008744L;
	
	/** 默认环境 */
	public static final String DEFAULT_PROFILE = "default";

	/** 条件 */
	private String profile;
	/** 编码 */
	private Charset charset;
	/** 是否使用变量 */
	private boolean useVar;
	/** 配置文件缓存 */
	private Map settingMap = new ConcurrentHashMap<>();
	
	//-------------------------------------------------------------------------------- Constructor start
	/**
	 * 默认构造，环境使用默认的：default，编码UTF-8，不使用变量
	 */
	public Profile() {
		this(DEFAULT_PROFILE);
	}
	
	/**
	 * 构造，编码UTF-8，不使用变量
	 * @param profile 环境
	 */
	public Profile(String profile) {
		this(profile, Setting.DEFAULT_CHARSET, false);
	}
	
	/**
	 * 构造
	 * @param profile 环境
	 * @param charset 编码
	 * @param useVar 是否使用变量
	 */
	public Profile(String profile, Charset charset, boolean useVar) {
		super();
		this.profile = profile;
		this.charset = charset;
		this.useVar = useVar;
	}
	//-------------------------------------------------------------------------------- Constructor end

	/**
	 * 获取当前环境下的配置文件
	 * @param name 文件名，如果没有扩展名，默认为.setting
	 * @return 当前环境下配置文件
	 */
	public Setting getSetting(String name){
		String nameForProfile = fixNameForProfile(name);
		Setting setting = settingMap.get(nameForProfile);
		if(null == setting){
			setting = new Setting(nameForProfile, charset, useVar);
			settingMap.put(nameForProfile, setting);
		}
		return setting;
	}
	
	/**
	 * 设置环境
	 * @param profile 环境
	 * @return 自身
	 */
	public Profile setProfile(String profile){
		this.profile = profile;
		return this;
	}
	
	/**
	 * 设置编码
	 * @param charset 编码
	 * @return 自身
	 */
	public Profile setCharset(Charset charset){
		this.charset = charset;
		return this;
	}
	
	/**
	 * 设置是否使用变量
	 * @param useVar 变量
	 * @return 自身
	 */
	public Profile setUseVar(boolean useVar){
		this.useVar = useVar;
		return this;
	}
	
	/**
	 * 清空所有环境的配置文件
	 * @return 自身
	 */
	public Profile clear(){
		this.settingMap.clear();
		return this;
	}
	
	//-------------------------------------------------------------------------------- Private method start
	/**
	 * 修正文件名
	 * @param name 文件名
	 * @return 修正后的文件名
	 */
	private String fixNameForProfile(String name){
		final String actralProfile = null == this.profile ? StrUtil.EMPTY : this.profile;
		if(StrUtil.isNotBlank(name) && false == name.contains(StrUtil.DOT)){
			return StrUtil.format("{}/{}.setting", actralProfile, name);
		}
		return StrUtil.format("{}/{}", actralProfile);
	}
	//-------------------------------------------------------------------------------- Private method end
}