java 使用ReentrantLock与Condition实现简单的消息队列

import java.util.LinkedList;
import java.util.Queue;
import java.util.Random;
import java.util.Scanner;
import java.util.concurrent.locks.Condition;
import java.util.concurrent.locks.ReentrantLock;

/**
 * @author luocc
 */
public class ConditionTest {

    private  ReentrantLock lock = new ReentrantLock();
    private  Condition c1 = lock.newCondition(); // 单数队列
    private  Condition c2 = lock.newCondition(); // 双数队列
    private Queue<Integer> q1 = new LinkedList();
    private Queue<Integer> q2 = new LinkedList();

    public void addTask(int i){
        lock.lock();
        try {
            if(i % 2 == 0){
                q2.offer(i);
                c2.signal();
                System.out.println(String.format("双数队列 添加数字：%d，唤醒双数",i));
            }else{
                q1.offer(i);
                c1.signal();
                System.out.println(String.format("单数队列 添加数字：%d，唤醒单数",i));
            }

        }catch (Exception e){
            e.printStackTrace();
        }finally {
            lock.unlock();
        }
    }

    public Integer readSignTask(){
        lock.lock();
        try {
            while (q1.isEmpty()){
                c1.await();
            }
            return q1.poll();
        }catch (Exception e){
            e.printStackTrace();
        }finally {
            lock.unlock();
        }
        return -1;
    }

    public Integer readDobTask(){
        lock.lock();
        try {
            while (q2.isEmpty()){
                c2.await();
            }
            return q2.poll();
        }catch (Exception e){
            e.printStackTrace();
        }finally {
            lock.unlock();
        }
        return -1;
    }

    public static void main(String[] args) {
        ConditionTest test = new ConditionTest();
        Random random = new Random();
//        Thread addTask = new Thread(() -> {
//            int i = 0;
//            while (i++ < 10) {
//                try {
//                    Thread.sleep(3000);
//                    test.addTask(random.nextInt(100));
//                } catch (InterruptedException e) {
//                    e.printStackTrace();
//                }
//            }
//        });
        Thread readSignTask = new Thread(() -> {
            while (true) {
                try {
                    Thread.sleep(5000);
                    Integer integer = test.readSignTask();
                    System.out.println(String.format("读取单数:%d",integer));
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
        });
        Thread readDobTask = new Thread(() -> {
            while (true) {
                try {
                    Thread.sleep(5000);
                    Integer integer = test.readDobTask();
                    System.out.println(String.format("读取双数:%d",integer));
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
        });
//        addTask.start();
        readSignTask.start();
        readDobTask.start();
//
        Scanner scanner = new Scanner(System.in);
        while (true){
            String next = scanner.next();
            try {
                int i = Integer.parseInt(next);
                test.addTask(i);
            }catch (NumberFormatException e){
                if("exit".equals(next)){
                    readDobTask.interrupt();
                    readSignTask.interrupt();
                    break;
                }else{
                    System.out.println(next + " : is not a integer");
                }
            }
        }
        System.out.println("main 结束");
        System.exit(0);
    }
}