schedule-pool模拟并行任务分片
模拟并行任务分片
代码部分:
package com.pool;
import com.alibaba.fastjson.JSON;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.Reader;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.Executors;
import java.util.concurrent.ScheduledThreadPoolExecutor;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicInteger;
/**
* Created on 2016/5/4.
*/
public class ScheduledPoolTest {
public static void main(String[] args) {
int len = 3;
// 定时任务线程池,长度为3
ScheduledThreadPoolExecutor executor = (ScheduledThreadPoolExecutor) Executors.newScheduledThreadPool(len);
// 添加3个任务,每5秒执行一次,并行处理
for (int i = 0; i < 3; i++) {
executor.scheduleAtFixedRate(new TaskThread(len, i), 0L, 5L, TimeUnit.SECONDS);
}
/**
* 手动关闭pool,没什么太大的用
*/
try (
Reader reader = new InputStreamReader(System.in);
BufferedReader in = new BufferedReader(reader)
) {
while (true) {
String line = in.readLine();
if (line.equalsIgnoreCase("no")) {
executor.shutdown();
break;
}
}
} catch (IOException e) {
e.printStackTrace();
}
}
}
/**
* 模拟切片任务
*/
class TaskThread implements Runnable {
private int increase = 0;
private List<Integer> list;
// 切片总数
private int len;
// 切片余数
private int mod;
private int size = 10000;
public TaskThread(int len, int mod) {
this.len = len;
this.mod = mod;
}
@Override
public void run() {
list = new ArrayList<>();
for (int i = increase * size; i < (increase + 1) * size; i++) {
// 根据 mod 分片
if (i % len == mod) {
list.add(i);
}
}
increase++;
System.out.println("**********************************************************");
System.out.println(Thread.currentThread().getName() + " : " + JSON.toJSONString(list));
System.out.println("**********************************************************");
}
}
运行效果:
**********************************************************
**********************************************************
**********************************************************
pool-1-thread-2 : [1,4,7,10,13,16,19,22,25,28,31,34,37,40,43,46,49,52,55,58,61,64,67,70,73,76,79,82,85,88,91,94,97]
**********************************************************
pool-1-thread-3 : [2,5,8,11,14,17,20,23,26,29,32,35,38,41,44,47,50,53,56,59,62,65,68,71,74,77,80,83,86,89,92,95,98]
**********************************************************
pool-1-thread-1 : [0,3,6,9,12,15,18,21,24,27,30,33,36,39,42,45,48,51,54,57,60,63,66,69,72,75,78,81,84,87,90,93,96,99]
**********************************************************
**********************************************************
pool-1-thread-1 : [102,105,108,111,114,117,120,123,126,129,132,135,138,141,144,147,150,153,156,159,162,165,168,171,174,177,180,183,186,189,192,195,198]
**********************************************************
**********************************************************
pool-1-thread-1 : [100,103,106,109,112,115,118,121,124,127,130,133,136,139,142,145,148,151,154,157,160,163,166,169,172,175,178,181,184,187,190,193,196,199]
**********************************************************
**********************************************************
pool-1-thread-1 : [101,104,107,110,113,116,119,122,125,128,131,134,137,140,143,146,149,152,155,158,161,164,167,170,173,176,179,182,185,188,191,194,197]
**********************************************************
no
close the pool
Process finished with exit code 0
只是模拟下在线程池中分片,用的是定时pool
1. 所有属性都是对象的,避免共享
2. 使用分片,确保数据不被多次处理(多个线程处理)
3. 使用多线程处理,提高效率
今天没看其它东东,昨天刚好想到这个,就写了个小demo,只是测试使用,ScheduledThreadPoolExecutor用的不多,一般定时机制都会选择quartz,更灵活与强大!