1 #region Copyright notice and license 2 3 // Copyright 2015-2016 gRPC authors. 4 // 5 // Licensed under the Apache License, Version 2.0 (the "License"); 6 // you may not use this file except in compliance with the License. 7 // You may obtain a copy of the License at 8 // 9 // http://www.apache.org/licenses/LICENSE-2.0 10 // 11 // Unless required by applicable law or agreed to in writing, software 12 // distributed under the License is distributed on an "AS IS" BASIS, 13 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 14 // See the License for the specific language governing permissions and 15 // limitations under the License. 16 17 #endregion 18 19 using System; 20 using System.Collections.Generic; 21 using System.Threading; 22 using System.Threading.Tasks; 23 using Google.Protobuf; 24 using Grpc.Core; 25 using Grpc.Core.Utils; 26 27 namespace Grpc.Testing 28 { 29 /// <summary> 30 /// Implementation of BenchmarkService server 31 /// </summary> 32 public class BenchmarkServiceImpl : BenchmarkService.BenchmarkServiceBase 33 { BenchmarkServiceImpl()34 public BenchmarkServiceImpl() 35 { 36 } 37 UnaryCall(SimpleRequest request, ServerCallContext context)38 public override Task<SimpleResponse> UnaryCall(SimpleRequest request, ServerCallContext context) 39 { 40 var response = new SimpleResponse { Payload = CreateZerosPayload(request.ResponseSize) }; 41 return Task.FromResult(response); 42 } 43 StreamingCall(IAsyncStreamReader<SimpleRequest> requestStream, IServerStreamWriter<SimpleResponse> responseStream, ServerCallContext context)44 public override async Task StreamingCall(IAsyncStreamReader<SimpleRequest> requestStream, IServerStreamWriter<SimpleResponse> responseStream, ServerCallContext context) 45 { 46 await requestStream.ForEachAsync(async request => 47 { 48 var response = new SimpleResponse { Payload = CreateZerosPayload(request.ResponseSize) }; 49 await responseStream.WriteAsync(response); 50 }); 51 } 52 CreateZerosPayload(int size)53 private static Payload CreateZerosPayload(int size) 54 { 55 return new Payload { Body = ByteString.CopyFrom(new byte[size]) }; 56 } 57 } 58 } 59