|
|
|
|
@ -1,11 +1,11 @@ |
|
|
|
|
#region Copyright notice and license |
|
|
|
|
// Copyright 2015, 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 |
|
|
|
|
@ -15,7 +15,7 @@ |
|
|
|
|
// * 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 |
|
|
|
|
@ -34,158 +34,203 @@ using System.Diagnostics; |
|
|
|
|
using System.Runtime.InteropServices; |
|
|
|
|
using Grpc.Core; |
|
|
|
|
|
|
|
|
|
namespace Grpc.Core.Internal |
|
|
|
|
{ |
|
|
|
|
internal delegate void CompletionCallbackDelegate(GRPCOpError error, IntPtr batchContextPtr); |
|
|
|
|
|
|
|
|
|
/// <summary> |
|
|
|
|
/// grpc_call from <grpc/grpc.h> |
|
|
|
|
/// </summary> |
|
|
|
|
internal class CallSafeHandle : SafeHandleZeroIsInvalid |
|
|
|
|
{ |
|
|
|
|
const uint GRPC_WRITE_BUFFER_HINT = 1; |
|
|
|
|
|
|
|
|
|
[DllImport("grpc_csharp_ext.dll")] |
|
|
|
|
static extern CallSafeHandle grpcsharp_channel_create_call(ChannelSafeHandle channel, CompletionQueueSafeHandle cq, string method, string host, Timespec deadline); |
|
|
|
|
|
|
|
|
|
[DllImport("grpc_csharp_ext.dll")] |
|
|
|
|
static extern GRPCCallError grpcsharp_call_cancel(CallSafeHandle call); |
|
|
|
|
|
|
|
|
|
[DllImport("grpc_csharp_ext.dll")] |
|
|
|
|
static extern GRPCCallError grpcsharp_call_cancel_with_status(CallSafeHandle call, StatusCode status, string description); |
|
|
|
|
|
|
|
|
|
[DllImport("grpc_csharp_ext.dll")] |
|
|
|
|
static extern GRPCCallError grpcsharp_call_start_unary(CallSafeHandle call, |
|
|
|
|
[MarshalAs(UnmanagedType.FunctionPtr)] CompletionCallbackDelegate callback, |
|
|
|
|
byte[] send_buffer, UIntPtr send_buffer_len, MetadataArraySafeHandle metadataArray); |
|
|
|
|
|
|
|
|
|
[DllImport("grpc_csharp_ext.dll")] |
|
|
|
|
static extern void grpcsharp_call_blocking_unary(CallSafeHandle call, CompletionQueueSafeHandle dedicatedCq, |
|
|
|
|
[MarshalAs(UnmanagedType.FunctionPtr)] CompletionCallbackDelegate callback, |
|
|
|
|
byte[] send_buffer, UIntPtr send_buffer_len, MetadataArraySafeHandle metadataArray); |
|
|
|
|
|
|
|
|
|
[DllImport("grpc_csharp_ext.dll")] |
|
|
|
|
static extern GRPCCallError grpcsharp_call_start_client_streaming(CallSafeHandle call, |
|
|
|
|
[MarshalAs(UnmanagedType.FunctionPtr)] CompletionCallbackDelegate callback, |
|
|
|
|
MetadataArraySafeHandle metadataArray); |
|
|
|
|
|
|
|
|
|
[DllImport("grpc_csharp_ext.dll")] |
|
|
|
|
static extern GRPCCallError grpcsharp_call_start_server_streaming(CallSafeHandle call, |
|
|
|
|
[MarshalAs(UnmanagedType.FunctionPtr)] CompletionCallbackDelegate callback, |
|
|
|
|
byte[] send_buffer, UIntPtr send_buffer_len, |
|
|
|
|
MetadataArraySafeHandle metadataArray); |
|
|
|
|
|
|
|
|
|
[DllImport("grpc_csharp_ext.dll")] |
|
|
|
|
static extern GRPCCallError grpcsharp_call_start_duplex_streaming(CallSafeHandle call, |
|
|
|
|
[MarshalAs(UnmanagedType.FunctionPtr)] CompletionCallbackDelegate callback, |
|
|
|
|
MetadataArraySafeHandle metadataArray); |
|
|
|
|
|
|
|
|
|
[DllImport("grpc_csharp_ext.dll")] |
|
|
|
|
static extern GRPCCallError grpcsharp_call_send_message(CallSafeHandle call, |
|
|
|
|
[MarshalAs(UnmanagedType.FunctionPtr)] CompletionCallbackDelegate callback, |
|
|
|
|
byte[] send_buffer, UIntPtr send_buffer_len); |
|
|
|
|
|
|
|
|
|
[DllImport("grpc_csharp_ext.dll")] |
|
|
|
|
static extern GRPCCallError grpcsharp_call_send_close_from_client(CallSafeHandle call, |
|
|
|
|
[MarshalAs(UnmanagedType.FunctionPtr)] CompletionCallbackDelegate callback); |
|
|
|
|
|
|
|
|
|
[DllImport("grpc_csharp_ext.dll")] |
|
|
|
|
static extern GRPCCallError grpcsharp_call_send_status_from_server(CallSafeHandle call, [MarshalAs(UnmanagedType.FunctionPtr)] CompletionCallbackDelegate callback, StatusCode statusCode, string statusMessage); |
|
|
|
|
|
|
|
|
|
[DllImport("grpc_csharp_ext.dll")] |
|
|
|
|
static extern GRPCCallError grpcsharp_call_recv_message(CallSafeHandle call, |
|
|
|
|
[MarshalAs(UnmanagedType.FunctionPtr)] CompletionCallbackDelegate callback); |
|
|
|
|
|
|
|
|
|
[DllImport("grpc_csharp_ext.dll")] |
|
|
|
|
static extern GRPCCallError grpcsharp_call_start_serverside(CallSafeHandle call, |
|
|
|
|
[MarshalAs(UnmanagedType.FunctionPtr)] CompletionCallbackDelegate callback); |
|
|
|
|
|
|
|
|
|
[DllImport("grpc_csharp_ext.dll")] |
|
|
|
|
static extern void grpcsharp_call_destroy(IntPtr call); |
|
|
|
|
|
|
|
|
|
private CallSafeHandle() |
|
|
|
|
{ |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
public static CallSafeHandle Create(ChannelSafeHandle channel, CompletionQueueSafeHandle cq, string method, string host, Timespec deadline) |
|
|
|
|
{ |
|
|
|
|
return grpcsharp_channel_create_call(channel, cq, method, host, deadline); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
public void StartUnary(byte[] payload, CompletionCallbackDelegate callback, MetadataArraySafeHandle metadataArray) |
|
|
|
|
{ |
|
|
|
|
AssertCallOk(grpcsharp_call_start_unary(this, callback, payload, new UIntPtr((ulong)payload.Length), metadataArray)); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
public void BlockingUnary(CompletionQueueSafeHandle dedicatedCq, byte[] payload, CompletionCallbackDelegate callback, MetadataArraySafeHandle metadataArray) |
|
|
|
|
{ |
|
|
|
|
grpcsharp_call_blocking_unary(this, dedicatedCq, callback, payload, new UIntPtr((ulong)payload.Length), metadataArray); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
public void StartClientStreaming(CompletionCallbackDelegate callback, MetadataArraySafeHandle metadataArray) |
|
|
|
|
{ |
|
|
|
|
AssertCallOk(grpcsharp_call_start_client_streaming(this, callback, metadataArray)); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
public void StartServerStreaming(byte[] payload, CompletionCallbackDelegate callback, MetadataArraySafeHandle metadataArray) |
|
|
|
|
{ |
|
|
|
|
AssertCallOk(grpcsharp_call_start_server_streaming(this, callback, payload, new UIntPtr((ulong)payload.Length), metadataArray)); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
public void StartDuplexStreaming(CompletionCallbackDelegate callback, MetadataArraySafeHandle metadataArray) |
|
|
|
|
{ |
|
|
|
|
AssertCallOk(grpcsharp_call_start_duplex_streaming(this, callback, metadataArray)); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
public void StartSendMessage(byte[] payload, CompletionCallbackDelegate callback) |
|
|
|
|
{ |
|
|
|
|
AssertCallOk(grpcsharp_call_send_message(this, callback, payload, new UIntPtr((ulong)payload.Length))); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
public void StartSendCloseFromClient(CompletionCallbackDelegate callback) |
|
|
|
|
{ |
|
|
|
|
AssertCallOk(grpcsharp_call_send_close_from_client(this, callback)); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
public void StartSendStatusFromServer(Status status, CompletionCallbackDelegate callback) |
|
|
|
|
{ |
|
|
|
|
AssertCallOk(grpcsharp_call_send_status_from_server(this, callback, status.StatusCode, status.Detail)); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
public void StartReceiveMessage(CompletionCallbackDelegate callback) |
|
|
|
|
{ |
|
|
|
|
AssertCallOk(grpcsharp_call_recv_message(this, callback)); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
public void StartServerSide(CompletionCallbackDelegate callback) |
|
|
|
|
{ |
|
|
|
|
AssertCallOk(grpcsharp_call_start_serverside(this, callback)); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
public void Cancel() |
|
|
|
|
{ |
|
|
|
|
AssertCallOk(grpcsharp_call_cancel(this)); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
public void CancelWithStatus(Status status) |
|
|
|
|
{ |
|
|
|
|
AssertCallOk(grpcsharp_call_cancel_with_status(this, status.StatusCode, status.Detail)); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
protected override bool ReleaseHandle() |
|
|
|
|
{ |
|
|
|
|
grpcsharp_call_destroy(handle); |
|
|
|
|
return true; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
private static void AssertCallOk(GRPCCallError callError) |
|
|
|
|
{ |
|
|
|
|
Trace.Assert(callError == GRPCCallError.GRPC_CALL_OK, "Status not GRPC_CALL_OK"); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
private static uint GetFlags(bool buffered) |
|
|
|
|
{ |
|
|
|
|
return buffered ? 0 : GRPC_WRITE_BUFFER_HINT; |
|
|
|
|
} |
|
|
|
|
namespace Grpc.Core.Internal { |
|
|
|
|
internal delegate void CompletionCallbackDelegate(GRPCOpError error, |
|
|
|
|
IntPtr batchContextPtr); |
|
|
|
|
|
|
|
|
|
/// <summary> |
|
|
|
|
/// grpc_call from <grpc/grpc.h> |
|
|
|
|
/// </summary> |
|
|
|
|
internal class CallSafeHandle : SafeHandleZeroIsInvalid { |
|
|
|
|
const uint GRPC_WRITE_BUFFER_HINT = 1; |
|
|
|
|
|
|
|
|
|
[DllImport("grpc_csharp_ext.dll")] static extern CallSafeHandle |
|
|
|
|
grpcsharp_channel_create_call(ChannelSafeHandle channel, |
|
|
|
|
CompletionQueueSafeHandle cq, string method, |
|
|
|
|
string host, Timespec deadline); |
|
|
|
|
|
|
|
|
|
[DllImport("grpc_csharp_ext.dll")] static extern GRPCCallError |
|
|
|
|
grpcsharp_call_cancel(CallSafeHandle call); |
|
|
|
|
|
|
|
|
|
[DllImport("grpc_csharp_ext.dll")] static extern GRPCCallError |
|
|
|
|
grpcsharp_call_cancel_with_status(CallSafeHandle call, StatusCode status, |
|
|
|
|
string description); |
|
|
|
|
|
|
|
|
|
[DllImport("grpc_csharp_ext.dll")] static extern GRPCCallError |
|
|
|
|
grpcsharp_call_start_unary( |
|
|
|
|
CallSafeHandle call, |
|
|
|
|
[MarshalAs(UnmanagedType.FunctionPtr)] CompletionCallbackDelegate |
|
|
|
|
callback, |
|
|
|
|
byte[] send_buffer, UIntPtr send_buffer_len, |
|
|
|
|
MetadataArraySafeHandle metadataArray); |
|
|
|
|
|
|
|
|
|
[DllImport("grpc_csharp_ext.dll")] static extern void |
|
|
|
|
grpcsharp_call_blocking_unary( |
|
|
|
|
CallSafeHandle call, CompletionQueueSafeHandle dedicatedCq, |
|
|
|
|
[MarshalAs(UnmanagedType.FunctionPtr)] CompletionCallbackDelegate |
|
|
|
|
callback, |
|
|
|
|
byte[] send_buffer, UIntPtr send_buffer_len, |
|
|
|
|
MetadataArraySafeHandle metadataArray); |
|
|
|
|
|
|
|
|
|
[DllImport("grpc_csharp_ext.dll")] static extern GRPCCallError |
|
|
|
|
grpcsharp_call_start_client_streaming( |
|
|
|
|
CallSafeHandle call, |
|
|
|
|
[MarshalAs(UnmanagedType.FunctionPtr)] CompletionCallbackDelegate |
|
|
|
|
callback, |
|
|
|
|
MetadataArraySafeHandle metadataArray); |
|
|
|
|
|
|
|
|
|
[DllImport("grpc_csharp_ext.dll")] static extern GRPCCallError |
|
|
|
|
grpcsharp_call_start_server_streaming( |
|
|
|
|
CallSafeHandle call, |
|
|
|
|
[MarshalAs(UnmanagedType.FunctionPtr)] CompletionCallbackDelegate |
|
|
|
|
callback, |
|
|
|
|
byte[] send_buffer, UIntPtr send_buffer_len, |
|
|
|
|
MetadataArraySafeHandle metadataArray); |
|
|
|
|
|
|
|
|
|
[DllImport("grpc_csharp_ext.dll")] static extern GRPCCallError |
|
|
|
|
grpcsharp_call_start_duplex_streaming( |
|
|
|
|
CallSafeHandle call, |
|
|
|
|
[MarshalAs(UnmanagedType.FunctionPtr)] CompletionCallbackDelegate |
|
|
|
|
callback, |
|
|
|
|
MetadataArraySafeHandle metadataArray); |
|
|
|
|
|
|
|
|
|
[DllImport("grpc_csharp_ext.dll")] static extern GRPCCallError |
|
|
|
|
grpcsharp_call_send_message( |
|
|
|
|
CallSafeHandle call, |
|
|
|
|
[MarshalAs(UnmanagedType.FunctionPtr)] CompletionCallbackDelegate |
|
|
|
|
callback, |
|
|
|
|
byte[] send_buffer, UIntPtr send_buffer_len); |
|
|
|
|
|
|
|
|
|
[DllImport("grpc_csharp_ext.dll")] static extern GRPCCallError |
|
|
|
|
grpcsharp_call_send_close_from_client( |
|
|
|
|
CallSafeHandle call, |
|
|
|
|
[MarshalAs(UnmanagedType.FunctionPtr)] CompletionCallbackDelegate |
|
|
|
|
callback); |
|
|
|
|
|
|
|
|
|
[DllImport("grpc_csharp_ext.dll")] static extern GRPCCallError |
|
|
|
|
grpcsharp_call_send_status_from_server( |
|
|
|
|
CallSafeHandle call, |
|
|
|
|
[MarshalAs(UnmanagedType.FunctionPtr)] CompletionCallbackDelegate |
|
|
|
|
callback, |
|
|
|
|
StatusCode statusCode, string statusMessage); |
|
|
|
|
|
|
|
|
|
[DllImport("grpc_csharp_ext.dll")] static extern GRPCCallError |
|
|
|
|
grpcsharp_call_recv_message( |
|
|
|
|
CallSafeHandle call, |
|
|
|
|
[MarshalAs(UnmanagedType.FunctionPtr)] CompletionCallbackDelegate |
|
|
|
|
callback); |
|
|
|
|
|
|
|
|
|
[DllImport("grpc_csharp_ext.dll")] static extern GRPCCallError |
|
|
|
|
grpcsharp_call_start_serverside( |
|
|
|
|
CallSafeHandle call, |
|
|
|
|
[MarshalAs(UnmanagedType.FunctionPtr)] CompletionCallbackDelegate |
|
|
|
|
callback); |
|
|
|
|
|
|
|
|
|
[DllImport("grpc_csharp_ext.dll")] static extern void |
|
|
|
|
grpcsharp_call_destroy(IntPtr call); |
|
|
|
|
|
|
|
|
|
private |
|
|
|
|
CallSafeHandle() {} |
|
|
|
|
|
|
|
|
|
public |
|
|
|
|
static CallSafeHandle Create(ChannelSafeHandle channel, |
|
|
|
|
CompletionQueueSafeHandle cq, string method, |
|
|
|
|
string host, Timespec deadline) { |
|
|
|
|
return grpcsharp_channel_create_call(channel, cq, method, host, deadline); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
public |
|
|
|
|
void StartUnary(byte[] payload, CompletionCallbackDelegate callback, |
|
|
|
|
MetadataArraySafeHandle metadataArray) { |
|
|
|
|
AssertCallOk(grpcsharp_call_start_unary( |
|
|
|
|
this, callback, payload, new UIntPtr((ulong)payload.Length), |
|
|
|
|
metadataArray)); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
public |
|
|
|
|
void BlockingUnary(CompletionQueueSafeHandle dedicatedCq, byte[] payload, |
|
|
|
|
CompletionCallbackDelegate callback, |
|
|
|
|
MetadataArraySafeHandle metadataArray) { |
|
|
|
|
grpcsharp_call_blocking_unary(this, dedicatedCq, callback, payload, |
|
|
|
|
new UIntPtr((ulong)payload.Length), |
|
|
|
|
metadataArray); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
public |
|
|
|
|
void StartClientStreaming(CompletionCallbackDelegate callback, |
|
|
|
|
MetadataArraySafeHandle metadataArray) { |
|
|
|
|
AssertCallOk( |
|
|
|
|
grpcsharp_call_start_client_streaming(this, callback, metadataArray)); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
public |
|
|
|
|
void StartServerStreaming(byte[] payload, |
|
|
|
|
CompletionCallbackDelegate callback, |
|
|
|
|
MetadataArraySafeHandle metadataArray) { |
|
|
|
|
AssertCallOk(grpcsharp_call_start_server_streaming( |
|
|
|
|
this, callback, payload, new UIntPtr((ulong)payload.Length), |
|
|
|
|
metadataArray)); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
public |
|
|
|
|
void StartDuplexStreaming(CompletionCallbackDelegate callback, |
|
|
|
|
MetadataArraySafeHandle metadataArray) { |
|
|
|
|
AssertCallOk( |
|
|
|
|
grpcsharp_call_start_duplex_streaming(this, callback, metadataArray)); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
public |
|
|
|
|
void StartSendMessage(byte[] payload, CompletionCallbackDelegate callback) { |
|
|
|
|
AssertCallOk(grpcsharp_call_send_message( |
|
|
|
|
this, callback, payload, new UIntPtr((ulong)payload.Length))); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
public |
|
|
|
|
void StartSendCloseFromClient(CompletionCallbackDelegate callback) { |
|
|
|
|
AssertCallOk(grpcsharp_call_send_close_from_client(this, callback)); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
public |
|
|
|
|
void StartSendStatusFromServer(Status status, |
|
|
|
|
CompletionCallbackDelegate callback) { |
|
|
|
|
AssertCallOk(grpcsharp_call_send_status_from_server( |
|
|
|
|
this, callback, status.StatusCode, status.Detail)); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
public |
|
|
|
|
void StartReceiveMessage(CompletionCallbackDelegate callback) { |
|
|
|
|
AssertCallOk(grpcsharp_call_recv_message(this, callback)); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
public |
|
|
|
|
void StartServerSide(CompletionCallbackDelegate callback) { |
|
|
|
|
AssertCallOk(grpcsharp_call_start_serverside(this, callback)); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
public |
|
|
|
|
void Cancel() { AssertCallOk(grpcsharp_call_cancel(this)); } |
|
|
|
|
|
|
|
|
|
public |
|
|
|
|
void CancelWithStatus(Status status) { |
|
|
|
|
AssertCallOk(grpcsharp_call_cancel_with_status(this, status.StatusCode, |
|
|
|
|
status.Detail)); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
protected |
|
|
|
|
override bool ReleaseHandle() { |
|
|
|
|
grpcsharp_call_destroy(handle); |
|
|
|
|
return true; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
private |
|
|
|
|
static void AssertCallOk(GRPCCallError callError) { |
|
|
|
|
Trace.Assert(callError == GRPCCallError.GRPC_CALL_OK, |
|
|
|
|
"Status not GRPC_CALL_OK"); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
private |
|
|
|
|
static uint GetFlags(bool buffered) { |
|
|
|
|
return buffered ? 0 : GRPC_WRITE_BUFFER_HINT; |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
} |
Craig Tiller
10 years ago