java多线程均摊任务

多线程均摊任务

学习记录

代码如下:

 public static void main(String[] args) {
        Date startDate = new Date();
        try {
            //创建一个任务提供给线程
            List<String> list = new ArrayList<>();
            for (int i = 0; i < 100; i++) {
                list.add(i + ",");
            }
            //创建5个线程并启动,可更改成任意数量线程
            System.out.println(new Test().list2Str(list, 5));
        } catch (Exception e) {
            e.printStackTrace();
        }finally {
            System.out.println("=====================耗时" + (new Date().getTime() - startDate.getTime()) / 1000 + "s");
        }
    }

    public String list2Str(final List<String> list, final int nThreads) throws Exception {
        if (list == null || list.isEmpty()) {
            return null;
        }
        int len = 0;
        for (String str : list) {
            len += str.length();
        }
        StringBuffer ret = new StringBuffer(len);

        final int size = list.size();
        ExecutorService executorService = Executors.newFixedThreadPool(nThreads);
        List<Future<String>> futures = new ArrayList<Future<String>>(nThreads);
        try {
            for (int i = 0; i < nThreads; i++) {
                int d = i;
                final int j = i;
                Callable<String> task = new Callable<String>() {
                    @Override
                    public String call() throws Exception {
                        int len = 0;
                        //读取数据总长度防止内存溢出
                        for (int n = size / nThreads * j; n < size / nThreads * (j + 1); n++) {
                            len += list.get(n).length();
                        }
                        //把数据按算法平均分给线程
                        StringBuffer sb = new StringBuffer(len);
                        for (int n = size / nThreads * j; n < size / nThreads * (j + 1); n++) {
                            sb.append(list.get(n));
                            System.out.println("第"+d+"个线程:"+list.get(n));
                        }
                        return sb.toString();
                    }
                };
                futures.add(executorService.submit(task));
            }

            for (Future<String> future : futures) {
                ret.append(future.get());
            }
        } finally {
            executorService.shutdown();
        }

        return ret.toString();
    }

结果如下: