当前位置 主页 > 网站技术 > 代码类 >

    利用.net core实现反向代理中间件的方法

    栏目:代码类 时间:2019-10-08 09:03

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

    使用起来也比较简单,放在Configure函数的最后面,直接使用Run来反向代理所有未被系统实现的请求即可。

      app.Run(Middleware.ReverseProxy.Invoke);

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

    总结

    以上就是这篇文章的全部内容了,希望本文的内容对大家的学习或者工作具有一定的参考学习价值,谢谢大家对IIS7站长之家的支持。