diff --git a/collector/src/main/java/com/navercorp/pinpoint/collector/cluster/zookeeper/DefaultZookeeperClient.java b/collector/src/main/java/com/navercorp/pinpoint/collector/cluster/zookeeper/DefaultZookeeperClient.java index cf7a4b75e..99ba6bb91 100644 --- a/collector/src/main/java/com/navercorp/pinpoint/collector/cluster/zookeeper/DefaultZookeeperClient.java +++ b/collector/src/main/java/com/navercorp/pinpoint/collector/cluster/zookeeper/DefaultZookeeperClient.java @@ -23,7 +23,6 @@ import com.navercorp.pinpoint.collector.cluster.zookeeper.exception.NoNodeExcept import com.navercorp.pinpoint.collector.cluster.zookeeper.exception.PinpointZookeeperException; import com.navercorp.pinpoint.collector.cluster.zookeeper.exception.TimeoutException; import com.navercorp.pinpoint.collector.cluster.zookeeper.exception.UnknownException; -import com.navercorp.pinpoint.common.util.concurrent.CommonState; import com.navercorp.pinpoint.common.util.concurrent.CommonStateContext; import com.navercorp.pinpoint.rpc.util.TimerFactory; import org.apache.zookeeper.CreateMode; @@ -92,7 +91,7 @@ public class DefaultZookeeperClient implements ZookeeperClient { @Override public void reconnectWhenSessionExpired() { - if (stateContext.getCurrentState() != CommonState.STARTED) { + if (!stateContext.isStarted()) { logger.warn("ZookeeperClient.reconnectWhenSessionExpired() failed. Error: Already closed."); return; } @@ -111,7 +110,7 @@ public class DefaultZookeeperClient implements ZookeeperClient { } ZooKeeper newZookeeper = createNewZookeeper(); - if (stateContext.getCurrentState() != CommonState.STARTED) { + if (!stateContext.isStarted()) { logger.warn("ZookeeperClient.reconnectWhenSessionExpired() failed. Error: Already closed."); return; } @@ -277,7 +276,7 @@ public class DefaultZookeeperClient implements ZookeeperClient { @Override public boolean isConnected() { - if (!watcher.isConnected() || stateContext.getCurrentState() != CommonState.STARTED) { + if (!watcher.isConnected() || !stateContext.isStarted()) { return false; } diff --git a/web/src/main/java/com/navercorp/pinpoint/web/cluster/zookeeper/ZookeeperClient.java b/web/src/main/java/com/navercorp/pinpoint/web/cluster/zookeeper/ZookeeperClient.java index 8d5c4cfdd..d68f804b2 100644 --- a/web/src/main/java/com/navercorp/pinpoint/web/cluster/zookeeper/ZookeeperClient.java +++ b/web/src/main/java/com/navercorp/pinpoint/web/cluster/zookeeper/ZookeeperClient.java @@ -17,7 +17,6 @@ package com.navercorp.pinpoint.web.cluster.zookeeper; -import com.navercorp.pinpoint.common.util.concurrent.CommonState; import com.navercorp.pinpoint.common.util.concurrent.CommonStateContext; import com.navercorp.pinpoint.rpc.util.TimerFactory; import com.navercorp.pinpoint.web.cluster.zookeeper.exception.AuthException; @@ -94,7 +93,7 @@ public class ZookeeperClient { } public void reconnectWhenSessionExpired() { - if (stateContext.getCurrentState() != CommonState.STARTED) { + if (!stateContext.isStarted()) { logger.warn("ZookeeperClient.reconnectWhenSessionExpired() failed. Error: Already closed."); return; } @@ -259,7 +258,7 @@ public class ZookeeperClient { } private void checkState() throws PinpointZookeeperException { - if (!zookeeperDataManager.isConnected() || stateContext.getCurrentState() != CommonState.STARTED) { + if (!zookeeperDataManager.isConnected() || !stateContext.isStarted()) { throw new ConnectionException("Instance must be connected."); } }