用.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);

這裏我這只是一個比較簡單的實現，大概就二三十行代碼，也不是很完善，但試了一下，基本要的功能也都有，感興趣的朋友能夠將其完善下，這個功能自己也只是一個過渡期使用的功能，後續使用過程當中若是有更新也會放上來。