package dk.thoerup.circuitbreaker;

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

import dk.thoerup.circuitbreaker.config.BreakerConfig;

public class AccountingCircuitBreaker extends CircuitBreaker {

	private AtomicInteger tripCount = new AtomicInteger(0); // how many times ahs the CB tripped  
	private AtomicInteger retripCount = new AtomicInteger(0);   
	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 lastRetrip = new AtomicLong(0);
	private AtomicLong lastFailure = new AtomicLong(0);
	private AtomicLong lastReset = new AtomicLong(0);
	
	private long lastResetCounters = 0;
	
	public AccountingCircuitBreaker(String name, BreakerConfig config) {
		super(name, config);

		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 retripBreaker() {
		super.retripBreaker();
		retripCount.incrementAndGet();
		lastRetrip.set( System.currentTimeMillis() );
	}
	
	
	@Override
	public void reset() {
		super.reset();
		lastReset.set( System.currentTimeMillis());
	}

	public int getTripCount() {
		return tripCount.get();
	}
	
	public int getRetripCount() {
		return retripCount.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 getLastRetrip() {
		return lastRetrip.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);
		retripCount.set(0);
		blockCount.set(0);
		totalFailureCount.set(0);
		totalCallCount.set(0);
		lastTrip.set(0);
		lastRetrip.set(0);
		lastReset.set(0);
		lastFailure.set(0);
		
		lastResetCounters = System.currentTimeMillis();
	}
	
}