- 1定义注解。我们可以定义一个标记注解,例如@Component,用来标记需要由框架扫描的类。
@Target(ElementType.TYPE) // 只能定义在类上
@Retention(RetentionPolicy.RUNTIME) // 保留运行阶段
public @interface Component {
}
- 2编写扫描器。扫描器用来扫描所有被@Component注解标记的类,并生成对应的对象。通常使用Java中的反射机制来实现扫描和创建对象。
public class ComponentScanner {
/**
* 扫描包,拿到有@Component的java对象
*
* @param
* @return
*/
public static List<Object> scan(String basePackage) throws InstantiationException, IllegalAccessException, ClassNotFoundException {
List<Object> componentList = new ArrayList<>(); // 过滤后存放携带@component的class
List<Class<?>> classes = ClassUtils.getClasses(basePackage); // 这是我们自定一Utils工具类
for (Class<?> aClass : classes) {
if(aClass.isAnnotationPresent(Component.class)){ // 判断是不是有那个@Component的class实例
Object instance = aClass.newInstance();
componentList.add(instance); // 将实例添加到componentList中
}
}
return componentList;
}
}
- 3 使用扫描器。通过调用扫描器的scan()方法,传入需要扫描的基础包名,即可获得所有被@Component注解标记的类的对象集合。
public class Application {
public static void main(String[] args) throws Exception {
String basePackage = "com.example";
List<Object> objectList = ComponentScanner.scan(basePackage);
/* 对象集合中包含已经生成的所有对象 */
for (Object object : objectList) {
// do something
}
}
}
在上述代码中,我们通过ComponentScanner类的scan()方法传入需要扫描类的基础包名,即可获取所有被@Component注解标记的类的对象集合。此时,框架帮助我们加载了所有被@Component注解标明的对象,并且可以使用反射机制调用它们的方法。
PS : 贡献一个ClassUtils.java
public class ClassUtils {
/**
* 拿到包下的所有class文件
* @param
* @return
*/
public static List<Class<?>> getClasses(String packageName) throws ClassNotFoundException {
List<Class<?>> classList = new ArrayList<Class<?>>();
String path = packageName.replace(".", "/"); // 拿到包路径 如com.qhx.zdy.java == com/qhx/zdy.java
URL url = Thread.currentThread().getContextClassLoader().getResource(path); // 反正,这里印象可以用IO替代
File dirFile = new File(url.getFile());
for (File file : dirFile.listFiles()) {
/* 如果是文件夹,则递归调用 */
if (file.isDirectory()) {
classList.addAll(getClasses(packageName + "." + file.getName()));
} else {
String className = file.getName().substring(0, file.getName().length() - 6);
classList.add(Class.forName(packageName + "." + className));
}
}
return classList;
}
}