##// END OF EJS Templates
Fixed high CPU usage of XREQ channel....
Fixed high CPU usage of XREQ channel. Previously, the XReqSocketChannel was consuming nearly 100% CPU. This was caused by the ioloop polling for POLLIN|POLLOUT|POLLERR always. Now we are only polling for POLLOUT when there is a message waiting to be sent. This seems to have fixed the problem.

File last commit:

r2695:c04e6487
r2695:c04e6487
Show More
kernelmanager.py
147 lines | 5.0 KiB | text/x-python | PythonLexer
""" Defines a KernelManager that provides signals and slots.
"""
# System library imports.
from PyQt4 import QtCore
import zmq
# IPython imports.
from IPython.zmq.kernelmanager import KernelManager, SubSocketChannel, \
XReqSocketChannel, RepSocketChannel
from util import MetaQObjectHasTraits
class QtSubSocketChannel(SubSocketChannel, QtCore.QObject):
# Emitted when any message is received.
message_received = QtCore.pyqtSignal(object)
# Emitted when a message of type 'pyout' or 'stdout' is received.
output_received = QtCore.pyqtSignal(object)
# Emitted when a message of type 'pyerr' or 'stderr' is received.
error_received = QtCore.pyqtSignal(object)
#---------------------------------------------------------------------------
# 'object' interface
#---------------------------------------------------------------------------
def __init__(self, *args, **kw):
""" Reimplemented to ensure that QtCore.QObject is initialized first.
"""
QtCore.QObject.__init__(self)
SubSocketChannel.__init__(self, *args, **kw)
#---------------------------------------------------------------------------
# 'SubSocketChannel' interface
#---------------------------------------------------------------------------
def call_handlers(self, msg):
""" Reimplemented to emit signals instead of making callbacks.
"""
# Emit the generic signal.
self.message_received.emit(msg)
# Emit signals for specialized message types.
msg_type = msg['msg_type']
if msg_type in ('pyout', 'stdout'):
self.output_received.emit(msg)
elif msg_type in ('pyerr', 'stderr'):
self.error_received.emit(msg)
def flush(self):
""" Reimplemented to ensure that signals are dispatched immediately.
"""
super(QtSubSocketChannel, self).flush()
QtCore.QCoreApplication.instance().processEvents()
class QtXReqSocketChannel(XReqSocketChannel, QtCore.QObject):
# Emitted when any message is received.
message_received = QtCore.pyqtSignal(object)
# Emitted when a reply has been received for the corresponding request type.
execute_reply = QtCore.pyqtSignal(object)
complete_reply = QtCore.pyqtSignal(object)
object_info_reply = QtCore.pyqtSignal(object)
#---------------------------------------------------------------------------
# 'object' interface
#---------------------------------------------------------------------------
def __init__(self, *args, **kw):
""" Reimplemented to ensure that QtCore.QObject is initialized first.
"""
QtCore.QObject.__init__(self)
XReqSocketChannel.__init__(self, *args, **kw)
#---------------------------------------------------------------------------
# 'XReqSocketChannel' interface
#---------------------------------------------------------------------------
def call_handlers(self, msg):
""" Reimplemented to emit signals instead of making callbacks.
"""
# Emit the generic signal.
self.message_received.emit(msg)
# Emit signals for specialized message types.
msg_type = msg['msg_type']
signal = getattr(self, msg_type, None)
if signal:
signal.emit(msg)
def _queue_request(self, msg, callback):
""" Reimplemented to skip callback handling.
"""
self.command_queue.put(msg)
self.add_io_state(zmq.POLLOUT)
class QtRepSocketChannel(RepSocketChannel, QtCore.QObject):
#---------------------------------------------------------------------------
# 'object' interface
#---------------------------------------------------------------------------
def __init__(self, *args, **kw):
""" Reimplemented to ensure that QtCore.QObject is initialized first.
"""
QtCore.QObject.__init__(self)
RepSocketChannel.__init__(self, *args, **kw)
class QtKernelManager(KernelManager, QtCore.QObject):
""" A KernelManager that provides signals and slots.
"""
__metaclass__ = MetaQObjectHasTraits
# Emitted when the kernel manager has started listening.
started_listening = QtCore.pyqtSignal()
# Emitted when the kernel manager has stopped listening.
stopped_listening = QtCore.pyqtSignal()
# Use Qt-specific channel classes that emit signals.
sub_channel_class = QtSubSocketChannel
xreq_channel_class = QtXReqSocketChannel
rep_channel_class = QtRepSocketChannel
#---------------------------------------------------------------------------
# 'KernelManager' interface
#---------------------------------------------------------------------------
def start_listening(self):
""" Reimplemented to emit signal.
"""
super(QtKernelManager, self).start_listening()
self.started_listening.emit()
def stop_listening(self):
""" Reimplemented to emit signal.
"""
super(QtKernelManager, self).stop_listening()
self.stopped_listening.emit()