Ejemplo n.º 1
0
 def testBasic(self):
   reap_cnt = [0]
   def callback(pid, status):
     val = reap_cnt.pop() + 1
     reap_cnt.append(val)
   pp  = PythonPrefork(on_reap_cb=callback)
   filename = 'test.txt'
   with file(filename, 'w') as fh:
     fh.write('0')
   
   while pp.signal_received is None:
     if pp.start(): continue
     with file(filename, 'r+') as fh:
       fcntl.flock(fh.fileno(), fcntl.LOCK_EX)
       num = int(fh.read()) + 1
       fh.seek(0, os.SEEK_SET)
       fh.write("%d" % (num))
       fcntl.flock(fh.fileno(), fcntl.LOCK_UN)
     if num == pp.max_workers:
       os.kill(pp.manager_pid, signal.SIGTERM)
     time.sleep(5)
     pp.finish()
   pp.wait_all_children()
   
   self.assertEqual(int(file(filename, 'r').read()), pp.max_workers)
   self.assertEqual(reap_cnt[0], pp.max_workers)
   os.remove(filename)
Ejemplo n.º 2
0
def main():
  pp = PythonPrefork()
  while not pp.signal_received:
    if pp.start(): continue
    run()
    pp.finish()
  pp.wait_all_children()