前些天在實現 MyBatis 批量插入時遇到了一個問題，當批量插入的數據量比較大時，會導致程序執行報錯，如下圖所示：
原因是 MySQL 只能執行一定長度的 SQL 語句，但當插入的數據量較多時，會生成一條很長的 SQL，這樣程序在執行時就會報錯。
要解決這個問題，有兩種方法：第一，設置 MySQL 可以執行 SQL 的最大長度；第二，將一個大 List 分成 N 個小 List 進行。由於無法准確的界定程序中最大的 SQL 長度，所以最優的解決方案還是第二種，於是就有了今天的這篇文章。
簡介
將一個 List 分成多個小 List 的過程，我們稱之為分片，當然也可以叫做“List 分隔”，選一個你喜歡的、好理解的叫法就行。
在 Java 中，分片的常見實現方法有以下幾種：
- 使用 Google 的 Guava 框架實現分片；
- 使用 Apache 的 commons 框架實現分片；
- 使用國產神級框架 Hutool 實現分片；
- 使用 JDK 8 中提供 Stream 實現分片；
- 自定義分片功能。
接下來我們分別來看。
1.Google Guava
先在項目的 pom.xml 中添加框架支持，增加以下配置：
<!-- google guava 工具類 -->
<!-- https://mvnrepository.com/artifact/com.google.guava/guava -->
<dependency>
<groupId>com.google.guava</groupId>
<artifactId>guava</artifactId>
<version>31.0.1-jre</version>
</dependency>
有了 Guava 框架之后，只需要使用 Lists.partition 方法即可實現分片，如下代碼所示：
import com.google.common.collect.Lists;
import java.util.Arrays;
import java.util.List;
/**
* Guava 分片
*/
public class PartitionByGuavaExample {
// 原集合
private static final List<String> OLD_LIST = Arrays.asList(
"唐僧,悟空,八戒,沙僧,曹操,劉備,孫權".split(","));
public static void main(String[] args) {
// 集合分片
List<List<String>> newList = Lists.partition(OLD_LIST, 3);
// 打印分片集合
newList.forEach(i -> {
System.out.println("集合長度：" + i.size());
});
}
}
以上代碼的執行結果如下圖所示：
2.apache commons
先在項目的 pom.xml 中添加框架支持，增加以下配置：
<!-- apache 集合工具類 -->
<!-- https://mvnrepository.com/artifact/org.apache.commons/commons-collections4 -->
<dependency>
<groupId>org.apache.commons</groupId>
<artifactId>commons-collections4</artifactId>
<version>4.4</version>
</dependency>
有了 commons 框架之后，只需要使用 ListUtils.partition 方法即可實現分片，如下代碼所示：
import org.apache.commons.collections4.ListUtils;
import java.util.Arrays;
import java.util.List;
/**
* commons.collections4 集合分片
*/
public class PartitionExample {
// 原集合
private static final List<String> OLD_LIST = Arrays.asList(
"唐僧,悟空,八戒,沙僧,曹操,劉備,孫權".split(","));
public static void main(String[] args) {
// 集合分片
List<List<String>> newList = ListUtils.partition(OLD_LIST, 3);
newList.forEach(i -> {
System.out.println("集合長度：" + i.size());
});
}
}
以上代碼的執行結果如下圖所示：
3.Hutool
先在項目的 pom.xml 中添加框架支持，增加以下配置：
<!-- 工具類 hutool -->
<!-- https://mvnrepository.com/artifact/cn.hutool/hutool-all -->
<dependency>
<groupId>cn.hutool</groupId>
<artifactId>hutool-all</artifactId>
<version>5.7.14</version>
</dependency>
有了 Hutool 框架之后，只需要使用 ListUtil.partition 方法即可實現分片，如下代碼所示：
import cn.hutool.core.collection.ListUtil;
import java.util.Arrays;
import java.util.List;
public class PartitionByHutoolExample {
// 原集合
private static final List<String> OLD_LIST = Arrays.asList(
"唐僧,悟空,八戒,沙僧,曹操,劉備,孫權".split(","));
public static void main(String[] args) {
// 分片處理
List<List<String>> newList = ListUtil.partition(OLD_LIST, 3);
newList.forEach(i -> {
System.out.println("集合長度：" + i.size());
});
}
}
以上代碼的執行結果如下圖所示：
4.JDK
Stream
通過 JDK 8 中的 Stream 來實現分片就無需添加任何框架了，具體的實現代碼如下：
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<Integer> OLD_LIST = Arrays.asList(
1, 2, 3, 4, 5, 6);
public static void main(String[] args) {
// 集合分片：將大於 3 和小於等於 3 的數據分別分為兩組
Map<Boolean, List<Integer>> newMap = OLD_LIST.stream().collect(
Collectors.partitioningBy(i -> i > 3)
);
// 打印結果
System.out.println(newMap);
}
}
以上代碼的執行結果如下圖所示：
此方式的優點的無需添加任何框架，但缺點是只能實現簡單的分片（將一個 List 分為兩個），並且要有明確的分片條件。比如本篇案例中設置的分片條件就是數組是否大於 3，如果大於 3 就會被歸為一組，否則就會被分到另一組。
5.自定義分片
如果你不想引入第三方框架，並且使用 Stream 也無法滿足你的需求，你就可以考慮自己寫代碼來實現分片功能了。因為此方式不常用，所以咱們這里只給出關鍵方法。
自定義分片功能的關鍵實現方法是 JDK 自帶的 subList 方法，如下圖所示：
使用示例如下：
import java.util.Arrays;
import java.util.List;
public class App {
private static final List<String> _OLD_LIST = Arrays.asList(
"唐僧,悟空,八戒,沙僧,曹操,劉備,孫權".split(","));
public static void main(String[] args) {
// 集合分隔
List<String> list = _OLD_LIST.subList(0, 3);
// 打印集合中的元素
list.forEach(i -> {
System.out.println(i);
});
}
}
以上代碼的執行結果如下圖所示：
總結
本文介紹了 5 種 List 分片的實現方法，其中最方便的實現方式是引入第三方框架，比如 Google 的 Guava、Apache 的 Commons 或者是國產開源的 Hutool 都可以，當然如果你的項目已經包含了以上任意一種，直接使用就行了。如果是簡單的分片就可以考慮使用 JDK 的 Stream 或者是 List 內置的 subList 方法來實現分片功能了。
關注公眾號「Java中文社群」查看更多 Java 總結性系列文章。