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

时间:2021-04-02 人气:0

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

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

总结

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

您可能感兴趣的文章:

加载全部内容