王海鱼 发表于 2024-11-10 23:40:08

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库即可轻松实现
import queue
import threading

import requests
from flask import Flask, request, jsonify

from log import log

# 初始化Flask应用
app = Flask(__name__, static_folder="dist")

# 捕获所有请求并转发
@app.route("/", defaults={"path": ""}, methods=["GET", "POST", "PUT", "DELETE"])
@app.route("/<path:path>", methods=["GET", "POST", "PUT", "DELETE"])
def catch_all(path):
    # 构建转发的目标 URL,我这里将请求都转发到10.8.0.126服务器上
    forward_url = f"http://10.8.0.126/{path}"
    # 转发请求,并传递请求头、数据以及文件(如果有)
    headers = {key: value for key, value in request.headers if key != "Host"}
    # 获取请求的数据,根据请求方法选择适当的数据处理方式
    data = request.get_json() if request.is_json else request.form or request.data

    json_data = data or {}
    try:
      response = requests.request(
            method=request.method,
            url=forward_url,
            headers=headers,
            json=json_data,
            files=request.files,
            params=request.args,
      )
    except requests.RequestException as e:
      return str(e), 500
    # 返回转发请求的响应
    return response.content, response.status_code, response.headers.items()

if __name__ == "__main__":
    app.run(debug=False, host="0.0.0.0", port=9095)

[*] 安装gunicorn库,部署转发服务程序(普通运行只供调试使用,不实用于生产情况)
pip install gunicorn

[*] 运行转发服务程序
gunicorn -b 0.0.0.0 -w 4 main:app
https://i-blog.csdnimg.cn/direct/a2a135c639084dbba09c1cb75672c7a3.png#pic_center
使用C++实现请求转发服务器

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

[*] 代码实现,这里捕捉所有请求使用了正则表达式,刚开始因为对这个库和正则不熟悉,卡了较长时间。
#include <iostream>
#include <httplib.h>
#include <string>

// 定义一个简单的路由处理器
void catch_all(const httplib::Request& req, httplib::Response& res)
{
    // 构建转发的目标 URL
    std::string path = req.path;
    std::cout << "path: " << path << std::endl;
    std::cout << "method: " << req.method << std::endl;
    std::string forward_url = "http://10.8.0.126/" + path;

    // 转发请求,并传递请求头、数据以及文件(如果有)
    httplib::Headers headers;
    headers = req.headers;
    headers.erase("Host");

    // 获取请求的数据,根据请求方法选择适当的数据处理方式
    std::string data;
    data = req.body;

    // 使用 httplib 客户端进行转发
    httplib::Client client("10.10.112.139", 9092);
    httplib::Result result;
    if (req.method == "GET") {
      result = client.Get(path, headers);
    } else if (req.method == "POST") {
      result = client.Post(path, headers, data, "application/json");
      std::cout << result->body << std::endl;
    } else if (req.method == "PUT") {
      result = client.Put(path, headers, data, "application/json");
    } else if (req.method == "DELETE") {
      result = client.Delete(path, headers);
    } else {
      res.set_content("Unsupported method", "text/plain");
      res.status = 405;
      return;
    }

    if (result && result->status == 200) {
      res.set_content(result->body, result->get_header_value("Content-Type"));
      res.status = result->status;
      for (const auto& header : result->headers) {
            res.headers.emplace(header.first, header.second);
      }
    } else {
      res.set_content("Forward request failed", "text/plain");
      res.status = 500;
    }
}

int main()
{
    httplib::Server svr;

    // 捕获所有请求并转发
    svr.Get(R"(/.*)", &catch_all);
    svr.Post(R"(/.*)", &catch_all);

    auto ret = svr.set_mount_point("/", "/home/narada/ems/www");

    svr.listen("0.0.0.0", 8080);
    return 0;
}


免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!更多信息从访问主页:qidao123.com:ToB企服之家,中国第一个企服评测及商务社交产业平台。
页: [1]
查看完整版本: Http请求转发服务器实现