Java 中 List 分片的 5 種方法！
作者 | 王磊
來源 | Java中文社群（ID：javacn666）
轉(zhuǎn)載請聯(lián)系授權(quán)（微信ID：GG_Stone）
前些天在實現(xiàn) MyBatis 批量插入時遇到了一個問題，當(dāng)批量插入的數(shù)據(jù)量比較大時，會導(dǎo)致程序執(zhí)行報錯，如下圖所示：
原因是 MySQL 只能執(zhí)行一定長度的 SQL 語句，但當(dāng)插入的數(shù)據(jù)量較多時，會生成一條很長的 SQL，這樣程序在執(zhí)行時就會報錯。
要解決這個問題，有兩種方法：第一，設(shè)置 MySQL 可以執(zhí)行 SQL 的最大長度；第二，將一個大 List 分成 N 個小 List 進行。由于無法準(zhǔn)確的界定程序中最大的 SQL 長度，所以最優(yōu)的解決方案還是第二種，于是就有了今天的這篇文章。
簡介
將一個 List 分成多個小 List 的過程，我們稱之為分片，當(dāng)然也可以叫做“List 分隔”，選一個你喜歡的、好理解的叫法就行。
在 Java 中，分片的常見實現(xiàn)方法有以下幾種：
使用 Google 的 Guava 框架實現(xiàn)分片； 使用 Apache 的 commons 框架實現(xiàn)分片； 使用國產(chǎn)神級框架 Hutool 實現(xiàn)分片； 使用 JDK 8 中提供 Stream 實現(xiàn)分片； 自定義分片功能。
接下來我們分別來看。
1.Google Guava
先在項目的 pom.xml 中添加框架支持，增加以下配置：
<dependency>
??<groupId>com.google.guavagroupId>
??<artifactId>guavaartifactId>
??<version>31.0.1-jreversion>
dependency>
有了 Guava 框架之后，只需要使用 Lists.partition 方法即可實現(xiàn)分片，如下代碼所示：
import?com.google.common.collect.Lists;
import?java.util.Arrays;
import?java.util.List;
/**
?*?Guava?分片
?*/
public?class?PartitionByGuavaExample?{
????//?原集合
????private?static?final?List?OLD_LIST?=?Arrays.asList(
????????????"唐僧,悟空,八戒,沙僧,曹操,劉備,孫權(quán)".split(","));
????public?static?void?main(String[]?args)?{
????????//?集合分片
????????List>?newList?=?Lists.partition(OLD_LIST,?3);
????????//?打印分片集合
????????newList.forEach(i?->?{
????????????System.out.println("集合長度："?+?i.size());
????????});
????}
}
以上代碼的執(zhí)行結(jié)果如下圖所示：
2.apache commons
先在項目的 pom.xml 中添加框架支持，增加以下配置：
<dependency>
??<groupId>org.apache.commonsgroupId>
??<artifactId>commons-collections4artifactId>
??<version>4.4version>
dependency>
有了 ?commons 框架之后，只需要使用 ListUtils.partition 方法即可實現(xiàn)分片，如下代碼所示：
import?org.apache.commons.collections4.ListUtils;
import?java.util.Arrays;
import?java.util.List;
/**
?*?commons.collections4?集合分片
?*/
public?class?PartitionExample?{
????//?原集合
????private?static?final?List?OLD_LIST?=?Arrays.asList(
????????????"唐僧,悟空,八戒,沙僧,曹操,劉備,孫權(quán)".split(","));
????public?static?void?main(String[]?args)?{
????????//?集合分片
????????List>?newList?=?ListUtils.partition(OLD_LIST,?3);
????????newList.forEach(i?->?{
????????????System.out.println("集合長度："?+?i.size());
????????});
????}
}
以上代碼的執(zhí)行結(jié)果如下圖所示：
3.Hutool
先在項目的 pom.xml 中添加框架支持，增加以下配置：
<dependency>
??<groupId>cn.hutoolgroupId>
??<artifactId>hutool-allartifactId>
??<version>5.7.14version>
dependency>
有了 Hutool 框架之后，只需要使用 ListUtil.partition 方法即可實現(xiàn)分片，如下代碼所示：
import?cn.hutool.core.collection.ListUtil;
import?java.util.Arrays;
import?java.util.List;
public?class?PartitionByHutoolExample?{
????//?原集合
????private?static?final?List?OLD_LIST?=?Arrays.asList(
????????????"唐僧,悟空,八戒,沙僧,曹操,劉備,孫權(quán)".split(","));
????public?static?void?main(String[]?args)?{
????????//?分片處理
????????List>?newList?=?ListUtil.partition(OLD_LIST,?3);
????????newList.forEach(i?->?{
????????????System.out.println("集合長度："?+?i.size());
????????});
????}
}
以上代碼的執(zhí)行結(jié)果如下圖所示：
4.JDK
Stream 通過 JDK 8 中的 ?Stream 來實現(xiàn)分片就無需添加任何框架了，具體的實現(xiàn)代碼如下：
import?java.util.Arrays;
import?java.util.List;
import?java.util.Map;
import?java.util.stream.Collectors;
/**
?*?JDK?Stream?Partition
?*/
public?class?PartitionByStreamExample?{
????//?原集合
????private?static?final?List?OLD_LIST?=?Arrays.asList(
????????????1,?2,?3,?4,?5,?6);
????public?static?void?main(String[]?args)?{
????????//?集合分片：將大于 3 和小于等于 3 的數(shù)據(jù)分別分為兩組
????????Map>?newMap?=?OLD_LIST.stream().collect(
????????????????Collectors.partitioningBy(i?->?i?>?3)
????????);
????????//?打印結(jié)果
????????System.out.println(newMap);
????}
}
以上代碼的執(zhí)行結(jié)果如下圖所示：
此方式的優(yōu)點的無需添加任何框架，但缺點是只能實現(xiàn)簡單的分片（將一個 List 分為兩個），并且要有明確的分片條件。比如本篇案例中設(shè)置的分片條件就是數(shù)組是否大于 3，如果大于 3 就會被歸為一組，否則就會被分到另一組。
5.自定義分片
如果你不想引入第三方框架，并且使用 Stream 也無法滿足你的需求，你就可以考慮自己寫代碼來實現(xiàn)分片功能了。因為此方式不常用，所以咱們這里只給出關(guān)鍵方法。
自定義分片功能的關(guān)鍵實現(xiàn)方法是 JDK 自帶的 subList 方法，如下圖所示：
使用示例如下：
import?java.util.Arrays;
import?java.util.List;
public?class?App?{
????private?static?final?List?_OLD_LIST?=?Arrays.asList(
????????????"唐僧,悟空,八戒,沙僧,曹操,劉備,孫權(quán)".split(","));
????public?static?void?main(String[]?args)?{
????????//?集合分隔
????????List?list?=?_OLD_LIST.subList(0,?3);
????????//?打印集合中的元素
????????list.forEach(i?->?{
????????????System.out.println(i);
????????});
????}
}
以上代碼的執(zhí)行結(jié)果如下圖所示：
總結(jié)
本文介紹了 5 種 List 分片的實現(xiàn)方法，其中最方便的實現(xiàn)方式是引入第三方框架，比如 Google 的 Guava、Apache 的 Commons 或者是國產(chǎn)開源的 Hutool 都可以，當(dāng)然如果你的項目已經(jīng)包含了以上任意一種，直接使用就行了。如果是簡單的分片就可以考慮使用 JDK 的 Stream 或者是 List 內(nèi)置的 subList 方法來實現(xiàn)分片功能了。
往期推薦
MyBatis原生批量插入的坑與解決方案！
4 種方法！檢查字符串是否為合法的日期格式
高并發(fā)下秒殺商品，必須知道的9個細節(jié)