thoerup/circuitbreaker/ClosedState.java

package dk.thoerup.circuitbreaker;

import java.util.concurrent.atomic.AtomicInteger;

import dk.thoerup.circuitbreaker.config.BreakerConfig;


public final class ClosedState implements CircuitBreakerState {

        final AtomicInteger failureCount = new AtomicInteger(0);
        
        
        BreakerConfig config;
        
        public void setThreshold(BreakerConfig config) {
                this.config = config;
        }

    public void preInvoke(CircuitBreaker circuitBreaker) throws Exception
    {
        // NO OP
    }

    public void postInvoke(CircuitBreaker circuitBreaker) throws Exception
    {
        resetFailureCount();
    }

    public void onError(CircuitBreaker circuitBreaker, Exception t) throws Exception
    {
        int currentCount = failureCount.incrementAndGet();
        int threshold = config.getTreshold();
        if(currentCount >= threshold)
            circuitBreaker.tripBreaker();
    }
    
    public void resetFailureCount() {
        failureCount.set(0);
    }

        public String getName() {
                return "Closed";
        }
        
        public int getFailureCount() {
                return failureCount.get();
        }
        
        public int getThreshold() {
                return config.getTreshold();
        }
}
1	torben	467	package dk.thoerup.circuitbreaker;
2	torben	393
3	torben	1306	import java.util.concurrent.atomic.AtomicInteger;
4	torben	393
5	torben	1306	import dk.thoerup.circuitbreaker.config.BreakerConfig;
6	torben	393
7
8	torben	1306
9	torben	424	public final class ClosedState implements CircuitBreakerState {
10	torben	393
11	torben	424	final AtomicInteger failureCount = new AtomicInteger(0);
12	torben	393
13	torben	1306
14			BreakerConfig config;
15
16			public void setThreshold(BreakerConfig config) {
17			this.config = config;
18	torben	393	}
19
20	torben	450	public void preInvoke(CircuitBreaker circuitBreaker) throws Exception
21	torben	393	{
22			// NO OP
23			}
24
25	torben	450	public void postInvoke(CircuitBreaker circuitBreaker) throws Exception
26	torben	393	{
27			resetFailureCount();
28			}
29
30	torben	450	public void onError(CircuitBreaker circuitBreaker, Exception t) throws Exception
31	torben	393	{
32			int currentCount = failureCount.incrementAndGet();
33	torben	1306	int threshold = config.getTreshold();
34	torben	393	if(currentCount >= threshold)
35			circuitBreaker.tripBreaker();
36			}
37
38	torben	456	public void resetFailureCount() {
39	torben	393	failureCount.set(0);
40			}
41	torben	397
42			public String getName() {
43			return "Closed";
44			}
45
46			public int getFailureCount() {
47			return failureCount.get();
48			}
49
50			public int getThreshold() {
51	torben	1306	return config.getTreshold();
52	torben	397	}
53	torben	393	}