Time: 2024-05-15 Wednesday 12:21:01
Author: Jackasher

组件扫描

package org.example.client;

import org.example.annoation.Component;

import java.io.File;
import java.lang.reflect.Constructor;
import java.net.URL;
import java.util.Arrays;
import java.util.HashMap;

public class ComponentScan {
    public static void main(String[] args) {
        HashMap<String, Object> beanMap = new HashMap<>();
        String packageName = "org.example.pojo";
        String packagePath = packageName.replaceAll("\\.", "/");
        System.out.println(packagePath);
        URL url = ClassLoader.getSystemClassLoader().getResource(packagePath);
        String path = url.getPath();
        System.out.println(path);
        File file = new File(path);
        File[] files = file.listFiles();
        Arrays.stream(files).forEach(f ->{

            try {
                System.out.println(f.getName().split("\\.")[0]);
                String className = packageName + "." + f.getName().split("\\.")[0];
                Class<?> aClass = Class.forName(className);
                if (aClass.isAnnotationPresent(Component.class)) {
                    Component annotation = aClass.getAnnotation(Component.class);
                    String id = annotation.value();
                    Object o = aClass.newInstance();
                    beanMap.put(id,o);

                }
            } catch (Exception e) {
                throw new RuntimeException(e);
            }


        });

        System.out.println(beanMap);
    }
}