业务代码优化杂谈

       最近在审查组员代码的时候发现了几个问题，这里整理出来分享给大家，希望大家看了后有不同意见欢迎拍砖。

      一、背景：分页接口中大量的模板代码。代码如下：

PageVO<UserSettle> result = new PageVO<>();
Page page = PageHelper.startPage(userSettleQueryDTO.getPage(), userSettleQueryDTO.getSize());
List<UserSettle> userSettles = userSettleMapper.queryAllByPage(userSettleQueryDTO);
PageInfo pageInfo = new PageInfo<>(page.getResult());
result.setPage(pageInfo.getPageNum());
result.setSize(pageInfo.getPageSize());
result.setTotal(pageInfo.getTotal());
result.setDataList(userSettles);
return result;

对于这种代码，直接告诉我这样写不合理。想到我在整理框架的时候，使用了统一的结果返回，所以这块提了一个建议。这里先直接上修改之后的代码

Page page = PageHelper.startPage(userSettleQueryDTO.getPage(), userSettleQueryDTO.getSize())
        .doSelectPage(() -> userSettleMapper.queryAllByPage(userSettleQueryDTO));
return page;

修改之后是不是就很简单了，那全局返回结果怎么处理的呢

@RestControllerAdvice
public class ResponseControllerAdvice implements ResponseBodyAdvice<Object> {


    @Override
    public boolean supports(MethodParameter methodParameter, Class<? extends HttpMessageConverter<?>> aClass) {
        //如果接口返回类型本身就是ResultVO那就没必要进行额外的操作，返回false
        String name = methodParameter.getMethod().getDeclaringClass().getName();
        return Objects.nonNull(name) && name.contains("xx.xx.controller") && !methodParameter.getParameterType().equals(ResultVO.class);
    }

    @Override
    public Object beforeBodyWrite(Object data, MethodParameter methodParameter, MediaType mediaType, Class<? extends HttpMessageConverter<?>> aClass, ServerHttpRequest serverHttpRequest, ServerHttpResponse serverHttpResponse) {
        // String类型不能直接包装，所以要进行些特别的处理
        if (methodParameter.getGenericParameterType().equals(String.class)) {
            ObjectMapper objectMapper = new ObjectMapper();
            try {
                // 将数据包装在ResultVO里后，再转换为json字符串响应给前端
                return objectMapper.writeValueAsString(new ResultVO<>(data));
            } catch (JsonProcessingException e) {
                throw new APIException("返回String类型错误");
            }
        }
        //对于分页返回结果处理
        if (methodParameter.getParameterType().equals(Page.class)) {
            Page page = (Page) data;
            PageVO result = new PageVO();
            result.setPage(page.getPageNum());
            result.setSize(page.getPageSize());
            result.setTotal(page.getTotal());
            result.setDataList(page.getResult());
            return new ResultVO<>(result);
        }
        // 将原本的数据包装在ResultVO里
        return new ResultVO<>(data);
    }
}

通过这个例子，我们可以发现如果在开发业务过程中存在大量模板代码，那就是时候去优化了。