src/jvm/backtype/storm/testing/AckFailMapTracker.java

package backtype.storm.testing;

import backtype.storm.utils.RegisteredGlobalState;
import java.util.HashSet;
import java.util.Set;

public class AckFailMapTracker implements AckFailDelegate {
    
    String _acked;
    String _failed;
    
    public AckFailMapTracker() {
        _acked = RegisteredGlobalState.registerState(new HashSet());
        _failed = RegisteredGlobalState.registerState(new HashSet());
    }
    
    public boolean isAcked(Object id) {
        return ((Set)RegisteredGlobalState.getState(_acked)).contains(id);
    }
    
    public boolean isFailed(Object id) {
        return ((Set)RegisteredGlobalState.getState(_failed)).contains(id);        
    }

    @Override
    public void ack(Object id) {
        ((Set)RegisteredGlobalState.getState(_acked)).add(id);
    }

    @Override
    public void fail(Object id) {
        ((Set)RegisteredGlobalState.getState(_failed)).add(id);
    }
    
}