java 包scan_Java扫描指定包中所有类

import java.io.File;

import java.io.FilenameFilter;

import java.io.IOException;

import java.net.JarURLConnection;

import java.net.URL;

import java.util.Enumeration;

import java.util.HashMap;

import java.util.Map;

import java.util.jar.JarEntry;

import java.util.jar.JarFile;

import com.cnp.andromeda.common.util.StringUtil;

/**

* @Author

* @Description 包扫描器

* @CopyRight

*/

public class ClassScanner{

private Map> classes           = new HashMap>();

private FilenameFilter        javaClassFilter;                                    // 类文件过滤器,只扫描一级类

private final String          CLASS_FILE_SUFFIX = ".class";                       // Java字节码文件后缀

private String                packPrefix;                                         // 包路径根路劲

public ClassScanner(){

javaClassFilter = new FilenameFilter(){

@Override

public boolean accept(File dir, String name){

// 排除内部内

return !name.contains("$");

}

};

}

/**

* @Title: scanning

* @Description 扫描指定包(本地)

* @param basePath 包所在的根路径

* @param packagePath 目标包路径

* @return Integer 被扫描到的类的数量

* @throws ClassNotFoundException

*/

public Integer scanning(String basePath, String packagePath) throws ClassNotFoundException{

packPrefix = basePath;

String packTmp = packagePath.replace('.', '/');

File dir = new File(basePath, packTmp);

// 不是文件夹

if(dir.isDirectory()){

scan0(dir);

}

return classes.size();

}

/**

* @Title: scanning

* @Description 扫描指定包, Jar或本地

* @param packagePath 包路径

* @param recursive 是否扫描子包

* @return Integer 类数量

*/

public Integer scanning(String packagePath, boolean recursive){

Enumeration dir;

String filePackPath = packagePath.replace('.', '/');

try{

// 得到指定路径中所有的资源文件

dir = Thread.currentThread().getContextClassLoader().getResources(filePackPath);

packPrefix = Thread.currentThread().getContextClassLoader().getResource("").getPath();

if(System.getProperty("file.separator").equals("\\")){

packPrefix = packPrefix.substring(1);

}

// 遍历资源文件

while(dir.hasMoreElements()){

URL url = dir.nextElement();

String protocol = url.getProtocol();

if("file".equals(protocol)){

File file = new File(url.getPath().substring(1));

scan0(file);

} else if("jar".equals(protocol)){

scanJ(url, recursive);

}

}

}

catch(Exception e){

throw new RuntimeException(e);

}

return classes.size();

}

/**

* @Title: scanJ

* @Description 扫描Jar包下所有class

* @param url jar-url路径

* @param recursive 是否递归遍历子包

* @throws IOException

* @throws ClassNotFoundException

*/

private void scanJ(URL url, boolean recursive) throws IOException, ClassNotFoundException{

JarURLConnection jarURLConnection = (JarURLConnection)url.openConnection();

JarFile jarFile = jarURLConnection.getJarFile();

// 遍历Jar包

Enumeration entries = jarFile.entries();

while(entries.hasMoreElements()){

JarEntry jarEntry = (JarEntry)entries.nextElement();

String fileName = jarEntry.getName();

if (jarEntry.isDirectory()) {

if (recursive) {

}

continue;

}

// .class

if(fileName.endsWith(CLASS_FILE_SUFFIX)){

String className = fileName.substring(0, fileName.indexOf('.')).replace('/', '.');

classes.put(className, Class.forName(className));

}

}

}

/**

* @Title: scan0

* @Description 执行扫描

* @param dir Java包文件夹

* @throws ClassNotFoundException

*/

private void scan0(File dir) throws ClassNotFoundException{

File[] fs = dir.listFiles(javaClassFilter);

for(int i = 0; fs != null && i 

File f = fs[i];

String path = f.getAbsolutePath();

// 跳过其他文件

if(path.endsWith(CLASS_FILE_SUFFIX)){

String className = StringUtil.getPackageByPath(f, packPrefix); // 获取包名

classes.put(className, Class.forName(className));

}

}

}

/**

* @Title: getClasses

* @Description 获取包中所有类

* @return Map<String,Class<?>> K:类全名, V:Class字节码

*/

public Map> getClasses(){

return classes;

}

public static void main(String[] args) throws ClassNotFoundException{

ClassScanner cs = new ClassScanner();

int c = cs.scanning("W:/IWiFI/Code/trunk/operation/target/classes/", "com/cnp/andromeda/common/util/");

System.out.println(c);

System.out.println(cs.getClasses().keySet());

ClassScanner cs2 = new ClassScanner();

int c2 = cs2.scanning("com.cnp.swarm", false);

System.out.println(c2);

System.out.println(cs2.getClasses().keySet());

}

}