1. 程式人生 > 程式設計 >Java Lock鎖多執行緒中實現流水線任務

Java Lock鎖多執行緒中實現流水線任務

  下面程式程式碼通過使用Lock鎖執行簡單的流水線任務:

import java.util.concurrent.locks.Condition;
import java.util.concurrent.locks.Lock;
import java.util.concurrent.locks.ReentrantLock;

/**
 * @author lzq
 * @data 2020/4/29 0029 - 下午 9:48
 */
public class TestLock {
  public static void main(String[] args) {
    DataSource dataSource=new DataSource();
    new Thread(() -> {
      for (int i=0;i<10;i++){
        dataSource.a();
      }
    },"A").start();

    new Thread(() -> {
      for (int i=0;i<10;i++){
        dataSource.b();
      }
    },"B").start();

    new Thread(() -> {
      for (int i=0;i<10;i++){
        dataSource.c();
      }
    },"C").start();
  }
}

class DataSource{
  private int x=1;
  private Lock lock=new ReentrantLock();
  private Condition condition1=lock.newCondition();
  private Condition condition2=lock.newCondition();
  private Condition condition3=lock.newCondition();

  public void a(){
    lock.lock();
    try {
      while(x!=1){
        condition1.await();
      }
      System.out.println(Thread.currentThread().getName()+":aaa");
      x=2;
      condition2.signal();
    } catch (InterruptedException e) {
      e.printStackTrace();
    } finally {
      lock.unlock();
    }
  }
  public void b(){
    lock.lock();
    try {
      while(x!=2){
        condition2.await();
      }
      System.out.println(Thread.currentThread().getName()+":bbb");
      x=3;
      condition3.signal();
    } catch (InterruptedException e) {
      e.printStackTrace();
    } finally {
      lock.unlock();
    }
  }
  public void c(){
    lock.lock();
    try {
      while(x!=3){
        condition3.await();
      }
      System.out.println(Thread.currentThread().getName()+":ccc");
      x=1;
      condition1.signal();
    } catch (InterruptedException e) {
      e.printStackTrace();
    } finally {
      lock.unlock();
    }
  }
}

    執行結果如下:

Java Lock鎖多執行緒中實現流水線任務

對於上面程式碼簡單分析:程式碼中包含簡單的生產者消費者流程和Lock實現三部曲,即重複判斷條件,執行邏輯,喚醒其他執行緒和產生鎖,加鎖,解鎖。注意這裡一點,條件判斷一定要重複判斷,不然可能會導致執行緒假醒影響結果。

因為當執行緒處於等待狀態時,執行緒會釋放資源,等到被喚醒的時候,從上次await的地方醒來繼續執行,這時條件判斷成立,執行await,其他執行緒再修改條件使得本執行緒被喚醒,此時本執行緒不會繼續判斷,而是繼續執行,如果使用迴圈判斷就能檢驗出條件被修改。

以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支援我們。