Spring-Boot中如何使用多线程处理任务

看到这个标题,相信不少人会感到疑惑,回忆你们自己的场景会发现,在 Spring 的项目中很少有使用多线程处理任务的,没错,大多数时候我们都是使用 Spring MVC 开发的 web 项目,默认的 Controller,Service,Dao 组件的作用域都是单实例,无状态,然后被并发多线程调用,那么如果我想使用多线程处理任务,该如何做呢?

比如如下场景:

使用 spring-boot 开发一个监控的项目,每个被监控的业务(可能是一个数据库表或者是一个 pid 进程)都会单独运行在一个线程中,有自己配置的参数,总结起来就是:

(1)多实例(多个业务,每个业务相互隔离互不影响)

(2)有状态(每个业务,都有自己的配置参数)

如果是非 spring-boot 项目,实现起来可能会相对简单点,直接 new 多线程启动,然后传入不同的参数类即可,在 spring 的项目中,由于 Bean 对象是 spring 容器管理的,你直接 new 出来的对象是没法使用的,就算你能 new 成功,但是 bean 里面依赖的其他组件比如 Dao,是没法初始化的,因为你饶过了 spring,默认的 spring 初始化一个类时,其相关依赖的组件都会被初始化,但是自己 new 出来的类,是不具备这种功能的,所以我们需要通过 spring 来获取我们自己的线程类,那么如何通过 spring 获取类实例呢,需要定义如下的一个类来获取 SpringContext 上下文:

/**
 * Created by Administrator on 2016/8/18.
 * 设置 Sping 的上下文
 */
@Component
public class ApplicationContextProvider implements ApplicationContextAware {
<span class="hljs-keyword">private</span> <span class="hljs-keyword">static</span> ApplicationContext context;

<span class="hljs-keyword">private</span> <span class="hljs-title function_">ApplicationContextProvider</span><span class="hljs-params">()</span>{}

<span class="hljs-meta">@Override</span>
<span class="hljs-keyword">public</span> <span class="hljs-keyword">void</span> <span class="hljs-title function_">setApplicationContext</span><span class="hljs-params">(ApplicationContext applicationContext)</span> <span class="hljs-keyword">throws</span> BeansException {
    context = applicationContext;
}

<span class="hljs-keyword">public</span>  <span class="hljs-keyword">static</span> &lt;T&gt; T <span class="hljs-title function_">getBean</span><span class="hljs-params">(String name,Class&lt;T&gt; aClass)</span>{
    <span class="hljs-keyword">return</span> context.getBean(name,aClass);
}

}

然后定义我们的自己的线程类,注意此类是原型作用域,不能是默认的单例:

@Component("mTask")
@Scope("prototype")
public class MoniotrTask extends Thread {
<span class="hljs-keyword">final</span> static <span class="hljs-type">Logger</span> logger= <span class="hljs-type">LoggerFactory</span>.getLogger(<span class="hljs-type">MoniotrTask</span>.<span class="hljs-keyword">class</span>);
<span class="hljs-comment">//参数封装</span>
<span class="hljs-keyword">private</span> <span class="hljs-type">Monitor</span> monitor;

public void setMonitor(<span class="hljs-type">Monitor</span> monitor) {
    <span class="hljs-keyword">this</span>.monitor = monitor;
}

<span class="hljs-meta">@Resource</span>(name = <span class="hljs-string">"greaterDaoImpl"</span>)
<span class="hljs-keyword">private</span> <span class="hljs-type">RuleDao</span> greaterDaoImpl;

<span class="hljs-meta">@Override</span>
public void run() {
    logger.info(<span class="hljs-string">"线程:"</span>+<span class="hljs-type">Thread</span>.currentThread().getName()+<span class="hljs-string">"运行中....."</span>);
}

}

写个测试例子,测试下使用 SpringContext 获取 Bean,查看是否是多实例:

/**
 * Created by Administrator on 2016/8/18.
 */
@RunWith(SpringJUnit4ClassRunner.class)
@SpringBootTest(classes =ApplicationMain.class)
public class SpingContextTest {
@Test
<span class="hljs-function"><span class="hljs-keyword">public</span> <span class="hljs-keyword">void</span> <span class="hljs-title">show</span>()throws Exception</span>{
    MoniotrTask m1=   ApplicationContextProvider.getBean(<span class="hljs-string">"mTask"</span>, MoniotrTask.<span class="hljs-keyword">class</span>);
    MoniotrTask m2=ApplicationContextProvider.getBean(<span class="hljs-string">"mTask"</span>, MoniotrTask.<span class="hljs-keyword">class</span>);
    MoniotrTask m3=ApplicationContextProvider.getBean(<span class="hljs-string">"mTask"</span>, MoniotrTask.<span class="hljs-keyword">class</span>);
    System.<span class="hljs-keyword">out</span>.println(m1+<span class="hljs-string">" =&gt; "</span>+m1.greaterDaoImpl);
    System.<span class="hljs-keyword">out</span>.println(m2+<span class="hljs-string">" =&gt; "</span>+m2.greaterDaoImpl);
    System.<span class="hljs-keyword">out</span>.println(m3+<span class="hljs-string">" =&gt; "</span>+m3.greaterDaoImpl);

}

}

运行结果如下:

[ INFO ] [2016-08-25 17:36:34] com.test.tools.SpingContextTest [57] - Started SpingContextTest in 2.902 seconds (JVM running for 4.196)
2016-08-25 17:36:34.842  INFO 8312 --- [           main] com.test.tools.SpingContextTest          : Started SpingContextTest in 2.902 seconds (JVM running for 4.196)
Thread[Thread-2,5,main] => com.xuele.bigdata.xalert.dao.rule.impl.GreaterDaoImpl@285f38f6
Thread[Thread-3,5,main] => com.xuele.bigdata.xalert.dao.rule.impl.GreaterDaoImpl@285f38f6
Thread[Thread-4,5,main] => com.xuele.bigdata.xalert.dao.rule.impl.GreaterDaoImpl@285f38f6

可以看到我们的监控类是多实例的,它里面的 Dao 是单实例的,这样以来我们就可以在 spring 中使用多线程处理我们的任务了。

如何启动我们的多线程任务类,可以专门定义一个组件类启动也可以在启动 Spring 的 main 方法中启动,下面看下,如何定义组件启动:

@Component
public class StartTask   {
final <span class="hljs-keyword">static</span> Logger logger= LoggerFactory.getLogger(StartTask.<span class="hljs-keyword">class</span>);

<span class="hljs-comment">//定义在构造方法完毕后,执行这个初始化方法</span>
@PostConstruct
<span class="hljs-function"><span class="hljs-keyword">public</span>  <span class="hljs-keyword">void</span> <span class="hljs-title">init</span>()</span>{

    final List&lt;Monitor&gt; list = ParseRuleUtils.parseRules();
    logger.info(<span class="hljs-string">"监控任务的总Task数:{}"</span>,list.size());
    <span class="hljs-keyword">for</span>(<span class="hljs-built_in">int</span> i=<span class="hljs-number">0</span>;i&lt;list.size();i++) {
        MoniotrTask moniotrTask=   ApplicationContextProvider.getBean(<span class="hljs-string">"mTask"</span>, MoniotrTask.<span class="hljs-keyword">class</span>);
        moniotrTask.setMonitor(list.<span class="hljs-keyword">get</span>(i));
        moniotrTask.start();
        logger.info(<span class="hljs-string">"第{}个监控task: {}启动 !"</span>,(i+<span class="hljs-number">1</span>),list.<span class="hljs-keyword">get</span>(i).getName());
    }

}

}

最后备忘下 logback.xml,里面可以配置相对和绝对的日志文件路径:

<!-- Logback configuration. See http://logback.qos.ch/manual/index.html -->
<configuration scan="true" scanPeriod="10 seconds">
  <!-- Simple file output -->
  <appender name="FILE" class="ch.qos.logback.core.rolling.RollingFileAppender">
    <!--<appender name="STDOUT" class="ch.qos.logback.core.ConsoleAppender">-->
    <!-- encoder defaults to ch.qos.logback.classic.encoder.PatternLayoutEncoder -->
    <encoder>
        <pattern>
            [ %-5level] [%date{yyyy-MM-dd HH🇲🇲ss}] %logger{96} [%line] - %msg%n
        </pattern>
        <charset>UTF-8</charset> <!-- 此处设置字符集 -->
    </encoder>
<span class="hljs-tag">&lt;<span class="hljs-name">rollingPolicy</span> <span class="hljs-attr">class</span>=<span class="hljs-string">"ch.qos.logback.core.rolling.TimeBasedRollingPolicy"</span>&gt;</span>
  <span class="hljs-comment">&lt;!-- rollover daily 配置日志所生成的目录以及生成文件名的规则,默认是相对路径 --&gt;</span>
  <span class="hljs-tag">&lt;<span class="hljs-name">fileNamePattern</span>&gt;</span>logs/xalert-%d{yyyy-MM-dd}.%i.log<span class="hljs-tag">&lt;/<span class="hljs-name">fileNamePattern</span>&gt;</span>
    <span class="hljs-comment">&lt;!--&lt;property name="logDir" value="E:/testlog" /&gt;--&gt;</span>
    <span class="hljs-comment">&lt;!--绝对路径定义--&gt;</span>
  <span class="hljs-comment">&lt;!--&lt;fileNamePattern&gt;${logDir}/logs/xalert-%d{yyyy-MM-dd}.%i.log&lt;/fileNamePattern&gt;--&gt;</span>
  <span class="hljs-tag">&lt;<span class="hljs-name">timeBasedFileNamingAndTriggeringPolicy</span>
      <span class="hljs-attr">class</span>=<span class="hljs-string">"ch.qos.logback.core.rolling.SizeAndTimeBasedFNATP"</span>&gt;</span>
    <span class="hljs-comment">&lt;!-- or whenever the file size reaches 64 MB --&gt;</span>
    <span class="hljs-tag">&lt;<span class="hljs-name">maxFileSize</span>&gt;</span>64 MB<span class="hljs-tag">&lt;/<span class="hljs-name">maxFileSize</span>&gt;</span>
  <span class="hljs-tag">&lt;/<span class="hljs-name">timeBasedFileNamingAndTriggeringPolicy</span>&gt;</span>
<span class="hljs-tag">&lt;/<span class="hljs-name">rollingPolicy</span>&gt;</span>


<span class="hljs-tag">&lt;<span class="hljs-name">filter</span> <span class="hljs-attr">class</span>=<span class="hljs-string">"ch.qos.logback.classic.filter.ThresholdFilter"</span>&gt;</span>
  <span class="hljs-tag">&lt;<span class="hljs-name">level</span>&gt;</span>DEBUG<span class="hljs-tag">&lt;/<span class="hljs-name">level</span>&gt;</span>
<span class="hljs-tag">&lt;/<span class="hljs-name">filter</span>&gt;</span>
<span class="hljs-comment">&lt;!-- Safely log to the same file from multiple JVMs. Degrades performance! --&gt;</span>
<span class="hljs-tag">&lt;<span class="hljs-name">prudent</span>&gt;</span>true<span class="hljs-tag">&lt;/<span class="hljs-name">prudent</span>&gt;</span>

</appender>

<!-- Console output -->
<appender name="STDOUT" class="ch.qos.logback.core.ConsoleAppender">
<!-- encoder defaults to ch.qos.logback.classic.encoder.PatternLayoutEncoder -->
<encoder>
<pattern>
[%-5level] [%date{yyyy-MM-dd HH🇲🇲ss}] %logger{96} [%line] - %msg%n
</pattern>
<charset>UTF-8</charset> <!-- 此处设置字符集 -->
</encoder>
<!-- Only log level WARN and above -->
<filter class="ch.qos.logback.classic.filter.ThresholdFilter">
<level>INFO</level>
</filter>
</appender>

<!-- Enable FILE and STDOUT appenders for all log messages.
By default, only log at level INFO and above. -->

<root level="INFO">
<appender-ref ref="STDOUT" />
<appender-ref ref="FILE" />

</root>

<!-- For loggers in the these namespaces, log at all levels. -->
<logger name="pedestal" level="ALL" />
<logger name="hammock-cafe" level="ALL" />
<logger name="user" level="ALL" />
<include resource="org/springframework/boot/logging/logback/base.xml"/>
<jmxConfigurator/>
</configuration

有什么问题可以扫码关注微信公众号:我是攻城师(woshigcs),在后台留言咨询。 技术债不能欠,健康债更不能欠, 求道之路,与君同行。