【编程技巧】Lists.partition批量处理集合数据

【编程技巧】Lists.partition批量处理集合数据

---

---

前言

你是否也有同样的苦恼？需要对集合做分批处理，比如批量插入数据库需要限制数量，每次插入指定条数等操作。废话不多说，老规矩，直接上代码

使用需要引入maven包

		<dependency>
            <groupId>com.google.guava</groupId>
            <artifactId>guava</artifactId>
            <version>22.0</version>
        </dependency>

---

一、原始写法

以下代码是我在批量插入时常用的代码，通过遍历集合用新的集合接收每一批插入的数据，看起来是不是逻辑又复杂又容易写出bug？

    public final static int BATCH_INSERT_SIZE = 100;

    public static void main(String[] args) {
        List<String> list = new ArrayList<>();

        int size = list.size();
        List<String> insertList = new ArrayList<>();

        int insertSize;
        for (String word : list) {
            insertList.add(word);
            insertSize = insertList.size();

            if (size > BATCH_INSERT_SIZE && BATCH_INSERT_SIZE > insertSize){continue;}

            if (size < BATCH_INSERT_SIZE && size > insertSize){continue;}

            batchInsert(insertList);
            insertList.clear();
            size -= BATCH_INSERT_SIZE;
            log.info("批量数量：{}",insertSize);
        }
    }

    private static void batchInsert(List<String> insertList) {
        System.out.println(insertList);
    }

二、Lists.partition优雅写法

使用Lists.partition一行代码搞定,就问你优雅不优雅？

public final static int BATCH_INSERT_SIZE = 100;

    public static void main(String[] args) {
        List<String> list = new ArrayList<>();

        List<List<String>> partitionList = Lists.partition(list, BATCH_INSERT_SIZE);
        for (List<String> partitions : partitionList) {
            batchInsert(partitions);
            log.info("批量插入数量：{}", partitions.size());
        }
    }

    private static void batchInsert(List<String> insertList) {
        System.out.println(insertList);
    }

---

总结

巧妙使用前人的轮子能够使代码变得更简洁优雅，合理的cv，哦不借鉴能让自己更强，然后造出属于自己的轮子。