多线程批处理---Fork/Join框架

Fork/Join框架是Java7提供的一个用于并行执行任务的框架, 是一个把大任务分割成若干个小任务,最终汇总每个小任务结果后得到大任务结果的框架。


下列代码场景是启动5个线程同时处理1w条记录的 list,最后把这个List遍历的的内容拼接起来。

package threadTest;

import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.ForkJoinPool;
import java.util.concurrent.Future;
import java.util.concurrent.RecursiveTask;

/**
 * @Auther: xzs
 * @Date: 2018/5/31 10:58
 * @Description:
 */
public class ThreadList {
    private List<String> list;
    public static void main(String[] args) {
        System.out.println(new ThreadList().run().length());
    }

    private void init(){
        //模拟初始化1W条数据
        list = new ArrayList<String>();
        for (int i = 0; i < 10000; i++) {
            list.add(i + "");
        }
    }

    public String run(){
        init();
        ForkJoinPool pool = new ForkJoinPool(5);
        Task task = new Task(list);
        Future<String> result =  pool.submit(task);
        String str = "";
        try {
            str = result.get();
        } catch (InterruptedException e) {
            e.printStackTrace();
        } catch (ExecutionException e) {
            e.printStackTrace();
        }
        return str;
    }

    class Task extends RecursiveTask<String> {
        private int size = 2000;
        private List<String> data;
        public Task(List<String> data){
            this.data = data;
        }
        @Override
        protected String compute() {
            StringBuffer sb = new StringBuffer();
            if(data.size() <= size){
                System.out.println("******************************** size:" + data.size());
                for (String str : data) {
                    sb.append(str);
                }
            }else{
                //细分成小任务
                List<Task> tasks = new ArrayList<Task>();
                for (int index = 0; index * size < data.size(); index++) {
                    Task task;
                    if((index + 1) * size > data.size()){
                        task = new Task(data.subList(index * size, data.size()));
                    }else{
                        task = new Task(data.subList(index * size, (index + 1) * size));
                    }
                    task.fork();
                    tasks.add(task);
                }
                for (Task task : tasks) {
                    sb.append(task.join());
                }
            }

            return sb.toString();
        }

    }
}
----- 借鉴大神整理
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值