org.tinygroup.loader.LoaderManager Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of org.tinygroup.loader Show documentation
There is a newer version: 2.2.3
/**
 *  Copyright (c) 1997-2013, www.tinygroup.org ([email protected]).
 *
 *  Licensed under the GPL, Version 3.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.gnu.org/licenses/gpl.html
 *
 *  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.tinygroup.loader;

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

public class LoaderManager {
	/**
	 * 添加的
	 */
	private static Map> loaders = new HashMap>();
	/**
	 * class与classloader的映射关系,便于直接查询
	 */
	private static Map classMap = new HashMap();
	/**
	 * loader与class的映射关系，便于直接查询
	 */
	private static Map> loaderMap = new HashMap>();

	private LoaderManager(){
		
	}
	/**
	 * 根据class名获取该class的classloader
	 * 
	 * @param className
	 * @return
	 * @throws ClassNotFoundException
	 */
	public static ClassLoader getLoader(String className)
			throws ClassNotFoundException {
		if (isSimple(className)) {
			return LoaderManager.class.getClassLoader();
		}
		if (classMap.containsKey(className)) {
			return classMap.get(className);
		}
		try {
			Class.forName(className);
			classMap.put(className, LoaderManager.class.getClassLoader());
			return LoaderManager.class.getClassLoader();
		} catch (ClassNotFoundException e) {

		}
		for (ClassLoader loader : loaders.keySet()) {
			try {
				loader.loadClass(className);
				classMap.put(className, loader);
				if (loaderMap.containsKey(loader)) {
					loaderMap.get(loader).add(className);
				} else {
					List classes = new ArrayList();
					classes.add(className);
					loaderMap.put(loader, classes);
				}
				return loader;
			} catch (ClassNotFoundException e) {

			}
		}
		throw new ClassNotFoundException(className);
	}

	/**
	 * 移除一个classloader
	 * 
	 * @param loader
	 */
	public static void removeClassLoader(ClassLoader loader) {
		if (!loaderMap.containsKey(loader) && !loaders.containsKey(loader)) {
			return;
		}
		loaders.remove(loader);
		List classes = loaderMap.remove(loader);

		if (classes == null)
			return;
		for (String className : classes) {
			if (classMap.get(className) == loader) {
				loaderMap.remove(className);
			}
		}
	}

	/**
	 * 根据class名获取class
	 * 
	 * @param className
	 * @return
	 * @throws ClassNotFoundException
	 */
	public static Class getClass(String className)
			throws ClassNotFoundException {
		if (isSimple(className)) {
			return getSimpleClass(className);
		}
		return getLoader(className).loadClass(className);
	}

	/**
	 * 添加classloader
	 * 
	 * @param loader
	 */
	public static void addClassLoader(ClassLoader loader, List jarFiles) {
		if (!loaders.containsKey(loader))
			loaders.put(loader, jarFiles);
	}

	public static List getLoaderFiles(ClassLoader loader) {
		return loaders.get(loader);
	}

	private static boolean isSimple(String className) {
		if ("int".equals(className)) {
			return true;
		} else if ("boolean".equals(className)) {
			return true;
		} else if ("float".equals(className)) {
			return true;
		} else if ("double".equals(className)) {
			return true;
		} else if ("long".equals(className)) {
			return true;
		} else if ("short".equals(className)) {
			return true;
		} else if ("char".equals(className)) {
			return true;
		} else if ("byte".equals(className)) {
			return true;
		}
		return false;
	}

	private static Class getSimpleClass(String className) {
		if ("int".equals(className)) {
			return int.class;
		} else if ("boolean".equals(className)) {
			return boolean.class;
		} else if ("float".equals(className)) {
			return float.class;
		} else if ("double".equals(className)) {
			return double.class;
		} else if ("long".equals(className)) {
			return long.class;
		} else if ("short".equals(className)) {
			return short.class;
		} else if ("char".equals(className)) {
			return char.class;
		} else if ("byte".equals(className)) {
			return byte.class;
		}
		throw new RuntimeException(className + "不是基本数据类型");
	}
}