self._dispatch_to_kernel(msg)
return msg['header']['msg_id']
def input(self, string):
if self.kernel is None:
raise RuntimeError('Cannot send input reply. No kernel exists.')
self.kernel.raw_input_str = string
def _dispatch_to_kernel(self, msg):
""" Send a message to the kernel and handle a reply.
"""
kernel = self.kernel
if kernel is None:
raise RuntimeError('Cannot send request. No kernel exists.')
stream = DummySocket()
self.session.send(stream, msg)
msg_parts = stream.recv_multipart()
kernel.dispatch_shell(stream, msg_parts)
idents, reply_msg = self.session.recv(stream, copy=False)
self.shell_channel.call_handlers_later(reply_msg)
#-----------------------------------------------------------------------------
# ABC Registration
#-----------------------------------------------------------------------------
KernelClientABC.register(InProcessKernelClient)
if self.kernel is None:
raise RuntimeError('Cannot send input reply. No kernel exists.')
self.kernel.raw_input_str = string
def is_complete(self, code):
msg = self.session.msg('is_complete_request', {'code': code})
self._dispatch_to_kernel(msg)
return msg['header']['msg_id']
def _dispatch_to_kernel(self, msg):
""" Send a message to the kernel and handle a reply.
"""
kernel = self.kernel
if kernel is None:
raise RuntimeError('Cannot send request. No kernel exists.')
stream = DummySocket()
self.session.send(stream, msg)
msg_parts = stream.recv_multipart()
kernel.dispatch_shell(stream, msg_parts)
idents, reply_msg = self.session.recv(stream, copy=False)
self.shell_channel.call_handlers_later(reply_msg)
#-----------------------------------------------------------------------------
# ABC Registration
#-----------------------------------------------------------------------------
KernelClientABC.register(InProcessKernelClient)
def _route_responses(self):
"""
Reads responses from the websocket and routes each to the appropriate channel queue based
on the message's channel. It does this for the duration of the class's lifetime until the
channels are stopped, at which time the socket is closed (unblocking the router) and
the thread terminates. If shutdown happens to occur while processing a response (unlikely),
termination takes place via the loop control boolean.
"""
try:
while not self._channels_stopped:
raw_message = self.channel_socket.recv()
if not raw_message:
break
response_message = json_decode(utf8(raw_message))
channel = response_message['channel']
self._channel_queues[channel].put_nowait(response_message)
except websocket.WebSocketConnectionClosedException:
pass # websocket closure most likely due to shutdown
except BaseException as be:
if not self._channels_stopped:
self.log.warning(
'Unexpected exception encountered ({})'.format(be))
self.log.debug('Response router thread exiting...')
KernelClientABC.register(HTTPKernelClient)
def _route_responses(self):
"""
Reads responses from the websocket and routes each to the appropriate channel queue based
on the message's channel. It does this for the duration of the class's lifetime until the
channels are stopped, at which time the socket is closed (unblocking the router) and
the thread terminates. If shutdown happens to occur while processing a response (unlikely),
termination takes place via the loop control boolean.
"""
try:
while not self._channels_stopped:
assert self.channel_socket is not None
raw_message = self.channel_socket.recv()
if not raw_message:
break
response_message = json_decode(utf8(raw_message))
channel = response_message["channel"]
assert self._channel_queues is not None
self._channel_queues[channel].put_nowait(response_message)
except websocket.WebSocketConnectionClosedException:
pass # websocket closure most likely due to shutdown
except BaseException as be:
if not self._channels_stopped:
self.log.warning(f"Unexpected exception encountered ({be})")
self.log.debug("Response router thread exiting...")
KernelClientABC.register(GatewayKernelClient)