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