java线程池方式执行run任务及在线程run方法中注入service

背景：接口1内需要调用接口2，接口2功能是执行一个长时间任务，执行完成后才会返回结果（比如执行5min后获取到执行结果），但是接口1内调接口2时不能等待5min后才返回结果，接口1需要返回自己的结果

方式：以线程池方式来执行任务，接口1触发接口2后不等待，直接返回自己的内容或做其他操作，将接口2的任务扔进线程池中任其执行，执行完了后可以接着做其他操作

一、线程池方式执行run任务

例子：

2个数据库表：report（存储报告连接）、task（任务表，包含任务执行的状态status和报告id）

1、创建业务接口controller

package com.controller;

import com.alibaba.fastjson.JSONObject;
import com.thtead.Worker;
import com.utils.ResultJson;
import com.utils.ResultGenerator;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

import java.util.concurrent.*;

@RestController
@RequestMapping("api/v1")
public class TestController {

    //创建线程池
    public static ThreadPoolExecutor executorService = new ThreadPoolExecutor(5, 5, 60L, TimeUnit.MILLISECONDS, new LinkedBlockingQueue<>(1));


    /**
     * 线程池执行任务
     * @return
     * @throws Exception
     */
    @PostMapping("/run")
    @ResponseBody
    public JSONObject post() throws Exception {

        //做自己的处理，比如先生成一条task记录，写入task表，将report id置空，status值为running
        JSONObject json =new JSONObject();
        json.put("taskid","111");
        //调用worker在线程池中执行任务
        executorService.submit(new Worker());   

        //不等待，做自己的处理，比如该接口返回task id
        return json;
    }


}

2、接口2：真正执行任务

package com.controller;

import com.alibaba.fastjson.JSONObject;
import org.springframework.web.bind.annotation.*;

@RestController
@RequestMapping("api/v1")
public class TestController2 {


    @RequestMapping(value = "/Report", method = RequestMethod.POST)
    @ResponseBody
    public JSONObject jacocoReport() throws Exception {

        Thread.sleep(6000);   //模拟任务执行，即需要长时间执行
        //模拟任务执行完后的返回内容
        JSONObject json =new JSONObject();
        json.put("key2","value2");
        json.put("key3","value3");
        JSONObject json2 =new JSONObject();
        json2.put("reporturl","http://111.231231.32321/3213123/312/31");
        json2.put("key2",json);
        return json2;

    }
}

3、创建worker，复写run

package com.thtead;

import com.alibaba.fastjson.JSONObject;
import com.controller.TestController2;


public class Worker implements Runnable {
    //用于传参    
    public Worker(String page){
        this.page1 = page;
    }
    String page1;

    @Override
    public void run(){
        try {
            TestController2 controller2 = new TestController2();
            JSONObject json = controller2.jacocoReport();   //执行任务
            System.out.println(json.get("reporturl"));
            System.out.println(page1);
            //这里可以继续执行其他操作：等待任务执行完获取到结果后，将结果接入report表，生成report id
            //再将report id写入task表，同时更改task表中的status为finish

        } catch (InterruptedException e) {
            e.printStackTrace();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

}

二、在线程run方法中注入service

 采坑： 在线程的run方法中使用@Autowired注解获得Service层实例，调用service层方法是报空指针：

 

 原因：在线程的run方法中使用注解@autoware注入的bean，会报空指针异常，原因是因为线程中为了线程安全，防注入。

解决：在bean工厂中拿实例

1、新增获取bean实例的工具类组件

package com.utils;

import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.stereotype.Component;

/**
 * 在线程的run方法中使用注解@autoware注入的bean，会报空指针异常，原因是因为线程中为了线程安全，防注入。
 * 获取bean实例的工具类组件
 */
@Component
public class GetBeanUtil implements ApplicationContextAware {
    //Spring应用上下文环境
    private static ApplicationContext applicationContext;
    /**
     * 实现ApplicationContextAware接口的回调方法，设置上下文环境
     */
    public void setApplicationContext(ApplicationContext context) {
        GetBeanUtil.applicationContext = context;
    }
    /**
     * 获取对象 这里重写了bean方法，起主要作用
     */
    public static Object getBean(String name) {
        return applicationContext.getBean(name);
    }

    public static ApplicationContext getApplicationContext() {
        return applicationContext;
    }
}

2、在run方法中使用工具类拿到Service层实例：

package com.thtead;

import com.alibaba.fastjson.JSONObject;
import com.controller.TestController2;
import com.service.ApiService;
import com.utils.GetBeanUtils;


public class Worker implements Runnable {

    //不要使用 @Autowired
    private ApiService apiService;

    public Worker(String page){
        this.page1 = page;
    }
    String page1;
    @Override
    public void run(){
        this.apiService = GetBeanUtils.getApplicationContext().getBean(ApiService.class);  //可以使用apiService里的接口
        try {
            TestController2 controller2 = new TestController2();
            JSONObject json = controller2.jacocoReport();
            System.out.println(json.get("reporturl"));
            System.out.println(page1);
            //这里可以继续执行其他操作：等待任务执行完获取到结果后，将结果接入report表，生成report id
            //再将report id写入task表，同时更改task表中的status为finish
            apiService.selectApiInfo("213123");
//             http  run -》reporturl
//                写库
        } catch (InterruptedException e) {
            e.printStackTrace();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

}