线程池ThreadPoolExecutor并行处理实现代码

线程池ThreadPoolExecutor并行处理实现代码

1、定义一个接口 Animal

package com.zh.vo;

public interface Animal {

void work();

}

2、定义一个实现类 Bird

package com.zh.vo;

public class Bird implements Animal {

@Override

public void work() {

int sum = 0;

for (int i = 0; i < 100000; i++) {

sum += i;

}

System.out.println("Bird calc: " + sum + " time: " + System.currentTimeMillis());

}

}

3、定义一个实现类 Cat

package com.zh.vo;

public class Cat implements Animal {

@Override

public void work() {

int sum = 0;

for (int i = 0; i < 100000; i++) {

sum += i;

}

System.out.println("Cat calc: " + sum + " time: " + System.currentTimeMillis());

}

}

4、定义一个实现类 Dog

package com.zh.vo;

public class Dog implements Animal {

@Override

public void work() {

int sum = 0;

for (int i = 0; i < 100000; i++) {

sum += i;

}

System.out.println("Dog calc: " + sum + " time: " + System.currentTimeMillis());

}

}

5、定义一个枚举类 AnimalEnum

package com.zh.enums;

import com.zh.vo.Bird;

import com.zh.vo.Cat;

import com.zh.vo.Dog;

public enum AnimalEnum {

CAT("cat", Cat.class), DOG("dog", Dog.class), BIRD("bird", Bird.class);

private String name;

private Class> clazz;

public String getName() {

return name;

}

public void setName(String name) {

this.name = name;

}

public Class> getClazz() {

return clazz;

}

public void setClazz(Class> clazz) {

this.clazz = clazz;

}

private AnimalEnum(String name, Class> clazz) {

this.name = name;

this.clazz = clazz;

}

public static void main(String[] args) {

// System.out.println(getName(DOG));

// System.out.println(getClazz(DOG));

AnimalEnum[] values = AnimalEnum.values();

System.out.println(values);

}

}

6、定义一个操作类 AnimalUtil

package com.zh.utils;

import java.util.HashMap;

import java.util.Map;

import com.zh.enums.AnimalEnum;

import com.zh.vo.Animal;

public enum AnimalUtil {

INSTANCE;

private static Map map = null;

private synchronized void init() {

map = new HashMap();

AnimalEnum[] values = AnimalEnum.values();

for (AnimalEnum animalEnum : values) {

Animal newInstance = null;

try {

newInstance = (Animal)animalEnum.getClazz().newInstance();

} catch (InstantiationException e) {

e.printStackTrace();

} catch (IllegalAccessException e) {

e.printStackTrace();

}

map.put(animalEnum, newInstance);

}

}

public Map getEnumMaps() {

if (map == null || map.isEmpty()) {

init();

}

return map;

}

}

7、定义一个测试主类使用 Future、Callable

package com.zh;

import java.util.HashMap;

import java.util.Map;

import java.util.Map.Entry;

import java.util.Set;

import java.util.concurrent.Callable;

import java.util.concurrent.ExecutionException;

import java.util.concurrent.Future;

import java.util.concurrent.LinkedBlockingQueue;

import java.util.concurrent.ThreadPoolExecutor;

import java.util.concurrent.TimeUnit;

import com.zh.enums.AnimalEnum;http://

import com.zh.utils.AnimalUtil;

import com.zh.vo.Animal;

/**

* @desc 测试线程池

* @author zhanhao

*/

public class ThreadPoolSubmitTest {

/**

* 定义线程池

*/

// Runtime.getRuntime().availableProcessors() * 2

private static ThreadPoolExecutor threadPoolExecutor =

new ThreadPoolExecutor(10, 20, 100, TimeUnit.MINUTES, new LinkedBlockingQueue());

/**

* @desc 执行主流程

* @param args

* @throws InterruptedException

* @throws ExecutionException

*/

public static void main(String[] args) throws InterruptedException, ExecutionException {

Map> futureMap = new HashMap>();

Set> entrySet = AnimalUtil.INSTANCE.getEnumMaps().entrySet();

for (Entry entry : entrySet) {

futureMap.put(entry.getKey().getName(), exec(entry.getKey()));

}

for (Entry> entry : futureMap.entrySet()) {

System.out.println(entry.getValue().get());

}

waitForAllThreadFinish();

threadPoolExecutor.shutdown();

}

/**

* @desc 讲任务提交到线程池中执行

* @param enums

* @return

*/

private static Future exec(AnimalEnum enums) {

return threadPoolExecutor.submit(new Callable() {

@Override

public String call() throws Exception {

Animal animal = AnimalUtil.INSTANCE.getEnumMaps().get(enums);

animal.work();

return Thread.currentThread().getName();

}

});

}

/**

* @desc 线程中有未完成的任务需等待完成

*/

private static void waitForAllThreadFinish() {

while (threadPoolExecutor.getQueue().size() > 0 || threadPoolExecutor.getActiveCount() > 0) {

try {

Thread.sleep(10);

} catch (InterruptedException e) {

e.printStackTrace();

}

}

}

}

8、执行结果

Dog calc: 704982704 time: 1574129306137

Bird calc: 704982704 time: 1574129306137

Cat calc: 704982704 time: 1574129306137

pool-1-thread-1

pool-1-thread-2

pool-1-thread-3

9、定义一个测试主类使用 Runnable

package com.zh;

import java.util.Map.Entry;

import java.util.Set;

import java.util.concurrent.ExecutionException;

import java.util.concurrent.LinkedBlockingQueue;

import java.util.concurrent.ThreadPoolExecutor;

import java.util.concurrent.TimeUnit;

import com.zh.enums.AnimalEnum;

import com.zh.utils.AnimalUtil;

import com.zh.vo.Animal;

/**

* @desc 测试线程池

* @author zhanhao

*/

public class ThreadPoolExecuteTest {

/**

* 定义线程池

*/

// Runtime.getRuntime().availableProcessors() * 2

private static ThreadPoolExecutor threadPoolExecutor =

new ThreadPoolExecutor(10, 20, 100, TimeUnit.MINUTES, new LinkedBlockingQueue());

/**

* @desc 执行主流程

* @param args

* @throws InterruptedException

* @throws ExecutionException

*/

public static void main(String[] args) throws InterruptedException, ExecutionException {

Set> entrySet = AnimalUtil.INSTANCE.getEnumMaps().entrySet();

for (Entry entry : entrySet) {

exec(entry.getKey());

}

waitForAllThreadFinish();

threadPoolExecutor.shutdown();

}

/**

* @desc 讲任务提交到线程池中执行

* @param enums

* @return

*/

private static void exec(AnimalEnum enums) {

threadPoolExecutor.execute(new Runnable() {

@Override

public void run() {

Animal animal = AnimalUtil.INSTANCE.getEnumMaps().get(enums);

animal.work();

System.out.println(Thread.currentThread().getName());

}

});

}

/**

* @desc 线程中有未完成的任务需等待完成

*/

private static void waitForAllThreadFinish() {

while (threadPoolExecutor.getQueue().size() > 0 || threadPoolExecutor.getActiveCount() > 0) {

try {

Thread.sleep(10);

} catch (InterruptedException e) {

e.printStackTrace();

}

}

}

}

10、执行结果

Bird calc: 704982704 time: 1574129356078

Dog calc: 704982704 time: 1574129356078

pool-1-thread-3

Cat calc: 704982704 time: 1574129356078

pool-1-thread-1

pool-1-thread-2

注：1、submit 方法可以有返回值 2、submit 底层调用execute方法

版权声明：本文内容由网络用户投稿，版权归原作者所有，本站不拥有其著作权，亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容，请联系我们jiasou666@gmail.com 处理，核实后本网站将在24小时内删除侵权内容。