搞点事情,使用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。
此文已由作者授权腾讯云技术社区发布,转载请注明文章出处