Java 中 List 分片的 5 種方法
前些天在實現 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 就會被歸為一組,否則就會被分到另一組。java培訓
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 方法來實現分片功能了。