goja.initialize.ctxbox.ClassSearcher Maven / Gradle / Ivy
The newest version!
/**
* Copyright (c) 2011-2013, kidzhou 周磊 ([email protected]) Licensed under the Apache
* License, Version 2.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.apache.org/licenses/LICENSE-2.0
* 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 goja.initialize.ctxbox;
import goja.core.StringPool;
import goja.core.kits.reflect.Reflect;
import com.jfinal.kit.PathKit;
import com.google.common.collect.Lists;
import org.apache.commons.lang3.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.io.File;
import java.io.IOException;
import java.util.Collections;
import java.util.Enumeration;
import java.util.List;
import java.util.jar.JarEntry;
import java.util.jar.JarFile;
public class ClassSearcher {
protected static final Logger LOG = LoggerFactory.getLogger(ClassSearcher.class);
private final List targets;
private String classpath = PathKit.getRootClassPath();
private String libDir =
PathKit.getWebRootPath() + File.separator + "WEB-INF" + File.separator + "lib";
private List scanPackages = Lists.newArrayList();
private boolean includeAllJarsInLib = false;
private List includeJars = Lists.newArrayList();
public ClassSearcher(Class target) {
targets = Lists.newArrayListWithCapacity(11);
this.targets.add(target);
}
public ClassSearcher(Class... targets) {
this.targets = Lists.newArrayListWithCapacity(targets.length);
Collections.addAll(this.targets, targets);
}
private static List> extraction(Class clazz, List classFileList) {
List> classList = Lists.newArrayList();
for (String classFile : classFileList) {
Class classInFile = Reflect.on(classFile).get();
if (clazz.isAssignableFrom(classInFile) && clazz != classInFile) {
classList.add(classInFile);
}
}
return classList;
}
public static ClassSearcher of(Class target) {
return new ClassSearcher(target);
}
public static ClassSearcher of(Class... targets) {
return new ClassSearcher(targets);
}
/**
* @param baseDirName 查找的文件夹路径
* @param targetFileName 需要查找的文件名
*/
private static List findFiles(String baseDirName, String targetFileName,
final String classpath) {
/**
* 算法简述： 从某个给定的需查找的文件夹出发，搜索该文件夹的所有子文件夹及文件，
* 若为文件，则进行匹配，匹配成功则加入结果集，若为子文件夹，则进队列。 队列不空，重复上述操作，队列为空，程序结束，返回结果。
*/
List classFiles = Lists.newArrayList();
File baseDir = new File(baseDirName);
if (!baseDir.exists() || !baseDir.isDirectory()) {
return classFiles;
} else {
String[] files = baseDir.list();
File file;
String fileName, open = classpath + File.separator, close = ".class";
int start, end;
if (files != null) {
for (String file_path : files) {
file = new File(baseDirName + File.separator + file_path);
if (file.isDirectory()) {
classFiles.addAll(
findFiles(baseDirName + File.separator + file_path, targetFileName, classpath));
} else {
if (wildcardMatch(targetFileName, file.getName())) {
fileName = file.getAbsolutePath();
start = fileName.indexOf(open);
end = fileName.indexOf(close, start + open.length());
// window 下会出现问题,正则替换问题
String className =
StringUtils.replace(fileName.substring(start + open.length(), end), File.separator,
".");
classFiles.add(className);
}
}
}
}
}
return classFiles;
}
/**
* 通配符匹配
*
* @param pattern 通配符模式
* @param fileName 待匹配的字符串
*/
private static boolean wildcardMatch(String pattern, String fileName) {
int patternLength = pattern.length();
int strLength = fileName.length();
int strIndex = 0;
char ch;
for (int patternIndex = 0; patternIndex < patternLength; patternIndex++) {
ch = pattern.charAt(patternIndex);
if (ch == '*') {
// 通配符星号*表示可以匹配任意多个字符
while (strIndex < strLength) {
if (wildcardMatch(pattern.substring(patternIndex + 1), fileName.substring(strIndex))) {
return true;
}
strIndex++;
}
} else if (ch == '?') {
// 通配符问号?表示匹配任意一个字符
strIndex++;
if (strIndex > strLength) {
// 表示str中已经没有字符匹配?了。
return false;
}
} else {
if ((strIndex >= strLength) || (ch != fileName.charAt(strIndex))) {
return false;
}
strIndex++;
}
}
return strIndex == strLength;
}
public List> search() {
List classFileList = Lists.newArrayList();
if (scanPackages.isEmpty()) {
classFileList = findFiles(classpath, "*.class", classpath);
} else {
for (String scanPackage : scanPackages) {
classFileList = findFiles(
classpath + File.separator + scanPackage.replaceAll("\\.", "\\" + File.separator),
"*.class", classpath);
}
}
classFileList.addAll(findjarFiles(libDir));
List> classes = Lists.newArrayList();
for (Class target : targets) {
classes.addAll(extraction(target, classFileList));
}
return classes;
}
/**
* 查找jar包中的class
*/
private List findjarFiles(String baseDirName) {
List classFiles = Lists.newArrayList();
File baseDir = new File(baseDirName);
if (!baseDir.exists() || !baseDir.isDirectory()) {
LOG.error("file search error:" + baseDirName + " is not a dir！");
} else {
File[] files = baseDir.listFiles();
if (files == null) {
return Collections.EMPTY_LIST;
}
for (File file : files) {
if (file.isDirectory()) {
classFiles.addAll(findjarFiles(file.getAbsolutePath()));
} else {
if (includeAllJarsInLib || includeJars.contains(file.getName())) {
JarFile localJarFile = null;
try {
localJarFile = new JarFile(new File(baseDirName + File.separator + file.getName()));
Enumeration entries = localJarFile.entries();
while (entries.hasMoreElements()) {
JarEntry jarEntry = entries.nextElement();
String entryName = jarEntry.getName();
if (scanPackages.isEmpty()) {
if (!jarEntry.isDirectory() && entryName.endsWith(".class")) {
String className = StringUtils.replace(entryName, StringPool.SLASH, ".")
.substring(0, entryName.length() - 6);
classFiles.add(className);
}
} else {
for (String scanPackage : scanPackages) {
scanPackage = scanPackage.replaceAll("\\.", "\\" + File.separator);
if (!jarEntry.isDirectory()
&& entryName.endsWith(".class")
&& entryName.startsWith(scanPackage)) {
String className = StringUtils.replace(entryName, File.separator, ".")
.substring(0, entryName.length() - 6);
classFiles.add(className);
}
}
}
}
} catch (IOException e) {
e.printStackTrace();
} finally {
try {
if (localJarFile != null) {
localJarFile.close();
}
} catch (IOException e) {
LOG.error("close jar file has error!", e);
}
}
}
}
}
}
return classFiles;
}
public ClassSearcher inJars(List jars) {
if (jars != null) {
includeJars.addAll(jars);
}
return this;
}
public ClassSearcher inJars(String... jars) {
if (jars != null) {
Collections.addAll(includeJars, jars);
}
return this;
}
public ClassSearcher includeAllJarsInLib(boolean includeAllJarsInLib) {
this.includeAllJarsInLib = includeAllJarsInLib;
return this;
}
public ClassSearcher classpath(String classpath) {
this.classpath = classpath;
return this;
}
public ClassSearcher libDir(String libDir) {
this.libDir = libDir;
return this;
}
public ClassSearcher scanPackages(List scanPaths) {
if (scanPaths != null) {
scanPackages.addAll(scanPaths);
}
return this;
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy