node和express搭建代理服务器(源码)

48次阅读

共计 3011 个字符,预计需要花费 8 分钟才能阅读完成。

实例 git 地址:https://github.com/xubaodian/… 本例用 node 和 express 搭建的代理服务器。,期望目标如下:1、开启某服务 A,该服务可实现若干功能, 例如普通的 restful 请求,文件上传,静态资源访问等等。2、开启 node 代理服务 B,指向服务 A,访问代理服务 B,可访问服务 A 的任意功能。就如下图所示:图中上半部分是直接访问服务,下班部分是通过代理服务器访问服务。使用代理服务器时,浏览器向代理服务器请求数据,代理服务器转发请求,并将接收到的数据返回给浏览器,即所有的数据都通过代理服务器转发。带着这个目标,我们就讲述下如何实现该功能。既然是请求和响应转发,那我们就了解一下,什么是请求。
请求和响应简述
http 请求和响应主要右报文头部、空行和报文主体三个部分组成。空行我们不用关心,其实对我们来说,只要完成报文头部和报文主体的转发,就可以说实现了代理功能。请求和响应通过代理的整个过程如下:1、代理服务器接收请求后,在将目标服务数据返回给浏览器前要保持请求。
2、提取请求路径、请求头、请求主体等数据。
3、以 2 中提取的数据为参数,向目标服务器发送请求。
4、接收目标服务器返回数据,提取响应头,响应主体等数据。
5、将 4 中的提取出来的数据返回给客户端(浏览器)。
6、断开连接。经过这几个步骤,就实现了代理。
代码实现
下面直接上代码,然后做一些讲解。代理函数如下:
const http = require(‘http’);
const querystring = require(‘querystring’);

// 获取请求的 cookie 和 query 等
let getHeader = (reqClient) => {
let headers = reqClient.headers;
headers.path = reqClient.path;
headers.query = reqClient.query;
headers.cookie = reqClient.get(‘cookie’) || ”;

return headers;
}

// 代理函数,options 是代理设置,包括目标服务器 ip,port 等
let proxy = (options) => {
let reqOptions = {
hostname: options.host,
port: options.port
}
// 返回请求处理函数,reqClient 浏览器的请求,resClient 是响应浏览器的对象
return function (reqClient, resClient) {
// 设置目标服务器的请求参数,头中的各项参数
let headers = getHeader(reqClient);
reqOptions.headers = reqClient.headers;
let query = [];
if (headers.query) {
Object.keys(headers.query).map(key => {
query.push(key + ‘=’ + headers.query[key]);
});
reqOptions.path = headers.path + (query.length === 0 ? ” : (‘?’ + query.join(‘&’)));

}
reqOptions.cookie = headers.cookie;
reqOptions.method = reqClient.method;
// 向目标服务器发送请求,reqProxy 是向目标服务器的请求,resProxy 是目标服务器的响应。
let reqProxy = http.request(reqOptions, (resProxy) => {
resProxy.setEncoding(‘utf8’);
// 设置返回 http 头
resClient.set(resProxy.headers);
resClient.status(resProxy.statusCode);
// 接收从目标服务器返回的数据
resProxy.on(‘data’, (chunk) => {
// 接收目标服务器数据后,以流的方式向浏览器返回数据
resClient.write(chunk);
});

// 接收目标服务器数据结束
resProxy.on(‘end’, () => {
// 向浏览器写数据结束。
resClient.end();
});
// 目标服务器响应错误
resProxy.on(‘error’, () => {
// 响应错误,结束向浏览器返回数据
resClient.end();
});
});

// 接收浏览器数据
reqClient.on(‘data’, (chunk) => {
// 以流的方式向目标服务器发送数据
reqProxy.write(chunk);
});
// 接收数据结束
reqClient.on(‘end’, () => {
// 向目标服务器写数据结束
reqProxy.end();
});

// 普通 JSON 数据代理
if (Object.keys(reqClient.body).length) {
reqProxy.write(querystring.stringify(reqClient.body));
reqProxy.end();
}
}
}

module.exports = proxy;
上面就是 node 代理的核心代码。支持普通的请求,静态资源代理,文件上传下载代理等功能。
git 地址:https://github.com/xubaodian/…
demo 中,核心代码在 common/proxy.js 里,我还实现了两个测试服务。
在 server 文件下的 app.js 和 app2.js 是两个服务的入口文件。
app2.js 是目标服务器,有三个测试页面 1、http://localhost:20000/json.html post 请求测试,对应 ’/json’ 接口,可发送数据,f12 查看请求是否成功 2、http://localhost:20000/upload.html 文件上传测试,对应接口 ’/upload’ 接口,上传文件,f12 查看请求是否成功,同时在服务器 upload 文件夹下会有文件。3、http://localhost:20000/get.html get 请求测试,对应接口 ’/get’,同样 f12 查看 app2 为目标服务器,有 3 个接口。1、’/upload’ 接口,测试文件上传功能,上传文件将放在 uploads 文件夹下,上传的文件,文件名是一个 uuid,没有后缀,添加后缀即可查看文件是完整。测试过,传 1G 的文件没问题,再大的文件没试过,有需要的可以试下 2、’/json’,测试 POST 请求。3、’/get’,测试 GET 请求。
app.js 为代理服务为器,监听端口为 18000, 将所有请求转发至 app2,即所有 app2 的接口静态资源,app 中访问时一致的。测试步骤:1、可开启目标服务器,通过三个页面测试功能。2、开启代理服务器,访问下面三个页面:http://localhost:18000/json.html http://localhost:18000/upload.htmlhttp://localhost:18000/get.html 测试同样的功能。若和步骤 1 实现同样功能,则代理服务功能已经实现了。经过测试,代理功能是没问题的。如果问题欢迎留言,或发送邮件至 472784995@qq.com。
至于性能,我没测过,因为我自己这边的应用场景,访问量都不大,可以使用。

正文完
 0