Refactor StateContext (#32909)

apache · Sep 16, 2024 · 28e3d2c · 28e3d2c
1 parent 973d027
commit 28e3d2c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 8 deletions.
diff --git a/mode/core/src/main/java/org/apache/shardingsphere/mode/state/StateContext.java b/mode/core/src/main/java/org/apache/shardingsphere/mode/state/StateContext.java
@@ -17,23 +17,19 @@
 
 package org.apache.shardingsphere.mode.state;
 
-import lombok.extern.slf4j.Slf4j;
 import org.apache.shardingsphere.infra.state.cluster.ClusterState;
 
 import java.util.concurrent.atomic.AtomicReference;
 
 /**
  * State context.
  */
-@Slf4j
 public final class StateContext {
 
-    private final AtomicReference<ClusterState> clusterState = new AtomicReference<>(ClusterState.OK);
+    private final AtomicReference<ClusterState> clusterState;
 
-    public StateContext(final ClusterState repositoryClusterState) {
-        if (clusterState.get() != repositoryClusterState) {
-            switchClusterState(repositoryClusterState);
-        }
+    public StateContext(final ClusterState clusterState) {
+        this.clusterState = new AtomicReference<>(clusterState);
     }
 
     /**

diff --git a/mode/core/src/test/java/org/apache/shardingsphere/mode/state/StateContextTest.java b/mode/core/src/test/java/org/apache/shardingsphere/mode/state/StateContextTest.java
@@ -28,7 +28,7 @@ class StateContextTest {
     private final StateContext stateContext = new StateContext(ClusterState.OK);
 
     @Test
-    void assertGetCurrentClusterState() {
+    void assertGetClusterState() {
         assertThat(stateContext.getClusterState(), is(ClusterState.OK));
     }