一、背景
平时业务开发可能有这样的一个场景:一个集合有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来构造测试对象,省时省力。
点击查看更多内容
1人点赞
评论
共同学习,写下你的评论
评论加载中...
作者其他优质文章
正在加载中
感谢您的支持,我会继续努力的~
扫码打赏,你说多少就多少
赞赏金额会直接到老师账户
支付方式
打开微信扫一扫,即可进行扫码打赏哦