模拟spring利用反射动态创建携带@Compont注解的对象

  1. 1定义注解。我们可以定义一个标记注解,例如@Component,用来标记需要由框架扫描的类。
@Target(ElementType.TYPE) // 只能定义在类上
@Retention(RetentionPolicy.RUNTIME) // 保留运行阶段
public @interface Component {
}
  1. 2编写扫描器。扫描器用来扫描所有被@Component注解标记的类,并生成对应的对象。通常使用Java中的反射机制来实现扫描和创建对象。
public class ComponentScanner {

   
    /**
     * 扫描包,拿到有@Component的java对象
     *
     * @param 
     * @return
     */
    public static List scan(String basePackage) throws InstantiationException, IllegalAccessException, ClassNotFoundException {

        List componentList = new ArrayList<>(); // 过滤后存放携带@component的class

        List> 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;
    }
}
 
  
  1. 3 使用扫描器。通过调用扫描器的scan()方法,传入需要扫描的基础包名,即可获得所有被@Component注解标记的类的对象集合。
public class Application { 
    public static void main(String[] args) throws Exception { 
        String basePackage = "com.example";
        List 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> getClasses(String packageName) throws ClassNotFoundException {

        List> classList = new ArrayList>();

        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;
    }
}

你可能感兴趣的:(Java,spring,java,后端)