没想到吧，Spring中还有一招集合注入的写法

旧城等待， 2023-09-23 17:18 79阅读 0赞

哈喽大家好啊，我是Hydra。

Spring作为项目中不可缺少的底层框架，提供的最基础的功能就是`bean`的管理了。`bean`的注入相信大家都比较熟悉了，但是有几种不太常用到的集合注入方式，可能有的同学会不太了解，今天我们就通过实例看看它的使用。

首先，声明一个接口：

public interface UserDao {
        String getName();
    }
    复制代码

然后定义两个类来分别实现这个接口，并通过`@Component`注解把`bean`放入spring容器中：

@Component
    public class UserDaoA implements UserDao {
        @Override
        public String getName() {
            return "Hydra";
        }
    }
    复制代码

@Component
    public class UserDaoB implements UserDao {
        @Override
        public String getName() {
            return "#公众号：码农参上";
        }
    }
    复制代码

准备工作完成后，我们看看几种不同类型的集合注入方式。

### Map注入 ###

首先来看`Map`类型的注入，直接在`Service`中注入一个`Map`，`key`为字符串类型，`value`为上面定义的接口类型。

@Service
    @AllArgsConstructor
    public class UserMapService {
        final Map<String, UserDao> userDaoMap;
    
        public Map<String,UserDao> getDaos(){
            return userDaoMap;
        }
    }
    复制代码

通过接口测试，查看这个`Map`中的内容：

![format_png][]

可以看到，`Map`中的`value`是实现了接口的实例对象，`key`则是`beanName`，可以通过`@Component`的`value`属性进行自定义。

修改`UserDaoA`，指定名称：

@Component(value = "Hydra")
    public class UserDaoA implements UserDao {...}
    复制代码

可以看到，`key`的值发生了改变：

![format_png 1][]

### List注入 ###

在`Service`中，这次注入泛型为接口`UserDao`类型的`List`。

@Service
    @AllArgsConstructor
    public class UserListService {
        private final List<UserDao> userDaoLists;
    
        public List<UserDao> getDaos(){
            return userDaoLists;
        }
    }
    复制代码

测试这个方法，查看`List`中的内容，是我们放入容器中的两个`bean`：

![format_png 2][]

我们知道，`List`是一个有序的数据结构，那么如果想要修改`List`中`bean`的排序，该如何做呢？

很简单，修改注入到spring容器中的两个`bean`，为它们添加`@Order`注解并指定加载顺序，数字越小越优先加载。

@Component
    @Order(1)
    public class UserDaoA implements UserDao {……}
    复制代码

@Component
    @Order(-1)
    public class UserDaoB implements UserDao {……}
    复制代码

修改完成后，再进行测试，可以看到`bean`的顺序发生了改变：

![format_png 3][]

### Set注入 ###

同样，也可以使用无序的`Set`注入`bean`，泛型指定为接口类型。

@Service
    @AllArgsConstructor
    public class UserSetService {
        private final Set<UserDao> userDaoSet;
    
        public Set<UserDao> getDaos(){
            return userDaoSet;
        }
    }
    复制代码

查看`Set`中的元素，和`List`相同，只不过顺序变为无序，不会因为`@Order`注解的值而改变：

![format_png 4][]

### 数组注入 ###

最后，我们再来看一下数组注入的方式：

@Service
    @AllArgsConstructor
    public class UserArrayService {
        private final  UserDao[] userDaoArray;
    
        public UserDao[] getDaos(){
            return userDaoArray;
        }
    }
    复制代码

查看数组中的元素：

![format_png 5][]

并且，和`List`比较类似的，数组中`bean`的排序会受到`@Order`注解数值的影响，有兴趣的同学可以自己尝试一下。

### 应用 ###

了解了这几种注入方式后，再简单提一下它的使用场景。例如，我们可以用`Map`注入实现策略模式，来替换代码中繁杂的`if/else`判断。例如，原始的代码中判断逻辑可能是这样的：

public String choice(String name){
        if (name.equals("auth")){
            return "Hydra";
        }else if (name.equals("official")){
            return "#公众号：码农参上";
        }    
        return null;
    }
    复制代码

使用策略模式进行改造，首先修改`beanName`：

@Component(value = "auth")
    public class UserDaoA implements UserDao {
        @Override
        public String getName() {
            return "Hydra";
        }
    }
    复制代码

@Component(value = "official")
    public class UserDaoB implements UserDao {
        @Override
        public String getName() {
            return "#公众号：码农参上";
        }
    }
    复制代码

再修改`Servie`中的方法，一行代码即可实现原有的`if/else`判断：

@Service
    @AllArgsConstructor
    public class TestService {
        final Map<String, UserDao> userDaoMap;
    
        public String choice2(String name){
            return userDaoMap.get(name).getName();
        };
    }
    复制代码

可能在这个例子中，这种写法的优点体现的不十分明显，但是当你有一个非常长的`if/else`判断时，这种模式能使你的代码看上去简洁很多，并且符合代码按照功能拆分的原则。

同理，如果你已经通过`@Order`注解定义好了`bean`的加载顺序，也可以将它理解为`bean`的优先级，例如我想要调用优先级最高的符合类型的`bean`的方法，那么完全可以这样写：

@Service
    @AllArgsConstructor
    public class TestService {
        final List<UserDao> userDaoLists;
        
        public String choiceFirst(){
            return userDaoLists.get(0).getName();
        };
    }
    复制代码

通过上面两个简单的例子可以看到，集合注入的方式使用起来非常灵活，我们可以在实际使用中，结合各种设计模式，写出实用而优雅的代码。

那么，这次的分享就到这里，我是Hydra，下篇文章再见。

> 作者简介，`码农参上`，一个热爱分享的公众号，有趣、深入、直接，与你聊聊技术。欢迎添加好友，进一步交流。

[format_png]: https://img-blog.csdnimg.cn/img_convert/6e9c0937bb1237a843f269f39333d003.webp?x-oss-process=image/format,png
[format_png 1]: https://img-blog.csdnimg.cn/img_convert/665201d12047516cd872c475f1ec4af6.webp?x-oss-process=image/format,png
[format_png 2]: https://img-blog.csdnimg.cn/img_convert/4685a8b023917fb011e544c4139f571f.webp?x-oss-process=image/format,png
[format_png 3]: https://img-blog.csdnimg.cn/img_convert/b16f45507ab50d43496b0e7c94cc64ad.webp?x-oss-process=image/format,png
[format_png 4]: https://img-blog.csdnimg.cn/img_convert/6e1068ad452416a5c3d9d2031bbefd0f.webp?x-oss-process=image/format,png
[format_png 5]: https://img-blog.csdnimg.cn/img_convert/6f260e1c166553e7432f1cec680aa15d.webp?x-oss-process=image/format,png