mirror of https://github.com/grpc/grpc.git
Guantao Liu
5 years ago
commit
81b4bc02db
18 changed files with 526 additions and 170 deletions
@ -0,0 +1,98 @@ |
||||
#region Copyright notice and license |
||||
|
||||
// Copyright 2019 The gRPC Authors |
||||
// |
||||
// Licensed under the Apache License, Version 2.0 (the "License"); |
||||
// you may not use this file except in compliance with the License. |
||||
// You may obtain a copy of the License at |
||||
// |
||||
// http://www.apache.org/licenses/LICENSE-2.0 |
||||
// |
||||
// Unless required by applicable law or agreed to in writing, software |
||||
// distributed under the License is distributed on an "AS IS" BASIS, |
||||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
// See the License for the specific language governing permissions and |
||||
// limitations under the License. |
||||
|
||||
#endregion |
||||
|
||||
|
||||
using System; |
||||
using System.Threading.Tasks; |
||||
|
||||
namespace Grpc.Core |
||||
{ |
||||
/// <summary> |
||||
/// Provides an abstraction over the callback providers |
||||
/// used by AsyncUnaryCall, AsyncDuplexStreamingCall, etc |
||||
/// </summary> |
||||
internal struct AsyncCallState |
||||
{ |
||||
readonly object responseHeadersAsync; // Task<Metadata> or Func<object, Task<Metadata>> |
||||
readonly object getStatusFunc; // Func<Status> or Func<object, Status> |
||||
readonly object getTrailersFunc; // Func<Metadata> or Func<object, Metadata> |
||||
readonly object disposeAction; // Action or Action<object> |
||||
readonly object callbackState; // arg0 for the callbacks above, if needed |
||||
|
||||
internal AsyncCallState( |
||||
Func<object, Task<Metadata>> responseHeadersAsync, |
||||
Func<object, Status> getStatusFunc, |
||||
Func<object, Metadata> getTrailersFunc, |
||||
Action<object> disposeAction, |
||||
object callbackState) |
||||
{ |
||||
this.responseHeadersAsync = responseHeadersAsync; |
||||
this.getStatusFunc = getStatusFunc; |
||||
this.getTrailersFunc = getTrailersFunc; |
||||
this.disposeAction = disposeAction; |
||||
this.callbackState = callbackState; |
||||
} |
||||
|
||||
internal AsyncCallState( |
||||
Task<Metadata> responseHeadersAsync, |
||||
Func<Status> getStatusFunc, |
||||
Func<Metadata> getTrailersFunc, |
||||
Action disposeAction) |
||||
{ |
||||
this.responseHeadersAsync = responseHeadersAsync; |
||||
this.getStatusFunc = getStatusFunc; |
||||
this.getTrailersFunc = getTrailersFunc; |
||||
this.disposeAction = disposeAction; |
||||
this.callbackState = null; |
||||
} |
||||
|
||||
internal Task<Metadata> ResponseHeadersAsync() |
||||
{ |
||||
var withState = responseHeadersAsync as Func<object, Task<Metadata>>; |
||||
return withState != null ? withState(callbackState) |
||||
: (Task<Metadata>)responseHeadersAsync; |
||||
} |
||||
|
||||
internal Status GetStatus() |
||||
{ |
||||
var withState = getStatusFunc as Func<object, Status>; |
||||
return withState != null ? withState(callbackState) |
||||
: ((Func<Status>)getStatusFunc)(); |
||||
} |
||||
|
||||
internal Metadata GetTrailers() |
||||
{ |
||||
var withState = getTrailersFunc as Func<object, Metadata>; |
||||
return withState != null ? withState(callbackState) |
||||
: ((Func<Metadata>)getTrailersFunc)(); |
||||
} |
||||
|
||||
internal void Dispose() |
||||
{ |
||||
var withState = disposeAction as Action<object>; |
||||
if (withState != null) |
||||
{ |
||||
withState(callbackState); |
||||
} |
||||
else |
||||
{ |
||||
((Action)disposeAction)(); |
||||
} |
||||
} |
||||
} |
||||
} |
||||
@ -0,0 +1,82 @@ |
||||
#region Copyright notice and license |
||||
|
||||
// Copyright 2019 The gRPC Authors |
||||
// |
||||
// Licensed under the Apache License, Version 2.0 (the "License"); |
||||
// you may not use this file except in compliance with the License. |
||||
// You may obtain a copy of the License at |
||||
// |
||||
// http://www.apache.org/licenses/LICENSE-2.0 |
||||
// |
||||
// Unless required by applicable law or agreed to in writing, software |
||||
// distributed under the License is distributed on an "AS IS" BASIS, |
||||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
// See the License for the specific language governing permissions and |
||||
// limitations under the License. |
||||
|
||||
#endregion |
||||
|
||||
using System.Threading.Tasks; |
||||
using NUnit.Framework; |
||||
|
||||
namespace Grpc.Core.Internal.Tests |
||||
{ |
||||
public class AsyncCallStateTest |
||||
{ |
||||
[Test] |
||||
public void Stateless() |
||||
{ |
||||
bool disposed = false; |
||||
Task<Metadata> responseHeaders = Task.FromResult(new Metadata()); |
||||
Metadata trailers = new Metadata(); |
||||
var state = new AsyncCallState(responseHeaders, () => new Status(StatusCode.DataLoss, "oops"), |
||||
() => trailers, () => disposed = true); |
||||
|
||||
Assert.AreSame(responseHeaders, state.ResponseHeadersAsync()); |
||||
|
||||
var status = state.GetStatus(); |
||||
Assert.AreEqual(StatusCode.DataLoss, status.StatusCode); |
||||
Assert.AreEqual("oops", status.Detail); |
||||
|
||||
Assert.AreSame(trailers, state.GetTrailers()); |
||||
|
||||
Assert.False(disposed); |
||||
state.Dispose(); |
||||
Assert.True(disposed); |
||||
} |
||||
|
||||
class State |
||||
{ |
||||
public bool disposed = false; |
||||
public Task<Metadata> responseHeaders = Task.FromResult(new Metadata()); |
||||
public Metadata trailers = new Metadata(); |
||||
public Status status = new Status(StatusCode.DataLoss, "oops"); |
||||
public void Dispose() { disposed = true; } |
||||
} |
||||
|
||||
[Test] |
||||
public void WithState() |
||||
{ |
||||
var callbackState = new State(); |
||||
|
||||
var state = new AsyncCallState( |
||||
obj => ((State)obj).responseHeaders, |
||||
obj => ((State)obj).status, |
||||
obj => ((State)obj).trailers, |
||||
obj => ((State)obj).Dispose(), |
||||
callbackState); |
||||
|
||||
Assert.AreSame(callbackState.responseHeaders, state.ResponseHeadersAsync()); |
||||
|
||||
var status = state.GetStatus(); |
||||
Assert.AreEqual(StatusCode.DataLoss, status.StatusCode); |
||||
Assert.AreEqual("oops", status.Detail); |
||||
|
||||
Assert.AreSame(callbackState.trailers, state.GetTrailers()); |
||||
|
||||
Assert.False(callbackState.disposed); |
||||
state.Dispose(); |
||||
Assert.True(callbackState.disposed); |
||||
} |
||||
} |
||||
} |
||||
Loading…
Reference in new issue