用.net core實現反向代理中間件


最近在將一些項目的rest api遷移到.net core中,最開始是用的Nginx做反向代理,將已經完成切換的部分切入系統,如下圖所示:

由於遷移過程中也在進行代碼重構,需要經常比較頻繁的測試,以保證能及時發現引入的問題。從而導致我們每遷移一部分都需要配置一次nginx的路由映射,保證遷移的功能能切入系統測試。

進行了一段時間后,發現經常配置Nginx一來比較麻煩,二來容易配錯;便想將這個反向代理的功能放在.net core程序中去,實現如下的功能:

  1. Rest請求直接發往.net core程序
  2. 如果該請求在.net core程序中實現,則執行請求並返回
  3. 如果未實現,將其請求老版接口的數據,並返回結果。

形成如下的一個結構:

試了一下,在.net core中實現這個功能比較簡單,加一個反向代理的中間件即可:   

public class ReverseProxy
    {
        static HttpClient _http = new HttpClient();

        public static async Task Invoke(HttpContext context)
        {
            var url = context.Request.Path.ToUriComponent();
            var uri = new Uri("http://localhost:8080/api" + url);

            var request   = CopyRequest(context, uri);
            var remoteRsp = await _http.SendAsync(request);
            var rsp       = context.Response;

            foreach (var header in remoteRsp.Headers)
            {
                rsp.Headers.Add(header.Key, header.Value.ToArray());
            }

            rsp.ContentType   = remoteRsp.Content.Headers.ContentType?.ToString();
            rsp.ContentLength = remoteRsp.Content.Headers.ContentLength;

            await remoteRsp.Content.CopyToAsync(rsp.Body);
        }


        static HttpRequestMessage CopyRequest(HttpContext context, Uri targetUri)
        {
            var req = context.Request;
            var requestMessage = new HttpRequestMessage()
            {
                Method     = new HttpMethod(req.Method),
                Content    = new StreamContent(req.Body),
                RequestUri = targetUri,
            };

            foreach (var header in req.Headers)
            {
                requestMessage.Content?.Headers.TryAddWithoutValidation(header.Key, header.Value.ToArray());
            }

            requestMessage.Headers.Host = targetUri.Host;

            return requestMessage;
        }
    }
View Code

使用起來也比較簡單,放在Configure函數的最后面,直接使用Run來反向代理所有未被系統實現的請求即可。

    app.Run(Middleware.ReverseProxy.Invoke);

這里我這只是一個比較簡單的實現,大概就二三十行代碼,也不是很完善,但試了一下,基本要的功能也都有,網上也有一些第三方框架可ProxyKit以使用。


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM