SpringBoot异步方法

Z时代
2024-01-10
分类：综合

前言

最近呢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&lt;&gt;(argNames.length);
        if(argNames.length &gt; 0 &amp;&amp; args.length &gt; 0){
            for (int i = 0; i &lt; 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&lt;&gt;(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&lt;&gt;(true);
    }
}