upstream/ipython Files · IPython/kernel/zmq/log.py

Shut down kernels in parallel...

Shut down kernels in parallel When stopping the notebook server, it currently sends a shutdown request to each kernel and then waits for the process to finish. This can be slow if you have several kernels running. This makes it issues all the shutdown requests before waiting on the processes, so shutdown happens in parallel. KernelManager (and MultiKernelManager) gain three new public API methods to allow this: * request_shutdown (promoted from a private method) * wait_shutdown (refactored out of shutdown_kernel) * cleanup (refactored out of shutdown_kernel)

Thomas Kluyver - - Load All Authors

File last commit:

r11130:de5468b5


                r16510:633371e5

Download file

             log.py
        
                    21 lines
            
             | 661 B
            
                | text/x-python
            
             |
                PythonLexer
            
             / IPython / kernel / zmq / log.py
          
                    History
                
                 |
                  Annotation
                 | Raw
                 |Copy content
                 |Copy permalink

      from logging import INFO, DEBUG, WARN, ERROR, FATAL

      from zmq.log.handlers import PUBHandler

      class EnginePUBHandler(PUBHandler):

          """A simple PUBHandler subclass that sets root_topic"""

          engine=None

          def __init__(self, engine, *args, **kwargs):

              PUBHandler.__init__(self,*args, **kwargs)

              self.engine = engine

          @property

          def root_topic(self):

              """this is a property, in case the handler is created

              before the engine gets registered with an id"""

              if isinstance(getattr(self.engine, 'id', None), int):

                  return "engine.%i"%self.engine.id

              else:

                  return "engine"

	Site-wide shortcuts
/	Use quick search box
g h	Goto home page
g g	Goto my private gists page
g G	Goto my public gists page
g 0-9	Goto bookmarked items from 0-9
n r	New repository page
n g	New gist page

	Repositories
g s	Goto summary page
g c	Goto changelog page
g f	Goto files page
g F	Goto files page with file search activated
g p	Goto pull requests page
g o	Goto repository settings
g O	Goto repository access permissions settings
t s	Toggle sidebar on some pages

				from logging import INFO, DEBUG, WARN, ERROR, FATAL

				from zmq.log.handlers import PUBHandler

				class EnginePUBHandler(PUBHandler):
				"""A simple PUBHandler subclass that sets root_topic"""
				engine=None

				def __init__(self, engine, args, *kwargs):
				PUBHandler.__init__(self,args, *kwargs)
				self.engine = engine

				@property
				def root_topic(self):
				"""this is a property, in case the handler is created
				before the engine gets registered with an id"""
				if isinstance(getattr(self.engine, 'id', None), int):
				return "engine.%i"%self.engine.id
				else:
				return "engine"