SignalingCountDownLatch.java

package org.rhq.server.metrics;

import java.util.concurrent.CountDownLatch;

/**
 * @author John Sanda
 */
public class SignalingCountDownLatch {

    private boolean aborted;

    private CountDownLatch latch;

    private String msg;

    public SignalingCountDownLatch(CountDownLatch latch) {
        this.latch = latch;
    }

    public void await() throws InterruptedException, AbortedException {
        latch.await();
        if (aborted) {
            throw new AbortedException(msg);
        }
    }

    public void abort(String msg) {
        aborted = true;
        this.msg = msg;
        latch.countDown();
    }

    public void countDown() {
        latch.countDown();
    }

}