Java LinkedBlockingQueue实现消息队列_linkedblockingqueue实现发送消息并消费

最近有个项目需要开发一个预约系统，系统涉及到发送短信验证码；一般用户点击发送验证码，发送请求到后端后，调用短信接口，成功后就返回响应的状态码给用户；但是这样的过程，有时候会因为短信接口响应慢，而导致前端响应慢；所以这里需要做一个简单的优化，当用户点击发送短信时，将我们的短信调用放入一个队列中，放入之后，即给前端响应；后面通过阻塞队列，取出队列内容，进行短信发送即可，这样可以更好的提升系统的性能和用户体验度；

一、创建短信生产者

public class MessageProducer implements Runnable {
    private BlockingQueue<String> queue;
    private String phone;

    public MessageProducer(BlockingQueue<String> queue, String phone) {
        this.queue = queue;
        this.phone = phone;
    }

    @Override
    public void run() {
        try {
            queue.put(phone);
            System.out.println("手机号码：" + phone);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }

    }
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20

二、短信消费者

public class MessageConsumer implements Runnable {
    private BlockingQueue<String> queue;

    public MessageConsumer(BlockingQueue<String> queue) {
        this.queue = queue;

    }

    @Override
    public void run() {
        while (true) {
            System.out.println("等待中------------");
            try {
                String phone = queue.take();
                System.out.println("发送短信:" + phone);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
            System.out.println("处理完成------------");
        }
    }
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22

三、测试

    public static void main(String[] args) {
        LinkedBlockingQueue<String> queue=new LinkedBlockingQueue<String>(10);
        ExecutorService executorService = Executors.newCachedThreadPool();
            executorService.submit(new MessageConsumer(queue));
        for (int i=0;i<6;i++){
            try {
                Thread.sleep(5000);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
            executorService.submit(new MessageProducer(queue,"消费者"+i));
        }
    }
1
2
3
4
5
6
7
8
9
10
11
12
13

结果：

当队列中没有消息时，它就会一直进行阻塞