package com.dy.common.util;
|
|
import java.io.File;
|
import java.io.FileNotFoundException;
|
import java.io.IOException;
|
import java.net.URI;
|
import java.util.ArrayList;
|
import java.util.Enumeration;
|
import java.util.List;
|
import java.util.jar.JarEntry;
|
import java.util.jar.JarFile;
|
import java.util.logging.Logger;
|
|
@SuppressWarnings("unused")
|
public class ClassScan {
|
|
private static final Logger logger = Logger.getLogger(ClassScan.class.getName());
|
|
|
/**
|
* 在指定的URL路径中搜索Class
|
*
|
* @param uris 要搜索的java文件URI,不能用URL类型，因为URL路径中有中文将失败
|
* @param rootPackage 要搜索的java文件的包路径
|
*/
|
public List<String> searchClassFromUrl(URI[] uris, String rootPackage){
|
List<String> classNames = new ArrayList<>() ;
|
for (URI uri : uris) {
|
try {
|
File file = new File(uri.getPath());
|
if (!file.exists()){
|
throw new FileNotFoundException();
|
}
|
this.doSearchClass(file, rootPackage ,classNames);
|
} catch (Exception ex) {
|
logger.throwing(ClassScan.class.getName(), "searchClassFromUrl", ex);
|
}
|
}
|
return classNames ;
|
}
|
|
/**
|
* 在指定的jar文件中搜索Class
|
* @param file 要搜索的jar文件
|
*/
|
public List<String> searchClassFromArchive(JarFile file) {
|
List<String> classNames = new ArrayList<>() ;
|
for (Enumeration<JarEntry> enumeration = file.entries(); enumeration.hasMoreElements();) {
|
JarEntry entry = enumeration.nextElement();
|
if (entry.getName().endsWith(".class")) {
|
classNames.add(file.getName()) ;
|
}
|
}
|
return classNames ;
|
}
|
|
/**
|
* 搜索指定文件夹下层级里的class
|
* @param dir 文件夹
|
* @param rootPackage 根包
|
* @param classNames 类名称集合
|
*/
|
private void doSearchClass(File dir, String rootPackage, List<String> classNames) {
|
String path ;
|
if (dir.isDirectory()) {
|
File[] fs = dir.listFiles() ;
|
if(fs != null && fs.length > 0){
|
for (File f : fs){
|
//只搜当前文件夹中文件
|
this.doSearchClass(f, rootPackage, classNames);
|
}
|
}
|
return;
|
}
|
if (dir.getName().endsWith(".class")) {
|
path = dir.getAbsolutePath() ;
|
path = path.replace('/', '.') ;
|
path = path.replace('\\', '.') ;
|
path = path.substring(path.indexOf(rootPackage)) ;
|
classNames.add(path) ;
|
} else if (dir.getName().endsWith(".zip") || dir.getName().endsWith(".jar")) {
|
try {
|
this.searchClassFromArchive(new JarFile(dir));
|
} catch (IOException ex) {
|
logger.throwing(ClassScan.class.getName(), "doSearchClass", ex);
|
}
|
}
|
}
|
|
}
|
