final ConcurrentMap<String, AtomicInteger> secondCounts =
new ConcurrentHashMap<String, AtomicInteger>();
final HelixIPCService secondIPC =
new NettyHelixIPCService(new NettyHelixIPCService.Config().setInstanceName(
secondNode.getInstanceName()).setPort(secondPort));
secondIPC.registerCallback(messageType, new HelixIPCCallback() {
@Override
public void onMessage(HelixMessageScope scope, UUID messageId, ByteBuf message) {
String key = scope.getPartition() + ":" + scope.getState();
secondCounts.putIfAbsent(key, new AtomicInteger());
secondCounts.get(key).incrementAndGet();