1. 场景描述
有这么一种常用的场景：比如从数据库中查询出订单信息，信息中只包含创建订单的用户ID，没有用户名，但前端界面需要展示出用户名，那么就要去用户服务那边根据用户ID拼接用户名，一般代码如下：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
|
public void fillUserName(List<Order> orders) { List<Long> userIds = orders.stream() .map(Order::getUserId) .collect(Collectors.toList()); List<User> users = userService.findByIds(userIds); Map<Long, String> userNameMap = users.stream() .collect(Collectors.toMap(User::getId, User::getName)); for (Order order : orders) { String userName = userNameMap.getOrDefault(order.getId(), null); order.setUserName(userName); }} |
然后接着，又有了一个新模块，叫合同，然后也需要拼接用户名，代码如下：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
|
public void fillUserName(List<Contract> contracts) { List<Long> userIds = contracts.stream() .map(Contracts::getUserId) .collect(Collectors.toList()); List<User> users = userService.findByIds(userIds); Map<Long, String> userNameMap = users.stream() .collect(Collectors.toMap(User::getId, User::getName)); for (Contracts contract : contracts) { String userName = userNameMap.getOrDefault(contract.getId(), null); contract.setUserName(userName); }} |
其实上述两段代码大部分都是相似的，如果之后类似的模块越来越多，那么项目中就会到处散落着这样的代码，导致这块代码逻辑的维护成本非常高（一处改，处处改）。
2. 抽象过程
首先看一下fillUserName这个方法实现的主要步骤：
- 获取用户ID的集合（不同类型的对象获取用户ID的方式不同）
- 根据用户ID的集合查询对应的用户
- 将用户集合转成用户ID为Key，用户名为Value的Map
- 遍历对象，填充用户名（不同类型的对象填充用户名的方式不同）
通过以上步骤可以看出，不同的模块在实现上有以下几个不同点：
- 处理的对象（一个是订单，一个是合同）
- 获取用户ID的行为（订单和合同获取用户ID的方式是未知的）
- 填充用户名的行为（订单和合同填充用户名的方式也是未知的）
因此，我们抽象出来的方法应该包含以上3个参数，其中第一个是对象集合Collection<T> datas，第二个是获取用户ID的函数Function<T, Long> idGetter，第三个是填充用户名的函数BiConsumer<T, String> nameSetter。
3. 代码实现
首先在UserService中实现fillUserName抽象方法
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
|
public <T> void fillUserName(Collection<T> datas, Function<T, Long> idGetter, BiConsumer<T, String> nameSetter) { List<Long> userIds = datas.stream() .map(idGetter) .collect(Collectors.toList()); List<User> users = findByIds(userIds); Map<Long, String> userNameMap = users.stream() .collect(Collectors.toMap(User::getId, User::getName)); for (T data : datas) { String userName = userNameMap.getOrDefault(idGetter.apply(data), null); nameSetter.accept(data, userName); }} |
然后Order模块的fillUserName方法实现如下：
|
1
2
3
|
public void fillUserName(List<Order> orders) { userService.fillUserName(orders, Order::getUserId, Order::setUserName);} |
然后Contract模块的fillUserName方法实现如下：
|
1
2
3
|
public void fillUserName(List<Contract> contracts) { userService.fillUserName(contracts, Contract::getUserId, Contract::setUserName);} |
现在，如果调用用户服务的逻辑发生变化，只用修改userService中的fillUserName方法就可以了，实现了与各个模块的解耦。
4. 总结
- 在出现代码copy现象的时候就应该考虑如何把公共代码抽象出来了
- 抽象过程中，列出实现的主要步骤，甄别出哪些是各个实现都是相同的，不同的地方有哪些？
- 不同的地方区分出是对象还是行为，如果是对象，则通过普通参数传入；如果是行为，则通过函数传入
- 面向对象编程是对对象的抽象，函数式编程则是对行为进行抽象