Merge pull request #317 from dleen/master

Fix handling of ZMQ messages received before initialization is complete.

Author: rolweber

kernel_gateway/services/kernels/pool.py

@@ -6,9 +6,10 @@
 
 from tornado.locks import Semaphore
 from tornado import gen
+from traitlets.config.configurable import LoggingConfigurable
 
 
-class KernelPool(object):
+class KernelPool(LoggingConfigurable):
     """Spawns a pool of kernels.
 
     The kernel pool is responsible for clean up and shutdown of individual
@@ -119,6 +120,10 @@ def _on_reply(self, kernel_id, session, msg_list):
         msg_list : list
             List of 0mq messages
         """
+        if not kernel_id in self.on_recv_funcs:
+            self.log.warning(
+                "Could not find callback for kernel_id: {}".format(kernel_id))
+            return
         idents, msg_list = session.feed_identities(msg_list)
         msg = session.deserialize(msg_list)
         self.on_recv_funcs[kernel_id](msg)