com.orientechnologies.agent.operation.NodesManager Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of agent Show documentation
The newest version!
package com.orientechnologies.agent.operation;

import com.orientechnologies.agent.cloud.processor.tasks.request.NewEnterpriseStatsTask;
import com.orientechnologies.agent.cloud.processor.tasks.response.EnterpriseStatsResponse;
import com.orientechnologies.common.concur.lock.OInterruptedException;
import com.orientechnologies.orient.server.distributed.ODistributedRequest;
import com.orientechnologies.orient.server.distributed.ODistributedServerManager;
import com.orientechnologies.orient.server.distributed.operation.NodeOperation;
import com.orientechnologies.orient.server.distributed.operation.NodeOperationTask;
import java.io.IOException;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

public class NodesManager {

  private ODistributedServerManager manager;

  public NodesManager(final ODistributedServerManager manager) {
    this.manager = manager;
    initCommands();
  }

  private void initCommands() {
    NodeOperationTask.register(
        2, () -> new NewEnterpriseStatsTask(), () -> new EnterpriseStatsResponse());
  }

  public List sendAll(NodeOperation task) {
    Set servers = manager.getActiveServers();
    long requestId = manager.getNextMessageIdCounter();
    OperationResponseManager responseManager = new OperationResponseManager(servers);
    ODistributedRequest req =
        new ODistributedRequest(
            manager, manager.getLocalNodeId(), requestId, null, new NodeOperationTask(task));
    for (String server : servers) {
      try {
        manager.getRemoteServer(server).sendRequest(req);
        manager.getMessageService().registerRequest(requestId, responseManager);
      } catch (IOException e) {
        responseManager.removeServerBecauseUnreachable(server);
      }
    }
    try {
      responseManager.waitForSynchronousResponses();
    } catch (InterruptedException e) {
      throw new OInterruptedException(e.getMessage());
    }
    return responseManager.getResponses();
  }

  public OperationResponseFromNode send(final String nodeName, final NodeOperation task) {
    Set nodes = new HashSet<>();
    nodes.add(nodeName);
    OperationResponseManager responseManager = new OperationResponseManager(nodes);
    try {
      long requestId = manager.getNextMessageIdCounter();
      ODistributedRequest req =
          new ODistributedRequest(
              manager, manager.getLocalNodeId(), requestId, null, new NodeOperationTask(task));
      manager.getRemoteServer(nodeName).sendRequest(req);

      manager.getMessageService().registerRequest(requestId, responseManager);
    } catch (IOException e) {
      responseManager.removeServerBecauseUnreachable(nodeName);
    }
    try {
      responseManager.waitForSynchronousResponses();
    } catch (InterruptedException e) {
      throw new OInterruptedException(e.getMessage());
    }
    return responseManager.getResponses().size() > 0 ? responseManager.getResponses().get(0) : null;
  }
}