Examples with AgentWorkContext com.thoughtworks.go.remote.work.AgentWorkContext used on opensource projects

Example 1 with AgentWorkContext

use of com.thoughtworks.go.remote.work.AgentWorkContext in project gocd by gocd.

the class AgentWebSocketClientController method process.

void process(Message message) throws InterruptedException {
    switch(message.getAction()) {
        case cancelBuild:
            cancelJobIfThereIsOneRunning();
            cancelBuild();
            break;
        case setCookie:
            String cookie = MessageEncoding.decodeData(message.getData(), String.class);
            getAgentRuntimeInfo().setCookie(cookie);
            LOG.info("Got cookie: {}", cookie);
            break;
        case assignWork:
            cancelJobIfThereIsOneRunning();
            Work work = MessageEncoding.decodeWork(message.getData());
            LOG.debug("Got work from server: [{}]", work.description());
            runner = new JobRunner();
            try {
                runner.run(work, new AgentWorkContext(agentIdentifier(), new BuildRepositoryRemoteAdapter(runner, webSocketSessionHandler), manipulator, getAgentRuntimeInfo(), packageRepositoryExtension, scmExtension, taskExtension, artifactExtension, pluginRequestProcessorRegistry));
            } finally {
                getAgentRuntimeInfo().idle();
                updateServerAgentRuntimeInfo();
            }
            break;
        case build:
            cancelBuild();
            BuildSettings buildSettings = MessageEncoding.decodeData(message.getData(), BuildSettings.class);
            runBuild(buildSettings);
            break;
        case reregister:
            LOG.warn("Reregister: invalidate current agent certificate fingerprint {} and stop websocket webSocketClient.", getAgentRegistry().uuid());
            webSocketSessionHandler.stop();
            sslInfrastructureService.invalidateAgentCertificate();
            break;
        case acknowledge:
            webSocketSessionHandler.acknowledge(message);
            break;
        default:
            throw new RuntimeException("Unknown action: " + message.getAction());
    }
}

Example 2 with AgentWorkContext

use of com.thoughtworks.go.remote.work.AgentWorkContext in project gocd by gocd.

the class AgentHTTPClientController method retrieveWork.

void retrieveWork() {
    final BuildRepositoryRemote client = remote();
    AgentIdentifier agentIdentifier = agentIdentifier();
    LOG.debug("[Agent Loop] {} is checking for work from Go", agentIdentifier);
    Work work;
    try {
        getAgentRuntimeInfo().idle();
        work = client.getWork(getAgentRuntimeInfo());
        if (!(work instanceof NoWork)) {
            LOG.debug("[Agent Loop] Got work from server: [{}]", work.description());
        }
        runner = new JobRunner();
        final AgentWorkContext agentWorkContext = new AgentWorkContext(agentIdentifier, client, manipulator, getAgentRuntimeInfo(), packageRepositoryExtension, scmExtension, taskExtension, artifactExtension, pluginRequestProcessorRegistry);
        runner.run(work, agentWorkContext);
    } catch (UnregisteredAgentException e) {
        LOG.warn("[Agent Loop] Invalid agent certificate with fingerprint {}. Registering with server on next iteration.", e.getUuid());
        sslInfrastructureService.invalidateAgentCertificate();
    } finally {
        getAgentRuntimeInfo().idle();
    }
}