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

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

哈喽大家好啊,我是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中的内容:

图片

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

修改UserDaoA,指定名称:

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

可以看到,key的值发生了改变:

图片

List注入

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

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

public List<UserDao> 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注入

同样,也可以使用无序的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<UserDao> userDaoLists;

public String choiceFirst(){
return userDaoLists.get(0).getName();
};
}

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

责任编辑:武晓燕 来源: 码农参上
相关推荐

2022-03-21 08:55:53

RocketMQ客户端过滤机制

2023-08-10 08:00:00

2018-01-26 23:23:23

JDBC MySQL数据库

2021-01-27 18:13:35

日志nginx信息

2017-12-26 15:41:26

2021-08-01 12:34:01

跳水人工智能AI

2019-04-12 09:24:46

Spring Clou服务注册

2021-12-06 08:59:40

RabbitMQ单例写法

2012-12-28 13:47:36

Raspberry PGeek

2017-02-09 17:00:00

iOSSwiftKVC

2022-01-05 17:13:28

监控HTTPS网站

2019-03-08 10:08:41

网络程序猿代码

2020-08-14 08:19:25

Shell命令行数据

2021-11-29 05:37:24

Windows Def操作系统微软

2009-04-28 07:48:29

盖茨打工基金会

2018-10-22 15:29:50

2023-09-07 06:48:38

Intel显卡AMD

2022-11-02 07:46:31

GoFrameGcache缓存

2020-12-31 06:12:38

Siri Windows电脑

2023-10-31 12:29:25

模型训练
点赞
收藏

51CTO技术栈公众号