H2O 生成

/*
 * 思路：
 * 1. 使用流式 API 实现线程同步。创建两个流分别处理氢和氧的释放。
 * 2. 使用两个流控制每个原子对应的释放方法（releaseHydrogen 和 releaseOxygen）。
 * 3. 使用 Collectors.partitioningBy 进行分组，保证每次三个原子成组处理。
 */

import java.util.concurrent.Semaphore;
import java.util.stream.IntStream;

public class H2OStream {
    private Semaphore hydrogenSemaphore;
    private Semaphore oxygenSemaphore;
    private Semaphore barrier;

    public H2OStream() {
        hydrogenSemaphore = new Semaphore(2);
        oxygenSemaphore = new Semaphore(1);
        barrier = new Semaphore(0);
    }

    public void releaseHydrogen(Runnable releaseHydrogen) throws InterruptedException {
        hydrogenSemaphore.acquire();
        barrier.release();
        releaseHydrogen.run();
        if (barrier.availablePermits() == 3) {
            barrier.acquire(3);
            hydrogenSemaphore.release(2);
            oxygenSemaphore.release(1);
        }
    }

    public void releaseOxygen(Runnable releaseOxygen) throws InterruptedException {
        oxygenSemaphore.acquire();
        barrier.release();
        releaseOxygen.run();
        if (barrier.availablePermits() == 3) {
            barrier.acquire(3);
            hydrogenSemaphore.release(2);
            oxygenSemaphore.release(1);
        }
    }

    public void releaseWater(String water) {
        water.chars().forEach(ch -> {
            if (ch == 'H') {
                try {
                    releaseHydrogen(() -> System.out.print("H"));
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            } else if (ch == 'O') {
                try {
                    releaseOxygen(() -> System.out.print("O"));
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
        });
    }
}