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() + ":登陆成功";
    }

}

测试一下

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

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

模拟入库

将日志保存到数据库。

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);

}

测试一下

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

使用@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);
    }
}

从新启动项目测试一下。

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

@EnableAsync
@SpringBootApplication
public class LogApplication {

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

}

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

这下可好启动都失败了。

  不要慌，先看一眼错误信息。由于有些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);
    }

}

从新启动下再测试一下。

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

有返回值的方法

  因为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);
    }
}

注意的点

• 使用以前须要在启动类开启@EnableAsync。
• 只能在自身以外调用，在本类调用是无效的。
• 全部的类都须要交由Spring容器进行管理。

总结

  @Async标注的方法，称之为异步方法；这些方法将在执行的时候，将会在独立的线程中被执行，调用者无需等待它的完成，便可继续其余的操做。

  虽然本身维护线程池也是能够实现相应的功能，可是我仍是推荐使用SpringBoot自带的异步方法，简单方便，只须要@Async和@EnableAsync就能够了。

结尾

  为何外卖小哥能看懂我写的代码？难道我之后也要去xxx?

  若是以为对你有帮助，能够多多评论，多多点赞哦，也能够到个人主页看看，说不定有你喜欢的文章，也能够随手点个关注哦，谢谢。