lxy/redkale
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

优化

Browse Source
This commit is contained in:
redkale
2023-11-30 16:53:38 +08:00
parent c25001c5e4
commit 9dff5967fb
1 changed files with 63 additions and 54 deletions
Download Patch File Download Diff File Expand all files Collapse all files

117
src/main/java/org/redkale/boot/ClassFilter.java
View File

@@ -48,13 +48,13 @@ public final class ClassFilter<T> {
private Class<? extends Annotation> annotationClass;//符合的注解。不为空时，扫描结果的class必须包含该注解 private Class<? extends Annotation> annotationClass;//符合的注解。不为空时，扫描结果的class必须包含该注解
private Pattern[] includePatterns; //符合的classname正则表达式 private Pattern[] includePatterns; //符合的className正则表达式
private Pattern[] excludePatterns;//拒绝的classname正则表达式 private Pattern[] excludePatterns;//拒绝的className正则表达式
private Set<String> privilegeIncludes; //特批符合条件的classname private Set<String> privilegeIncludes; //特批符合条件的className
private Set<String> privilegeExcludes;//特批拒绝条件的classname private Set<String> privilegeExcludes;//特批拒绝条件的className
private List<ClassFilter> ors; //或关系的其他ClassFilter private List<ClassFilter> ors; //或关系的其他ClassFilter
@@ -157,12 +157,12 @@ public final class ClassFilter<T> {
* 自动扫描地过滤指定的class * 自动扫描地过滤指定的class
* *
* @param property AnyValue * @param property AnyValue
* @param clazzname String * @param clazzName String
* @param url URL * @param url URL
*/ */
@SuppressWarnings("unchecked") @SuppressWarnings("unchecked")
public final void filter(AnyValue property, String clazzname, URL url) { public final void filter(AnyValue property, String clazzName, URL url) {
filter(property, clazzname, true, url); filter(property, clazzName, true, url);
} }
/** /**
@@ -252,34 +252,34 @@ public final class ClassFilter<T> {
/** /**
* 判断class是否有效 * 判断class是否有效
* *
* @param classname String * @param className String
* *
* @return boolean * @return boolean
*/ */
public boolean accept(String classname) { public boolean accept(String className) {
return accept(null, classname); return accept(null, className);
} }
/** /**
* 判断class是否有效 * 判断class是否有效
* *
* @param property AnyValue * @param property AnyValue
* @param classname String * @param className String
* *
* @return boolean * @return boolean
*/ */
public boolean accept(AnyValue property, String classname) { public boolean accept(AnyValue property, String className) {
boolean r = accept0(property, classname); boolean r = accept0(property, className);
if (r && ands != null) { if (r && ands != null) {
for (ClassFilter filter : ands) { for (ClassFilter filter : ands) {
if (!filter.accept(property, classname)) { if (!filter.accept(property, className)) {
return false; return false;
} }
} }
} }
if (!r && ors != null) { if (!r && ors != null) {
for (ClassFilter filter : ors) { for (ClassFilter filter : ors) {
if (filter.accept(filter.conf, classname)) { if (filter.accept(filter.conf, className)) {
return true; return true;
} }
} }
@@ -287,29 +287,29 @@ public final class ClassFilter<T> {
return r; return r;
} }
private boolean accept0(AnyValue property, String classname) { private boolean accept0(AnyValue property, String className) {
if (this.refused) { if (this.refused) {
return false; return false;
} }
if (this.privilegeIncludes != null && this.privilegeIncludes.contains(classname)) { if (this.privilegeIncludes != null && this.privilegeIncludes.contains(className)) {
return true; return true;
} }
if (this.privilegeExcludes != null && this.privilegeExcludes.contains(classname)) { if (this.privilegeExcludes != null && this.privilegeExcludes.contains(className)) {
return false; return false;
} }
if (classname.startsWith("java.") || classname.startsWith("javax.")) { if (className.startsWith("java.") || className.startsWith("javax.")) {
return false; return false;
} }
if (excludePatterns != null) { if (excludePatterns != null) {
for (Pattern reg : excludePatterns) { for (Pattern reg : excludePatterns) {
if (reg.matcher(classname).matches()) { if (reg.matcher(className).matches()) {
return false; return false;
} }
} }
} }
if (includePatterns != null) { if (includePatterns != null) {
for (Pattern reg : includePatterns) { for (Pattern reg : includePatterns) {
if (reg.matcher(classname).matches()) { if (reg.matcher(className).matches()) {
return true; return true;
} }
} }
@@ -576,67 +576,76 @@ public final class ClassFilter<T> {
* @throws IOException 异常 * @throws IOException 异常
*/ */
public static void load(final File excludeFile, RedkaleClassLoader loader, final ClassFilter... filters) throws IOException { public static void load(final File excludeFile, RedkaleClassLoader loader, final ClassFilter... filters) throws IOException {
List<URL> urlfiles = new ArrayList<>(2); List<URL> urlFiles = new ArrayList<>(2);
List<URL> urljares = new ArrayList<>(2); List<URL> urlJares = new ArrayList<>(2);
final URL exurl = excludeFile != null ? excludeFile.toURI().toURL() : null; final URL exurl = excludeFile != null ? excludeFile.toURI().toURL() : null;
for (URL url : loader.getAllURLs()) { for (URL url : loader.getAllURLs()) {
if (exurl != null && exurl.sameFile(url)) { if (exurl != null && exurl.sameFile(url)) {
continue; continue;
} }
if (url.getPath().endsWith(".jar")) { if (url.getPath().endsWith(".jar")) {
urljares.add(url); urlJares.add(url);
} else { } else {
urlfiles.add(url); urlFiles.add(url);
} }
} }
List<File> files = new ArrayList<>(); List<File> files = new ArrayList<>();
boolean debug = logger.isLoggable(Level.FINEST); boolean debug = logger.isLoggable(Level.FINEST);
StringBuilder debugstr = new StringBuilder(); StringBuilder debugstr = new StringBuilder();
for (final URL url : urljares) { for (final URL url : urlJares) {
Set<String> classes = cache.get(url); Set<String> classes = cache.get(url);
if (classes == null) { if (classes == null) {
classes = new LinkedHashSet<>(); classes = new LinkedHashSet<>();
try (JarFile jar = new JarFile(URLDecoder.decode(url.getFile(), StandardCharsets.UTF_8))) { try (JarFile jar = new JarFile(URLDecoder.decode(url.getFile(), StandardCharsets.UTF_8))) {
Enumeration<JarEntry> it = jar.entries(); Enumeration<JarEntry> it = jar.entries();
while (it.hasMoreElements()) { while (it.hasMoreElements()) {
String entryname = it.nextElement().getName().replace('/', '.'); String entryName = it.nextElement().getName().replace('/', '.');
if (entryname.endsWith(".class") && entryname.indexOf('$') < 0) { if (entryName.endsWith(".class") && entryName.indexOf('$') < 0) {
String classname = entryname.substring(0, entryname.length() - 6); String className = entryName.substring(0, entryName.length() - 6);
if (classname.startsWith("javax.") || classname.startsWith("com.sun.")) { if (className.startsWith("javax.") || className.startsWith("com.sun.")) {
continue; continue;
} }
//常见的jar跳过 //常见的jar跳过
if (classname.startsWith("org.redkaledyn.")) { if (className.startsWith("org.redkaledyn.")) {
break; //redkale动态生成的类 break; //redkale动态生成的类
} }
if (classname.startsWith("com.mysql.")) { if (className.startsWith("org.junit.")) {
break; break;
} }
if (classname.startsWith("org.junit.")) { if (className.startsWith("org.openjfx.")) {
break; break;
} }
if (classname.startsWith("org.openjfx.")) { if (className.startsWith("org.mariadb.")) {
break; break;
} }
if (classname.startsWith("org.mariadb.")) { if (className.startsWith("com.mysql.")) {
break; break;
} }
if (classname.startsWith("oracle.jdbc.")) { if (className.startsWith("oracle.jdbc.")) {
break; break;
} }
if (classname.startsWith("org.postgresql.")) { if (className.startsWith("org.postgresql.")) {
break; break;
} }
if (classname.startsWith("com.microsoft.sqlserver.")) { if (className.startsWith("com.microsoft.sqlserver.")) {
break; break;
} }
classes.add(classname); if (className.startsWith("org.apache.")) {
break;
}
if (className.startsWith("io.netty.")) {
break;
}
if (className.startsWith("io.vertx.")) {
break;
}
classes.add(className);
if (debug) { if (debug) {
debugstr.append(classname).append("\r\n"); debugstr.append(className).append("\r\n");
} }
for (final ClassFilter filter : filters) { for (final ClassFilter filter : filters) {
if (filter != null) { if (filter != null) {
filter.filter(null, classname, url); filter.filter(null, className, url);
} }
} }
} }
@@ -644,16 +653,16 @@ public final class ClassFilter<T> {
} }
cache.put(url, classes); cache.put(url, classes);
} else { } else {
for (String classname : classes) { for (String className : classes) {
for (final ClassFilter filter : filters) { for (final ClassFilter filter : filters) {
if (filter != null) { if (filter != null) {
filter.filter(null, classname, url); filter.filter(null, className, url);
} }
} }
} }
} }
} }
for (final URL url : urlfiles) { for (final URL url : urlFiles) {
Set<String> classes = cache.get(url); Set<String> classes = cache.get(url);
if (classes == null) { if (classes == null) {
classes = new LinkedHashSet<>(); classes = new LinkedHashSet<>();
@@ -664,38 +673,38 @@ public final class ClassFilter<T> {
if (cs.isEmpty()) { if (cs.isEmpty()) {
files.clear(); files.clear();
File root = new File(url.getFile()); File root = new File(url.getFile());
String rootpath = root.getPath(); String rootPath = root.getPath();
loadClassFiles(excludeFile, root, files); loadClassFiles(excludeFile, root, files);
for (File f : files) { for (File f : files) {
String classname = f.getPath().substring(rootpath.length() + 1, f.getPath().length() - 6).replace(File.separatorChar, '.'); String className = f.getPath().substring(rootPath.length() + 1, f.getPath().length() - 6).replace(File.separatorChar, '.');
if (classname.startsWith("javax.") || classname.startsWith("com.sun.")) { if (className.startsWith("javax.") || className.startsWith("com.sun.")) {
continue; continue;
} }
classes.add(classname); classes.add(className);
if (debug) { if (debug) {
debugstr.append(classname).append("\r\n"); debugstr.append(className).append("\r\n");
} }
for (final ClassFilter filter : filters) { for (final ClassFilter filter : filters) {
if (filter != null) { if (filter != null) {
filter.filter(null, classname, url); filter.filter(null, className, url);
} }
} }
} }
} else { } else {
for (String classname : classes) { for (String className : classes) {
for (final ClassFilter filter : filters) { for (final ClassFilter filter : filters) {
if (filter != null) { if (filter != null) {
filter.filter(null, classname, url); filter.filter(null, className, url);
} }
} }
} }
} }
cache.put(url, classes); cache.put(url, classes);
} else { } else {
for (String classname : classes) { for (String className : classes) {
for (final ClassFilter filter : filters) { for (final ClassFilter filter : filters) {
if (filter != null) { if (filter != null) {
filter.filter(null, classname, url); filter.filter(null, className, url);
} }
} }
} }