java 注解实现一个可配置线程池的方法示例

/** * <h1>线程池核心配置(<b style="color:#CD0000">基本线程池数量、最大线程池数量、队列初始容量、线程连接保持活动秒数（默认60s）</b>)</h1> *  * <blockquote><code> * <table border="1px" style="border-color:gray;" width="100%"><tbody> * <tr><th style="color:green;text-align:left;"> * 属性名称 * </th><th style="color:green;text-align:left;"> * 属性含义 * </th></tr> * <tr><td> * queueCapacity * </td><td> * 基本线程池数量 * </td></tr> * <tr><td> * count * </td><td> * 最大线程池数量 * </td></tr> * <tr><td> * maxCount * </td><td> * 队列初始容量 * </td></tr> * <tr><td> * aliveSec * </td><td> * 线程连接保持活动秒数（默认60s） * </td></tr> * </tbody></table> * </code></blockquote>  */public class PoolConfig {  private int queueCapacity = 200;  private int count = 0;  private int maxCount = 0;  private int aliveSec;  public int getQueueCapacity() { return queueCapacity; }   public void setQueueCapacity(int queueCapacity) { this.queueCapacity = queueCapacity; }  public void setCount(int count) { this.count = count; }  public void setMaxCount(int maxCount) { this.maxCount = maxCount; }  public void setAliveSec(int aliveSec) { this.aliveSec = aliveSec; }  public int getCount() { return count; }  public int getMaxCount() { return maxCount; }  public int getAliveSec() { return aliveSec; }}

import java.util.ArrayList;import java.util.HashMap;import java.util.List;import java.util.Map; import org.springframework.boot.context.properties.ConfigurationProperties;import org.springframework.stereotype.Component; /** * <h1>线程池配置(<b style="color:#CD0000">线程池核心配置、各个业务处理的任务数量</b>)</h1> *  * <blockquote><code> * <table border="1px" style="border-color:gray;" width="100%"><tbody> * <tr><th style="color:green;text-align:left;"> * 属性名称 * </th><th style="color:green;text-align:left;"> * 属性含义 * </th></tr> * <tr><td> * pool * </td><td> * 线程池核心配置 * 【{@link PoolConfig}】 * </td></tr> * <tr><td> * count * </td><td> * 线程池各个业务任务初始的任务数 * </td></tr> * </tbody></table> * </code></blockquote>  */@Component@ConfigurationProperties(prefix="thread")public class ThreadPoolConfig {  private PoolConfig pool = new PoolConfig();  Map<String, Integer> count = new HashMap<>();  public PoolConfig getPool() { return pool; }  public void setPool(PoolConfig pool) { this.pool = pool; }  public Map<String, Integer> getCount() { return count; } }

@Target(ElementType.TYPE)@Retention(RetentionPolicy.RUNTIME)@Documented@Componentpublic @interface ExcutorTask {  /** * The value may indicate a suggestion for a logical ExcutorTask name, * to be turned into a Spring bean in case of an autodetected ExcutorTask . * @return the suggested ExcutorTask name, if any */ String value() default ""; }

public class Beans {  private static final char PREFIX = '.';  public static ConcurrentMap<String, String> scanBeanClassNames(){ ConcurrentMap<String, String> beanClassNames = new ConcurrentHashMap<>(); ClassPathScanningCandidateComponentProvider provider = new ClassPathScanningCandidateComponentProvider(false);    provider.addIncludeFilter(new AnnotationTypeFilter(ExcutorTask.class));   for(Package pkg : Package.getPackages()){   String basePackage = pkg.getName();     Set<BeanDefinition> components = provider.findCandidateComponents(basePackage);      for (BeanDefinition component : components) {     String beanClassName = component.getBeanClassName();     try {    Class<?> clazz = Class.forName(component.getBeanClassName());    boolean isAnnotationPresent = clazz.isAnnotationPresent(ZimaTask.class);    if(isAnnotationPresent){     ZimaTask task = clazz.getAnnotation(ExcutorTask.class);     String aliasName = task.value();     if(aliasName != null && !"".equals(aliasName)){     beanClassNames.put(aliasName, component.getBeanClassName());     }    }    } catch (ClassNotFoundException e) {    e.printStackTrace();    }     beanClassNames.put(beanClassName.substring(beanClassName.lastIndexOf(PREFIX) + 1), component.getBeanClassName());     }   }   return beanClassNames;  } }

@Componentpublic class TaskPool {  public ThreadPoolTaskExecutor poolTaskExecutor;  @Autowired  private ThreadPoolConfig threadPoolConfig;  @Autowired  private ApplicationContext context;  private final Integer MAX_POOL_SIZE = 2000;  private PoolConfig poolCfg;  private Map<String, Integer> tasksCount;  private ConcurrentMap<String, String> beanClassNames;  @PostConstruct  public void init() {  beanClassNames = Beans.scanBeanClassNames();      poolTaskExecutor = new ThreadPoolTaskExecutor();      poolCfg = threadPoolConfig.getPool();  tasksCount = threadPoolConfig.getCount();  int corePoolSize = poolCfg.getCount(),   maxPoolSize = poolCfg.getMaxCount(),   queueCapacity = poolCfg.getQueueCapacity(),   minPoolSize = 0, maxCount = (corePoolSize << 1);  for(String taskName : tasksCount.keySet()){  minPoolSize += tasksCount.get(taskName); }  if(corePoolSize > 0){  if(corePoolSize <= minPoolSize){  corePoolSize = minPoolSize;  } }else{  corePoolSize = minPoolSize; }  if(queueCapacity > 0){  poolTaskExecutor.setQueueCapacity(queueCapacity); }  if(corePoolSize > 0){  if(MAX_POOL_SIZE < corePoolSize){  corePoolSize = MAX_POOL_SIZE;  }  poolTaskExecutor.setCorePoolSize(corePoolSize); }  if(maxPoolSize > 0){  if(maxPoolSize <= maxCount){  maxPoolSize = maxCount;  }  if(MAX_POOL_SIZE < maxPoolSize){  maxPoolSize = MAX_POOL_SIZE;  }  poolTaskExecutor.setMaxPoolSize(maxPoolSize); }  if(poolCfg.getAliveSec() > 0){  poolTaskExecutor.setKeepAliveSeconds(poolCfg.getAliveSec()); }  poolTaskExecutor.initialize();  }   public void execute(Class<?>... clazz){ int i = 0, len = tasksCount.size(); for(; i < len; i++){  Integer taskCount = tasksCount.get(i);  for(int t = 0; t < taskCount; t++){  try{   Object taskObj = context.getBean(clazz[i]);   if(taskObj != null){   poolTaskExecutor.execute((Runnable) taskObj);   }  }catch(Exception ex){   ex.printStackTrace();  }  } }  }   public void execute(String... args){   int i = 0, len = tasksCount.size(); for(; i < len; i++){  Integer taskCount = tasksCount.get(i);  for(int t = 0; t < taskCount; t++){  try{   Object taskObj = null;   if(context.containsBean(args[i])){   taskObj = context.getBean(args[i]);   }else{   if(beanClassNames.containsKey(args[i].toLowerCase())){    Class<?> clazz = Class.forName(beanClassNames.get(args[i].toLowerCase()));    taskObj = context.getBean(clazz);   }   }   if(taskObj != null){   poolTaskExecutor.execute((Runnable) taskObj);   }  }catch(Exception ex){   ex.printStackTrace();  }  } }  }  public void execute(){ for(String taskName : tasksCount.keySet()){  Integer taskCount = tasksCount.get(taskName);  for(int t = 0; t < taskCount; t++){  try{   Object taskObj = null;   if(context.containsBean(taskName)){   taskObj = context.getBean(taskName);   }else{   if(beanClassNames.containsKey(taskName)){    Class<?> clazz = Class.forName(beanClassNames.get(taskName));    taskObj = context.getBean(clazz);   }   }   if(taskObj != null){   poolTaskExecutor.execute((Runnable) taskObj);   }  }catch(Exception ex){   ex.printStackTrace();  }  } }  }  }

#配置执行的task线程数thread.count.NeedExcutorTask=4#最大存活时间thread.pool.aliveSec=300000#其他配置同理

@Configurationpublic class TaskManager {  @Resource private TaskPool taskPool;  @PostConstruct public void executor(){ taskPool.execute(); }}

@ExcutorTaskpublic class NeedExcutorTask implements Runnable{  @Override public void run() {    Thread.sleep(1000L);    log.info("====== 任务执行 =====")  }}