Search in sources :

Example 11 with Comm

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

the class CommMsgHandler method handleMsg.

private void handleMsg(Message message) {
    Map<String, Serializable> commMap = message.getContent();
    Comm comm = kernel.getComm(getString(commMap, COMM_ID));
    logger.debug("Comm message handling, target name: " + (comm != null ? comm.getTargetName() : "undefined"));
    if (comm != null) {
        comm.handleMsg(message);
    }
}
Also used : Serializable(java.io.Serializable) Comm(com.twosigma.beakerx.kernel.comm.Comm)

Example 12 with Comm

use of com.twosigma.beakerx.kernel.comm.Comm 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)

Example 13 with Comm

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

the class CommOpenHandler method readComm.

protected Comm readComm(Map<String, Serializable> map) {
    Comm ret = new Comm(getString(map, COMM_ID), getString(map, TARGET_NAME));
    ret.setData((HashMap<String, Serializable>) map.get(DATA));
    ret.setTargetModule(getString(map, TARGET_MODULE));
    return ret;
}
Also used : Serializable(java.io.Serializable) Comm(com.twosigma.beakerx.kernel.comm.Comm)

Aggregations

Comm (com.twosigma.beakerx.kernel.comm.Comm)13 Serializable (java.io.Serializable)7 HashMap (java.util.HashMap)5 Message (com.twosigma.beakerx.message.Message)4 ConcurrentHashMap (java.util.concurrent.ConcurrentHashMap)3 KernelTest (com.twosigma.beakerx.KernelTest)2 Header (com.twosigma.beakerx.message.Header)2 Map (java.util.Map)2 KernelHandler (com.twosigma.beakerx.handler.KernelHandler)1 KernelHandlerWrapper.wrapBusyIdle (com.twosigma.beakerx.handler.KernelHandlerWrapper.wrapBusyIdle)1 JupyterHandlerTest (com.twosigma.beakerx.jupyter.handler.JupyterHandlerTest)1 SimpleEvaluationObject (com.twosigma.beakerx.jvm.object.SimpleEvaluationObject)1 KernelFunctionality (com.twosigma.beakerx.kernel.KernelFunctionality)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 LinkedHashMap (java.util.LinkedHashMap)1 List (java.util.List)1 Before (org.junit.Before)1 Test (org.junit.Test)1