Java遍历包中所有类-终续
上一篇中，我向大家讲述了遍历jar包时所遇到的困难，本篇将向大家分享最终版代码。
package com.itkt.mtravel.hotel.util;import java.io.File;import java.net.URL;import java.net.URLClassLoader;import java.util.ArrayList;import java.util.Enumeration;import java.util.List;import java.util.jar.JarEntry;import java.util.jar.JarFile;public class PackageUtil {public static void main(String[] args) throws Exception {String packageName = "com.wang.vo.request.hotel";// List<String> classNames = getClassName(packageName);List<String> classNames = getClassName(packageName, false);if (classNames != null) {for (String className : classNames) {System.out.println(className);}}}/** * 获取某包下（包括该包的所有子包）所有类 * @param packageName 包名 * @return 类的完整名称 */public static List<String> getClassName(String packageName) {return getClassName(packageName, true);}/** * 获取某包下所有类 * @param packageName 包名 * @param childPackage 是否遍历子包 * @return 类的完整名称 */public static List<String> getClassName(String packageName, boolean childPackage) {List<String> fileNames = null;ClassLoader loader = Thread.currentThread().getContextClassLoader();String packagePath = packageName.replace(".", "/");URL url = loader.getResource(packagePath);if (url != null) {String type = url.getProtocol();if (type.equals("file")) {fileNames = getClassNameByFile(url.getPath(), null, childPackage);} else if (type.equals("jar")) {fileNames = getClassNameByJar(url.getPath(), childPackage);}} else {fileNames = getClassNameByJars(((URLClassLoader) loader).getURLs(), packagePath, childPackage);}return fileNames;}/** * 从项目文件获取某包下所有类 * @param filePath 文件路径 * @param className 类名集合 * @param childPackage 是否遍历子包 * @return 类的完整名称 */private static List<String> getClassNameByFile(String filePath, List<String> className, boolean childPackage) {List<String> myClassName = new ArrayList<String>();File file = new File(filePath);File[] childFiles = file.listFiles();for (File childFile : childFiles) {if (childFile.isDirectory()) {if (childPackage) {myClassName.addAll(getClassNameByFile(childFile.getPath(), myClassName, childPackage));}} else {String childFilePath = childFile.getPath();if (childFilePath.endsWith(".class")) {childFilePath = childFilePath.substring(childFilePath.indexOf("\\classes") + 9, childFilePath.lastIndexOf("."));childFilePath = childFilePath.replace("\\", ".");myClassName.add(childFilePath);}}}return myClassName;}/** * 从jar获取某包下所有类 * @param jarPath jar文件路径 * @param childPackage 是否遍历子包 * @return 类的完整名称 */private static List<String> getClassNameByJar(String jarPath, boolean childPackage) {List<String> myClassName = new ArrayList<String>();String[] jarInfo = jarPath.split("!");String jarFilePath = jarInfo[0].substring(jarInfo[0].indexOf("/"));String packagePath = jarInfo[1].substring(1);try {JarFile jarFile = new JarFile(jarFilePath);Enumeration<JarEntry> entrys = jarFile.entries();while (entrys.hasMoreElements()) {JarEntry jarEntry = entrys.nextElement();String entryName = jarEntry.getName();if (entryName.endsWith(".class")) {if (childPackage) {if (entryName.startsWith(packagePath)) {entryName = entryName.replace("/", ".").substring(0, entryName.lastIndexOf("."));myClassName.add(entryName);}} else {int index = entryName.lastIndexOf("/");String myPackagePath;if (index != -1) {myPackagePath = entryName.substring(0, index);} else {myPackagePath = entryName;}if (myPackagePath.equals(packagePath)) {entryName = entryName.replace("/", ".").substring(0, entryName.lastIndexOf("."));myClassName.add(entryName);}}}}} catch (Exception e) {e.printStackTrace();}return myClassName;}/** * 从所有jar中搜索该包，并获取该包下所有类 * @param urls URL集合 * @param packagePath 包路径 * @param childPackage 是否遍历子包 * @return 类的完整名称 */private static List<String> getClassNameByJars(URL[] urls, String packagePath, boolean childPackage) {List<String> myClassName = new ArrayList<String>();if (urls != null) {for (int i = 0; i < urls.length; i++) {URL url = urls[i];String urlPath = url.getPath();// 不必搜索classes文件夹if (urlPath.endsWith("classes/")) {continue;}String jarPath = urlPath + "!/" + packagePath;myClassName.addAll(getClassNameByJar(jarPath, childPackage));}}return myClassName;}}由于我们并不确定jar包生成时采用的哪种方式，如果采用默认生成jar包的方式，那我们通过Thread.currentThread().getContextClassLoader().getResource()是获取不到的，因此我增加了从所有jar包中搜索提供的包域名，这样功能就完善了很多。
那么就此关于“如何遍历包中所有类”就结束了，PackageUtil这个类的功能还有些少，不排除日后进一步完善的可能，如果大家关于这个util有什么新的需求或者建议，随时欢迎大家提出。发现bug的，也请及时通知我以便改进。
============友情链接============
Java遍历jar包所有类-后续 http://blog.csdn.net/wangpeng047/article/details/8202353
Java遍历包中所有类 http://blog.csdn.net/wangpeng047/article/details/8124390