Python 多线程技巧 用threading.Event代替time.sleep()

果我们碰到了这样的场景：线程1做一件时间较长的事情，等做完了，主程序去测试其状态是不是真的更改了。

通常比较容易想到的是：

1. 线程1里，用大循环来模拟这个长时间的函数，比如for i in range(1, 10000)等等；

2. 主程序里，用比如time.sleep(60)去等待，默许60秒应该已经够线程1去run了。

显然，这样做是不严谨的，因为没办法确切控制线程1的时间，所以测试程序可能会fail掉。

好的解决办法是用threading.Event的wait()和set()来精确控制线程。

见代码如下：

线程1：Work

主线程：跟踪测试is_done的状态。

view plain copy to clipboard print ?

1. #!/usr/bin/python  
2. # -*- coding: utf-8 -*-  
3. import threading  
4. def DoWork(work, callback):  
5.   def Task():  
6.     print 'work begin...'  
7.     work()  
8.     print 'work done!'  
9.     print 'callback begin...'  
10.     callback()  
11.     print 'callback done!'  
12.   t = threading.Thread(target=Task)  
13.   t.start()  
14. def TestWorkAndDone():  
15.   is_started = threading.Event()  
16.   can_done = threading.Event()  
17.   is_done = threading.Event()  
18.   def Work():  
19.     """it's not a good idea to use huge for loop here to kill time, 
20.        because the start and end time are not under the control. 
21.     """  
22.     print '1'  
23.     is_started.set()  
24.     print '2'  
25.     # block here until can_done.set() is called.  
26.     can_done.wait(timeout=60)  # .await() in Java  
27.     print '7'  
28.   DoWork(work=Work, callback=lambda:is_done.set())  
29.   print '3'  
30.   # block here until is_started.set() is called.  
31.   is_started.wait(timeout=60)  
32.   print '4'  
33.   if is_started.isSet():  
34.     print 'SUCCESS: Work started'  
35.   if is_done.isSet():  
36.     print 'FAILD: Work hasnot done'  
37.   print '5'  
38.   can_done.set()  # Work() is unblock.  
39.   print '6'  
40.   # block here until callback() is called.  
41.   # no need to use time.sleep(60) here.  
42.   is_done.wait(timeout=60)  
43.   if is_done.isSet():  
44.     print '<mce:script type="text/javascript" src="http://hi.images.csdn.net/js/blog/tiny_mce/themes/advanced/langs/zh.js" mce_src="http://hi.images.csdn.net/js/blog/tiny_mce/themes/advanced/langs/zh.js"></mce:script><mce:script type="text/javascript" src="http://hi.images.csdn.net/js/blog/tiny_mce/plugins/syntaxhl/langs/zh.js" mce_src="http://hi.images.csdn.net/js/blog/tiny_mce/plugins/syntaxhl/langs/zh.js"></mce:script>SUCCESS: Work done'  
45. # main  

1. TestWorkAndDone()  

https://www.cnblogs.com/zhengyuxin/articles/1956425.html