thoerup/curcuitbreaker/ClosedState.java

package dk.thoerup.curcuitbreaker;

import java.util.concurrent.atomic.*;


public class ClosedState implements CircuitBreakerState {

        AtomicInteger failureCount = new AtomicInteger(0);
        AtomicInteger failureThreshold = new AtomicInteger(0);
        
        public void setThreshold(int threshold) {
                failureThreshold.set( threshold );
        }

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

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

    public void onError(CircuitBreaker circuitBreaker, Throwable t) throws Throwable
    {
        int currentCount = failureCount.incrementAndGet();
        int threshold = failureThreshold.get();
        if(currentCount >= threshold)
            circuitBreaker.tripBreaker();
    }
    
    private void resetFailureCount() {
        failureCount.set(0);
    }
}
1	package dk.thoerup.curcuitbreaker;
2
3	import java.util.concurrent.atomic.*;
4
5
6
7	public class ClosedState implements CircuitBreakerState {
8
9	AtomicInteger failureCount = new AtomicInteger(0);
10	AtomicInteger failureThreshold = new AtomicInteger(0);
11
12	public void setThreshold(int threshold) {
13	failureThreshold.set( threshold );
14	}
15
16	public void preInvoke(CircuitBreaker circuitBreaker) throws Throwable
17	{
18	// NO OP
19	}
20
21	public void postInvoke(CircuitBreaker circuitBreaker) throws Throwable
22	{
23	resetFailureCount();
24	}
25
26	public void onError(CircuitBreaker circuitBreaker, Throwable t) throws Throwable
27	{
28	int currentCount = failureCount.incrementAndGet();
29	int threshold = failureThreshold.get();
30	if(currentCount >= threshold)
31	circuitBreaker.tripBreaker();
32	}
33
34	private void resetFailureCount() {
35	failureCount.set(0);
36	}
37	}