Merge remote-tracking branch 'origin/fix_getState(Replicator-r)' into fix_getState(Replicator-r)

Author: contrueCT

hugegraph-commons/hugegraph-common/pom.xml

@@ -223,6 +223,12 @@
             <groupId>org.projectlombok</groupId>
             <artifactId>lombok</artifactId>
         </dependency>
+        <dependency>
+            <groupId>com.alipay.sofa</groupId>
+            <artifactId>jraft-core</artifactId>
+            <version>1.3.14</version>
+            <scope>compile</scope>
+        </dependency>
     </dependencies>
 
     <dependencyManagement>

hugegraph-commons/hugegraph-common/src/main/java/org/apache/hugegraph/util/RaftReflectionUtil.java

@@ -0,0 +1,90 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.apache.hugegraph.util;
+
+import com.alipay.sofa.jraft.Node;
+import com.alipay.sofa.jraft.ReplicatorGroup;
+import com.alipay.sofa.jraft.core.Replicator;
+import com.alipay.sofa.jraft.entity.PeerId;
+import com.alipay.sofa.jraft.util.ThreadId;
+
+import lombok.extern.slf4j.Slf4j;
+import lombok.var;
+
+@Slf4j
+public class RaftReflectionUtil {
+
+    public static Replicator.State getReplicatorState(Node node, PeerId peerId) {
+        if (node == null || peerId == null) {
+            return null;
+        }
+
+        // Get ReplicatorGroup from Node
+        var clz = node.getClass();
+        ReplicatorGroup replicateGroup = null;
+        try {
+            var f = clz.getDeclaredField("replicatorGroup");
+            f.setAccessible(true);
+            try {
+                replicateGroup = (ReplicatorGroup) f.get(node);
+            } finally {
+                f.setAccessible(false);
+            }
+        } catch (NoSuchFieldException | IllegalAccessException e) {
+            log.info("getReplicatorGroup: error {}", e.getMessage());
+            return null;
+        }
+
+        if (replicateGroup == null) {
+            return null;
+        }
+
+        ThreadId threadId = replicateGroup.getReplicator(peerId);
+        if (threadId == null) {
+            return null;
+        } else {
+            Replicator r = (Replicator) threadId.lock();
+            if (r == null) {
+                return Replicator.State.Probe;
+            }
+            Replicator.State result = null;
+
+            // Get state from Replicator
+            try {
+                var replicatorClz = r.getClass();
+                try {
+                    var f = replicatorClz.getDeclaredField("state");
+                    f.setAccessible(true);
+                    try {
+                        result = (Replicator.State) f.get(r);
+                    } finally {
+                        f.setAccessible(false);
+                    }
+                } catch (NoSuchFieldException | IllegalAccessException e) {
+                    log.info("getReplicatorState: error {}", e.getMessage());
+                    result = null;
+                }
+            }
+            finally {
+                threadId.unlock();
+            }
+            return result;
+        }
+    }
+
+}

hugegraph-pd/hg-pd-core/pom.xml

@@ -83,5 +83,11 @@
             <artifactId>gson</artifactId>
             <version>2.8.9</version>
         </dependency>
+        <dependency>
+            <groupId>org.apache.hugegraph</groupId>
+            <artifactId>hugegraph-common</artifactId>
+            <version>1.7.0</version>
+            <scope>compile</scope>
+        </dependency>
     </dependencies>
 </project>

hugegraph-pd/hg-pd-core/src/main/java/org/apache/hugegraph/pd/raft/RaftEngine.java

@@ -60,6 +60,8 @@
 import io.netty.channel.ChannelHandler;
 import lombok.extern.slf4j.Slf4j;
 
+import org.apache.hugegraph.util.RaftReflectionUtil;
+
 @Slf4j
 public class RaftEngine {
 

hugegraph-store/hg-store-core/src/main/java/org/apache/hugegraph/store/PartitionEngine.java

@@ -99,6 +99,8 @@
 import lombok.Getter;
 import lombok.extern.slf4j.Slf4j;
 
+import org.apache.hugegraph.util.RaftReflectionUtil;
+
 /**
  * Partition processing engine
  */