重构日记一——flatMap与builder模式

   最近在做一个老系统的优化工作，由于里面的代码实在惨不忍睹，所以在做业务逻辑梳理及系统优化的同时，也开始做起了重构工作。由于没有充足的时间去做重新设计，毕竟在阿里白天要和沙雕产品撕逼，晚上要做正事，留给你优化系统的时间基本上少之又少，因此只能先从一些点开始，逐步进行，由点到面，记录下近期的重构工作。废话不多说，进入正题。

   厂里有好多系统都是跑了好多年了，很多代码如果没问题就不会有人来重构，像这样的JDK1.6前的代码，在我们日常的老系统中几乎随地可见，冗长恶心，极易重复，IDEA到处飘黄，为了这样的for循环去专门写个方法或工具类做抽象又小题大做，而且也比较难复用。这段代码的目的是从reasons这样一个数组里，取出所有reasonId的列表并去重，我们先把几个类型定义列出来，不必要的代码先省略了。

1. 用flatMap化解嵌套循环

        Set<Long> reasonIds = Sets.newHashSet();
        for (RejectItemReasonDO rejectDO : reasons) {
            List<RejectMapDO> reasonMap = rejectDO.getRejectMapList();
            if (CollectionUtils.isNotEmpty(reasonMap)) {
                for (RejectMapDO rejectInfo : reasonMap) {
                    reasonIds.addAll(rejectInfo.getRejectIds());
                }
            }
        }

    reasons是一个如下类型定义的列表：

public class RejectItemReasonDO extends AbstractBaseDO {

    private static final long serialVersionUID = -111111111L;

    @Setter
    @Getter
    private List<RejectMapDO> rejectMapList    = Lists.newArrayList();

    ...
}

RejectMapDO 定义

@Data
public class RejectMapDO extends BaseDO {

    private static final long serialVersionUID = -181828238283L;

    /**
     * 原因ID
     */
    private Set<Long>         rejectIds        = Sets.newHashSet();
    
    ...
}

   让我们用stream和flatMap改写一下，在这里用flapMap将数据结构中两层的list，化解嵌套循环，代码一下子就干净了很多。

Set<Long> reasonIds = rejectItemReasonDOS.stream().flatMap(rejectItemReasonDO -> rejectItemReasonDO.getRejectMapList()
                .stream()).filter(Objects::nonNull).flatMap(rejectMapDO -> rejectMapDO.getRejectIds()
                .stream()).filter(Objects::nonNull).collect(Collectors.toSet());

   

    有没有发现有什么问题？

    这里加了两个 .filter(Objects::nonNull) 过滤空元素而原代码中似乎只有一个，为啥要加两个？ 原代码中使用的是 reasonIds.addAll(rejectInfo.getRejectIds())，list的addAll方法如果碰到空元素是会抛异常的，我们不希望这样，因此在重构中顺带修复了一个可能的bug，增加代码的健壮性，当然你也可以说我的数据来源保证了不会为空，那也可以，但是在实际代码的编写中，原则上是不能相信他人和数据的，你懂的。

　2. 用builder模式改写复杂参数构造方式

　　我们的代码中有许多对象成员变量较多，在构建参数的时候，往往会有许多get/set操作，极为丑陋。这个时候我们想到了builder模式，builder的主要功能就是用来构建复杂对象，分离对象的表示和实现，从而让代码更整洁。示例代码如下，在需要构建的复杂对象中，创建一个static的builder，或者专门为这个复杂对象创建一个builder类，通过builder的操作封装参数类的操作。

      参数类及builder定义：

@Data
public class GlobalPublishAuditContext implements Serializable {

    private static final long serialVersionUID = 7663807161349892L;

    private Long productId;

    private Long sellerId;

    private String operator;

    private Boolean isPublish;

    private String  source;

    private Integer newItemStatus;
 
    private IqcExpand newIqcExpand;

    private Integer newItemSubStatus;
 
    private ProductForAuditDTO productAuditDTO;

    private Map<String, SkuForAuditDTO> skuMap;

    private Boolean needSendMTeeAuditMsg;

    private Boolean needSendScmAuditMsg;

    private Boolean needDoPostApproveIfQcSkip;

    private Boolean needManualCheck;

    private Boolean isImageEdit;

    private Boolean isFirstActive;
    /**
     *
     */
    private Map<String, String> extension;

    /**
     * new builder
     * @return
     */
    public static GlobalPublishAuditContextBuilder builder(){
        return new GlobalPublishAuditContextBuilder();
    }

    /**
     * builder for GlobalPublishAuditContext
     */
    public static class GlobalPublishAuditContextBuilder {

        private GlobalPublishAuditContext context;

        private ProductForAuditDTO productForAuditDTO;

        private IqcExpand newIqcExpand;

        private Map<String, String> extension;

        private Map<String, SkuForAuditDTO> skuMap;


        public GlobalPublishAuditContextBuilder() {
            context = new GlobalPublishAuditContext();
        }
    
       ...
...
}

　　使用Demo如下，是不是比各种get/set操作清爽多了。好了，先写到这里，清理一波烂代码后继续。

GlobalPublishAuditContext productAuditContext = GlobalPublishAuditContext.builder()
                .isPublish(request.isPublish())
                .sellerId(request.getSellerId())
                .productId(request.getProductId())
                .needManualCheck(request.isNeedManualCheck())
                .newItemSubStatus(ItemAuditStatus.APPROVED.getValue()).build();