Java中M个元素的集合每N个分成一组通用方案

一、背景

平时业务开发可能有这样的一个场景：一个集合有M个元素，需要每N个元素一起处理。

虽然这个功能非常简单，直接用for循环或者while可以写，其实可以封装成工具类，方便后续复用。

二、源码

/**
 * 将size为M的集合以每eachPieceSize为单位分隔成N组
 *
 * @param data          数据
 * @param eachPieceSize 每组的数量
 * @param <T>           泛型
 * @return N个分组
 */
public static <T> List<List<T>> splitToPieces(Collection<T> data, int eachPieceSize) {
    if (CollectionUtils.isEmpty(data)) {
        return new ArrayList<>(0);
    }
    if (eachPieceSize <= 0) {
        throw new IllegalArgumentException("参数错误");
    }
    List<List<T>> result = new ArrayList<>();
    for (int index = 0; index < data.size(); index += eachPieceSize) {
        result.add(data.stream().skip(index).limit(eachPieceSize).collect(Collectors.toList()));
    }
    return result;
}

单元测试

@Test
public void test() {
    // 生成26个String作为元素的list
    List<String> randomValue = new EasyRandom().objects(String.class, 26).collect(Collectors.toList());
    // 每5个一组
    List<List<String>> lists = ListUtil.splitToPieces(randomValue, 5);

    int size = lists.size();
    Assert.assertEquals(6, size);
    Assert.assertEquals(1, lists.get(size - 1).size());
}

三、总结

遇到一些可以复用的场景可以考虑封装成工具类。

测试时可以考虑用EasyRandom来构造测试对象，省时省力。