用.net core实现反向代理中间件

最近在将一些项目的rest api迁移到.net core中，最开始是用的Nginx做反向代理，将已经完成切换的部分切入系统，如下图所示：

由于迁移过程中也在进行代码重构，需要经常比较频繁的测试，以保证能及时发现引入的问题。从而导致我们每迁移一部分都需要配置一次nginx的路由映射，保证迁移的功能能切入系统测试。

进行了一段时间后，发现经常配置Nginx一来比较麻烦，二来容易配错；便想将这个反向代理的功能放在.net core程序中去，实现如下的功能：

Rest请求直接发往.net core程序
如果该请求在.net core程序中实现，则执行请求并返回
如果未实现，将其请求老版接口的数据，并返回结果。

形成如下的一个结构：

试了一下，在.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以使用。