private static Message initMessage(JupyterMessages type, Message message) { Message reply = new Message(new Header(type, message.getHeader().getSession())); reply.setParentHeader(message.getHeader()); reply.setIdentities(message.getIdentities()); return reply; }
public String asJson() { return MessageSerializer.toJson(this); }
private void handleControlMsg() { Message message = readMessage(controlSocket); JupyterMessages type = message.getHeader().getTypeEnum(); if (type.equals(SHUTDOWN_REQUEST)) { Message reply = new Message(new Header(SHUTDOWN_REPLY, message.getHeader().getSession())); reply.setParentHeader(message.getHeader()); reply.setContent(message.getContent()); sendMsg(controlSocket, Collections.singletonList(reply)); shutdown(); } }
private static Message copyMessage(Message origin) { String header = toJson(origin.getHeader()); Message copy = new Message(parse(header, Header.class)); for (byte[] list : origin.getIdentities()) { copy.getIdentities().add(list.clone()); } String parent = toJson(origin.getParentHeader()); String metadata = toJson(origin.getMetadata()); String content = toJson(origin.getContent()); copy.setParentHeader(parse(parent, Header.class)); copy.setMetadata(parse(metadata, LinkedHashMap.class)); copy.setContent(parse(content, LinkedHashMap.class)); return copy; }
protected Message createReplyMessage(Message message, Serializable responceData) { Message ret = null; if (message != null) { ret = new Message(new Header(COMM_MSG, message.getHeader().getSession())); Map<String, Serializable> commMap = message.getContent(); HashMap<String, Serializable> map = new HashMap<>(); map.put(COMM_ID, getString(commMap, COMM_ID)); map.put(DATA, responceData); ret.setContent(map); } return ret; }
private TryResult addImplicits(Message parent, String codeToExecute) { SimpleEvaluationObject seo = createSimpleEvaluationObject( codeToExecute, kernel, new Message(new Header(JupyterMessages.COMM_MSG, parent.getHeader().getSession())), 1); return kernel.executeCode(codeToExecute, seo); }
public static Message getExecuteRequestMessage(String code) { Header header = new Header(JupyterMessages.COMM_MSG,"session1"); Message message = new Message(header); header.setTypeEnum(JupyterMessages.EXECUTE_REQUEST); HashMap<String, Serializable> content = new HashMap<>(); content.put("code", code); message.setContent(content); return message; }
private void publishMessage(KernelFunctionality kernel, Message message, int executionCount) { for (Message msg : responses) { msg.setParentHeader(message.getHeader()); } kernel.publish(responses); }
private String getMessageTarget(Message message){ Map content = message.getContent(); return content == null ? null : (String) message.getContent().get(TARGET_NAME); } }
public static Header initHeader(JupyterMessages jupyterMessages) { Header header = new Header(jupyterMessages, "sessionId" + jupyterMessages.getName()); header.setId("messageId"); header.setUsername("username"); header.setVersion("5.0"); return header; }
@Override public String toString() { return "Type = " + header.getType(); }
@Before public void setUp() throws Exception { message = MessageTest.createMessage(); }
public static boolean isErrorMessage(Message message) { return message.type().equals(JupyterMessages.ERROR); } }
public JupyterMessages type() { return (header != null && header.getTypeEnum() != null) ? header.getTypeEnum() : null; }
private void handleMsg(Message message) { logger.debug("Processing kernel info request"); Message reply = new Message(new Header(KERNEL_INFO_REPLY, message.getHeader().getSession())); reply.setContent(content()); reply.setParentHeader(message.getHeader()); reply.setIdentities(message.getIdentities()); send(reply); }
private TryResult initSparkContextInShell(KernelFunctionality kernel, Message parent) { String addSc = String.format(("import com.twosigma.beakerx.widget.SparkVariable\n" + "val %s = SparkVariable.getSparkSession()\n" + "val %s = %s.sparkContext\n" + "import org.apache.spark.SparkContext._\n" + "import %s.implicits._\n" + "import %s.sql\n" + "import org.apache.spark.sql.functions._\n"), SPARK_SESSION_NAME, SPARK_CONTEXT_NAME, SPARK_SESSION_NAME, SPARK_SESSION_NAME, SPARK_SESSION_NAME); SimpleEvaluationObject seo = createSimpleEvaluationObject(addSc, kernel, new Message(new Header(JupyterMessages.COMM_MSG, parent.getHeader().getSession())), 1); return kernel.executeCode(addSc, seo); }
public static Map getContent(Message message) { return message.getContent(); }