Examples with Header com.twosigma.beakerx.message.Header used on opensource projects

Example 16 with Header

use of com.twosigma.beakerx.message.Header in project beakerx by twosigma.

the class JavaKernelInfoHandlerTest method handle_sentMessageHasHeaderTypeIsKernelInfoReply.

@Test
public void handle_sentMessageHasHeaderTypeIsKernelInfoReply() throws Exception {
    // when
    handler.handle(message);
    // then
    Message sentMessage = kernel.getSentMessages().get(0);
    Header header = sentMessage.getHeader();
    Assertions.assertThat(header).isNotNull();
    Assertions.assertThat(header.getType()).isEqualTo(KERNEL_INFO_REPLY.getName());
}

Example 17 with Header

use of com.twosigma.beakerx.message.Header in project beakerx by twosigma.

the class KotlinKernelInfoHandlerTest method handle_sentMessageHasHeaderTypeIsKernelInfoReply.

@Test
public void handle_sentMessageHasHeaderTypeIsKernelInfoReply() throws Exception {
    // when
    handler.handle(message);
    // then
    Message sentMessage = kernel.getSentMessages().get(0);
    Header header = sentMessage.getHeader();
    Assertions.assertThat(header).isNotNull();
    Assertions.assertThat(header.getType()).isEqualTo(KERNEL_INFO_REPLY.getName());
}

Example 18 with Header

use of com.twosigma.beakerx.message.Header in project beakerx by twosigma.

the class Comm method doOpen.

private void doOpen(Message parentMessage, Buffer buffer) {
    Message message = new Message();
    message.setHeader(new Header(COMM_OPEN, parentMessage != null ? parentMessage.getHeader().getSession() : null));
    if (parentMessage != null) {
        message.setParentHeader(parentMessage.getHeader());
    }
    HashMap<String, Serializable> map = new HashMap<>();
    map.put(COMM_ID, getCommId());
    map.put(TARGET_NAME, getTargetName());
    HashMap<String, Serializable> state = new HashMap<>();
    state.put(STATE, data.getData());
    state.put(METHOD, (Serializable) data.getData().get(METHOD));
    if (!buffer.isEmpty()) {
        state.put(BUFFER_PATHS, buffer.getBufferPaths());
        message.setBuffers(buffer.getBuffers());
    }
    map.put(DATA, state);
    map.put(METADATA, metadata);
    map.put(TARGET_MODULE, getTargetModule());
    message.setContent(map);
    message.setMetadata(buildMetadata());
    kernel.publish(singletonList(message));
    kernel.addComm(getCommId(), this);
}

Example 19 with Header

use of com.twosigma.beakerx.message.Header 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);
}

Example 20 with Header

use of com.twosigma.beakerx.message.Header in project beakerx by twosigma.

the class MessageFactoryTest method getExecuteRequestMessage.

public static Message getExecuteRequestMessage(String code) {
    Message message = new Message();
    Header header = new Header();
    header.setTypeEnum(JupyterMessages.EXECUTE_REQUEST);
    message.setHeader(header);
    HashMap<String, Serializable> content = new HashMap<>();
    content.put("code", code);
    message.setContent(content);
    return message;
}