java的List分批工具
-
在项目中,我们可能有些时候需要对List进行分割。
-
比如一个List里面有2021个元素。
-
那么我需要按照100个为1批(最后不足100为1批),将其分为21组
这里提供两种思路。
思路1. 纯手工写一个工具类
import java.util.ArrayList;
import java.util.List;
public class SplitList {
public static void main(String[] args) {
List<Integer> list = generateList(2021);
List<List<Integer>> split = split(list, 100);
System.out.println(split);
}
public static <T> List<List<T>> split(List<T> list, int batchSize) {
List<List<T>> res = new ArrayList<>();
int k = 0;
List<T> temp = new ArrayList<>();
for (T t : list) {
temp.add(t);
k++;
if (k >= batchSize) {
k = 0;
res.add(temp);
temp = new ArrayList<>();
}
}
if (!temp.isEmpty()) {
res.add(temp);
}
return res;
}
private static List<Integer> generateList(int count) {
List<Integer> list = new ArrayList<>();
for (int i = 0; i < count; i++) {
list.add(i);
}
return list;
}
}
思路2 引入第三方的jar包,若项目为maven管理,也可以采用这种方式
pom.xml中添加如下依赖
<dependency>
<groupId>com.google.guava</groupId>
<artifactId>guava</artifactId>
<version>23.0</version>
</dependency>
import com.google.common.collect.Lists;
import java.util.ArrayList;
import java.util.List;
public class SplitApi {
public static void main(String[] args) {
List<Integer> list = generateList(2021);
List<List<Integer>> split = Lists.partition(list,100);
System.out.println(split);
}
private static List<Integer> generateList(int count) {
List<Integer> list = new ArrayList<>();
for (int i = 0; i < count; i++) {
list.add(i);
}
return list;
}
}