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

这里我这只是一个比较简单的实现，大概就二三十行代码，也不是很完善，但试了一下，基本要的功能也都有，感兴趣的朋友能够将其完善下，这个功能自己也只是一个过渡期使用的功能，后续使用过程当中若是有更新也会放上来。