多线程下使用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();
        }
    }
}