--- CircuitBreaker/src/dk/thoerup/curcuitbreaker/AccountingCircuitBreaker.java	2009/10/07 04:17:45	404
+++ CircuitBreaker/src/dk/thoerup/circuitbreaker/AccountingCircuitBreaker.java	2010/03/08 08:38:36	621
@@ -1,6 +1,7 @@
-package dk.thoerup.curcuitbreaker;
+package dk.thoerup.circuitbreaker;
 
 import java.util.concurrent.atomic.AtomicInteger;
+import java.util.concurrent.atomic.AtomicLong;
 
 public class AccountingCircuitBreaker extends CircuitBreaker {
 
@@ -8,26 +9,30 @@
 	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 long lastResetCounters = 0;
 	
 	public AccountingCircuitBreaker(String name, int threshold, long timeoutMS) {
 		super(name, threshold, timeoutMS);
+
+		resetCounters();
 	}
 
 	@Override
-	public Object invoke(CircuitInvocation invocation) throws Throwable {
+	public Object invoke(CircuitInvocation invocation) throws Exception {
 		Object result;
 		try {
 			totalCallCount.incrementAndGet();
 			result = super.invoke(invocation);
-		} catch (Throwable t) {
-			if (t instanceof CircuitBreakerException) {
+		} catch (Exception e) {
+			if (e instanceof CircuitBreakerException) {
 				blockCount.incrementAndGet();
 			} else {
 				totalFailureCount.incrementAndGet();
+				lastFailure.set( System.currentTimeMillis() );
 			}
-			throw t;
+			throw e;
 		}
 		
 		return result;
@@ -37,6 +42,7 @@
 	public void tripBreaker() {
 		super.tripBreaker();
 		tripCount.incrementAndGet();
+		lastTrip.set( System.currentTimeMillis() );
 	}
 	
 	public int getTripCount() {
@@ -55,6 +61,14 @@
 		return lastResetCounters;
 	}
 	
+	public long getLastTrip() {
+		return lastTrip.get();
+	}
+	
+	public long getLastFailure() {
+		return lastFailure.get();
+	}
+	
 	public int getTotalCallCount() {
 		return totalCallCount.get();
 	}
@@ -64,6 +78,8 @@
 		blockCount.set(0);
 		totalFailureCount.set(0);
 		totalCallCount.set(0);
+		lastTrip.set(0);
+		lastFailure.set(0);
 		
 		lastResetCounters = System.currentTimeMillis();
 	}