--- CircuitBreaker/src/dk/thoerup/circuitbreaker/OpenState.java 2009/10/22 06:01:35 467 +++ CircuitBreaker/src/dk/thoerup/circuitbreaker/OpenState.java 2011/04/19 15:22:09 1306 @@ -2,14 +2,17 @@ import java.util.concurrent.atomic.AtomicLong; +import dk.thoerup.circuitbreaker.config.BreakerConfig; + public final class OpenState implements CircuitBreakerState { final private AtomicLong tripTime = new AtomicLong(0); - final private AtomicLong timeout = new AtomicLong(0); - public void setTimeout(long timeout) { - this.timeout.set( timeout ); + BreakerConfig config; + + public void setTimeout(BreakerConfig config) { + this.config = config; } public void preInvoke(CircuitBreaker circuitBreaker) throws Exception @@ -17,7 +20,7 @@ long now = System.currentTimeMillis(); long elapsed = now - tripTime.get(); - if(elapsed > timeout.get()) + if(elapsed > config.getTimeout()) { circuitBreaker.attemptReset(); } @@ -48,7 +51,7 @@ } public long getTimeout() { - return timeout.get(); + return config.getTimeout(); } public long getElapsed() {