[[432699]]

本文轉載自微信公眾號「Java中文社群」，作者磊哥  。轉載本文請聯(lián)系Java中文社群公眾號。

前些天在實現(xiàn) MyBatis 批量插入時遇到了一個問題，當批量插入的數據量比較大時，會導致程序執(zhí)行報錯，如下圖所示：

原因是 MySQL 只能執(zhí)行一定長度的 SQL 語句，但當插入的數據量較多時，會生成一條很長的 SQL，這樣程序在執(zhí)行時就會報錯。

要解決這個問題，有兩種方法：第一，設置 MySQL 可以執(zhí)行 SQL 的最大長度;第二，將一個大 List 分成 N 個小 List 進行。由于無法準確的界定程序中最大的 SQL 長度，所以最優(yōu)的解決方案還是第二種，于是就有了今天的這篇文章。

簡介

將一個 List 分成多個小 List 的過程，我們稱之為分片，當然也可以叫做“List 分隔”，選一個你喜歡的、好理解的叫法就行。

在 Java 中，分片的常見實現(xiàn)方法有以下幾種：

1. 使用 Google 的 Guava 框架實現(xiàn)分片;
2. 使用 Apache 的 commons 框架實現(xiàn)分片;
3. 使用國產神級框架 Hutool 實現(xiàn)分片;
4. 使用 JDK 8 中提供 Stream 實現(xiàn)分片;
5. 自定義分片功能。

接下來我們分別來看。

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 方法即可實現(xiàn)分片，如下代碼所示：

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()); 
        }); 
    } 
} 

以上代碼的執(zhí)行結果如下圖所示：

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 方法即可實現(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<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()); 
        }); 
    } 
} 

以上代碼的執(zhí)行結果如下圖所示：

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 方法即可實現(xiàn)分片，如下代碼所示：

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()); 
        }); 
    } 
} 

以上代碼的執(zhí)行結果如下圖所示：

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<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); 
    } 
} 

以上代碼的執(zhí)行結果如下圖所示：

圖片此方式的優(yōu)點的無需添加任何框架，但缺點是只能實現(xiàn)簡單的分片(將一個 List 分為兩個)，并且要有明確的分片條件。比如本篇案例中設置的分片條件就是數組是否大于 3，如果大于 3 就會被歸為一組，否則就會被分到另一組。

5.自定義分片

如果你不想引入第三方框架，并且使用 Stream 也無法滿足你的需求，你就可以考慮自己寫代碼來實現(xiàn)分片功能了。因為此方式不常用，所以咱們這里只給出關鍵方法。

自定義分片功能的關鍵實現(xiàn)方法是 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); 
        }); 
    } 
} 

以上代碼的執(zhí)行結果，如下圖所示：

總結 

本文介紹了 5 種 List 分片的實現(xiàn)方法，其中最方便的實現(xiàn)方式是引入第三方框架，比如 Google 的 Guava、Apache 的 Commons 或者是國產開源的 Hutool 都可以，當然如果你的項目已經包含了以上任意一種，直接使用就行了。如果是簡單的分片就可以考慮使用 JDK 的 Stream 或者是 List 內置的 subList 方法來實現(xiàn)分片功能了。