线程池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;
import com.zh.utils.AnimalUtil;
import com.zh.vo.Animal;


public class ThreadPoolSubmitTest {

  
  // Runtime.getRuntime().availableProcessors() * 2
  private static ThreadPoolExecutor threadPoolExecutor =
    new ThreadPoolExecutor(10, 20, 100, TimeUnit.MINUTES, new linkedBlockingQueue());

  
  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();
  }

  
  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();
      }
    });
  }

  
  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;


public class ThreadPoolExecuteTest {

  
  // Runtime.getRuntime().availableProcessors() * 2
  private static ThreadPoolExecutor threadPoolExecutor =
    new ThreadPoolExecutor(10, 20, 100, TimeUnit.MINUTES, new linkedBlockingQueue());

  
  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();
  }

  
  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());
      }
    });
  }

  
  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方法

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持考高分网。