Search in sources :

Example 1 with COMMS

use of com.twosigma.beakerx.kernel.comm.Comm.COMMS in project beakerx by twosigma.

the class CommInfoHandler method handleMsg.

private void handleMsg(Message message) {
    logger.debug("Processing CommInfoHandler");
    Message reply = new Message();
    reply.setHeader(new Header(COMM_INFO_REPLY, message.getHeader().getSession()));
    HashMap<String, Serializable> content = new HashMap<>();
    content.put(COMMS, new HashMap<String, Serializable>());
    String target = getMessageTarget(message);
    kernel.getCommHashSet().stream().map(hash -> kernel.getComm(hash)).filter(comm -> target == null || target.isEmpty() || comm.getTargetName().equals(target)).forEach(comm -> {
        HashMap<String, Serializable> commRepDetails = new HashMap<>();
        commRepDetails.put(TARGET_NAME, comm.getTargetName());
        ((HashMap<String, Serializable>) content.get(COMMS)).put(comm.getCommId(), commRepDetails);
    });
    reply.setContent(content);
    reply.setParentHeader(message.getHeader());
    reply.setIdentities(message.getIdentities());
    send(reply);
}
Also used : Logger(org.slf4j.Logger) TARGET_NAME(com.twosigma.beakerx.kernel.comm.Comm.TARGET_NAME) Comm(com.twosigma.beakerx.kernel.comm.Comm) COMM_INFO_REPLY(com.twosigma.beakerx.kernel.msg.JupyterMessages.COMM_INFO_REPLY) LoggerFactory(org.slf4j.LoggerFactory) HashMap(java.util.HashMap) Message(com.twosigma.beakerx.message.Message) COMMS(com.twosigma.beakerx.kernel.comm.Comm.COMMS) Serializable(java.io.Serializable) KernelHandlerWrapper.wrapBusyIdle(com.twosigma.beakerx.handler.KernelHandlerWrapper.wrapBusyIdle) KernelHandler(com.twosigma.beakerx.handler.KernelHandler) Map(java.util.Map) KernelFunctionality(com.twosigma.beakerx.kernel.KernelFunctionality) Header(com.twosigma.beakerx.message.Header) Serializable(java.io.Serializable) Message(com.twosigma.beakerx.message.Message) Header(com.twosigma.beakerx.message.Header) HashMap(java.util.HashMap)

Aggregations

KernelHandler (com.twosigma.beakerx.handler.KernelHandler)1 KernelHandlerWrapper.wrapBusyIdle (com.twosigma.beakerx.handler.KernelHandlerWrapper.wrapBusyIdle)1 KernelFunctionality (com.twosigma.beakerx.kernel.KernelFunctionality)1 Comm (com.twosigma.beakerx.kernel.comm.Comm)1 COMMS (com.twosigma.beakerx.kernel.comm.Comm.COMMS)1 TARGET_NAME (com.twosigma.beakerx.kernel.comm.Comm.TARGET_NAME)1 COMM_INFO_REPLY (com.twosigma.beakerx.kernel.msg.JupyterMessages.COMM_INFO_REPLY)1 Header (com.twosigma.beakerx.message.Header)1 Message (com.twosigma.beakerx.message.Message)1 Serializable (java.io.Serializable)1 HashMap (java.util.HashMap)1 Map (java.util.Map)1 Logger (org.slf4j.Logger)1 LoggerFactory (org.slf4j.LoggerFactory)1