add serializationScope and refactoring for efficiency

src/csharp/Grpc.Core.Tests/ContextualMarshallerTest.cs

@@ -52,6 +52,8 @@ namespace Grpc.Core.Tests
                     }
                     if (str == "SERIALIZE_TO_NULL")
                     {
+                        // TODO: test for not calling complete Complete() (that resulted in null payload before...)
+                        // TODO: test for calling Complete(null byte array)
                         return;
                     }
                     var bytes = System.Text.Encoding.UTF8.GetBytes(str);

src/csharp/Grpc.Core/Internal/AsyncCall.cs

@@ -95,10 +95,10 @@ namespace Grpc.Core.Internal
                         readingDone = true;
                     }
 
-                    var payload = UnsafeSerialize(msg);
-
+                    using (var serializationScope = DefaultSerializationContext.GetInitializedThreadLocalScope())
                     using (var metadataArray = MetadataArraySafeHandle.Create(details.Options.Headers))
                     {
+                        var payload = UnsafeSerialize(msg, serializationScope.Context); // do before metadata array?
                         var ctx = details.Channel.Environment.BatchContextPool.Lease();
                         try
                         {
@@ -160,11 +160,14 @@ namespace Grpc.Core.Internal
                     halfcloseRequested = true;
                     readingDone = true;
 
-                    var payload = UnsafeSerialize(msg);
+                    //var payload = UnsafeSerialize(msg);
 
                     unaryResponseTcs = new TaskCompletionSource<TResponse>();
+                    using (var serializationScope = DefaultSerializationContext.GetInitializedThreadLocalScope())
                     using (var metadataArray = MetadataArraySafeHandle.Create(details.Options.Headers))
                     {
+                        var payload = UnsafeSerialize(msg, serializationScope.Context); // do before metadata array?
+
                         call.StartUnary(UnaryResponseClientCallback, payload, GetWriteFlagsForCall(), metadataArray, details.Options.Flags);
                         callStartedOk = true;
                     }
@@ -235,11 +238,15 @@ namespace Grpc.Core.Internal
 
                     halfcloseRequested = true;
 
-                    var payload = UnsafeSerialize(msg);
+                    //var payload = UnsafeSerialize(msg);
 
                     streamingResponseCallFinishedTcs = new TaskCompletionSource<object>();
+                    
+                    using (var serializationScope = DefaultSerializationContext.GetInitializedThreadLocalScope())
                     using (var metadataArray = MetadataArraySafeHandle.Create(details.Options.Headers))
                     {
+                        var payload = UnsafeSerialize(msg, serializationScope.Context); // do before metadata array?
+                        
                         call.StartServerStreaming(ReceivedStatusOnClientCallback, payload, GetWriteFlagsForCall(), metadataArray, details.Options.Flags);
                         callStartedOk = true;
                     }

src/csharp/Grpc.Core/Internal/AsyncCallBase.cs

@@ -115,8 +115,9 @@ namespace Grpc.Core.Internal
         /// </summary>
         protected Task SendMessageInternalAsync(TWrite msg, WriteFlags writeFlags)
         {
-            var payload = UnsafeSerialize(msg);
-
+            using (var serializationScope = DefaultSerializationContext.GetInitializedThreadLocalScope())
+            {
+                var payload = UnsafeSerialize(msg, serializationScope.Context); // do before metadata array?
                 lock (myLock)
                 {
                     GrpcPreconditions.CheckState(started);
@@ -134,6 +135,7 @@ namespace Grpc.Core.Internal
                     return streamingWriteTcs.Task;
                 }
             }
+        }
 
         /// <summary>
         /// Initiates reading a message. Only one read operation can be active at a time.
@@ -213,20 +215,12 @@ namespace Grpc.Core.Internal
         /// </summary>
         protected abstract Task CheckSendAllowedOrEarlyResult();
 
-        protected SliceBufferSafeHandle UnsafeSerialize(TWrite msg)
-        {
-            DefaultSerializationContext context = null;
-            try
+        // runs the serializer, propagating any exceptions being thrown without modifying them
+        protected SliceBufferSafeHandle UnsafeSerialize(TWrite msg, DefaultSerializationContext context)
         {
-                context = DefaultSerializationContext.GetInitializedThreadLocal();
             serializer(msg, context);
             return context.GetPayload();
         }
-            finally
-            {
-                context?.Reset();
-            }
-        }
 
         protected Exception TryDeserialize(IBufferReader reader, out TRead msg)
         {

src/csharp/Grpc.Core/Internal/AsyncCallServer.cs

@@ -129,7 +129,9 @@ namespace Grpc.Core.Internal
         /// </summary>
         public Task SendStatusFromServerAsync(Status status, Metadata trailers, ResponseWithFlags? optionalWrite)
         {
-            var payload = optionalWrite.HasValue ? UnsafeSerialize(optionalWrite.Value.Response) : null;
+            using (var serializationScope = DefaultSerializationContext.GetInitializedThreadLocalScope())
+            {
+                var payload = optionalWrite.HasValue ? UnsafeSerialize(optionalWrite.Value.Response, serializationScope.Context) : null;
                 var writeFlags = optionalWrite.HasValue ? optionalWrite.Value.WriteFlags : default(WriteFlags);
 
                 lock (myLock)
@@ -153,6 +155,7 @@ namespace Grpc.Core.Internal
                     return sendStatusFromServerTcs.Task;
                 }
             }
+        }
 
         /// <summary>
         /// Gets cancellation token that gets cancelled once close completion

src/csharp/Grpc.Core/Internal/DefaultSerializationContext.cs

@@ -29,8 +29,7 @@ namespace Grpc.Core.Internal
             new ThreadLocal<DefaultSerializationContext>(() => new DefaultSerializationContext(), false);
 
         bool isComplete;
-        //byte[] payload;
-        SliceBufferSafeHandle sliceBuffer;
+        SliceBufferSafeHandle sliceBuffer = SliceBufferSafeHandle.Create();
 
         public DefaultSerializationContext()
         {
@@ -42,12 +41,10 @@ namespace Grpc.Core.Internal
             GrpcPreconditions.CheckState(!isComplete);
             this.isComplete = true;
 
-            GetBufferWriter();
             var destSpan = sliceBuffer.GetSpan(payload.Length);
             payload.AsSpan().CopyTo(destSpan);
             sliceBuffer.Advance(payload.Length);
             sliceBuffer.Complete();
-            //this.payload = payload;
         }
 
         /// <summary>
@@ -55,11 +52,6 @@ namespace Grpc.Core.Internal
         /// </summary>
         public override IBufferWriter<byte> GetBufferWriter()
         {
-            if (sliceBuffer == null)
-            {
-                // TODO: avoid allocation..
-                sliceBuffer = SliceBufferSafeHandle.Create();
-            }
             return sliceBuffer;
         }
 
@@ -81,17 +73,35 @@ namespace Grpc.Core.Internal
         public void Reset()
         {
             this.isComplete = false;
-            //this.payload = null;
-            this.sliceBuffer = null;  // reset instead...
+            this.sliceBuffer.Reset();
         }
 
-        public static DefaultSerializationContext GetInitializedThreadLocal()
+        // Get a cached thread local instance of deserialization context
+        // and wrap it in a disposable struct that allows easy resetting
+        // via "using" statement.
+        public static UsageScope GetInitializedThreadLocalScope()
         {
             var instance = threadLocalInstance.Value;
-            instance.Reset();
-            return instance;
+            return new UsageScope(instance);
         }
 
+        public struct UsageScope : IDisposable
+        {
+            readonly DefaultSerializationContext context;
+
+            public UsageScope(DefaultSerializationContext context)
+            {
+                this.context = context;
+            }
 
+            public DefaultSerializationContext Context => context;
+
+            // TODO: add Serialize method...
+
+            public void Dispose()
+            {
+                context.Reset();
+            }
+        }
     }
 }