implement accurate timeout function

Author: halfcrazy

slack_bot/app.py

@@ -1,8 +1,10 @@
 # coding=utf-8
+import time
 import os
 import re
+from multiprocessing.pool import ThreadPool
 
-from flask import Flask
+from flask import Flask, g
 
 from flask_slackbot import SlackBot
 
@@ -37,10 +39,23 @@ def create_app(config=None):
     slackbot.set_handler(callback)
     slackbot.filter_outgoing(_filter)
 
+    def _set_time():
+        while True:
+            g.time = time.time()
+
+    @app.before_request
+    def start_time_it():
+        g.pool = ThreadPool(processes=1)
+        g.pool.apply_async(_set_time)
+
+    @app.after_request
+    def end_time_it():
+        g.pool.close()
+
     return app
 
 
-@timeout(30.0)
+@timeout(g, 30.0)
 def callback(kwargs):
     s = convert2str(kwargs['text'])
     trigger_word = convert2str(kwargs['trigger_word'])

slack_bot/plugins/github_issue.py

@@ -37,7 +37,8 @@ def handle(data):
 
 
 if __name__ == '__main__':
-    from flask import Flask
-    app = Flask(__name__)
-    app.config['org_name'] = 'python-cn'
-    print handle(None)
+    # from flask import Flask
+    # app = Flask(__name__)
+    # app.config['org_name'] = 'python-cn'
+    # print handle(None)
+    pass

slack_bot/utils.py

@@ -1,19 +1,24 @@
 #!/usr/bin/env python
 # -*- coding: utf-8 -*-
+import time
 from multiprocessing import TimeoutError
 from multiprocessing.pool import ThreadPool
 from functools import wraps
 
 
-def timeout(seconds):
+def timeout(g, seconds, default="timeout"):
     def decorator(fn):
         @wraps(fn)
         def wrapper(*args, **kwargs):
             pool = ThreadPool(processes=1)
             async_result = pool.apply_async(fn, args=args, kwds=kwargs)
             try:
-                return async_result.get(seconds)
+                # the time cost before start fn
+                cost_time = time.time() - g.time
+                return async_result.get(seconds - cost_time)
             except TimeoutError:
-                return kwargs.pop('default', {'text': 'timeout'})
+                return default() if callable(default) else default
+            finally:
+                pool.close()
         return wrapper
     return decorator