交通信号在Java中的多线程实现

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

public class TrafficSignal {
    private final Lock lock = new ReentrantLock();
    private final Condition greenLight = lock.newCondition();
    private final Condition yellowLight = lock.newCondition();
    private final Condition redLight = lock.newCondition();
    private Signal currentSignal = Signal.GREEN;

    public void changeToGreen() {
        lock.lock();
        try {
            while (currentSignal != Signal.YELLOW) {
                greenLight.await();
            }
            currentSignal = Signal.GREEN;
            System.out.println("Green light is on.");
            yellowLight.signal();
        } catch (InterruptedException e) {
            e.printStackTrace();
        } finally {
            lock.unlock();
        }
    }

    public void changeToYellow() {
        lock.lock();
        try {
            while (currentSignal != Signal.RED) {
                yellowLight.await();
            }
            currentSignal = Signal.YELLOW;
            System.out.println("Yellow light is on.");
            redLight.signal();
        } catch (InterruptedException e) {
            e.printStackTrace();
        } finally {
            lock.unlock();
        }
    }

    public void changeToRed() {
        lock.lock();
        try {
            while (currentSignal != Signal.GREEN) {
                redLight.await();
            }
            currentSignal = Signal.RED;
            System.out.println("Red light is on.");
            greenLight.signal();
        } catch (InterruptedException e) {
            e.printStackTrace();
        } finally {
            lock.unlock();
        }
    }

    private enum Signal {
        GREEN,
        YELLOW,
        RED
    }
}

public class Main {
    public static void main(String[] args) {
        TrafficSignal trafficSignal = new TrafficSignal();

        Thread greenThread = new Thread(() -> {
            while (true) {
                trafficSignal.changeToGreen();
                try {
                    Thread.sleep(5000); // 绿灯持续5秒
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
        });

        Thread yellowThread = new Thread(() -> {
            while (true) {
                trafficSignal.changeToYellow();
                try {
                    Thread.sleep(2000); // 黄灯持续2秒
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
        });

        Thread redThread = new Thread(() -> {
            while (true) {
                trafficSignal.changeToRed();
                try {
                    Thread.sleep(3000); // 红灯持续3秒
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
        });

        greenThread.start();
        yellowThread.start();
        redThread.start();
    }
}