--- CircuitBreaker/src/dk/thoerup/curcuitbreaker/CircuitBreakerManager.java 2009/10/09 06:45:40 426 +++ CircuitBreaker/src/dk/thoerup/circuitbreaker/CircuitBreakerManager.java 2010/10/03 17:03:08 1153 @@ -1,5 +1,6 @@ -package dk.thoerup.curcuitbreaker; +package dk.thoerup.circuitbreaker; +import java.util.Iterator; import java.util.Map; import java.util.Set; import java.util.concurrent.ConcurrentHashMap; @@ -28,10 +29,21 @@ public CircuitBreaker getCircuitBreaker(String name) { return breakers.get(name); } + + public void removeCircuitBreaker(String name) { + breakers.remove(name); + } public void addCircuitBreaker(CircuitBreaker cb) { breakers.put(cb.getName(), cb); } + public void shutdown() { + for (Iterator it = breakers.keySet().iterator(); it.hasNext(); ) { + String key = it.next(); + breakers.get(key).shutdown(); + it.remove(); + } + } }