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

原创：微信公众号 码农参上，欢迎分享，转载请保留出处。

哈喽大家好啊，我是Hydra。

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

首先，声明一个接口：

public interface UserDao {
    String getName();
}
1
2
3

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

@Component
public class UserDaoA implements UserDao {
    @Override
    public String getName() {
        return "Hydra";
    }
}
1
2
3
4
5
6
7

@Component
public class UserDaoB implements UserDao {
    @Override
    public String getName() {
        return "#公众号：码农参上";
    }
}
1
2
3
4
5
6
7

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

Map注入

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

@Service
@AllArgsConstructor
public class UserMapService {
    final Map<String, UserDao> userDaoMap;

    public Map<String,UserDao> getDaos(){
        return userDaoMap;
    }
}
1
2
3
4
5
6
7
8
9

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

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

修改UserDaoA，指定名称：

@Component(value = "Hydra")
public class UserDaoA implements UserDao {...}
1
2

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

List注入

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

@Service
@AllArgsConstructor
public class UserListService {
    private final List<UserDao> userDaoLists;

    public List<UserDao> getDaos(){
        return userDaoLists;
    }
}
1
2
3
4
5
6
7
8
9

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

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

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

@Component
@Order(1)
public class UserDaoA implements UserDao {……}
1
2
3

@Component
@Order(-1)
public class UserDaoB implements UserDao {……}
1
2
3

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

Set注入

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

@Service
@AllArgsConstructor
public class UserSetService {
    private final Set<UserDao> userDaoSet;

    public Set<UserDao> getDaos(){
        return userDaoSet;
    }
}
1
2
3
4
5
6
7
8
9

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

数组注入

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

@Service
@AllArgsConstructor
public class UserArrayService {
    private final  UserDao[] userDaoArray;

    public UserDao[] getDaos(){
        return userDaoArray;
    }
}
1
2
3
4
5
6
7
8
9

查看数组中的元素：

并且，和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;
}
1
2
3
4
5
6
7
8

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

@Component(value = "auth")
public class UserDaoA implements UserDao {
    @Override
    public String getName() {
        return "Hydra";
    }
}
1
2
3
4
5
6
7

@Component(value = "official")
public class UserDaoB implements UserDao {
    @Override
    public String getName() {
        return "#公众号：码农参上";
    }
}
1
2
3
4
5
6
7

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

@Service
@AllArgsConstructor
public class TestService {
    final Map<String, UserDao> userDaoMap;

    public String choice2(String name){
        return userDaoMap.get(name).getName();
    };
}
1
2
3
4
5
6
7
8
9

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

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

@Service
@AllArgsConstructor
public class TestService {
    final List<UserDao> userDaoLists;
    
    public String choiceFirst(){
        return userDaoLists.get(0).getName();
    };
}
1
2
3
4
5
6
7
8
9

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

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

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