Http请求转发服务器实现

Http请求转发服务器实现

需求场景

云服务器通过VPN连接了现场的n台工控机，每台工控机上都在跑web程序，如今我想通过公网直接访问工控机上的web服务，给客户检察现场的具体运行情况，而不是让客户再装一个VPN，简化操作。
如今我们有以下几种方案可以实现：
1.使用代理，将每台工控机的web服务端口代理到云服务器上的一个空闲端口，这种方式的优点是不需要额外写代码实现，只需要通过配置代理即可，缺点是云服务器上端口占用过多，每打开现场的网址url上面都会带一个端标语。
2.使用Http请求转发服务器，将现场和他对应的VPN的ip地址映射起来存到数据库里，前端先请求服务器当地IP地址(127.0.0.1)，当通过主界面跳转到某个现场时，通过数据库查询现场VPN的IP地址，然后将之后的请求都转发到对应的VPN地址，即可获取对应现场的数据
具体实现

使用Python实现请求转发服务器

• 使用Python实现请求转发服务器非常简单，用flask库即可轻松实现
  
  1. import queue
  2. import threading
  4. import requests
  5. from flask import Flask, request, jsonify
  7. from log import log
  9. # 初始化Flask应用
  10. app = Flask(__name__, static_folder="dist")
  12. # 捕获所有请求并转发
  13. @app.route("/", defaults={"path": ""}, methods=["GET", "POST", "PUT", "DELETE"])
  14. @app.route("/<path:path>", methods=["GET", "POST", "PUT", "DELETE"])
  15. def catch_all(path):
  16.     # 构建转发的目标 URL,我这里将请求都转发到10.8.0.126服务器上
  17.     forward_url = f"http://10.8.0.126/{path}"
  18.     # 转发请求，并传递请求头、数据以及文件（如果有）
  19.     headers = {key: value for key, value in request.headers if key != "Host"}
  20.     # 获取请求的数据，根据请求方法选择适当的数据处理方式
  21.     data = request.get_json() if request.is_json else request.form or request.data
  23.     json_data = data or {}
  24.     try:
  25.         response = requests.request(
  26.             method=request.method,
  27.             url=forward_url,
  28.             headers=headers,
  29.             json=json_data,
  30.             files=request.files,
  31.             params=request.args,
  32.         )
  33.     except requests.RequestException as e:
  34.         return str(e), 500
  35.     # 返回转发请求的响应
  36.     return response.content, response.status_code, response.headers.items()
  38. if __name__ == "__main__":
  39.     app.run(debug=False, host="0.0.0.0", port=9095)

  复制代码
• 安装gunicorn库，部署转发服务程序(普通运行只供调试使用，不实用于生产情况)
  
  1. pip install gunicorn

  复制代码
• 运行转发服务程序
  
  1. gunicorn -b 0.0.0.0 -w 4 main:app

  复制代码
  
  
  

使用C++实现请求转发服务器

在用Python完成请求转发服务器的开辟后，方案却被领导否决了，因为后台程序都是用C++写的，用Python写不好融合，然后我继续开始研究C++服务端的方案，先采用cpp-httplib库实现。

• 代码实现，这里捕捉所有请求使用了正则表达式，刚开始因为对这个库和正则不熟悉，卡了较长时间。
  
  1. #include <iostream>
  2. #include <httplib.h>
  3. #include <string>
  5. // 定义一个简单的路由处理器
  6. void catch_all(const httplib::Request& req, httplib::Response& res)
  7. {
  8.     // 构建转发的目标 URL
  9.     std::string path = req.path;
  10.     std::cout << "path: " << path << std::endl;
  11.     std::cout << "method: " << req.method << std::endl;
  12.     std::string forward_url = "http://10.8.0.126/" + path;
  14.     // 转发请求，并传递请求头、数据以及文件（如果有）
  15.     httplib::Headers headers;
  16.     headers = req.headers;
  17.     headers.erase("Host");
  19.     // 获取请求的数据，根据请求方法选择适当的数据处理方式
  20.     std::string data;
  21.     data = req.body;
  23.     // 使用 httplib 客户端进行转发
  24.     httplib::Client client("10.10.112.139", 9092);
  25.     httplib::Result result;
  26.     if (req.method == "GET") {
  27.         result = client.Get(path, headers);
  28.     } else if (req.method == "POST") {
  29.         result = client.Post(path, headers, data, "application/json");
  30.         std::cout << result->body << std::endl;
  31.     } else if (req.method == "PUT") {
  32.         result = client.Put(path, headers, data, "application/json");
  33.     } else if (req.method == "DELETE") {
  34.         result = client.Delete(path, headers);
  35.     } else {
  36.         res.set_content("Unsupported method", "text/plain");
  37.         res.status = 405;
  38.         return;
  39.     }
  41.     if (result && result->status == 200) {
  42.         res.set_content(result->body, result->get_header_value("Content-Type"));
  43.         res.status = result->status;
  44.         for (const auto& header : result->headers) {
  45.             res.headers.emplace(header.first, header.second);
  46.         }
  47.     } else {
  48.         res.set_content("Forward request failed", "text/plain");
  49.         res.status = 500;
  50.     }
  51. }
  53. int main()
  54. {
  55.     httplib::Server svr;
  57.     // 捕获所有请求并转发
  58.     svr.Get(R"(/.*)", &catch_all);
  59.     svr.Post(R"(/.*)", &catch_all);
  61.     auto ret = svr.set_mount_point("/", "/home/narada/ems/www");
  63.     svr.listen("0.0.0.0", 8080);
  64.     return 0;
  65. }

  复制代码

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。