我们在使用java中的集合(Collection、List、Set等)的时候,可能因为某种限制需要将大集合按照固定的长度拆分为N个小的集合。
如:有一个List<String> list 里面有1099个原始,我需要按照固定长度50,来拆分这个集合为很多小集合,每个集合的原始是50个元素,可能最后一个小集合的长度不到50.
如何做呢?网上有不少资料,代码如下:
/**
* 拆分集合
*
* @param <T> 泛型对象
* @param resList 需要拆分的集合
* @param subListLength 每个子集合的元素个数
* @return 返回拆分后的各个集合组成的列表
* 代码里面用到了guava和common的结合工具类
**/
public static <T> List<List<T>> split(List<T> resList, int subListLength) {
if (CollectionUtils.isEmpty(resList) || subListLength <= 0) {
return Lists.newArrayList();
}
List<List<T>> ret = Lists.newArrayList();
int size = resList.size();
if (size <= subListLength) {
// 数据量不足 subListLength 指定的大小
ret.add(resList);
} else {
int pre = size / subListLength;
int last = size % subListLength;
// 前面pre个集合,每个大小都是 subListLength 个元素
for (int i = 0; i < pre; i++) {
List<T> itemList = Lists.newArrayList();
for (int j = 0; j < subListLength; j++) {
itemList.add(resList.get(i * subListLength + j));
}
ret.add(itemList);
}
// last的进行处理
if (last > 0) {
List<T> itemList = Lists.newArrayList();
for (int i = 0; i < last; i++) {
itemList.add(resList.get(pre * subListLength + i));
}
ret.add(itemList);
}
}
return ret;
}
// 运行代码
public static void main(String[] args) {
List<String> list = Lists.newArrayList();
int size = 1099;
for (int i = 0; i < size; i++) {
list.add("hello-" + i);
}
List<List<String>> temps = split(list, 6);
int j = 0;
for (List<String> obj : temps) {
System.out.println(String.format("row:%s -> size:%s,data:%s", ++j, obj.size(), obj));
}
}
其实我平日用的更多的是guava框架,它自带了一个方法可以实现上面的集合切割的功能。代码如下:
public static void main(String[] args) {
List<String> list = Lists.newArrayList();
int size = 1099;
for (int i = 0; i < size; i++) {
list.add("hello-" + i);
}
// 切割大集合到指定的长度:11
List<List<String>> rsList = Lists.partition(list, 11);
int i = 0;
for (List<String> obj : rsList) {
System.out.println(String.format("row:%s -> size:%s,data:%s", ++i, obj.size(), obj));
}
}
推荐大家使用guava的功能,它们都是经过严格的单元测试的代码。