JAVA多线程(四) Executor并发框架向RabbitMQ推送消息

2022-12-28 19:13:30

github代码地址: https://github.com/showkawa/springBoot_2017/tree/master/spb-demo/spb-brian-query-service

假设一个需求用户点击某个页面，我们后台需要向MQ推送信信息

1，模拟的MQ服务，我这边使用RabbitMQ (关于MQ 发送和监听消息可以参考我的博客：SpringBoot消息中间件RabbitMQ)

//后台监听消息    
@RabbitListener(queues = "brian.test")

    public void receiveMessage(User user){

        logger.info("接收到MQ的消息体: " + user);

    }

2.向IOC容器中注册一个ThreadPoolTaskExecutor实例

    @Bean

    public ThreadPoolTaskExecutor  brianThreadPool(){

        ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();

        //核心线程数

        executor.setCorePoolSize(8);

        //最大线程数

        executor.setMaxPoolSize(16);

        //队列中最大的数

        executor.setQueueCapacity(8);

        //县城名称前缀

        executor.setThreadNamePrefix("brianThreadPool_");

        //rejectionPolicy：当pool已经达到max的时候，如何处理新任务

        //callerRuns：不在新线程中执行任务，而是由调用者所在的线程来执行

        //对拒绝task的处理策略

        executor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());

        //线程空闲后最大的存活时间

        executor.setKeepAliveSeconds(60);

        //初始化加载

        executor.initialize();

        return executor;

    }

3.实现线程池并发推送消息

    /**

     * 多线程推送消息到MQ服务

     */

    public void sendMessageByThredPool(User user) throws ExecutionException, InterruptedException {

            Future<String> future = executor.submit(() -> {

                sendMessageService.sendMessage("brian","mymq",user);
                logger.info("线程 [ " + Thread.currentThread().getName() + " ] 推送消息到MQ成功! " + new Date());

                return Thread.currentThread().getName();

            });

    }

4. Controller层的调用

    @PostMapping("/loop/sendMsg/userInfo")

    public ResponseEntity addUserInfo2MQ(@RequestBody User user) throws ExecutionException, InterruptedException {

        brianService.sendMessageByThredPool(user);

        return new ResponseEntity(user, HttpStatus.OK);

    }

5.利用postman 做压力测试，测试接口

5.1 postman做loop压力测试，需要单创建一个Collections来测试，并且当前Collections值允许放一个测试用例，比如我下面的loopSendUserInfo