java8 批量处理

java8 批量处理

 

import java.util.AbstractMap;

import java.util.List;
import java.util.Optional;
import java.util.function.Consumer;
import java.util.stream.Collectors;
import java.util.stream.IntStream;

public static <T> void divideBatchHandler(List<T> dataList, Consumer<List<T>> consumer, int size) {
	Optional.ofNullable(dataList)
			.ifPresent(list -> IntStream.range(0, list.size())
					.mapToObj(i -> new AbstractMap.SimpleImmutableEntry<>(i, list.get(i)))// 给数据编号
					.collect(Collectors.groupingBy(e -> e.getKey() / size,
							Collectors.mapping(Map.Entry::getValue, Collectors.toList()))) // 按编号分批并合并编号对应的值
					.values().parallelStream()// 并行处理
					.forEach(consumer) // 执行处理函数
			);
}

 

调用：

 

@Test
public void batchHandle() {
	/* 批量处理测试 */
	List<String> list = new ArrayList<>();//projectMongodbDao.findAll();
	list.add("a");
	list.add("b");
	list.add("c");
	list.add("d");
	// 批量处理
	CommonUtils.divideBatchHandler(list, (x) -> print(x), 3);
}

private void print(List<String> list) {
	System.out.println(list);
}