代码简洁之道

2022-06-12

这篇文章主要是记录日常开发中的一些代码片段，寻找是否有更简洁的写法，仅供参考。

1 比较两个数的大小

场景举例：比较两款竞品的订单下单总数，得出较大的那个数。

一般写法，用 if 判断：

if (number1 > number2) {
    return number1;
}
return number2;

简洁写法，用已经封装好的方法：

1	return Math.max(number1, number2);

2 找出集合中的最大值

场景举例：在一个订单系统中，获取一个月内最大的日下单量。

一般写法，for 循环加上 if 判断：

List<Integer> listNumber = getDayCountList();
Integer maxValue = 0;
for (int i = 0; i < listNumber.size(); i++) {
    if (listNumber.get(i) > maxValue) {
        maxValue = listNumber.get(i);
    }
}

简洁写法：

1 2	List<Integer> listNumber = getDayCountList(); return Collections.max(listNumber);

3 判断集合是否为空

一般写法：

List<Integer> listNumber = getDayCountList();
if (null != listNumber && listNumber.size() > 0) {
    // ...
}

简洁写法：

List<Integer> listNumber = getDayCountList();
if (!CollectionUtils.isEmpty(listNumber)) {
    // ...
}

4 打印出数组中所有的值

一般写法：

int[] intArray = new int[]{1, 2, 3, 4, 5, 6, 7, 8, 9};
for (int i : intArray) {
    LOGGER.info(i);
}

简洁写法：

1 2	int[] intArray = new int[]{1, 2, 3, 4, 5, 6, 7, 8, 9}; LOGGER.info(Arrays.toString(intArray));

5 转换时间格式

一般写法，对时间的格式进行转换后，再覆盖原值：

List<Map<String, Object>> listData = getListByTime();
SimpleDateFormat dateFormatOrigin = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
SimpleDateFormat dateFormatTarget = new SimpleDateFormat("yyyy-MM-dd'T'HH:mm:ss");
Date dateTime = new Date();
for (Map<String, Object> map : listData) {
    String createdTime = String.valueOf(map.get("createdTime"));
    dateTime = dateFormatOrigin.parse(createdTime);
    createdTime = dateFormatTarget.format(dateTime);
    map.put("createdTime", createdTime);
}

简洁写法，从 SQL 语句入手，在查询时就将时间格式转换：

1	SELECT DATE_FORMAT(createdTime, '%Y-%m-%dT%T') AS createdTime FROM tb_user

6 筛选集合里的数据

一般写法，用 for 循环：

List<User> listUser = getUserList();
List<Long> listUserId = new ArrayList<>();
List<String> listUserName = new ArrayList<>();
for (User user : listUser) {
    Long userId = user.getId();
    String name = user.getName();
    listUserId.add(userId);
    listUserName.add(name);
}

简洁写法，用 stream 流：

1
2
3

List<User> listUser = getUserList();
List<Long> listUserId = listUser.stream().map(User::getId).collect(Collectors.toList());
List<String> listUserName = listUser.stream().map(User::getName).collect(Collectors.toList());

7 对集合里的数据去重

一般写法，用 for 循环：

List<String> listData = getDataList();
for (int i = 0; i < listData.size() - 1; i++) {
    for (int j = listData.size() - 1; j > i; j--) {
        if (listData.get(j).equals(listData.get(i))) {
            listData.remove(j);
        }
    }
}

简洁写法，用 stream 流：

1 2	List<String> listData = getDataList(); listData = listData.stream().distinct().collect(Collectors.toList());

8 复制 Map 数据并新增元素

场景举例：原先的 Map 数据在其他的业务中还有用到，不能直接更改。

一般写法：

Map<String, Object> originMap = getMapData();
Map<String, Object> targetMap = new HashMap<>();
targetMap.put("userId", originMap.get("userId"));
targetMap.put("userName", originMap.get("userName"));
targetMap.put("phone", originMap.get("phone"));
targetMap.put("email", originMap.get("email"));
targetMap.put("orderId", originMap.get("orderId"));
targetMap.put("createdTime", originMap.get("createdTime"));
// 新增元素
targetMap.put("status", "1");
targetMap.put("deleted", "1");

简洁写法：

Map<String, Object> originMap = getMapData();
Map<String, Object> targetMap = new HashMap<>();
targetMap.putAll(originMap);
// 新增元素
targetMap.put("status", "1");
targetMap.put("deleted", "1");

参考文章：头铁