com.kaka.Startup Maven / Gradle / Ivy
package com.kaka;
import com.kaka.notice.detector.*;
import com.kaka.numerical.NumericDetector;
import com.kaka.util.ClassScaner;
import java.util.*;
/**
* 启动器，其中包含类扫描及事件通知模型的注册
*
* @author zkpursuit
*/
public abstract class Startup {
final Map detectorMap = Collections.synchronizedMap(new LinkedHashMap<>());
/**
* 构造方法
*
* @param registers 事件通知模型注册器
*/
public Startup(IDetector... registers) {
addDetector(new NumericDetector());
addDetector(new ProxyDetector());
addDetector(new CommandDetector());
addDetector(new MediatorDetector());
for (IDetector detector : registers) {
addDetector(detector);
}
}
/**
* 添加事件通知模型注册器
*
* @param detector 事件通知模型注册器
*/
final protected void addDetector(IDetector detector) {
String name = detector.name();
if (name == null) {
name = detector.getClass().getTypeName();
}
detectorMap.put(name, detector);
}
/**
* 初始化
*
* @param loader 从此类加载器中扫描加载类
* @param packages .分割的包名
* @return 扫描到的类集合
*/
final protected Set> scan(ClassLoader loader, String... packages) {
Set delIdxs = new HashSet<>();
//过滤子包和相同的包
for (int i = 0; i < packages.length; i++) {
String selectPackage = packages[i];
for (int j = i + 1; j < packages.length; j++) {
String currPackage = packages[j];
if (selectPackage.length() < currPackage.length()) {
int idx = currPackage.indexOf(selectPackage);
if (idx >= 0) {
idx += selectPackage.length();
if (currPackage.charAt(idx) == '.') {
delIdxs.add(j);
}
}
} else if (selectPackage.length() > currPackage.length()) {
int idx = selectPackage.indexOf(currPackage);
if (idx >= 0) {
idx += currPackage.length();
if (selectPackage.charAt(idx) == '.') {
delIdxs.add(i);
}
}
} else if (selectPackage.equals(currPackage)) {
delIdxs.add(j);
}
}
}
if (loader == null) {
loader = Thread.currentThread().getContextClassLoader();
}
Set> classes = new HashSet<>();
for (int i = 0; i < packages.length; i++) {
if (!delIdxs.contains(i)) {
Set> _classes = ClassScaner.getClasses(loader, packages[i]);
if (!_classes.isEmpty()) {
classes.addAll(_classes);
}
}
}
detectorMap.forEach((String name, IDetector detector) -> {
classes.forEach(detector::discern);
if (detector instanceof PriorityDetector) {
((PriorityDetector) detector).centralizeProcess();
}
});
return classes;
}
/**
* 初始化
*
* @param packages .分割的包名
* @return 扫描到的类集合
*/
final protected Set> scan(String... packages) {
return scan(null, packages);
}
}