配置文件读取工具类

package io.flysium.framework.util;


/**
 * 配置文件读取工具类
 */
public final class PropertiesUtils {

	private static Logger log = LoggerFactory.getLogger(PropertiesUtils.class);

	private static final String CLASSPATH_PREFIX = "classpath:";
	private static final String FILE_PREFIX = "file:";

	private PropertiesUtils() {
	}

	/**
	 * 
	 * @param vmName
	 *            启动应用是通过 -D设置的参数名称，如 CONFING_PATH、dubbo.protocol.port等
	 * @param classPathFileName
	 *            在classpath 下面的文件名，不带后缀，如 config.rocketmq
	 * @param fileName
	 *            文件全名，带后缀，如rocketmq.properties
	 * @return
	 */
	public static ResourceBundle getBundle(String vmName, String classPathFileName, String fileName) {
		ResourceBundle resource = null;
		String configPath = System.getProperty(vmName);// 获取vm动态参数
		if (configPath == null) {// 如果没有配置vm参数，根据classpath方式来获取
			resource = ResourceBundle.getBundle(classPathFileName, Locale.CHINA);
		} else {
			try {
				// 根据jvm参数配置的绝对路径获取
				if (configPath.startsWith(FILE_PREFIX)) {
					configPath = configPath.substring(FILE_PREFIX.length()) + File.separator + fileName;
				} else if (configPath.startsWith(CLASSPATH_PREFIX)) {
					configPath += "/" + fileName;
				}
				resource = getResourceBundle(configPath);
			} catch (Exception e) {
				log.error(e.getMessage(), e);
			}
		}
		return resource;
	}
	/**
	 * 注意使用完关闭inputstream流
	 * 
	 * @param vmName
	 *            启动应用是通过 -D设置的参数名称，如 CONFING_PATH、dubbo.protocol.port等
	 * @param fileName
	 *            文件全名，带后缀，如rocketmq.properties
	 * @return
	 */
	public static InputStream getBundleInputStream(String vmName, String fileName) {
		InputStream in = null;
		String configPath = System.getProperty(vmName);// 获取vm动态参数
		try {
			if (configPath == null) {// 如果没有配置vm参数，根据classpath方式来获取
				in = getResourceInputStream(fileName);
			} else {
				// 根据jvm参数配置的绝对路径获取
				if (configPath.startsWith(FILE_PREFIX)) {
					configPath = configPath.substring(FILE_PREFIX.length()) + File.separator + fileName;
				} else if (configPath.startsWith(CLASSPATH_PREFIX)) {
					configPath += "/" + fileName;
				}
				in = getResourceInputStream(configPath);
			}
		} catch (Exception e) {
			log.error(e.getMessage(), e);
		}
		return in;
	}

	/**
	 * 获取配置文件值（优先获取vm参数的值）
	 * 
	 * @param resource
	 * @param key
	 * @return
	 */
	public static String getString(ResourceBundle resource, String key) {
		String vmKey = System.getProperty(key);// 获取vm动态参数
		if (vmKey != null) {
			return vmKey;// 如果vm参数配置了该key，则优取该值
		}
		return resource.getString(key);
	}

	/**
	 * 支持读取在jar中的资源文件
	 * 
	 * @param resourceLocation
	 * @return
	 * @throws IOException
	 * @author SvenAugustus(蔡政滦) e-mail: SvenAugustus@outlook.com
	 * @version 2016年7月30日
	 */
	public static ResourceBundle getResourceBundle(String resourceLocation) throws IOException {
		Assert.notNull(resourceLocation, "Resource location must not be null");
		ResourceBundle resource = null;
		InputStream in = null;
		try {
			in = getResourceInputStream(resourceLocation);
			resource = new PropertyResourceBundle(in);
		} finally {
			if (in != null) {
				try {
					in.close();
				} catch (IOException e) {
					log.error(e.getMessage(), e);
				}
			}
		}
		return resource;
	}

	/**
	 * 支持读取在jar中的资源文件
	 * 
	 * @param resourceLocation
	 * @return
	 * @throws IOException
	 * @author SvenAugustus(蔡政滦) e-mail: SvenAugustus@outlook.com
	 * @version 2016年7月30日
	 */
	public static InputStream getResourceInputStream(String resourceLocation) throws IOException {
		Assert.notNull(resourceLocation, "Resource location must not be null");
		InputStream in = null;
		if (resourceLocation.startsWith(CLASSPATH_PREFIX)) {
			String path = resourceLocation.substring(CLASSPATH_PREFIX.length());
			String description = (new StringBuilder()).append("class path resource [").append(path).append("]")
					.toString();
			ClassLoader cl = ClassUtils.getDefaultClassLoader();
			URL url = cl == null ? ClassLoader.getSystemResource(path) : cl.getResource(path);
			Assert.notNull(url, (new StringBuilder()).append(description)
					.append(" cannot be resolved to absolute file path because it does not exist").toString());

			if ("jar".equals(url.getProtocol())) {
				in = new ClassPathResource(path).getInputStream();
			} else if ("file".equals(url.getProtocol())) {
				File resourceFile = null;
				try {
					resourceFile = new File(ResourceUtils.toURI(url).getSchemeSpecificPart());
				} catch (URISyntaxException ex) {
					log.error(ex.getMessage(), ex);
					resourceFile = new File(url.getFile());
				}
				in = new FileInputStream(resourceFile);
			} else {
				throw new FileNotFoundException(
						(new StringBuilder()).append(description).append(" cannot be resolved to absolute file path ")
								.append("because it does not reside in the file system: ").append(url).toString());
			}
		}
		if (in == null) {
			File resourceFile = null;
			try {
				resourceFile = ResourceUtils.getFile(new URL(resourceLocation));
			} catch (MalformedURLException ex) {
				log.error(ex.getMessage(), ex);
				resourceFile = new File(resourceLocation);
			}
			in = new FileInputStream(resourceFile);
		}
		return in;
	}

}