首页 > 代码库 > 搞点事情,使用node搭建反向代理

搞点事情,使用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 = http://www.mamicode.com/‘‘
        res.on(‘data‘, function(chunk){
            resData = http://www.mamicode.com/resData + chunk.toString();"hljs-string">‘end‘, function(chunk){
            try {
                resData = http://www.mamicode.com/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。

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

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