java producer consumer

Main.java

package com.example;

import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.*;

public class Main {
    static void run(int inputSize, int batchSize) throws Exception {
        ExecutorService executor = Executors.newCachedThreadPool();
        BlockingQueue<Integer> q = new LinkedBlockingQueue<>(2 * batchSize);

        Future<Void> p = executor.submit(() -> {
            for (int i = 0; i < inputSize; i++) {
                q.put(i);
            }
            return null;
        });

        while (!p.isDone() || !q.isEmpty()) {
            executor.submit(() -> {
                List<Integer> lst = new ArrayList<>();
                q.drainTo(lst, batchSize);
                System.out.println(lst);
                return null;
            }).get();
        }
        executor.shutdown();
    }

    public static void main(String[] args) throws Exception {
        run(18, 5);
        run(0, 2);
        run(10, 5);
        run(10, 4);
    }
}