thoerup/circuitbreaker/AccountingCircuitBreaker.java

package dk.thoerup.circuitbreaker;

import java.util.concurrent.atomic.AtomicInteger;
import java.util.concurrent.atomic.AtomicLong;

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 totalFailureCount = new AtomicInteger(0); //how many times has the backend thrown an exception
        private AtomicInteger totalCallCount = new AtomicInteger(0);
        
        private AtomicLong lastTrip = new AtomicLong(0);
        private AtomicLong lastFailure = new AtomicLong(0);
        private AtomicLong lastReset = new AtomicLong(0);
        
        private long lastResetCounters = 0;
        
        public AccountingCircuitBreaker(String name, int threshold, long timeoutMS) {
                super(name, threshold, timeoutMS);

                resetCounters();
        }

        @Override
        public Object invoke(CircuitInvocation invocation) throws Exception {
                Object result;
                try {
                        totalCallCount.incrementAndGet();
                        result = super.invoke(invocation);
                } catch (Exception e) {
                        if (e instanceof CircuitBreakerException) {
                                blockCount.incrementAndGet();
                        } else {
                                totalFailureCount.incrementAndGet();
                                lastFailure.set( System.currentTimeMillis() );
                        }
                        throw e;
                }
                
                return result;
        }

        @Override
        public void tripBreaker() {
                super.tripBreaker();
                tripCount.incrementAndGet();
                lastTrip.set( System.currentTimeMillis() );
        }
        
        
        
        @Override
        public void reset() {
                super.reset();
                lastReset.set( System.currentTimeMillis());
        }

        public int getTripCount() {
                return tripCount.get();
        }
        
        public int getBlockCount() {
                return blockCount.get();        
        }
        
        public int getTotalFailureCount() {
                return totalFailureCount.get();
        }
        
        public long getLastResetCounters() {
                return lastResetCounters;
        }
        
        public long getLastTrip() {
                return lastTrip.get();
        }
        
        public long getLastFailure() {
                return lastFailure.get();
        }
        
        public long getLastReset() {
                return lastReset.get();
        }
        
        public int getTotalCallCount() {
                return totalCallCount.get();
        }
        
        public void resetCounters() {
                tripCount.set(0);
                blockCount.set(0);
                totalFailureCount.set(0);
                totalCallCount.set(0);
                lastTrip.set(0);
                lastReset.set(0);
                lastFailure.set(0);
                
                lastResetCounters = System.currentTimeMillis();
        }
        
}
1	package dk.thoerup.circuitbreaker;
2
3	import java.util.concurrent.atomic.AtomicInteger;
4	import java.util.concurrent.atomic.AtomicLong;
5
6	public class AccountingCircuitBreaker extends CircuitBreaker {
7
8	private AtomicInteger tripCount = new AtomicInteger(0); // how many times ahs the CB tripped
9	private AtomicInteger blockCount = new AtomicInteger(0); //how many times has this CB blocked a call that would otherwise go to the backend
10	private AtomicInteger totalFailureCount = new AtomicInteger(0); //how many times has the backend thrown an exception
11	private AtomicInteger totalCallCount = new AtomicInteger(0);
12
13	private AtomicLong lastTrip = new AtomicLong(0);
14	private AtomicLong lastFailure = new AtomicLong(0);
15	private AtomicLong lastReset = new AtomicLong(0);
16
17	private long lastResetCounters = 0;
18
19	public AccountingCircuitBreaker(String name, int threshold, long timeoutMS) {
20	super(name, threshold, timeoutMS);
21
22	resetCounters();
23	}
24
25	@Override
26	public Object invoke(CircuitInvocation invocation) throws Exception {
27	Object result;
28	try {
29	totalCallCount.incrementAndGet();
30	result = super.invoke(invocation);
31	} catch (Exception e) {
32	if (e instanceof CircuitBreakerException) {
33	blockCount.incrementAndGet();
34	} else {
35	totalFailureCount.incrementAndGet();
36	lastFailure.set( System.currentTimeMillis() );
37	}
38	throw e;
39	}
40
41	return result;
42	}
43
44	@Override
45	public void tripBreaker() {
46	super.tripBreaker();
47	tripCount.incrementAndGet();
48	lastTrip.set( System.currentTimeMillis() );
49	}
50
51
52
53	@Override
54	public void reset() {
55	super.reset();
56	lastReset.set( System.currentTimeMillis());
57	}
58
59	public int getTripCount() {
60	return tripCount.get();
61	}
62
63	public int getBlockCount() {
64	return blockCount.get();
65	}
66
67	public int getTotalFailureCount() {
68	return totalFailureCount.get();
69	}
70
71	public long getLastResetCounters() {
72	return lastResetCounters;
73	}
74
75	public long getLastTrip() {
76	return lastTrip.get();
77	}
78
79	public long getLastFailure() {
80	return lastFailure.get();
81	}
82
83	public long getLastReset() {
84	return lastReset.get();
85	}
86
87	public int getTotalCallCount() {
88	return totalCallCount.get();
89	}
90
91	public void resetCounters() {
92	tripCount.set(0);
93	blockCount.set(0);
94	totalFailureCount.set(0);
95	totalCallCount.set(0);
96	lastTrip.set(0);
97	lastReset.set(0);
98	lastFailure.set(0);
99
100	lastResetCounters = System.currentTimeMillis();
101	}
102
103	}