linyiqun commented on a change in pull request #480: HDDS-2927. Cache EndPoint
tasks instead of creating them all the time in RunningDatanodeState
URL: https://github.com/apache/hadoop-ozone/pull/480#discussion_r370452344
##########
File path:
hadoop-hdds/container-service/src/main/java/org/apache/hadoop/ozone/container/common/states/datanode/RunningDatanodeState.java
##########
@@ -49,14 +52,62 @@
private final SCMConnectionManager connectionManager;
private final Configuration conf;
private final StateContext context;
- private CompletionService<EndpointStateMachine.EndPointStates> ecs;
+ private CompletionService<EndPointStates> ecs;
+ /** Cache the end point task per end point per end point state. */
+ private Map<String, Map<EndPointStates,
+ Callable<EndPointStates>>> endpointTasks;
public RunningDatanodeState(Configuration conf,
SCMConnectionManager connectionManager,
StateContext context) {
this.connectionManager = connectionManager;
this.conf = conf;
this.context = context;
+ initEndPointTask();
+ }
+
+ /**
+ * Initialize end point tasks corresponding to each end point,
+ * each end point state.
+ */
+ private void initEndPointTask() {
+ endpointTasks = new HashMap<String, Map<EndPointStates,
+ Callable<EndPointStates>>>();
+ for (EndpointStateMachine endpoint : connectionManager.getValues()) {
+ Map<EndPointStates, Callable<EndPointStates>> endpointTaskForState
+ = new HashMap<EndPointStates, Callable<EndPointStates>>();
+
+ for (EndPointStates state : EndPointStates.values()) {
+ Callable<EndPointStates> endPointTask = null;
+ switch (endpoint.getState()) {
+ case GETVERSION:
+ endPointTask = new VersionEndpointTask(endpoint, conf,
+ context.getParent().getContainer());
+ break;
+ case REGISTER:
+ endPointTask = RegisterEndpointTask.newBuilder()
+ .setConfig(conf)
+ .setEndpointStateMachine(endpoint)
+ .setContext(context)
+ .setDatanodeDetails(context.getParent().getDatanodeDetails())
+ .setOzoneContainer(context.getParent().getContainer())
+ .build();
+ break;
+ case HEARTBEAT:
+ endPointTask = HeartbeatEndpointTask.newBuilder()
+ .setConfig(conf)
+ .setEndpointStateMachine(endpoint)
+ .setDatanodeDetails(context.getParent().getDatanodeDetails())
+ .setContext(context)
+ .build();
+ break;
+ case SHUTDOWN:
+ break;
+ }
+ endpointTaskForState.put(state, endPointTask);
Review comment:
Have removed SHUTDOWN switch case.
----------------------------------------------------------------
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.
For queries about this service, please contact Infrastructure at:
[email protected]
With regards,
Apache Git Services
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
