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
        
        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 void resetCounters() {
                tripCount.set(0);
                blockCount.set(0);
                failureCount.set(0);
        }
        
}
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	public AccountingCircuitBreaker(String name, int threshold, long timeoutMS) {
12	super(name, threshold, timeoutMS);
13	}
14
15	@Override
16	public Object invoke(CircuitInvocation invocation) throws Throwable {
17	Object result;
18	try {
19	result = super.invoke(invocation);
20	} catch (Throwable t) {
21	if (t instanceof CircuitBreakerException) {
22	blockCount.incrementAndGet();
23	} else {
24	failureCount.incrementAndGet();
25	}
26	throw t;
27	}
28
29	return result;
30	}
31
32	@Override
33	public void tripBreaker() {
34	super.tripBreaker();
35	tripCount.incrementAndGet();
36	}
37
38	public int getTripCount() {
39	return tripCount.get();
40	}
41
42	public int getBlockCount() {
43	return blockCount.get();
44	}
45
46	public int getFailureCount() {
47	return failureCount.get();
48	}
49
50	public void resetCounters() {
51	tripCount.set(0);
52	blockCount.set(0);
53	failureCount.set(0);
54	}
55
56	}