diff --git a/src/csharp/Grpc.Core/Internal/ServerCallHandler.cs b/src/csharp/Grpc.Core/Internal/ServerCallHandler.cs index 98995a0862e..f9bf40f2372 100644 --- a/src/csharp/Grpc.Core/Internal/ServerCallHandler.cs +++ b/src/csharp/Grpc.Core/Internal/ServerCallHandler.cs @@ -21,6 +21,7 @@ using System.Collections.Generic; using System.Linq; using System.Threading; using System.Threading.Tasks; +using Grpc.Core.Interceptors; using Grpc.Core.Internal; using Grpc.Core.Logging; using Grpc.Core.Utils; @@ -32,7 +33,12 @@ namespace Grpc.Core.Internal Task HandleCall(ServerRpcNew newRpc, CompletionQueueSafeHandle cq); } - internal class UnaryServerCallHandler : IServerCallHandler + internal interface IInterceptableCallHandler + { + IServerCallHandler Intercept(Interceptor interceptor); + } + + internal class UnaryServerCallHandler : IServerCallHandler, IInterceptableCallHandler where TRequest : class where TResponse : class { @@ -74,7 +80,7 @@ namespace Grpc.Core.Internal { if (!(e is RpcException)) { - Logger.Warning(e, "Exception occured in handler."); + Logger.Warning(e, "Exception occured in handler or interceptors."); } status = HandlerUtils.GetStatusFromExceptionAndMergeTrailers(e, context.ResponseTrailers); } @@ -89,9 +95,14 @@ namespace Grpc.Core.Internal } await finishedTask.ConfigureAwait(false); } + + IServerCallHandler IInterceptableCallHandler.Intercept(Interceptor interceptor) + { + return new UnaryServerCallHandler(method, (request, context) => interceptor.UnaryServerHandler(request, context, handler)); + } } - internal class ServerStreamingServerCallHandler : IServerCallHandler + internal class ServerStreamingServerCallHandler : IServerCallHandler, IInterceptableCallHandler where TRequest : class where TResponse : class { @@ -131,7 +142,7 @@ namespace Grpc.Core.Internal { if (!(e is RpcException)) { - Logger.Warning(e, "Exception occured in handler."); + Logger.Warning(e, "Exception occured in handler or interceptors."); } status = HandlerUtils.GetStatusFromExceptionAndMergeTrailers(e, context.ResponseTrailers); } @@ -147,9 +158,14 @@ namespace Grpc.Core.Internal } await finishedTask.ConfigureAwait(false); } + + IServerCallHandler IInterceptableCallHandler.Intercept(Interceptor interceptor) + { + return new ServerStreamingServerCallHandler(method, (request, responseStream, context) => interceptor.ServerStreamingServerHandler(request, responseStream, context, handler)); + } } - internal class ClientStreamingServerCallHandler : IServerCallHandler + internal class ClientStreamingServerCallHandler : IServerCallHandler, IInterceptableCallHandler where TRequest : class where TResponse : class { @@ -189,7 +205,7 @@ namespace Grpc.Core.Internal { if (!(e is RpcException)) { - Logger.Warning(e, "Exception occured in handler."); + Logger.Warning(e, "Exception occured in handler or interceptor."); } status = HandlerUtils.GetStatusFromExceptionAndMergeTrailers(e, context.ResponseTrailers); } @@ -205,9 +221,14 @@ namespace Grpc.Core.Internal } await finishedTask.ConfigureAwait(false); } + + IServerCallHandler IInterceptableCallHandler.Intercept(Interceptor interceptor) + { + return new ClientStreamingServerCallHandler(method, (requestStream, context) => interceptor.ClientStreamingServerHandler(requestStream, context, handler)); + } } - internal class DuplexStreamingServerCallHandler : IServerCallHandler + internal class DuplexStreamingServerCallHandler : IServerCallHandler, IInterceptableCallHandler where TRequest : class where TResponse : class { @@ -245,7 +266,7 @@ namespace Grpc.Core.Internal { if (!(e is RpcException)) { - Logger.Warning(e, "Exception occured in handler."); + Logger.Warning(e, "Exception occured in handler or interceptor."); } status = HandlerUtils.GetStatusFromExceptionAndMergeTrailers(e, context.ResponseTrailers); } @@ -260,6 +281,11 @@ namespace Grpc.Core.Internal } await finishedTask.ConfigureAwait(false); } + + IServerCallHandler IInterceptableCallHandler.Intercept(Interceptor interceptor) + { + return new DuplexStreamingServerCallHandler(method, (requestStream, responseStream, context) => interceptor.DuplexStreamingServerHandler(requestStream, responseStream, context, handler)); + } } internal class UnimplementedMethodCallHandler : IServerCallHandler diff --git a/src/csharp/Grpc.Core/ServerServiceDefinition.cs b/src/csharp/Grpc.Core/ServerServiceDefinition.cs index 59868c1f402..3e6c12884b7 100644 --- a/src/csharp/Grpc.Core/ServerServiceDefinition.cs +++ b/src/csharp/Grpc.Core/ServerServiceDefinition.cs @@ -19,7 +19,10 @@ using System; using System.Collections.Generic; using System.Collections.ObjectModel; +using System.Linq; +using Grpc.Core.Interceptors; using Grpc.Core.Internal; +using Grpc.Core.Utils; namespace Grpc.Core { @@ -45,6 +48,28 @@ namespace Grpc.Core } } + /// + /// Returns a instance that + /// intercepts calls to the underlying service handler via the given interceptor. + /// This is an EXPERIMENTAL API. + /// + /// The interceptor to register on service. + public ServerServiceDefinition Intercept(Interceptor interceptor) + { + GrpcPreconditions.CheckNotNull(interceptor, "interceptor"); + return new ServerServiceDefinition(CallHandlers.ToDictionary( + x => x.Key, x => + { + var value = x.Value; + var interceptable = value as IInterceptableCallHandler; + if (interceptable == null) + { + return value; + } + return interceptable.Intercept(interceptor); + })); + } + /// /// Creates a new builder object for ServerServiceDefinition. ///