Examples with Command com.cloud.agent.api.Command used on opensource projects

Example 6 with Command

use of com.cloud.agent.api.Command in project cloudstack by apache.

the class DirectAgentTest method testDownloadTemplate.

@Test
public void testDownloadTemplate() {
    ImageStoreTO image = Mockito.mock(ImageStoreTO.class);
    PrimaryDataStoreTO primaryStore = Mockito.mock(PrimaryDataStoreTO.class);
    Mockito.when(primaryStore.getUuid()).thenReturn(getLocalStorageUuid());
    // Mockito.when(image.get).thenReturn(primaryStore);
    ImageStoreTO imageStore = Mockito.mock(ImageStoreTO.class);
    Mockito.when(imageStore.getProtocol()).thenReturn("http");
    TemplateObjectTO template = Mockito.mock(TemplateObjectTO.class);
    Mockito.when(template.getPath()).thenReturn(getTemplateUrl());
    Mockito.when(template.getDataStore()).thenReturn(imageStore);
    // Mockito.when(image.getTemplate()).thenReturn(template);
    // CopyTemplateToPrimaryStorageCmd cmd = new
    // CopyTemplateToPrimaryStorageCmd(image);
    Command cmd = null;
    try {
        agentMgr.send(hostId, cmd);
    } catch (AgentUnavailableException e) {
        // TODO Auto-generated catch block
        e.printStackTrace();
    } catch (OperationTimedoutException e) {
        // TODO Auto-generated catch block
        e.printStackTrace();
    }
}

Example 7 with Command

use of com.cloud.agent.api.Command in project cloudstack by apache.

the class AgentManagerImpl method connectAgent.

protected void connectAgent(final Link link, final Command[] cmds, final Request request) {
    // send startupanswer to agent in the very beginning, so agent can move on without waiting for the answer for an undetermined time, if we put this logic into another
    // thread pool.
    final StartupAnswer[] answers = new StartupAnswer[cmds.length];
    Command cmd;
    for (int i = 0; i < cmds.length; i++) {
        cmd = cmds[i];
        if (cmd instanceof StartupRoutingCommand || cmd instanceof StartupProxyCommand || cmd instanceof StartupSecondaryStorageCommand || cmd instanceof StartupStorageCommand) {
            answers[i] = new StartupAnswer((StartupCommand) cmds[i], 0, getPingInterval());
            break;
        }
    }
    Response response = null;
    response = new Response(request, answers[0], _nodeId, -1);
    try {
        link.send(response.toBytes());
    } catch (final ClosedChannelException e) {
        s_logger.debug("Failed to send startupanswer: " + e.toString());
    }
    _connectExecutor.execute(new HandleAgentConnectTask(link, cmds, request));
}

Example 8 with Command

use of com.cloud.agent.api.Command in project cloudstack by apache.

the class AgentManagerImpl method checkForCommandsAndTag.

/**
     * @param commands
     * @return
     */
private Command[] checkForCommandsAndTag(final Commands commands) {
    final Command[] cmds = commands.toCommands();
    assert cmds.length > 0 : "Ask yourself this about a hundred times.  Why am I  sending zero length commands?";
    setEmptyAnswers(commands, cmds);
    for (final Command cmd : cmds) {
        tagCommand(cmd);
    }
    return cmds;
}

Example 9 with Command

use of com.cloud.agent.api.Command in project cloudstack by apache.

the class AgentManagerImpl method send.

@Override
public Answer[] send(final Long hostId, final Commands commands, int timeout) throws AgentUnavailableException, OperationTimedoutException {
    assert hostId != null : "Who's not checking the agent id before sending?  ... (finger wagging)";
    if (hostId == null) {
        throw new AgentUnavailableException(-1);
    }
    if (timeout <= 0) {
        timeout = Wait.value();
    }
    if (CheckTxnBeforeSending.value()) {
        if (!noDbTxn()) {
            throw new CloudRuntimeException("We do not allow transactions to be wrapped around commands sent to be executed on remote agents.  " + "We cannot predict how long it takes a command to complete.  " + "The transaction may be rolled back because the connection took too long.");
        }
    } else {
        assert noDbTxn() : "I know, I know.  Why are we so strict as to not allow txn across an agent call?  ...  Why are we so cruel ... Why are we such a dictator .... Too bad... Sorry...but NO AGENT COMMANDS WRAPPED WITHIN DB TRANSACTIONS!";
    }
    final Command[] cmds = checkForCommandsAndTag(commands);
    final AgentAttache agent = getAttache(hostId);
    if (agent == null || agent.isClosed()) {
        throw new AgentUnavailableException("agent not logged into this management server", hostId);
    }
    final Request req = new Request(hostId, agent.getName(), _nodeId, cmds, commands.stopOnError(), true);
    req.setSequence(agent.getNextSequence());
    final Answer[] answers = agent.send(req, timeout);
    notifyAnswersToMonitors(hostId, req.getSequence(), answers);
    commands.setAnswers(answers);
    return answers;
}

Example 10 with Command

use of com.cloud.agent.api.Command in project cloudstack by apache.

the class ClusteredAgentManagerImpl method propagateAgentEvent.

public Boolean propagateAgentEvent(final long agentId, final Event event) throws AgentUnavailableException {
    final String msPeer = getPeerName(agentId);
    if (msPeer == null) {
        return null;
    }
    if (s_logger.isDebugEnabled()) {
        s_logger.debug("Propagating agent change request event:" + event.toString() + " to agent:" + agentId);
    }
    final Command[] cmds = new Command[1];
    cmds[0] = new ChangeAgentCommand(agentId, event);
    final String ansStr = _clusterMgr.execute(msPeer, agentId, _gson.toJson(cmds), true);
    if (ansStr == null) {
        throw new AgentUnavailableException(agentId);
    }
    final Answer[] answers = _gson.fromJson(ansStr, Answer[].class);
    if (s_logger.isDebugEnabled()) {
        s_logger.debug("Result for agent change is " + answers[0].getResult());
    }
    return answers[0].getResult();
}