Index: third_party/grpc/src/csharp/Grpc.Core/Internal/ServerRequestStream.cs |
diff --git a/third_party/protobuf/csharp/src/Google.Protobuf.JsonDump/Program.cs b/third_party/grpc/src/csharp/Grpc.Core/Internal/ServerRequestStream.cs |
similarity index 53% |
copy from third_party/protobuf/csharp/src/Google.Protobuf.JsonDump/Program.cs |
copy to third_party/grpc/src/csharp/Grpc.Core/Internal/ServerRequestStream.cs |
index e8a6073e2bcb6b6f47dafb1f67f31de9b4a8671a..e7be82c31857e44e3830fc52a647cfbb537fd50c 100644 |
--- a/third_party/protobuf/csharp/src/Google.Protobuf.JsonDump/Program.cs |
+++ b/third_party/grpc/src/csharp/Grpc.Core/Internal/ServerRequestStream.cs |
@@ -1,7 +1,7 @@ |
#region Copyright notice and license |
-// Protocol Buffers - Google's data interchange format |
-// Copyright 2008 Google Inc. All rights reserved. |
-// https://developers.google.com/protocol-buffers/ |
+ |
+// 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 |
@@ -28,45 +28,56 @@ |
// 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.IO; |
+using System.Collections.Generic; |
+using System.Threading; |
+using System.Threading.Tasks; |
-namespace Google.Protobuf.ProtoDump |
+namespace Grpc.Core.Internal |
{ |
- /// <summary> |
- /// Small utility to load a binary message and dump it in JSON format. |
- /// </summary> |
- internal class Program |
+ internal class ServerRequestStream<TRequest, TResponse> : IAsyncStreamReader<TRequest> |
+ where TRequest : class |
+ where TResponse : class |
{ |
- private static int Main(string[] args) |
+ readonly AsyncCallServer<TRequest, TResponse> call; |
+ TRequest current; |
+ |
+ public ServerRequestStream(AsyncCallServer<TRequest, TResponse> call) |
{ |
- if (args.Length != 2) |
- { |
- Console.Error.WriteLine("Usage: Google.Protobuf.JsonDump <descriptor type name> <input data>"); |
- Console.Error.WriteLine("The descriptor type name is the fully-qualified message name,"); |
- Console.Error.WriteLine("including assembly e.g. ProjectNamespace.Message,Company.Project"); |
- return 1; |
- } |
- Type type = Type.GetType(args[0]); |
- if (type == null) |
- { |
- Console.Error.WriteLine("Unable to load type {0}.", args[0]); |
- return 1; |
- } |
- if (!typeof(IMessage).IsAssignableFrom(type)) |
+ this.call = call; |
+ } |
+ |
+ public TRequest Current |
+ { |
+ get |
{ |
- Console.Error.WriteLine("Type {0} doesn't implement IMessage.", args[0]); |
- return 1; |
+ if (current == null) |
+ { |
+ throw new InvalidOperationException("No current element is available."); |
+ } |
+ return current; |
} |
- IMessage message = (IMessage) Activator.CreateInstance(type); |
- using (var input = File.OpenRead(args[1])) |
+ } |
+ |
+ public async Task<bool> MoveNext(CancellationToken token) |
+ { |
+ if (token != CancellationToken.None) |
{ |
- message.MergeFrom(input); |
+ throw new InvalidOperationException("Cancellation of individual reads is not supported."); |
} |
- Console.WriteLine(message); |
- return 0; |
+ var taskSource = new AsyncCompletionTaskSource<TRequest>(); |
+ call.StartReadMessage(taskSource.CompletionDelegate); |
+ var result = await taskSource.Task.ConfigureAwait(false); |
+ this.current = result; |
+ return result != null; |
+ } |
+ |
+ public void Dispose() |
+ { |
+ // TODO(jtattermusch): implement the semantics of stream disposal. |
} |
} |
-} |
+} |