Net Core中 使用Middleware 實現反向代理

有這樣的一個需求,咱們要攔截某些特定的請求,並將它們從新定向到另外一臺服務器中,然而客戶端並不知情。api

在NetCore中咱們能夠用中間件來實現,瀏覽器

首先建立項目:

  我這裏只有2.1 Version 的服務器

  

添加ProxyMiddleware

  

 

ProxyMiddleware內容以下:

   代碼很少有興趣的朋友能夠調試一下。這裏還能夠有不少的方向擴展。app

 public class ProxyMiddleware
    {
        private static readonly HttpClient _httpClient = new HttpClient();
        private readonly RequestDelegate _nextRequestDelegate;
        private static readonly Uri _targetUri = new Uri("https://www.cnblogs.com/");
        public ProxyMiddleware(RequestDelegate nextMiddleware)
        {
            _nextRequestDelegate = nextMiddleware;
        }
        public async Task Invoke(HttpContext context)
        {
            bool validateUri = false; 
            if (context.Request.Path.StartsWithSegments("/api/values", out var Path))
            {
                validateUri = true;
            }
            if (validateUri == true)
            {
                var targetRequestMessage = CreateTargetMessage(context);
                using (var responseMessage = await _httpClient.SendAsync(targetRequestMessage))
                {
                    context.Response.StatusCode = (int)responseMessage.StatusCode;
                    CloneResponseHeadersIntoContext(context, responseMessage);
                    await responseMessage.Content.CopyToAsync(context.Response.Body);
                }
                return;
            }
            await _nextRequestDelegate(context);
        }
        private void CloneRequestContentAndHeaders(HttpContext context, HttpRequestMessage requestMessage)
        {
            foreach (var header in context.Request.Headers)
            {
                requestMessage.Content?.Headers.TryAddWithoutValidation(header.Key, header.Value.ToArray());
            }
        }
        private HttpRequestMessage CreateTargetMessage(HttpContext context)
        {
            var requestMessage = new HttpRequestMessage();
            CloneRequestContentAndHeaders(context, requestMessage);
            requestMessage.RequestUri = _targetUri;
            requestMessage.Headers.Host = _targetUri.Host;
            requestMessage.Method = new HttpMethod(context.Request.Method);
            return requestMessage;
        }
        private void CloneResponseHeadersIntoContext(HttpContext context, HttpResponseMessage responseMessage)
        {
            foreach (var header in responseMessage.Headers)
            {
                context.Response.Headers[header.Key] = header.Value.ToArray();
            }
            foreach (var header in responseMessage.Content.Headers)
            {
                context.Response.Headers[header.Key] = header.Value.ToArray();
            }
            context.Response.Headers.Remove("Transfer-Encoding");
        }
    }

添加管道

        // This method gets called by the runtime. Use this method to configure the HTTP request pipeline.
        public void Configure(IApplicationBuilder app, IHostingEnvironment env)
        {
            if (env.IsDevelopment())
            {
                app.UseDeveloperExceptionPage();
            }
            app.UseMiddleware<ProxyMiddleware>();
            app.UseMvc();
        }

運行結果:

  你們能夠注意瀏覽器網址,以及顯示的內容就能夠了,(樣式沒了)async

  

 

 

 代碼解釋:

  全部的描述在代碼中,這裏我只是標出這點代碼的重點學習

        建立靜態HttpClient鏈接,減小鏈接池數量
     private static readonly HttpClient _httpClient = new HttpClient();
      
private readonly RequestDelegate _nextRequestDelegate;
     新的目標服務器
private static readonly Uri _targetUri = new Uri("https://www.cnblogs.com/"); public ProxyMiddleware(RequestDelegate nextMiddleware) { _nextRequestDelegate = nextMiddleware; }
        
     全部的工做將由 Invoke執行
     public async Task Invoke(HttpContext context) { bool validateUri = false; if (context.Request.Path.StartsWithSegments("/api/values", out var Path)) { validateUri = true; } if (validateUri == true) { var targetRequestMessage = CreateTargetMessage(context); using (var responseMessage = await _httpClient.SendAsync(targetRequestMessage)) { context.Response.StatusCode = (int)responseMessage.StatusCode; CloneResponseHeadersIntoContext(context, responseMessage); await responseMessage.Content.CopyToAsync(context.Response.Body); } return; } await _nextRequestDelegate(context); }
        private void CloneResponseHeadersIntoContext(HttpContext context, HttpResponseMessage responseMessage)
        {
            foreach (var header in responseMessage.Headers)
            {
                context.Response.Headers[header.Key] = header.Value.ToArray();
            }
            foreach (var header in responseMessage.Content.Headers)
            {
                context.Response.Headers[header.Key] = header.Value.ToArray();
            }
       這裏有一個坑你們注意了,有興趣的同窗能夠調查研究一下,要是介紹的話能夠單獨開一篇了 context.Response.Headers.Remove(
"Transfer-Encoding"); }

 

有不足之處 但願你們指出相互學習。ui

相關文章
相關標籤/搜索