如何優(yōu)雅的關閉 Java線程池

簡介 在開發(fā)中使用線程池去執(zhí)行異步任務是比較普遍的操作 ，然而雖然有些異步操作我們并不十分要求可靠性和實時性，但總歸業(yè)務還是需要的。如果在每次的服務發(fā)版過程中，我們不去介入線程池的停機邏輯
，那么很有可能就會造成線程池中隊列的任務還未執(zhí)行完成，自然就會造成數(shù)據(jù)的丟失。

探究

注意，本文所有前提是對進程進行下線時使用的是kill -15

我們知道Spring已經(jīng)實現(xiàn)了自己的優(yōu)雅停機方案  ，詳細請參考org.springframework.context.support.AbstractApplicationContext#registerShutdownHook ，然后主要看調(diào)用的org.springframework.context.support.AbstractApplicationContext#doClose
， 在這個方法里定義了容器銷毀的執(zhí)行順序

protected void doClose() {   // Check whether an actual close attempt is necessary...  if (this.active.get() && this.closed.compareAndSet(false, true)) {    if (logger.isDebugEnabled()) {     logger.debug("Closing " + this);   }   LiveBeansView.unregisterApplicationContext(this);   try {     // Publish shutdown event.    publishEvent(new ContextClosedEvent(this));   }   catch (Throwable ex) {     logger.warn("Exception thrown from ApplicationListener handling ContextClosedEvent", ex);   }   // Stop all Lifecycle beans, to avoid delays during inpidual destruction.   if (this.lifecycleProcessor != null) {     try {      this.lifecycleProcessor.onClose();    }    catch (Throwable ex) {      logger.warn("Exception thrown from LifecycleProcessor on context close", ex);    }   }   // Destroy all cached singletons in the context's BeanFactory.   destroyBeans();   // Close the state of this context itself.   closeBeanFactory();   // Let subclasses do some final clean-up if they wish...   onClose();   // Reset local application listeners to pre-refresh state.   if (this.earlyApplicationListeners != null) {     this.applicationListeners.clear();    this.applicationListeners.addAll(this.earlyApplicationListeners);   }   // Switch to inactive.   this.active.set(false);  } }

我們先主要關注下destroyBeans這個方法
，看bean的銷毀邏輯是什么，然后看到了下面的一個bean的銷毀順序邏輯,具體方法在org.springframework.beans.factory.support.DefaultSingletonBeanRegistry#destroySingletons

private final MapdisposableBeans = new LinkedHashMap<>();  public void destroySingletons() {   if (logger.isTraceEnabled()) {    logger.trace("Destroying singletons in " + this);  }  synchronized (this.singletonObjects) {    this.singletonsCurrentlyInDestruction = true;  }  String[] disposableBeanNames;  synchronized (this.disposableBeans) {    disposableBeanNames = StringUtils.toStringArray(this.disposableBeans.keySet());  }  for (int i = disposableBeanNames.length - 1; i >= 0; i--) {    destroySingleton(disposableBeanNames[i]);  }  this.containedBeanMap.clear();  this.dependentBeanMap.clear();  this.dependenciesForBeanMap.clear();  clearSingletonCache(); }

可以看到最至關重要的就是一個屬性disposableBeans,這個屬性是一個LinkedHashMap， 因此屬性是有序的  ，所以銷毀的時候也是按照某種規(guī)則保持和放入一樣的順序進行銷毀的
，現(xiàn)在就是要確認這個屬性里到底存的是什么。

經(jīng)過調(diào)試發(fā)現(xiàn) ，在創(chuàng)建bean的org.springframework.beans.factory.support.AbstractAutowireCapableBeanFactory#doCreateBean方法中
，會調(diào)用一個方法org.springframework.beans.factory.support.AbstractBeanFactory#registerDisposableBeanIfNecessary ， 在這個方法中會調(diào)用org.springframework.beans.factory.support.DefaultSingletonBeanRegistry#registerDisposableBean然后將當前創(chuàng)建的bean放入到屬性disposableBeans中，那么現(xiàn)在來看一下放入的邏輯什么？

相關代碼貼一下

protected void registerDisposableBeanIfNecessary(String beanName, Object bean, RootBeanDefinition mbd) {   AccessControlContext acc = (System.getSecurityManager() != null ? getAccessControlContext() : null);  if (!mbd.isPrototype() && requiresDestruction(bean, mbd)) {     if (mbd.isSingleton()) {       // Register a DisposableBean implementation that performs all destruction      // work for the given bean: DestructionAwareBeanPostProcessors,      // DisposableBean interface, custom destroy method.      registerDisposableBean(beanName,                             new DisposableBeanAdapter(bean, beanName, mbd, getBeanPostProcessors(), acc));    }    else {       // A bean with a custom scope...      Scope scope = this.scopes.get(mbd.getScope());      if (scope == null) {         throw new IllegalStateException("No Scope registered for scope name '" + mbd.getScope() + "'");      }      scope.registerDestructionCallback(beanName,                                        new DisposableBeanAdapter(bean, beanName, mbd, getBeanPostProcessors(), acc));    }  }}

org.springframework.beans.factory.support.AbstractBeanFactory#requiresDestruction

protected boolean requiresDestruction(Object bean, RootBeanDefinition mbd) {   return (bean != null &&    (DisposableBeanAdapter.hasDestroyMethod(bean, mbd) || (hasDestructionAwareBeanPostProcessors() &&      DisposableBeanAdapter.hasApplicableProcessors(bean, getBeanPostProcessors())))); }

經(jīng)過兩個方法可以看到如果一個bean的scope是singleton并且這個bean實現(xiàn)了org.springframework.beans.factory.DisposableBean這個接口的destroy()方法，那么就會滿足條件 。

現(xiàn)在可以確定一點
，如果我們將線程池交給Spring管理，并且實現(xiàn)它的close方法