com.weicoder.common.util.ResourceUtil Maven / Gradle / Ivy

Go to download
package com.weicoder.common.util;

import java.io.File;
import java.io.InputStream;
import java.net.URL;
import java.util.Enumeration;
import java.util.List;

import com.weicoder.common.constants.StringConstants;
import com.weicoder.common.constants.SystemConstants;
import com.weicoder.common.io.FileUtil;
import com.weicoder.common.lang.Lists;

/**
 * 资源工具累类
 * @author WD
 */
public final class ResourceUtil {
	/**
	 * 尝试加载工程类下的资源文件
	 * @param name 文件名 相对路径
	 * @return 文件 如果不存在返回 null
	 */
	public static File newFile(String name) {
		try {
			return new File(getResource(name).toURI());
		} catch (Exception e) {
			return FileUtil.newFile(SystemConstants.USER_DIR + StringConstants.BACKSLASH + name);
		}
	}

	/**
	 * 尝试加载资源
	 * @param resourceName 资源文件名
	 * @return URL资源
	 */
	public static URL getResource(String resourceName) {
		// 获得资源URL 使用当前线程
		URL url = Thread.currentThread().getContextClassLoader().getResource(resourceName);
		// 如果获得的资源为null
		if (url == null) {
			// 使用本类加载
			url = ClassLoader.getSystemClassLoader().getResource(resourceName);
			// 如果为空
			if (url == null) {
				url = ClassLoader.getSystemResource(resourceName);
			}
		}
		// 如果url还为空 做资源的名的判断重新调用方法
		if (url == null && !EmptyUtil.isEmpty(resourceName) && (!resourceName.startsWith(StringConstants.BACKSLASH))) {
			return getResource(StringConstants.BACKSLASH + resourceName);
		}
		// 返回资源
		return url;
	}

	/**
	 * 尝试加载资源
	 * @param resourceName 资源文件名
	 * @return URL资源
	 */
	public static List getResources(String resourceName) {
		// 声明列表
		List urls = Lists.newList();
		try {
			// 获得资源URL 使用当前线程
			for (Enumeration u = Thread.currentThread().getContextClassLoader().getResources(resourceName); u
					.hasMoreElements();) {
				urls.add(u.nextElement());
			}
			// 如果获得的资源为null
			if (EmptyUtil.isEmpty(urls)) {
				// 使用本类加载
				for (Enumeration u = ClassLoader.getSystemClassLoader().getResources(resourceName); u
						.hasMoreElements();) {
					urls.add(u.nextElement());
				}
				// 如果为空
				if (EmptyUtil.isEmpty(urls)) {
					for (Enumeration u = ClassLoader.getSystemResources(resourceName); u.hasMoreElements();) {
						urls.add(u.nextElement());
					}
				}
			}
			// 如果url还为空 做资源的名的判断重新调用方法
			if (EmptyUtil.isEmpty(urls) && !EmptyUtil.isEmpty(resourceName)
					&& (!resourceName.startsWith(StringConstants.BACKSLASH))) {
				return getResources(StringConstants.BACKSLASH + resourceName);
			}
		} catch (Exception e) {}
		// 返回资源
		return urls;
	}

	/**
	 * 加载资源
	 * @param name 资源名
	 * @return 输入流
	 */
	public static InputStream loadResource(String name) {
		// 声明流
		InputStream in = ClassLoader.getSystemResourceAsStream(name);
		// 判断流为空
		if (in == null) {
			// 使用当前线程来加载资源
			in = Thread.currentThread().getContextClassLoader().getResourceAsStream(name);
			// 判断流为空
			if (in == null) {
				// 使用当前类来加载资源
				in = ClassLoader.getSystemClassLoader().getResourceAsStream(name);
			}
		}
		// 返回流
		return in;
	}

	private ResourceUtil() {}
}