Class Helper updated. It can search for classes correctly.
This commit is contained in:
parent
b28129ab97
commit
413441a26a
|
|
@ -1,9 +1,9 @@
|
|||
package haflow.utility;
|
||||
|
||||
import java.io.File;
|
||||
import java.io.IOException;
|
||||
import java.net.URL;
|
||||
import java.net.URLClassLoader;
|
||||
import java.net.URLDecoder;
|
||||
import java.util.ArrayList;
|
||||
import java.util.Enumeration;
|
||||
import java.util.List;
|
||||
|
|
@ -14,54 +14,82 @@ import org.springframework.stereotype.Component;
|
|||
|
||||
@Component
|
||||
public class ClassHelper {
|
||||
private List<String> getClassNameByFile(String packageName,
|
||||
String filePath, List<String> className,
|
||||
boolean searchInChildPackage) throws Exception {
|
||||
|
||||
public List<String> getClassNames(String packageName,
|
||||
boolean searchInChildPackage) {
|
||||
try {
|
||||
List<String> classNames = new ArrayList<String>();
|
||||
String decodedFilePath = URLDecoder.decode(filePath, "utf-8");
|
||||
File file = new File(decodedFilePath);
|
||||
File[] classFiles = file.listFiles();
|
||||
for (File classFile : classFiles) {
|
||||
if (classFile.isDirectory()) {
|
||||
if (searchInChildPackage) {
|
||||
classNames.addAll(getClassNameByFile(packageName,
|
||||
classFile.getPath(), classNames,
|
||||
searchInChildPackage));
|
||||
URLClassLoader classLoader = (URLClassLoader) Thread
|
||||
.currentThread().getContextClassLoader();
|
||||
URL[] urls = classLoader.getURLs();
|
||||
for (URL url : urls) {
|
||||
if (url.getProtocol().equals("file")) {
|
||||
File file = new File(url.getFile());
|
||||
String root = file.getPath();
|
||||
if (file.isDirectory()) {
|
||||
classNames.addAll(this.getClassNamesFromDirectory(root,
|
||||
packageName, searchInChildPackage, file, true));
|
||||
} else if (file.getName().endsWith(".jar")) {
|
||||
classNames.addAll(this.getClassNameFromJar(packageName,
|
||||
searchInChildPackage, file));
|
||||
}
|
||||
}
|
||||
}
|
||||
return classNames;
|
||||
} catch (Exception e) {
|
||||
e.printStackTrace();
|
||||
return null;
|
||||
}
|
||||
}
|
||||
|
||||
private List<String> getClassNamesFromDirectory(String root,
|
||||
String packageName, boolean searchInChildPackage, File directory,
|
||||
boolean continueSearch) {
|
||||
List<String> classNames = new ArrayList<String>();
|
||||
File[] files = directory.listFiles();
|
||||
for (File file : files) {
|
||||
if (file.isDirectory()) {
|
||||
if (searchInChildPackage || continueSearch) {
|
||||
boolean needToContinue = continueSearch;
|
||||
if (file.getPath().endsWith(packageName.replace(".", "/"))) {
|
||||
needToContinue = needToContinue & false;
|
||||
}
|
||||
classNames.addAll(this.getClassNamesFromDirectory(root,
|
||||
packageName, searchInChildPackage, file,
|
||||
needToContinue));
|
||||
}
|
||||
} else {
|
||||
String classFilePath = classFile.getPath();
|
||||
if (classFilePath.endsWith(".class")) {
|
||||
classFilePath = classFilePath.replace("\\", ".");
|
||||
classFilePath = classFilePath.replace("/", ".");
|
||||
classFilePath = classFilePath.substring(
|
||||
classFilePath.indexOf("classes") + 8,
|
||||
classFilePath.lastIndexOf("."));
|
||||
|
||||
classNames.add(classFilePath);
|
||||
if (file.getPath().endsWith(".class")) {
|
||||
String classFileName = file.getPath().replace(root + "/",
|
||||
"");
|
||||
if (classFileName.startsWith(packageName.replace(".", "/"))) {
|
||||
String className = classFileName.substring(0,
|
||||
classFileName.length() - ".class".length())
|
||||
.replace("/", ".");
|
||||
classNames.add(className);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
return classNames;
|
||||
}
|
||||
|
||||
private List<String> getClassNameByJar(String jarPath,
|
||||
boolean searchInChildPackage) throws Exception {
|
||||
private List<String> getClassNameFromJar(String packageName,
|
||||
boolean searchInChildPackage, File file) {
|
||||
List<String> classNames = new ArrayList<String>();
|
||||
String[] jarInfo = jarPath.split("!");
|
||||
String jarFilePath = jarInfo[0].substring(jarInfo[0].indexOf("/"));
|
||||
String decodedJarFilePath = URLDecoder.decode(jarFilePath, "utf-8");
|
||||
String packagePath = jarInfo[1].substring(1);
|
||||
|
||||
JarFile jarFile = new JarFile(decodedJarFilePath);
|
||||
Enumeration<JarEntry> entrys = jarFile.entries();
|
||||
while (entrys.hasMoreElements()) {
|
||||
JarEntry jarEntry = entrys.nextElement();
|
||||
JarFile jarFile = null;
|
||||
try {
|
||||
jarFile = new JarFile(file);
|
||||
Enumeration<JarEntry> jarEntries = jarFile.entries();
|
||||
while (jarEntries.hasMoreElements()) {
|
||||
JarEntry jarEntry = jarEntries.nextElement();
|
||||
String entryName = jarEntry.getName();
|
||||
if (entryName.endsWith(".class")) {
|
||||
String packagePath = packageName.replace(".", "/") + "/";
|
||||
if (searchInChildPackage) {
|
||||
if (entryName.startsWith(packagePath)) {
|
||||
entryName = entryName.replace("/", ".").substring(0,
|
||||
entryName.lastIndexOf("."));
|
||||
entryName = entryName.replace("/", ".").substring(
|
||||
0, entryName.lastIndexOf("."));
|
||||
classNames.add(entryName);
|
||||
}
|
||||
} else {
|
||||
|
|
@ -73,64 +101,24 @@ public class ClassHelper {
|
|||
entryPath = entryName;
|
||||
}
|
||||
if (entryPath.equals(packagePath)) {
|
||||
entryName = entryName.replace("/", ".").substring(0,
|
||||
entryName.lastIndexOf("."));
|
||||
entryName = entryName.replace("/", ".").substring(
|
||||
0, entryName.lastIndexOf("."));
|
||||
classNames.add(entryName);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
jarFile.close();
|
||||
return classNames;
|
||||
}
|
||||
|
||||
private List<String> getClassNameByJars(URL[] jarFiles, String packagePath,
|
||||
boolean searchInChildPackage) throws Exception {
|
||||
List<String> classNames = new ArrayList<String>();
|
||||
if (jarFiles != null) {
|
||||
for (int i = 0; i < jarFiles.length; i++) {
|
||||
URL jarFileUrl = jarFiles[i];
|
||||
String jarFilePath = jarFileUrl.getPath();
|
||||
if (jarFilePath.endsWith("classes/")) {
|
||||
continue;
|
||||
}
|
||||
String jarPath = jarFilePath + "!/" + packagePath;
|
||||
classNames.addAll(this.getClassNameByJar(jarPath,
|
||||
searchInChildPackage));
|
||||
}
|
||||
}
|
||||
return classNames;
|
||||
}
|
||||
|
||||
public List<String> getClassNames(String packageName,
|
||||
boolean searchInchildPackage) {
|
||||
try {
|
||||
List<String> classNames = null;
|
||||
ClassLoader classLoader = Thread.currentThread()
|
||||
.getContextClassLoader();
|
||||
String packagePath = packageName.replace(".", "/");
|
||||
URL packageUrl = classLoader.getResource(packagePath);
|
||||
if (packageUrl != null) {
|
||||
String protocol = packageUrl.getProtocol();
|
||||
if (protocol.equals("file")) {
|
||||
classNames = this.getClassNameByFile(packageName,
|
||||
packageUrl.getPath(), null, searchInchildPackage);
|
||||
} else if (protocol.equals("jar")) {
|
||||
classNames = this.getClassNameByJar(packageUrl.getPath(),
|
||||
searchInchildPackage);
|
||||
} else {
|
||||
classNames = null;
|
||||
}
|
||||
} else {
|
||||
classNames = this.getClassNameByJars(
|
||||
((URLClassLoader) classLoader).getURLs(), packagePath,
|
||||
searchInchildPackage);
|
||||
}
|
||||
return classNames;
|
||||
} catch (Exception e) {
|
||||
} catch (IOException e) {
|
||||
e.printStackTrace();
|
||||
} finally {
|
||||
if (jarFile != null) {
|
||||
try {
|
||||
jarFile.close();
|
||||
} catch (IOException e) {
|
||||
e.printStackTrace();
|
||||
return null;
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
return classNames;
|
||||
}
|
||||
}
|
||||
|
|
|
|||
|
|
@ -0,0 +1,20 @@
|
|||
package haflow.test;
|
||||
|
||||
import haflow.utility.ClassHelper;
|
||||
|
||||
import org.junit.Test;
|
||||
|
||||
import java.util.List;
|
||||
|
||||
public class ClassHelperTest {
|
||||
@Test
|
||||
public void test() {
|
||||
ClassHelper classHelper = new ClassHelper();
|
||||
List<String> classes = classHelper.getClassNames("haflow.module", true);
|
||||
if (classes != null) {
|
||||
for (String c : classes) {
|
||||
System.out.println(c);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue