FabInfra-Mirror/libs.capnproto-dotnetcore_Runtime
1
0
Fork 0
mirror of https://github.com/FabInfra/capnproto-dotnetcore_Runtime.git synced 2025-03-12 14:51:41 +01:00
Code Issues Packages Projects Releases Wiki Activity

PendingAnswer should also use StrictlyOrderedAwaitTask<T>

Browse Source
This commit is contained in:
Christian Köllner 2020-04-11 21:21:17 +02:00
parent 19b36a1643
commit 197817a7d7
1 changed files with 79 additions and 84 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

81
Capnp.Net.Runtime/Rpc/PendingAnswer.cs
View File

@ -1,4 +1,5 @@
using System;
using Capnp.Util;
using System;
using System.Collections.Generic;
using System.Threading;
using System.Threading.Tasks;
@ -9,7 +10,7 @@ namespace Capnp.Rpc
{
readonly CancellationTokenSource? _cts;
readonly TaskCompletionSource<AnswerOrCounterquestion> _cancelCompleter;
readonly Task<AnswerOrCounterquestion> _answerTask;
readonly StrictlyOrderedAwaitTask<AnswerOrCounterquestion> _answerTask;
public PendingAnswer(Task<AnswerOrCounterquestion> callTask, CancellationTokenSource? cts)
{
@ -23,19 +24,16 @@ namespace Capnp.Rpc
_cts = cts;
_cancelCompleter = new TaskCompletionSource<AnswerOrCounterquestion>();
_answerTask = CancelableAwaitWhenReady();
_answerTask = CancelableAwaitWhenReady().EnforceAwaitOrder();
Chain(async t =>
TakeCapTableOwnership();
}
async void TakeCapTableOwnership()
{
var aorcq = default(AnswerOrCounterquestion);
try
{
aorcq = await t;
}
catch
{
}
var aorcq = await _answerTask;
if (aorcq.Answer != null)
{
@ -43,11 +41,35 @@ namespace Capnp.Rpc
{
foreach (var cap in aorcq.Answer.Caps)
{
cap?.AddRef();
cap.AddRef();
}
}
}
});
}
catch
{
}
}
async void ReleaseCapTableOwnership()
{
try
{
var aorcq = await _answerTask;
if (aorcq.Answer != null)
{
if (aorcq.Answer.Caps != null)
{
foreach (var cap in aorcq.Answer.Caps)
{
cap?.Release();
}
}
}
}
catch
{
}
}
public CancellationToken CancellationToken => _cts?.Token ?? CancellationToken.None;
@ -60,18 +82,16 @@ namespace Capnp.Rpc
_cancelCompleter.SetCanceled();
}
public void Chain(Action<Task<AnswerOrCounterquestion>> func)
public void Chain(Action<StrictlyOrderedAwaitTask<AnswerOrCounterquestion>> func)
{
func(_answerTask);
}
public void Chain(PromisedAnswer.READER rd, Action<Task<Proxy>> func)
{
Chain(t =>
{
async Task<Proxy> EvaluateProxy()
{
var aorcq = await t;
var aorcq = await _answerTask;
if (aorcq.Answer != null)
{
@ -128,37 +148,12 @@ namespace Capnp.Rpc
}
func(EvaluateProxy());
});
}
public void Dispose()
{
_cts?.Dispose();
Chain(async t =>
{
AnswerOrCounterquestion aorcq;
try
{
aorcq = await t;
}
catch
{
return;
}
if (aorcq.Answer != null)
{
if (aorcq.Answer.Caps != null)
{
foreach (var cap in aorcq.Answer.Caps)
{
cap?.Release();
}
}
}
});
ReleaseCapTableOwnership();
}
}
}