转载

SpringBoot异步方法

前言

最近呢xxx接到了一个任务，是需要把AOP打印出的请求日志，给保存到数据库。xxx一看这个简单啊，不就是保存到数据库嘛。一顿操作猛如虎，过了20分钟就把这个任务完成了。xxx作为一个优秀的程序员，发现这样同步保存会增加了接口的响应时间。这肯定难不倒xxx，当即决定使用多线程来处理这个问题。终于在临近饭点完成了。准备边吃边欣赏自己的杰作时，外卖小哥临时走来了一句，搞这样麻烦干啥，你加个 @Async 不就可以了。

实现一个精简版的请求日志输出。

LogAspect

@Slf4j
@Aspect
@Component
public class LogAspect {

    @Pointcut("execution(* com.hxh.log.controller.*.*(..)))")
    public void saveLog(){}

    @Before("saveLog()")
    public void saveLog(JoinPoint joinPoint) {
        // 获取HttpServletRequest
        ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
        assert attributes != null;

        HttpServletRequest request = attributes.getRequest();

        MethodSignature signature = (MethodSignature) joinPoint.getSignature();

        //获取请求参数
        String[] argNames = signature.getParameterNames();
        Object[] args = joinPoint.getArgs();

        log.info("请求路径:{},请求方式:{},请求参数:{},IP:{}",request.getRequestURI(),
                request.getMethod(),
                getRequestParam(argNames,args),
                request.getRemoteAddr());
    }

    /**
     * 组装请求参数
     * @param argNames 参数名称
     * @param args 参数值
     * @return 返回JSON串
     */
    private String getRequestParam(String[] argNames, Object[] args){
        HashMap<String,Object> params = new HashMap<>(argNames.length);
        if(argNames.length > 0 && args.length > 0){
            for (int i = 0; i < argNames.length; i++) {
                params.put(argNames[i] , args[i]);
            }
        }
        return JSON.toJSONString(params);
    }

}

LoginController

@RestController
public class LoginController {

    @PostMapping("/login")
    public String login(@RequestBody LoginForm loginForm){
        return loginForm.getUsername() + ":登录成功";
    }

}

测试一下

将项目启动然后测试一下。

SpringBoot异步方法

控制台已经打印出了请求日志。

SpringBoot异步方法

模拟入库

将日志保存到数据库。

LogServiceImpl

@Slf4j
@Service
public class LogServiceImpl implements LogService {

    @Override
    public void saveLog(RequestLog requestLog) throws InterruptedException {
        // 模拟入库需要的时间
        Thread.sleep(2000);
        log.info("请求日志保存成功:{}",requestLog);
    }
}

改造一下LogAspect添加日志入库

@Before("saveLog()")
public void saveLog(JoinPoint joinPoint) throws InterruptedException {
    // 获取HttpServletRequest
    ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
    assert attributes != null;

    HttpServletRequest request = attributes.getRequest();

    MethodSignature signature = (MethodSignature) joinPoint.getSignature();

    //获取请求参数
    String[] argNames = signature.getParameterNames();
    Object[] args = joinPoint.getArgs();

    log.info("请求路径:{},请求方式:{},请求参数:{},IP:{}",request.getRequestURI(),
            request.getMethod(),
            getRequestParam(argNames,args),
            request.getRemoteAddr());

    // 日志入库
    RequestLog requestLog = new RequestLog();
    requestLog.setRequestUrl(request.getRequestURI());
    requestLog.setRequestType(request.getMethod());
    requestLog.setRequestParam(request.getRequestURI());
    requestLog.setIp(request.getRemoteAddr());
    logService.saveLog(requestLog);

}

测试一下

SpringBoot异步方法

控制台已经打印出了请求日志。

SpringBoot异步方法

使用 `@Async`

由于保存日志消耗了2s，导致接口的响应时间也增加了2s。这样的结果显然不是我想要的。所以我们就按外卖小哥的方法，在 LogServiceImpl.saveLog() 上加一个 @Async 试试。

@Slf4j
@Service
public class LogServiceImpl implements LogService {

    @Async
    @Override
    public void saveLog(RequestLog requestLog) throws InterruptedException {
        // 模拟入库需要的时间
        Thread.sleep(2000);
        log.info("请求日志保存成功:{}",requestLog);
    }
}

重新启动项目测试一下。

SpringBoot异步方法

发现耗时还是2s多，这外卖小哥在瞎扯吧，于是转身进入了 baidu 的知识海洋遨游，发现要在启动类加个 @EnableAsync 。

@EnableAsync
@SpringBootApplication
public class LogApplication {

    public static void main(String[] args)  {
        SpringApplication.run(LogApplication.class, args);
    }

}

启动一下项目再来测试一下。

SpringBoot异步方法

这下可好启动都失败了。

SpringBoot异步方法

不要慌，先看一眼错误信息。因为有些service使用了CGLib这种动态代理而不是JDK原生的代理，导致问题的出现。所以我们需要给 @EnableAsync 加上 proxyTargetClass=true 。

@Slf4j
@EnableAsync(proxyTargetClass=true)
@SpringBootApplication
public class LogApplication {

    public static void main(String[] args)  {
        SpringApplication.run(LogApplication.class, args);
    }

}

重新启动下再测试一下。

SpringBoot异步方法

这下就成功了嘛，接口响应耗时变成了 324ms ，已经不像之前消耗 2s 那样了。

SpringBoot异步方法

有返回值的方法

由于 saveLog() 是没有返回值，假如碰到有返回值的情况该咋办呢？使用 Future<T> 即可。

@Slf4j
@Service
public class LogServiceImpl implements LogService {

    @Async
    @Override
    public Future<Boolean> saveLog(RequestLog requestLog) throws InterruptedException {
        // 模拟入库需要的时间
        Thread.sleep(2000);
        log.info("请求日志保存成功:{}",requestLog);
        return new AsyncResult<>(true);
    }
}

配置线程池

既然是异步方法，肯定是用其他的线程执行的，当然可以配置相应的线程池了。

@Configuration
public class ThreadConfig {

    /**
     * 日志异步保存输出线程池
     * @return 返回线程池
     */
    @Bean("logExecutor")
    public Executor taskExecutor() {
        ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
        executor.setCorePoolSize(5);
        executor.setMaxPoolSize(10);
        executor.setQueueCapacity(200);
        executor.setKeepAliveSeconds(60);
        executor.setThreadNamePrefix("logExecutor-");
        executor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());
        executor.setWaitForTasksToCompleteOnShutdown(true);
        executor.setAwaitTerminationSeconds(60);
        return executor;
    }
}

在使用 @Async 的时候指定对应的线程池就好了。

@Slf4j
@Service
public class LogServiceImpl implements LogService {

    @Override
    @Async("logExecutor")
    public Future<Boolean> saveLog(RequestLog requestLog) throws InterruptedException {
        // 模拟入库需要的时间
        Thread.sleep(2000);
        log.info("请求日志保存成功:{}",requestLog);
        return new AsyncResult<>(true);
    }
}