Flask 接口转发网页打开超时|极客笔记

Flask 接口转发网页打开超时

在开发一个基于 Flask 框架的 Web 服务时，经常会涉及到接口转发的需求，即在前端页面上发起请求到 Flask 后端服务器，再由 Flask 后端服务器转发请求到其他服务端接口的情况。然而，当网络环境不稳定或者目标接口响应速度较慢时，就会出现网页打开超时的问题。本文将详细讨论如何在 Flask 中进行接口转发，并处理网页打开超时的情况。

Flask 接口转发

在 Flask 中实现接口转发非常简单，只需要使用 requests 库即可。下面是一个简单的示例，演示了如何在 Flask 中实现接口转发：

from flask import Flask, request, jsonify
import requests

app = Flask(__name__)

@app.route('/forward', methods=['POST'])
def forward():
    data = request.json
    url = data['url']
    headers = data.get('headers', {})
    params = data.get('params', {})

    response = requests.get(url, headers=headers, params=params)

    return jsonify(response.json())

if __name__ == '__main__':
    app.run(port=5000)

在上面的代码中，我们定义了一个 /forward 路由，接收 POST 请求，并从请求的 json 数据中获取目标接口的 URL、headers 和 params，并使用 requests 库向目标接口发起请求，将响应数据返回给前端。

网页打开超时处理

当网络环境较差或者目标接口响应速度较慢时，会导致网页打开超时的问题。为了避免这种情况的发生，我们可以在 Flask 代码中设置超时时间，当接口转发请求超过指定的时间仍未返回响应时，主动断开连接，避免出现网页打开超时的问题。

下面是一个修改后的示例，演示了如何在 Flask 中设置接口转发请求的超时时间：

from flask import Flask, request, jsonify
import requests

app = Flask(__name__)

@app.route('/forward', methods=['POST'])
def forward():
    data = request.json
    url = data['url']
    headers = data.get('headers', {})
    params = data.get('params', {})

    try:
        response = requests.get(url, headers=headers, params=params, timeout=5)  # 设置超时时间为 5 秒
        return jsonify(response.json())
    except requests.exceptions.Timeout:
        return jsonify({'error': 'Request timeout'})

if __name__ == '__main__':
    app.run(port=5000)

在上面的代码中，我们通过 timeout=5 参数将接口转发请求的超时时间设置为 5 秒，当请求超过这个时间仍未返回响应时，会捕获 requests.exceptions.Timeout 异常，返回一个包含错误信息的 JSON 响应。