zyxxoo commented on code in PR #1943:
URL:
https://github.com/apache/incubator-hugegraph/pull/1943#discussion_r946479746
##########
hugegraph-core/src/main/java/com/baidu/hugegraph/election/RoleElectionStateMachineImpl.java:
##########
@@ -0,0 +1,281 @@
+package com.baidu.hugegraph.election;
+
+import java.util.Optional;
+import java.util.concurrent.locks.LockSupport;
+
+import com.baidu.hugegraph.util.E;
+
+public class RoleElectionStateMachineImpl implements RoleElectionStateMachine {
+
+ private volatile boolean shutdown = false;
+ private Config config;
+
+ private volatile RoleState state;
+
+ private final MetaDataAdapter metaDataAdapter;
+
+ public RoleElectionStateMachineImpl(Config config, MetaDataAdapter
adapter) {
+ this.config = config;
+ this.metaDataAdapter = adapter;
+ this.state = new UnKnownState(null);
+ }
+
+ @Override
+ public void shutdown() {
+ this.shutdown = true;
+ }
+
+ @Override
+ public void apply(StateMachineCallback stateMachineCallback) {
+ int failCount = 0;
+ StateMachineContextImpl context = new StateMachineContextImpl(this);
+ while (!this.shutdown) {
+ E.checkArgumentNotNull(this.state, "State don't be null");
+ try {
+ this.state = state.transform(context);
+ Callback runnable = this.state.callback(stateMachineCallback);
+ runnable.call(context);
+ failCount = 0;
+ } catch (Throwable e) {
+ stateMachineCallback.error(context, e);
+ failCount ++;
+ if (failCount >= this.config.exceedsFailCount()) {
+ this.state = new SafeState(context.epoch());
+ Callback runnable =
this.state.callback(stateMachineCallback);
+ runnable.call(context);
+ }
+ }
+ }
+ }
+
+ private interface RoleState {
+
+ RoleState transform(StateMachineContext context);
+
+ Callback callback(StateMachineCallback callback);
+
+ static void heartBeatPark(StateMachineContext context) {
+ long heartBeatIntervalSecond =
context.config().heartBeatIntervalSecond();
+ LockSupport.parkNanos(heartBeatIntervalSecond * 1_000_000_000);
+ }
+
+ static void randomPark(StateMachineContext context) {
+ long randomTimeout = context.config().randomTimeoutMillisecond();
+ LockSupport.parkNanos(randomTimeout * 1_000_000);
+ }
+ }
+
+ @FunctionalInterface
+ private interface Callback {
+
+ void call(StateMachineContext context);
+ }
+
+ private static class UnKnownState implements RoleState {
Review Comment:
UnKnown equal Unknown
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
