搞定Spring6——轻松玩转自定义注解，实现简易版IOC容器

2024-01-10 23:49:45

简易IOC容器：告别依赖注入的烦恼

在Java开发中，依赖注入和IOC（控制反转）是一种强大的设计模式，可以帮助管理对象之间的依赖关系。通过使用IOC容器，我们可以简化代码结构，提高可维护性和可复用性。

Spring 6：自定义注解实现IOC容器

Spring 6框架为自定义IOC容器提供了强大支持。我们可以自定义注解，标记需要注入的对象，并利用Spring的IOC特性轻松实现简易版容器。

创建自定义注解

首先，我们需要一个注解标记需要注入的类，如下所示：

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

此注解表示注入的类在运行时仍然有效，并且只能用于字段。

创建IOC容器

接着，我们创建IOC容器管理对象：

public class SimpleIocContainer {

    private Map<Class<?>, Object> beanMap = new HashMap<>();

    public void registerBean(Class<?> clazz, Object bean) {
        beanMap.put(clazz, bean);
    }

    public <T> T getBean(Class<T> clazz) {
        return (T) beanMap.get(clazz);
    }
}

该容器使用Map存储对象，key为类类型，value为对象本身。registerBean方法用于注册对象，getBean方法用于获取对象。

使用注解和IOC容器

现在，我们可以使用自定义注解和IOC容器实现依赖注入：

public class UserService {

    @AutoWired
    private UserRepository userRepository;

    public void saveUser(User user) {
        userRepository.save(user);
    }
}

我们用@AutoWired注解标记userRepository字段，表示需要注入UserRepository对象。

在main方法中，我们可以使用IOC容器注入对象：

public static void main(String[] args) {
    SimpleIocContainer iocContainer = new SimpleIocContainer();
    iocContainer.registerBean(UserRepository.class, new UserRepository());
    UserService userService = iocContainer.getBean(UserService.class);
    userService.saveUser(new User());
}