搞点事情，使用node搭建反向代理

导语

最近有个需求，需要对业务管理后台的操作记录进行上报。一般这种上报需求都是又后台同学来做比较合适的。但是因为后台人力的原因。这个工作落到了我这个小前端的头上。这里记录下做这个需求踩的一些坑。

一、实现反向代理

做为一个前端工程师，写代理脚本第一选择肯定是node。不过在此之前，要把请求代理到机器A上面的node服务上面。这里使用了tnginx。在nginx.config文件里面添加以下配置并重启。把cgi域名下的请求，代理到机器上面的8000端口node 服务。

server{
                listen 80;
                server_name cgi.qqcomic.oa.com admin.cgi.qqcomic.oa.com;
                access_log /usr/local/services/tnginx_1_0_0-1.0/access.log;
                location / {
                      proxy_pass http://127.0.0.1:8000;
                      proxy_set_header        X-Real-IP $remote_addr;
                      proxy_set_header        X-Forwarded-For $proxy_add_x_forwarded_for;
                      proxy_set_header        Host $http_host;
                    }
            }

然后使用node的http-proxy模块，起一个代理server，就像这样。

    var httpProxy = require('http-proxy');
    var http = require('http')
    http.createServer(function(req, res) {
           proxy.web(req, res, { target: http://10.213.167.135});
    }).listen(8000);

本机使用fiddler 代理cgi域名到测试机器A的ip，刷新一下，成功访问到cgi内容。(ps: 注意idc机器是没有dns解析服务的，这里需要在 /etc/hosts文件上面添加相关域名的ip地址)

二、获取请求的相关数据

成功实现请求代理是一个好的开始。现在，需要开始搞点事情了。首先，我们需要获取请求的参数，这些参数可能是在url里面，也可能是在POST实体里面。url里面的参数很容易拿到，只需要读取req对象的url就能获取。POST实体里面的数据获取比较麻烦，因为POST请求的种类比较多，手动解析比较麻烦。这里使用了formidable模块来解析。然后把解析完成的结果挂载在req对象上面，方便后面获取。

  function getBody(request){
        var formidable = require('formidable');
        var form = new formidable.IncomingForm(),fields = {};
        //..巴拉巴拉，解析出参数，挂载在request对象上面
    }

除了请求的参数，我们还需要获取cgi回包的数据，这样才能判断这个请求是不是有效的。获取回包数据，可以在res对象上面监听data事件，拼接回包数据。类似这样封装一个方法:

    function getResRawData(res,callback){
        var resData = ''
        res.on('data', function(chunk){
            resData = resData + chunk.toString();
        });
        res.on('end', function(chunk){
            try {
                resData = JSON.parse(resData);
                callback(null,resData)
            }catch (e){
                callback(e)
            }    
        });
        res.on('error', function(err){
                callback(err)
        });
    }

比较坑一点的是，回包可能会被gzip压缩，这样我们上面代码得到的会是乱码。因此处理回包的时候，要判断回包的content-encoding是不是gzip，如果是gzip的话，需要使用node的zlib模块进行解压。

三、进行数据上报

获取了请求的参数和回包内容，我们就可以进行数据上报了，上报的时机应该是在代理请求回包之后。http-proxy模块提供了proxyRes事件给我们监听，我们可以在这个事件的回调函数里面，获取回包的内容，并调用上报方法，使用node的request模块进行数据上报。类似这样:

    var request = require('request');
    proxy.on('proxyRes', function (proxyRes,req,res) {
        //获取回包内容
        getResRawData(proxyRes,function(err,data){
            //发起请求上报
            request.post(data, function(err,httpResponse,body){ 
                console.log(body)
               })
        })
    });

四、划分路由模块

系统的cgi可能不只是一个命令，可能不只是一种回包格式。所以我们需要添加一个路由模块，把不同请求，映射到对应的处理器上面。可以比较简单的根据正则匹配url，返回不同的模块字符串，然后在代理请求回包后，根据模块字符串require这些模块去处理对应的请求。类似这样:

  http.createServer(function(req, res) {
        var route = router(req);//根据请求url，返回对应的模块字符串
        //根据请求，获取处理请求的模块
        var target = (route && route.target) || null;
        if(!target){
            res.end('bad request')
        }else{
           //注入配置到req对象里面，后面会用到
            req.routerConfig = route
            proxy.web(req, res, { target: target});
        }
    }).listen(8000);

    proxy.on('proxyRes', function (proxyRes,req,res) {
        //获取回包内容
        getResRawData(proxyRes,function(err,data){
             //根据路由配置，加载对应的处理器去处理请求
             if(req.routerConfig && req.routerConfig.handle){
                action = require(req.routerConfig.handle);
                action.handle(req,data)
            }
        })
    });

附：系统设计流程图

五、小结

有了node之后，前端有了更大的舞台，可以帮助解决一些后台的工作。这次的需求只是一个小小的应该例子，后续我们还可以在这个proxy server的基础上，添加白名单做权限限制，限制某些rtx用户只能操作固定的cgi。

此文已由作者授权腾讯云技术社区发布，转载请注明文章出处