From 834a3d29a656b5f1e2313e1608bc6f60c2fd21cd Mon Sep 17 00:00:00 2001
From: mgravell <marc.gravell@gmail.com>
Date: Tue, 2 Jul 2019 14:24:24 +0100
Subject: [PATCH] capture the server too

---
 src/csharp/Grpc.Core/Server.cs | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/csharp/Grpc.Core/Server.cs b/src/csharp/Grpc.Core/Server.cs
index 81b4b59528b..8705f52666f 100644
--- a/src/csharp/Grpc.Core/Server.cs
+++ b/src/csharp/Grpc.Core/Server.cs
@@ -334,7 +334,7 @@ namespace Grpc.Core
         /// <summary>
         /// Selects corresponding handler for given call and handles the call.
         /// </summary>
-        private async Task HandleCallAsync(ServerRpcNew newRpc, CompletionQueueSafeHandle cq, Action<CompletionQueueSafeHandle> continuation)
+        private async Task HandleCallAsync(ServerRpcNew newRpc, CompletionQueueSafeHandle cq, Action<Server, CompletionQueueSafeHandle> continuation)
         {
             try
             {
@@ -351,7 +351,7 @@ namespace Grpc.Core
             }
             finally
             {
-                continuation(cq);
+                continuation(this, cq);
             }
         }
 
@@ -374,7 +374,7 @@ namespace Grpc.Core
                     // Don't await, the continuations will run on gRPC thread pool once triggered
                     // by cq.Next().
                     #pragma warning disable 4014
-                    HandleCallAsync(newRpc, cq, state => AllowOneRpc(state));
+                    HandleCallAsync(newRpc, cq, (server, state) => server.AllowOneRpc(state));
                     #pragma warning restore 4014
                 }
             }