Merge pull request Yelp#12 from jnb/pre-fork-init

Invoke pre-fork initialization function.

zygote/worker.py

@@ -68,7 +68,7 @@ def __init__(self, sock, basepath, module, args):
 
         os.chdir(basepath)
         sys.path.insert(0, basepath)
-        t = __import__(module, [], [], ['get_application'], 0)
+        t = __import__(module, [], [], ['initialize', 'get_application'], 0)
 
         self.sock = sock
 
@@ -86,6 +86,11 @@ def __init__(self, sock, basepath, module, args):
 
         signal.signal(signal.SIGCHLD, self.reap_child)
 
+        # If there is an initialize function defined then call it.
+        if hasattr(t, 'initialize'):
+            self.log.info('initializing zygote')
+            t.initialize(*self.args)
+
         self.log.info('new zygote started')
 
     def handle_control(self, fd, events):