1. 程式人生 > 其它 >Java 中 List 分片的 5 種方法

Java 中 List 分片的 5 種方法

前些天在實現 MyBatis 批量插入時遇到了一個問題,當批量插入的資料量比較大時,會導致程式執行報錯,如下圖所示:

原因是 MySQL 只能執行一定長度的 SQL 語句,但當插入的資料量較多時,會生成一條很長的 SQL,這樣程式在執行時就會報錯。

要解決這個問題,有兩種方法:第一,設定 MySQL 可以執行 SQL 的最大長度;第二,將一個大 List 分成 N 個小 List 進行。由於無法準確的界定程式中最大的 SQL 長度,所以最優的解決方案還是第二種,於是就有了今天的這篇文章。

簡介

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

在 Java 中,分片的常見實現方法有以下幾種:

  1. 使用 Google 的 Guava 框架實現分片;
  2. 使用 Apache 的 commons 框架實現分片;
  3. 使用國產神級框架 Hutool 實現分片;
  4. 使用 JDK 8 中提供 Stream 實現分片;
  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 方法即可實現分片,如下程式碼所示:

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 方法來實現分片功能了。