当前位置：首页 > article >正文

java 大集合切分成一个集合中有多个小集合

article 2024/10/30 18:45:06

直接上代码

List<OtsAttendance> otsAttendanceList = new ArrayList<>();
举个例子
otsAttendanceList 该集合里面有很多个对象。
那么如果批量新增的话，不太好，需要拆分成,
（一个集合，然后泛型又是一个集合）
例如说,otsAttendanceList  有五万条数据。拆分成
List<List<OtsAttendance>> batchedLists
batchedLists 有50条，而里面的List<OtsAttendance>有1000条数据。

下面是代码

这个代码就不写注释了，相信大家都能看懂。
List<String> jobNums = otsAttendanceList.stream().map(OtsAttendance::getJobNum).collect(Collectors.toList());

List<List<String>> batchedJobNums = batchList(jobNums, 1000);

batchedJobNums.forEach(batch -> {
	this.remove(new LambdaQueryWrapper<OtsAttendance>().in(OtsAttendance::getJobNum, batch)
		.ge(OtsAttendance::getCheckDate, fromDate)
		.le(OtsAttendance::getCheckDate, toDate));
});

List<List<OtsAttendance>> batchedLists = batchList(otsAttendanceList, 1000);

batchedLists.forEach(batch -> this.saveBatch(batch));

下面写这个batchList拆分方法的代码

private <T> List<List<T>> batchList(List<T> list, int batchSize) {
	List<List<T>> batchedLists = new ArrayList<>();
	int size = list.size();
	int index = 0;

	while (index < size) {
		int endIndex = Math.min(index + batchSize, size);
		batchedLists.add(list.subList(index, endIndex));
		index = endIndex;
	}

	return batchedLists;
}

我记得这个拆分的方法(batchList)是有其他jdk类的方法可以替换的，有时间我这边在找一下。