分享

线程之间通信 等待(wait)和通知(notify)

 WindySky 2018-03-09

线程通信概念:

    线程是操作系统中独立的个体,但这些个体如果不经过特殊的处理就不能成为一个整体,线程之间的通信就成为整体的必用方式之一。当线程存在通信指挥,系统间的交互性会更强大,在提高CPU利用率的同时还会对线程任务在处理过程中进行有效的把控与监督。

为了支持多线程之间的协作,JDK提供了两个非常重要的接口线程等待wait()方法和通知notify()方法。这两个方法并不是在Thread类中的,而是输出Object类。这也意味着任何对象都可以调用这2个方法。

我们先看一个简单的例子:

复制代码
 1 public class ListAdd1 {
 2     private volatile static List list = new ArrayList();
 3     public void add(){
 4         list.add("jianzh5");
 5     }
 6     public int size(){
 7         return list.size();
 8     }
 9 
10     public static void main(String[] args) {
11         final ListAdd1 list1 = new ListAdd1();
12         Thread t1 = new Thread(new Runnable() {
13             @Override
14             public void run() {
15                 try {
16                     for(int i = 0; i <10; i++){
17                         list1.add();
18                         System.out.println("当前线程:" + Thread.currentThread().getName() + "添加了一个元素..");
19                         Thread.sleep(500);
20                     }
21                 } catch (InterruptedException e) {
22                     e.printStackTrace();
23                 }
24             }
25         }, "t1");
26 
27         Thread t2 = new Thread(new Runnable() {
28             @Override
29             public void run() {
30                 while(true){
31                     if(list1.size() == 5){
32                         System.out.println("当前线程收到通知:" + Thread.currentThread().getName() + " list size = 5 线程停止..");
33                         throw new RuntimeException();
34                     }
35                 }
36             }
37         }, "t2");
38         t1.start();
39         t2.start();
40     }
41 }
复制代码

代码很简单,这是在没使用JDK线程协作时的做法。线程t2一直在死循环,当list的size等于5时退出t2,t1则继续运行。

这样其实也可以是说线程之间的协作,但是问题就是t2会一直循环运行,浪费了CPU资源(PS:list必须使用关键字volatile修饰)。

我们再看使用wait和notify时的代码:

复制代码
 1 public class ListAdd2 {
 2     private volatile static List list = new ArrayList();
 3 
 4     public void add(){
 5         list.add("jianzh5");
 6     }
 7     public int size(){
 8         return list.size();
 9     }
10 
11     public static void main(String[] args) {
12 
13         final ListAdd2 list2 = new ListAdd2();
14         final byte[] lock = new byte[0];
15         Thread t1 = new Thread(new Runnable() {
16             @Override
17             public void run() {
18                 try {
19                     synchronized (lock) {
20                         System.out.println("t1启动..");
21                         for(int i = 0; i <10; i++){
22                             list2.add();
23                             System.out.println("当前线程:" + Thread.currentThread().getName() + "添加了一个元素..");
24                             Thread.sleep(500);
25                             if(list2.size() == 5){
26                                 System.out.println("已经发出通知..");
27                                 lock.notify();
28                             }
29                         }
30                     }
31                 } catch (InterruptedException e) {
32                     e.printStackTrace();
33                 }
34 
35             }
36         }, "t1");
37 
38         Thread t2 = new Thread(new Runnable() {
39             @Override
40             public void run() {
41                 synchronized (lock) {
42                     System.out.println("t2启动..");
43                     if(list2.size() != 5){
44                         try {
45                             lock.wait();
46                         } catch (InterruptedException e) {
47                             e.printStackTrace();
48                         }
49                     }
50                     System.out.println("当前线程:" + Thread.currentThread().getName() + "收到通知线程停止..");
51                     throw new RuntimeException();
52                 }
53             }
54         }, "t2");
55         t2.start();
56         t1.start();
57     }
58 }
复制代码

这里首先创建了一个的byte[]对象lock,然后线程t1,t2使用synchronzied关键字同步lock对象。线程t1一直往list添加元素,当元素大小等于5的时候调用lock.notify()方法通知lock对象。线程t2在size不等于5的时候一直处于等待状态。

这里使用byte[0]数组是因为JVM创建byte[0]所占用的空间比普通的object对象小,而花费的代价也最小。

运行结果如下:

看到这里可能会有疑问,为什么t1通知了t2线程运行而结果却是t1先运行完后t2再运行。

说明如下:

1、wait() 和 notify()必须配合synchrozied关键字使用,无论是wait()还是notify()都需要首先获取目标对象的一个监听器。

2、wait()释放锁,而notify()不释放锁。

 线程t2一开始处于wait状态,这时候释放了锁所以t1可以一直执行,而t1在notify的时候并不会释放锁,所以t1还会继续运行。 

 

知识拓展

现在我们来探讨一下有界阻塞队列的实现原理并模拟一下它的实现 :

1、有界队列顾名思义是有容器大小限制的

2、当调用put()方法时,如果此时容器的长度等于限定的最大长度,那么该方法需要阻塞直到队列可以有空间容纳下添加的元素

3、当调用take()方法时,如果此时容器的长度等于最小长度0,那么该方法需要阻塞直到队列中有了元素能够取出

4、put() 和 take()方法是需要协作的,能够及时通知状态进行插入和移除操作

根据以上阻塞队列的几个属性,我们可以使用wait 和notify实现以下它的实现原理:

 

复制代码
/**
 * 自定义大小的阻塞容器
 */
public class MyQueue {
    //1、初始化容器
    private final LinkedList<Object> list = new LinkedList<>();
    //2、定义计数器
    private AtomicInteger count = new AtomicInteger(0);
    //3、设定容器的上限和下限
    private final int minSize = 0;
    private final int maxSize;

    //4、构造器
    public MyQueue(int size) {
        this.maxSize = size;
    }

    //5、定义锁对象
    private final Object lock = new Object();

    //6、阻塞增加方法
    public void put(Object obj) {
        synchronized (lock) {
            while (count.get() == this.maxSize) {
                try {
                    lock.wait();
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
            //加入元素 计数器累加 唤醒取数线程可以取数
            list.add(obj);
            count.incrementAndGet();
            lock.notify();
            System.out.println("新增的元素:" + obj);
        }
    }

    public Object take() {
        Object result = null;
        synchronized (lock) {
            while (count.get() == this.minSize) {
                try {
                    lock.wait();
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
            //移除元素 计数器递减 唤醒添加的线程可以添加元素
            result = list.removeFirst();
            count.decrementAndGet();
            lock.notify();
        }
        return result;
    }

    public int getSize() {
        return this.count.get();
    }

    public static void main(String[] args) {
        final MyQueue myQueue = new MyQueue(5);
        myQueue.put("a");
        myQueue.put("b");
        myQueue.put("c");
        myQueue.put("d");
        myQueue.put("e");

        System.out.println("当前队列长度:" + myQueue.getSize());
        Thread t1 = new Thread(new Runnable() {
            @Override public void run() {
                myQueue.put("f");
                myQueue.put("g");
            }
        }, "t1");

        t1.start();

        Thread t2 = new Thread(new Runnable() {
            @Override public void run() {
                Object obj = myQueue.take();
                System.out.println("移除的元素为:"+obj);
                Object obj2 = myQueue.take();
                System.out.println("移除的元素为:"+obj2);
            }
        },"t2");

        try {
            TimeUnit.SECONDS.sleep(5);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }

        t2.start();
    }
}
复制代码

 

实现过程如下:

1、通过构造器初始化指定容器的大小

2、程序内部有一个AtomicInteger的计数器,当调用put()操作时此计数器加1;当调用take()方法时此计数器减1

3、在进行相应的take()和put()方法时会使用while判断进行阻塞,会一直处于wait状态,并在可以进行操作的时候唤醒另外一个线程可以进行相应的操作。

4、将此代码运行可以看到相应的效果。

 

 

 

 

    本站是提供个人知识管理的网络存储空间,所有内容均由用户发布,不代表本站观点。请注意甄别内容中的联系方式、诱导购买等信息,谨防诈骗。如发现有害或侵权内容,请点击一键举报。
    转藏 分享 献花(0

    0条评论

    发表

    请遵守用户 评论公约

    类似文章 更多