随笔 - 813  文章 - 8 评论 - 938 trackbacks - 19

最近在将一些项目的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);

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

posted on 2019-09-11 00:04 天方 阅读(...) 评论(...) 编辑 收藏