函数式设计思想--拼接用户名

1. 场景描述

有这么一种常用的场景：比如从数据库中查询出订单信息，信息中只包含创建订单的用户ID，没有用户名，但前端界面需要展示出用户名，那么就要去用户服务那边根据用户ID拼接用户名，一般代码如下：

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);
    }
}

然后接着，又有了一个新模块，叫合同，然后也需要拼接用户名，代码如下：

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抽象方法

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方法实现如下：

public void fillUserName(List<Order> orders) {
    userService.fillUserName(orders, Order::getUserId, Order::setUserName);
}

然后Contract模块的fillUserName方法实现如下：

public void fillUserName(List<Contract> contracts) {
    userService.fillUserName(contracts, Contract::getUserId, Contract::setUserName);
}

现在，如果调用用户服务的逻辑发生变化，只用修改userService中的fillUserName方法就可以了，实现了与各个模块的解耦。

4. 总结

在出现代码copy现象的时候就应该考虑如何把公共代码抽象出来了
抽象过程中，列出实现的主要步骤，甄别出哪些是各个实现都是相同的，不同的地方有哪些？
不同的地方区分出是对象还是行为，如果是对象，则通过普通参数传入；如果是行为，则通过函数传入
面向对象编程是对对象的抽象，函数式编程则是对行为进行抽象