Merge pull request #47 from halfcrazy/handle-timeout

handle timeout request, avoid heroku 503 error

Author: dongweiming

slack_bot/app.py

@@ -10,6 +10,7 @@
 import settings
 import plugins
 from ext import redis_store, cache
+from utils import timeout
 from plugins.utils import convert2str, replaced
 
 
@@ -40,6 +41,7 @@ def create_app(config=None):
     return app
 
 
+@timeout(30.0)
 def callback(kwargs, app):
     s = convert2str(kwargs['text'])
     trigger_word = convert2str(kwargs['trigger_word'])

slack_bot/utils.py

@@ -0,0 +1,19 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+from multiprocessing import TimeoutError
+from multiprocessing.pool import ThreadPool
+from functools import wraps
+
+
+def timeout(seconds):
+    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)
+            except TimeoutError:
+                return kwargs.pop('default', {'text': 'timeout'})
+        return wrapper
+    return decorator