哈喽大家好啊,我是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
类型的注入,直接在Service
中注入一个Map
,key
为字符串类型,value
为上面定义的接口类型。
- @Service
- @AllArgsConstructor
- public class UserMapService {
- final Map<String, UserDao> userDaoMap;
-
- public Map<String,UserDao> getDaos(){
- return userDaoMap;
- }
- }
- 复制代码
通过接口测试,查看这个Map
中的内容:
可以看到,Map
中的value
是实现了接口的实例对象,key
则是beanName
,可以通过@Component
的value
属性进行自定义。
修改UserDaoA
,指定名称:
- @Component(value = "Hydra")
- public class UserDaoA implements UserDao {...}
- 复制代码
可以看到,key
的值发生了改变:
在Service
中,这次注入泛型为接口UserDao
类型的List
。
- @Service
- @AllArgsConstructor
- public class UserListService {
- private final List
userDaoLists; -
- public List
getDaos(){ - return userDaoLists;
- }
- }
- 复制代码
测试这个方法,查看List
中的内容,是我们放入容器中的两个bean
:
我们知道,List
是一个有序的数据结构,那么如果想要修改List
中bean
的排序,该如何做呢?
很简单,修改注入到spring容器中的两个bean
,为它们添加@Order
注解并指定加载顺序,数字越小越优先加载。
- @Component
- @Order(1)
- public class UserDaoA implements UserDao {……}
- 复制代码
- @Component
- @Order(-1)
- public class UserDaoB implements UserDao {……}
- 复制代码
修改完成后,再进行测试,可以看到bean
的顺序发生了改变:
同样,也可以使用无序的Set
注入bean
,泛型指定为接口类型。
- @Service
- @AllArgsConstructor
- public class UserSetService {
- private final Set<UserDao> userDaoSet;
-
- public Set<UserDao> getDaos(){
- return userDaoSet;
- }
- }
- 复制代码
查看Set
中的元素,和List
相同,只不过顺序变为无序,不会因为@Order
注解的值而改变:
最后,我们再来看一下数组注入的方式:
- @Service
- @AllArgsConstructor
- public class UserArrayService {
- private final UserDao[] userDaoArray;
-
- public UserDao[] getDaos(){
- return userDaoArray;
- }
- }
- 复制代码
查看数组中的元素:
并且,和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
userDaoLists; -
- public String choiceFirst(){
- return userDaoLists.get(0).getName();
- };
- }
- 复制代码
通过上面两个简单的例子可以看到,集合注入的方式使用起来非常灵活,我们可以在实际使用中,结合各种设计模式,写出实用而优雅的代码。
那么,这次的分享就到这里,我是Hydra,下篇文章再见。
作者简介,
码农参上
,一个热爱分享的公众号,有趣、深入、直接,与你聊聊技术。欢迎添加好友,进一步交流。