极客教程Flask 超时控制
在开发Web应用程序时，经常需要处理一些复杂耗时的任务，如请求外部API、计算密集型操作等。为了避免用户长时间等待或请求超时，我们可以在Flask应用程序中实现超时控制来确保应用的稳定性和性能。
背景知识
在Flask中，可以使用timeout装饰器或before_request钩子函数对请求进行超时控制。timeout装饰器可以设置一个时间限制，如果请求处理时间超过该限制，会自动返回超时错误。before_request钩子函数可以在请求处理之前设置一个超时处理逻辑，如手动中断请求或返回自定义错误信息。
使用 timeout 装饰器
下面是一个使用timeout装饰器实现超时控制的示例：
from flask import Flask
from flask import request
from flask import jsonify
import timeout_decorator
app = Flask(__name__)
@app.route('/api/calculate', methods=['POST'])
@timeout_decorator.timeout(5)
def calculate():
data = request.get_json()
num1 = data['num1']
num2 = data['num2']
result = num1 + num2
return jsonify({'result': result})
if __name__ == '__main__':
app.run()
在上面的示例中，我们定义了一个计算两个数相加的API接口/api/calculate，并使用timeout_decorator.timeout(5)设置了超时时间为5秒。如果API请求处理时间超过5秒，会自动返回超时错误。
使用 before_request 钩子函数
除了使用timeout装饰器，我们还可以使用before_request钩子函数实现超时控制。下面是一个使用before_request钩子函数实现超时控制的示例：
from flask import Flask
from flask import request
from flask import jsonify
import signal
app = Flask(__name__)
def handle_timeout(signum, frame):
return jsonify({'error': 'Request Timeout'}), 408
@app.before_request
def before_request():
signal.signal(signal.SIGALRM, handle_timeout)
signal.alarm(5)
@app.route('/api/calculate', methods=['POST'])
def calculate():
data = request.get_json()
num1 = data['num1']
num2 = data['num2']
result = num1 + num2
signal.alarm(0)
return jsonify({'result': result})
if __name__ == '__main__':
app.run()
在上面的示例中，我们使用before_request钩子函数设置了一个5秒的超时时间。当请求处理时间超过5秒时，会触发handle_timeout函数返回超时错误。
总结
通过上面的示例代码，我们详细介绍了在Flask应用程序中实现超时控制的两种方法：使用timeout装饰器和before_request钩子函数。这些方法可以帮助我们有效地避免请求超时或长时间等待，确保Web应用程序的稳定性和性能。在实际开发中，根据具体需求选择合适的超时控制方式，并适时调整超时时间，以提高用户体验和系统性能。