Examples with SlowMessage com.bakdata.conquery.models.messages.SlowMessage used on opensource projects

Example 1 with SlowMessage

use of com.bakdata.conquery.models.messages.SlowMessage in project conquery by bakdata.

the class ManagerNode method messageReceived.

@Override
public void messageReceived(IoSession session, Object message) throws Exception {
    ConqueryMDC.setLocation("ManagerNode[" + session.getLocalAddress().toString() + "]");
    if (message instanceof MessageToManagerNode) {
        MessageToManagerNode mrm = (MessageToManagerNode) message;
        log.trace("ManagerNode received {} from {}", message.getClass().getSimpleName(), session.getRemoteAddress());
        ReactingJob<MessageToManagerNode, NetworkMessageContext.ManagerNodeNetworkContext> job = new ReactingJob<>(mrm, new NetworkMessageContext.ManagerNodeNetworkContext(jobManager, new NetworkSession(session), datasetRegistry, config.getCluster().getBackpressure()));
        // TODO: 01.07.2020 FK: distribute messages/jobs to their respective JobManagers (if they have one)
        if (mrm.isSlowMessage()) {
            ((SlowMessage) mrm).setProgressReporter(job.getProgressReporter());
            jobManager.addSlowJob(job);
        } else {
            jobManager.addFastJob(job);
        }
    } else {
        log.error("Unknown message type {} in {}", message.getClass(), message);
    }
}

Example 2 with SlowMessage

use of com.bakdata.conquery.models.messages.SlowMessage in project conquery by bakdata.

the class ForwardToWorker method react.

@Override
public void react(ShardNodeNetworkContext context) throws Exception {
    Worker worker = Objects.requireNonNull(context.getWorkers().getWorker(workerId));
    ConqueryMDC.setLocation(worker.toString());
    WorkerMessage message = deserializeMessage(messageRaw, context.getWorkers().getBinaryMapper());
    if (message instanceof SlowMessage) {
        ((SlowMessage) message).setProgressReporter(progressReporter);
    }
    message.react(worker);
}

Example 3 with SlowMessage

use of com.bakdata.conquery.models.messages.SlowMessage in project conquery by bakdata.

the class ShardNode method messageReceived.

@Override
public void messageReceived(IoSession session, Object message) throws Exception {
    setLocation(session);
    if (!(message instanceof MessageToShardNode)) {
        log.error("Unknown message type {} in {}", message.getClass(), message);
        return;
    }
    MessageToShardNode srm = (MessageToShardNode) message;
    log.trace("{} recieved {} from {}", getName(), message.getClass().getSimpleName(), session.getRemoteAddress());
    ReactingJob<MessageToShardNode, ShardNodeNetworkContext> job = new ReactingJob<>(srm, context);
    if (((Message) message).isSlowMessage()) {
        ((SlowMessage) message).setProgressReporter(job.getProgressReporter());
        jobManager.addSlowJob(job);
    } else {
        jobManager.addFastJob(job);
    }
}