【Java】List集合按数量分组

有时候，我们需要将大的集合按指定的数量分割成若干个小集合。（比如：集合作为SQL中IN的参数，而SQL又有长度限制，所以需要分批分几次进行查询）

虽然此需求感觉不常见，但偶也写过几次类似的方法，故记录之。

更新于2017年：其实Guava库有个已有的方法实现此需求：Lists.partition(List<T> list, int size)

v2，更新于2016-01-20

v1的代码使用后发现有问题，如果对分组后的一子集作删除操作，其他子集用迭代器遍历时会出现ConcurrentModificationException。

修改后的代码如下:

import java.util.ArrayList;
import java.util.List;

public class CollectionGroupUtil {

    public static List groupListByQuantity(List list, int quantity) {
        if (list == null || list.size() == 0) {
            return list;
        }

        if (quantity <= 0) {
            new IllegalArgumentException("Wrong quantity.");
        }

        List wrapList = new ArrayList();
        int count = 0;
        while (count < list.size()) {
            wrapList.add(new ArrayList(list.subList(count, (count + quantity) > list.size() ? list.size() : count + quantity)));
            count += quantity;
        }

        return wrapList;
    }

}

import java.util.ArrayList;
import java.util.List;

import org.junit.Test;

public class CollectionGroupUtilTest {

    /**
     * 大于分组数量的情况
     */
    @Test
    public void test() {
        List<String> allList = new ArrayList<String>();
        for (int i = 1; i <= 504; i++) {
            allList.add(i  + "");
        }

        List<List<String>> groupList = CollectionGroupUtil.groupListByQuantity(allList, 50);
        int i = 0;
        List<String> list = null;
        for (int c = 0; c < groupList.size(); c++) {
            list = groupList.get(c);

            System.out.println("第" + (c + 1) + "组： ");
            for (String temp : list) {
                System.out.print(temp + ", ");
            }
            System.out.println();
        }
    }

    /**
     * 小于分组数量的情况
     */
    @Test
    public void test2() {
        List<String> allList = new ArrayList<String>();
        for (int i = 1; i <= 45; i++) {
            allList.add(i  + "");
        }

        List<List<String>> groupList = CollectionGroupUtil.groupListByQuantity(allList, 50);
        int i = 0;
        List<String> list = null;
        for (int c = 0; c < groupList.size(); c++) {
            list = groupList.get(c);

            System.out.println("第" + (c + 1) + "组： ");
            for (String temp : list) {
                System.out.print(temp + ", ");
            }
            System.out.println();
        }
    }

    /**
     * 集合只有一个记录的情况
     */
    @Test
    public void test3() {
        List<String> allList = new ArrayList<String>();
        for (int i = 1; i <= 1; i++) {
            allList.add(i  + "");
        }

        List<List<String>> groupList = CollectionGroupUtil.groupListByQuantity(allList, 50);
        int i = 0;
        List<String> list = null;
        for (int c = 0; c < groupList.size(); c++) {
            list = groupList.get(c);

            System.out.println("第" + (c + 1) + "组： ");
            for (String temp : list) {
                System.out.print(temp + ", ");
            }
            System.out.println();
        }
    }

    /**
     * 空集合的情况
     */
    @Test
    public void test4() {
        List<List<String>> groupList = CollectionGroupUtil.groupListByQuantity(null, 50);
        System.out.println(groupList);

        groupList = CollectionGroupUtil.groupListByQuantity(new ArrayList(), 50);
        System.out.println(groupList);
    }

    /**
     * 集合刚满一个分组的情况
     */
    @Test
    public void test5() {
        List<String> allList = new ArrayList<String>();
        for (int i = 1; i <= 50; i++) {
            allList.add(i  + "");
        }

        List<List<String>> groupList = CollectionGroupUtil.groupListByQuantity(allList, 50);
        int i = 0;
        List<String> list = null;
        for (int c = 0; c < groupList.size(); c++) {
            list = groupList.get(c);

            System.out.println("第" + (c + 1) + "组： ");
            for (String temp : list) {
                System.out.print(temp + ", ");
            }
            System.out.println();
        }
    }

    /**
     * 出现ConcurrentModificationException的情况
     */
    @Test
    public void test6() {
        List<String> allList = new ArrayList<String>();
        for (int i = 1; i <= 55; i++) {
            allList.add(i  + "");
        }

        List<List<String>> groupList = CollectionGroupUtil.groupListByQuantity(allList, 50);
        groupList.get(0).remove(0);

        int i = 0;
        List<String> list = null;
        for (int c = 0; c < groupList.size(); c++) {
            list = groupList.get(c);

            System.out.println("第" + (c + 1) + "组： ");
            for (String temp : list) {
                System.out.print(temp + ", ");
            }
            System.out.println();
        }
    }

}

以简单的方式重现ConcurrentModificationException异常：

import java.util.ArrayList;
import java.util.List;

public class SeeConcurrentModificationExceptionInSubList {

    public static void main(String[] args) {
        List<String> originalList = new ArrayList<String>();
        for (int i = 1; i <= 5; i++) {
            originalList.add(i + "");
        }

        List<String> subList1 = originalList.subList(0, 3);
        List<String> subList2 = originalList.subList(3, originalList.size());

        /* 按坐标删除 */
        subList1.remove(0);

        /* 用迭代器删除 */
        /*
        Iterator<String> i = subList1.iterator();
        i.next();
        i.remove();
        */

        System.out.println("subList1 -> ");
        for (String tempStr : subList1) {
            System.out.println(tempStr);
        }

        System.out.println("subList2 -> ");
        for (String tempStr : subList2) {
            System.out.println(tempStr);
        }

    }

}

v1，更新于2016-01-18

工具类

import java.util.ArrayList;
import java.util.List;

public class CollectionGroupUtil {

    public static List groupListByQuantity(List list, int quantity) {
        if (list == null || list.size() == 0) {
            return list;
        }

        if (quantity <= 0) {
            new IllegalArgumentException("Wrong quantity.");
        }

        List wrapList = new ArrayList();
        int count = 0;
        while (count < list.size()) {
            wrapList.add(list.subList(count, (count + quantity) > list.size() ? list.size() : count + quantity));
            count += quantity;
        }

        return wrapList;
    }

}

测试类

import java.util.ArrayList;
import java.util.List;

import org.junit.Test;

public class CollectionGroupUtilTest {

    /**
     * 大于分组数量的情况
     */
    @Test
    public void test() {
        List<String> allList = new ArrayList<String>();
        for (int i = 1; i <= 504; i++) {
            allList.add(i  + "");
        }

        List<List<String>> groupList = CollectionGroupUtil.groupListByQuantity(allList, 50);
        int i = 0;
        List<String> list = null;
        for (int c = 0; c < groupList.size(); c++) {
            list = groupList.get(c);

            System.out.println("第" + (c + 1) + "组： ");
            for (String temp : list) {
                System.out.print(temp + ", ");
            }
            System.out.println();
        }
    }

    /**
     * 小于分组数量的情况
     */
    @Test
    public void test2() {
        List<String> allList = new ArrayList<String>();
        for (int i = 1; i <= 45; i++) {
            allList.add(i  + "");
        }

        List<List<String>> groupList = CollectionGroupUtil.groupListByQuantity(allList, 50);
        int i = 0;
        List<String> list = null;
        for (int c = 0; c < groupList.size(); c++) {
            list = groupList.get(c);

            System.out.println("第" + (c + 1) + "组： ");
            for (String temp : list) {
                System.out.print(temp + ", ");
            }
            System.out.println();
        }
    }

    /**
     * 集合只有一个记录的情况
     */
    @Test
    public void test3() {
        List<String> allList = new ArrayList<String>();
        for (int i = 1; i <= 1; i++) {
            allList.add(i  + "");
        }

        List<List<String>> groupList = CollectionGroupUtil.groupListByQuantity(allList, 50);
        int i = 0;
        List<String> list = null;
        for (int c = 0; c < groupList.size(); c++) {
            list = groupList.get(c);

            System.out.println("第" + (c + 1) + "组： ");
            for (String temp : list) {
                System.out.print(temp + ", ");
            }
            System.out.println();
        }
    }

    /**
     * 空集合的情况
     */
    @Test
    public void test4() {
        List<List<String>> groupList = CollectionGroupUtil.groupListByQuantity(null, 50);
        System.out.println(groupList);

        groupList = CollectionGroupUtil.groupListByQuantity(new ArrayList(), 50);
        System.out.println(groupList);
    }

    /**
     * 集合刚满一个分组的情况
     */
    @Test
    public void test5() {
        List<String> allList = new ArrayList<String>();
        for (int i = 1; i <= 50; i++) {
            allList.add(i  + "");
        }

        List<List<String>> groupList = CollectionGroupUtil.groupListByQuantity(allList, 50);
        int i = 0;
        List<String> list = null;
        for (int c = 0; c < groupList.size(); c++) {
            list = groupList.get(c);

            System.out.println("第" + (c + 1) + "组： ");
            for (String temp : list) {
                System.out.print(temp + ", ");
            }
            System.out.println();
        }
    }

}