简单的手动实现spring中的自动装配案例

简简单单的实现一个spring中的自动装配和容器管理的小骚操作。

1，创建AutoSetBean.java

使用injectBeans静态方法，可以扫描指定包下的所有带MyInject注解的字段，如果在beans的Map中存在这个字段的实例化类，则执行装配。

import java.lang.reflect.Field;
import java.util.Set;
import org.example.test.custominj.myAnno.MyInject;
import org.reflections.Reflections;
import org.reflections.scanners.Scanners;
import org.reflections.util.ConfigurationBuilder;

public class AutoSetBean {

    public static void injectBeans(String packageName) {
        // 使用Reflections库来扫描指定包名下的所有类
        Reflections reflections = new Reflections(new ConfigurationBuilder()
                .forPackages(packageName)
                .addScanners(Scanners.FieldsAnnotated)
        );

        // 获取所有带有@MyInject注解的字段
        Set<Field> fields = reflections.getFieldsAnnotatedWith(MyInject.class);

        for (Field field : fields) {
            try {
                // 获取字段所在的类
                Class<?> declaringClass = field.getDeclaringClass();
                // 设置字段可访问
                field.setAccessible(true);

                // 如果MyBean.beans中已经存在这个类的实例，则直接使用
                Object instance = MyBean.beans.get(declaringClass);
                if (instance == null) {
                    // 如果不存在这个类的实例，则创建一个新的实例
                    instance = declaringClass.getDeclaredConstructor().newInstance();
                    MyBean.beans.put(declaringClass, instance);
                }

                // 获取需要注入的字段的类型
                Class<?> fieldType = field.getType();

                // 从MyBean.beans中获取该类型的实例
                Object beanToInject = MyBean.beans.get(fieldType);

                if (beanToInject != null) {
                    // 注入实例
                    field.set(instance, beanToInject);
                }
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
    }
}

2, 创建MyBean.java

通过使用registerBean静态方法，扫描传入的包名，检查其是否有@MyRegister注解，并将所有有此注解的类实例化后装入beans中，以备后续的装配。

import org.example.test.custominj.myAnno.MyRegister;
import org.reflections.Reflections;
import org.reflections.scanners.Scanners;
import org.reflections.util.ConfigurationBuilder;

import java.util.HashMap;
import java.util.Map;
import java.util.Set;

public class MyBean {
    public static Map<Class<?>, Object> beans = new HashMap<>();

    public static void registBean(String packageName) {
        // 配置Reflections库
        Reflections reflections = new Reflections(new ConfigurationBuilder()
                .forPackages(packageName)
                .addScanners(Scanners.TypesAnnotated)
        );

        // 获取所有带有@MyRegister注解的类
        Set<Class<?>> annotatedClasses = reflections.getTypesAnnotatedWith(MyRegister.class);

        for (Class<?> clazz : annotatedClasses) {
            try {
                // 实例化对象并放入beans Map中
                beans.put(clazz, clazz.getDeclaredConstructor().newInstance());
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
    }
}

3, 创建MyInject.java 和MyRegister注解

配置两个自定义注解，用于标识是否执行装配。

import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;

@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.FIELD)
public @interface MyInject {
}

import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;

@Retention(RetentionPolicy.RUNTIME)
public @interface MyRegister {
}

4，创建测试的Client.java 和被注入的MyService.java

@MyRegister作用于类上，注释后可以将这个类放入beans中，

@MyInject作用于字段，可以实现从容器中找到特定类型的实例化对象后执行装配。

import org.example.test.custominj.myAnno.MyInject;
import org.example.test.custominj.myAnno.MyRegister;

@MyRegister
public class Client {
    @MyInject
    private MyService myService;

    public void doSomething() {
        myService.doSomething();
    }
}

import org.example.test.custominj.myAnno.MyRegister;


@MyRegister
public class MyService {
    public void doSomething() {
        System.out.println("do something");
    }
}

5，创建测试类

@Test 测试自定义的自动装配

    @Test
    public void test4() {
        MyBean.registBean("org.example.test.myinjtest");
        // 打印已注册的beans
        for (Map.Entry<Class<?>, Object> entry : MyBean.beans.entrySet()) {
            System.out.println(entry.getKey() + ": " + entry.getValue());
        }
        AutoSetBean.injectBeans("org.example.test.myinjtest");
        client = (Client) MyBean.beans.get(Client.class);
        client.doSomething();
    }

执行结果，成功执行Service下的dosomething方法，成功打印！