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

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

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

public static  List> splitToPieces(Collection data, int eachPieceSize) {
    if (CollectionUtils.isEmpty(data)) {
        return new ArrayList<>(0);
    }
    if (eachPieceSize <= 0) {
        throw new IllegalArgumentException("参数错误");
    }
    List> 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 randomValue = new EasyRandom().objects(String.class, 26).collect(Collectors.toList());
    // 每5个一组
    List> lists = ListUtil.splitToPieces(randomValue, 5);

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

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

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