添加请求标识

Author: liguoliang

src/DotNetCoreRpc.Client/RpcClient.cs

@@ -23,6 +23,7 @@ public T CreateClient<T>(string serviceName) where T : class
 
         public T CreateClient<T>(HttpClient httpClient) where T : class
         {
+            httpClient.DefaultRequestHeaders.Add("req-source", "dncrpc");
             HttpRequestInterceptor httpRequestInterceptor = new HttpRequestInterceptor(httpClient);
             return _proxyGenerator.CreateInterfaceProxyWithoutTarget<T>(httpRequestInterceptor);
         }

src/DotNetCoreRpc.Server/DncRpcIApplicationBuilderExtensions.cs

@@ -7,8 +7,10 @@ public static class DncRpcIApplicationBuilderExtensions
     {
         public static IApplicationBuilder UseDotNetCoreRpc(this IApplicationBuilder applicationBuilder)
         {
-            return applicationBuilder.UseWhen(context => string.Equals(context.Request.Method,"post",StringComparison.OrdinalIgnoreCase)
-            && context.Request.Path.Value.Contains("/DotNetCoreRpc/ServerRequest"),
+            return applicationBuilder.UseWhen(context => context.Request.Path.Value.Contains("/DotNetCoreRpc/ServerRequest")
+                && context.Request.Headers.ContainsKey("req-source")
+                && context.Request.Headers["req-source"] == "dncrpc"
+                && string.Equals(context.Request.Method, "post", StringComparison.OrdinalIgnoreCase), 
             appBuilder => appBuilder.UseMiddleware<DotNetCoreRpcMiddleware>());
         }
     }