node和express搭建代理服務器（源碼）

實例git 地址：https://github.com/xubaodian/...

本例用node和express搭建的代理服務器。，指望目標以下：

一、開啓某服務A，該服務可實現若干功能,例如普通的restful請求，文件上傳，靜態資源訪問等等。

二、開啓node代理服務B，指向服務A，訪問代理服務B，可訪問服務A的任意功能。

就以下圖所示：



圖中上半部分是直接訪問服務，下班部分是經過代理服務器訪問服務。

使用代理服務器時，瀏覽器向代理服務器請求數據，代理服務器轉發請求，並將接收到的數據返回給瀏覽器，即全部的數據都經過代理服務器轉發。
帶着這個目標，咱們就講述下如何實現該功能。

既然是請求和響應轉發，那咱們就瞭解一下，什麼是請求。

請求和響應簡述

http請求和響應主要右報文頭部、空行和報文主體三個部分組成。

空行咱們不用關心，其實對咱們來講，只要完成報文頭部和報文主體的轉發，就能夠說實現了代理功能。

請求和響應經過代理的整個過程以下：

一、代理服務器接收請求後，在將目標服務數據返回給瀏覽器前要保持請求。

二、提取請求路徑、請求頭、請求主體等數據。

三、以2中提取的數據爲參數，向目標服務器發送請求。

四、接收目標服務器返回數據，提取響應頭，響應主體等數據。

五、將4中的提取出來的數據返回給客戶端（瀏覽器）。

六、斷開鏈接。

通過這幾個步驟，就實現了代理。

代碼實現

下面直接上代碼，而後作一些講解。代理函數以下：

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是目標服務器，有三個測試頁面

一、http://localhost:20000/json.html post請求測試，對應'/json'接口，可發送數據，f12查看請求是否成功

二、http://localhost:20000/upload.html 文件上傳測試，對應接口'/upload'接口，上傳文件，f12查看請求是否成功，同時在服務器upload文件夾下會有文件。

三、http://localhost:20000/get.html get請求測試，對應接口'/get'，一樣f12查看

app2爲目標服務器，有3個接口。

一、'/upload'接口，測試文件上傳功能，上傳文件將放在uploads文件夾下，上傳的文件，文件名是一個uuid，沒有後綴，添加後綴便可查看文件是完整。測試過，傳1G的文件沒問題，再大的文件沒試過，有須要的能夠試下

二、'/json'，測試POST請求。

三、'/get'，測試GET請求。

app.js爲代理服務爲器，監聽端口爲18000,將全部請求轉發至app2，即全部app2的接口靜態資源，app中訪問時一致的。

測試步驟：
一、可開啓目標服務器，經過三個頁面測試功能。

二、開啓代理服務器，訪問下面三個頁面：

http://localhost:18000/json.html

http://localhost:18000/upload.html

http://localhost:18000/get.html

測試一樣的功能。若和步驟1實現一樣功能，則代理服務功能已經實現了。

通過測試，代理功能是沒問題的。

若是問題歡迎留言，或發送郵件至472784995@qq.com。

至於性能，我沒測過，由於我本身這邊的應用場景，訪問量都不大，可使用。