| Index: third_party/grpc/src/csharp/Grpc.Core/Internal/CallSafeHandle.cs
|
| diff --git a/third_party/grpc/src/csharp/Grpc.Core/Internal/CallSafeHandle.cs b/third_party/grpc/src/csharp/Grpc.Core/Internal/CallSafeHandle.cs
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..bc045b67b10b7ce70b143e3adef038665a6e81fd
|
| --- /dev/null
|
| +++ b/third_party/grpc/src/csharp/Grpc.Core/Internal/CallSafeHandle.cs
|
| @@ -0,0 +1,217 @@
|
| +#region Copyright notice and license
|
| +// Copyright 2015-2016, Google Inc.
|
| +// All rights reserved.
|
| +//
|
| +// Redistribution and use in source and binary forms, with or without
|
| +// modification, are permitted provided that the following conditions are
|
| +// met:
|
| +//
|
| +// * Redistributions of source code must retain the above copyright
|
| +// notice, this list of conditions and the following disclaimer.
|
| +// * Redistributions in binary form must reproduce the above
|
| +// copyright notice, this list of conditions and the following disclaimer
|
| +// in the documentation and/or other materials provided with the
|
| +// distribution.
|
| +// * Neither the name of Google Inc. nor the names of its
|
| +// contributors may be used to endorse or promote products derived from
|
| +// this software without specific prior written permission.
|
| +//
|
| +// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
| +// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
| +// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
| +// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
| +// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
| +// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
| +// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
| +// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
| +// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
| +// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
| +// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
| +
|
| +#endregion
|
| +using System;
|
| +using System.Diagnostics;
|
| +using System.Runtime.InteropServices;
|
| +using Grpc.Core;
|
| +using Grpc.Core.Utils;
|
| +using Grpc.Core.Profiling;
|
| +
|
| +namespace Grpc.Core.Internal
|
| +{
|
| + /// <summary>
|
| + /// grpc_call from <c>grpc/grpc.h</c>
|
| + /// </summary>
|
| + internal class CallSafeHandle : SafeHandleZeroIsInvalid, INativeCall
|
| + {
|
| + public static readonly CallSafeHandle NullInstance = new CallSafeHandle();
|
| + static readonly NativeMethods Native = NativeMethods.Get();
|
| +
|
| + const uint GRPC_WRITE_BUFFER_HINT = 1;
|
| + CompletionRegistry completionRegistry;
|
| + CompletionQueueSafeHandle completionQueue;
|
| +
|
| + private CallSafeHandle()
|
| + {
|
| + }
|
| +
|
| + public void Initialize(CompletionRegistry completionRegistry, CompletionQueueSafeHandle completionQueue)
|
| + {
|
| + this.completionRegistry = completionRegistry;
|
| + this.completionQueue = completionQueue;
|
| + }
|
| +
|
| + public void SetCredentials(CallCredentialsSafeHandle credentials)
|
| + {
|
| + Native.grpcsharp_call_set_credentials(this, credentials).CheckOk();
|
| + }
|
| +
|
| + public void StartUnary(UnaryResponseClientHandler callback, byte[] payload, MetadataArraySafeHandle metadataArray, WriteFlags writeFlags)
|
| + {
|
| + using (completionQueue.NewScope())
|
| + {
|
| + var ctx = BatchContextSafeHandle.Create();
|
| + completionRegistry.RegisterBatchCompletion(ctx, (success, context) => callback(success, context.GetReceivedStatusOnClient(), context.GetReceivedMessage(), context.GetReceivedInitialMetadata()));
|
| + Native.grpcsharp_call_start_unary(this, ctx, payload, new UIntPtr((ulong)payload.Length), metadataArray, writeFlags)
|
| + .CheckOk();
|
| + }
|
| + }
|
| +
|
| + public void StartUnary(BatchContextSafeHandle ctx, byte[] payload, MetadataArraySafeHandle metadataArray, WriteFlags writeFlags)
|
| + {
|
| + using (Profilers.ForCurrentThread().NewScope("CallSafeHandle.StartUnary"))
|
| + {
|
| + Native.grpcsharp_call_start_unary(this, ctx, payload, new UIntPtr((ulong)payload.Length), metadataArray, writeFlags)
|
| + .CheckOk();
|
| + }
|
| + }
|
| +
|
| + public void StartClientStreaming(UnaryResponseClientHandler callback, MetadataArraySafeHandle metadataArray)
|
| + {
|
| + using (completionQueue.NewScope())
|
| + {
|
| + var ctx = BatchContextSafeHandle.Create();
|
| + completionRegistry.RegisterBatchCompletion(ctx, (success, context) => callback(success, context.GetReceivedStatusOnClient(), context.GetReceivedMessage(), context.GetReceivedInitialMetadata()));
|
| + Native.grpcsharp_call_start_client_streaming(this, ctx, metadataArray).CheckOk();
|
| + }
|
| + }
|
| +
|
| + public void StartServerStreaming(ReceivedStatusOnClientHandler callback, byte[] payload, MetadataArraySafeHandle metadataArray, WriteFlags writeFlags)
|
| + {
|
| + using (completionQueue.NewScope())
|
| + {
|
| + var ctx = BatchContextSafeHandle.Create();
|
| + completionRegistry.RegisterBatchCompletion(ctx, (success, context) => callback(success, context.GetReceivedStatusOnClient()));
|
| + Native.grpcsharp_call_start_server_streaming(this, ctx, payload, new UIntPtr((ulong)payload.Length), metadataArray, writeFlags).CheckOk();
|
| + }
|
| + }
|
| +
|
| + public void StartDuplexStreaming(ReceivedStatusOnClientHandler callback, MetadataArraySafeHandle metadataArray)
|
| + {
|
| + using (completionQueue.NewScope())
|
| + {
|
| + var ctx = BatchContextSafeHandle.Create();
|
| + completionRegistry.RegisterBatchCompletion(ctx, (success, context) => callback(success, context.GetReceivedStatusOnClient()));
|
| + Native.grpcsharp_call_start_duplex_streaming(this, ctx, metadataArray).CheckOk();
|
| + }
|
| + }
|
| +
|
| + public void StartSendMessage(SendCompletionHandler callback, byte[] payload, WriteFlags writeFlags, bool sendEmptyInitialMetadata)
|
| + {
|
| + using (completionQueue.NewScope())
|
| + {
|
| + var ctx = BatchContextSafeHandle.Create();
|
| + completionRegistry.RegisterBatchCompletion(ctx, (success, context) => callback(success));
|
| + Native.grpcsharp_call_send_message(this, ctx, payload, new UIntPtr((ulong)payload.Length), writeFlags, sendEmptyInitialMetadata).CheckOk();
|
| + }
|
| + }
|
| +
|
| + public void StartSendCloseFromClient(SendCompletionHandler callback)
|
| + {
|
| + using (completionQueue.NewScope())
|
| + {
|
| + var ctx = BatchContextSafeHandle.Create();
|
| + completionRegistry.RegisterBatchCompletion(ctx, (success, context) => callback(success));
|
| + Native.grpcsharp_call_send_close_from_client(this, ctx).CheckOk();
|
| + }
|
| + }
|
| +
|
| + public void StartSendStatusFromServer(SendCompletionHandler callback, Status status, MetadataArraySafeHandle metadataArray, bool sendEmptyInitialMetadata)
|
| + {
|
| + using (completionQueue.NewScope())
|
| + {
|
| + var ctx = BatchContextSafeHandle.Create();
|
| + completionRegistry.RegisterBatchCompletion(ctx, (success, context) => callback(success));
|
| + Native.grpcsharp_call_send_status_from_server(this, ctx, status.StatusCode, status.Detail, metadataArray, sendEmptyInitialMetadata).CheckOk();
|
| + }
|
| + }
|
| +
|
| + public void StartReceiveMessage(ReceivedMessageHandler callback)
|
| + {
|
| + using (completionQueue.NewScope())
|
| + {
|
| + var ctx = BatchContextSafeHandle.Create();
|
| + completionRegistry.RegisterBatchCompletion(ctx, (success, context) => callback(success, context.GetReceivedMessage()));
|
| + Native.grpcsharp_call_recv_message(this, ctx).CheckOk();
|
| + }
|
| + }
|
| +
|
| + public void StartReceiveInitialMetadata(ReceivedResponseHeadersHandler callback)
|
| + {
|
| + using (completionQueue.NewScope())
|
| + {
|
| + var ctx = BatchContextSafeHandle.Create();
|
| + completionRegistry.RegisterBatchCompletion(ctx, (success, context) => callback(success, context.GetReceivedInitialMetadata()));
|
| + Native.grpcsharp_call_recv_initial_metadata(this, ctx).CheckOk();
|
| + }
|
| + }
|
| +
|
| + public void StartServerSide(ReceivedCloseOnServerHandler callback)
|
| + {
|
| + using (completionQueue.NewScope())
|
| + {
|
| + var ctx = BatchContextSafeHandle.Create();
|
| + completionRegistry.RegisterBatchCompletion(ctx, (success, context) => callback(success, context.GetReceivedCloseOnServerCancelled()));
|
| + Native.grpcsharp_call_start_serverside(this, ctx).CheckOk();
|
| + }
|
| + }
|
| +
|
| + public void StartSendInitialMetadata(SendCompletionHandler callback, MetadataArraySafeHandle metadataArray)
|
| + {
|
| + using (completionQueue.NewScope())
|
| + {
|
| + var ctx = BatchContextSafeHandle.Create();
|
| + completionRegistry.RegisterBatchCompletion(ctx, (success, context) => callback(success));
|
| + Native.grpcsharp_call_send_initial_metadata(this, ctx, metadataArray).CheckOk();
|
| + }
|
| + }
|
| +
|
| + public void Cancel()
|
| + {
|
| + Native.grpcsharp_call_cancel(this).CheckOk();
|
| + }
|
| +
|
| + public void CancelWithStatus(Status status)
|
| + {
|
| + Native.grpcsharp_call_cancel_with_status(this, status.StatusCode, status.Detail).CheckOk();
|
| + }
|
| +
|
| + public string GetPeer()
|
| + {
|
| + using (var cstring = Native.grpcsharp_call_get_peer(this))
|
| + {
|
| + return cstring.GetValue();
|
| + }
|
| + }
|
| +
|
| + protected override bool ReleaseHandle()
|
| + {
|
| + Native.grpcsharp_call_destroy(handle);
|
| + return true;
|
| + }
|
| +
|
| + private static uint GetFlags(bool buffered)
|
| + {
|
| + return buffered ? 0 : GRPC_WRITE_BUFFER_HINT;
|
| + }
|
| + }
|
| +}
|
|
|