thoerup/curcuitbreaker/AccountingCircuitBreaker.java

package dk.thoerup.curcuitbreaker;

import java.util.concurrent.atomic.AtomicInteger;

public class AccountingCircuitBreaker extends CircuitBreaker {

        private AtomicInteger tripCount = new AtomicInteger(0); // how many times ahs the CB tripped  
        private AtomicInteger blockCount = new AtomicInteger(0); //how many times has this CB blocked a call that would otherwise go to the backend
        private AtomicInteger failureCount = new AtomicInteger(0); //how many times has the backend thrown an exception
        
        private long lastReset = 0;
        
        public AccountingCircuitBreaker(String name, int threshold, long timeoutMS) {
                super(name, threshold, timeoutMS);
        }

        @Override
        public Object invoke(CircuitInvocation invocation) throws Throwable {
                Object result;
                try {
                        result = super.invoke(invocation);
                } catch (Throwable t) {
                        if (t instanceof CircuitBreakerException) {
                                blockCount.incrementAndGet();
                        } else {
                                failureCount.incrementAndGet();
                        }
                        throw t;
                }
                
                return result;
        }

        @Override
        public void tripBreaker() {
                super.tripBreaker();
                tripCount.incrementAndGet();
        }
        
        public int getTripCount() {
                return tripCount.get();
        }
        
        public int getBlockCount() {
                return blockCount.get();        
        }
        
        public int getFailureCount() {
                return failureCount.get();
        }
        
        public long getLastReset() {
                return lastReset;
        }
        
        public void resetCounters() {
                tripCount.set(0);
                blockCount.set(0);
                failureCount.set(0);
                
                lastReset = System.currentTimeMillis();
        }
        
}
1	package dk.thoerup.curcuitbreaker;
2
3	import java.util.concurrent.atomic.AtomicInteger;
4
5	public class AccountingCircuitBreaker extends CircuitBreaker {
6
7	private AtomicInteger tripCount = new AtomicInteger(0); // how many times ahs the CB tripped
8	private AtomicInteger blockCount = new AtomicInteger(0); //how many times has this CB blocked a call that would otherwise go to the backend
9	private AtomicInteger failureCount = new AtomicInteger(0); //how many times has the backend thrown an exception
10
11	private long lastReset = 0;
12
13	public AccountingCircuitBreaker(String name, int threshold, long timeoutMS) {
14	super(name, threshold, timeoutMS);
15	}
16
17	@Override
18	public Object invoke(CircuitInvocation invocation) throws Throwable {
19	Object result;
20	try {
21	result = super.invoke(invocation);
22	} catch (Throwable t) {
23	if (t instanceof CircuitBreakerException) {
24	blockCount.incrementAndGet();
25	} else {
26	failureCount.incrementAndGet();
27	}
28	throw t;
29	}
30
31	return result;
32	}
33
34	@Override
35	public void tripBreaker() {
36	super.tripBreaker();
37	tripCount.incrementAndGet();
38	}
39
40	public int getTripCount() {
41	return tripCount.get();
42	}
43
44	public int getBlockCount() {
45	return blockCount.get();
46	}
47
48	public int getFailureCount() {
49	return failureCount.get();
50	}
51
52	public long getLastReset() {
53	return lastReset;
54	}
55
56	public void resetCounters() {
57	tripCount.set(0);
58	blockCount.set(0);
59	failureCount.set(0);
60
61	lastReset = System.currentTimeMillis();
62	}
63
64	}