Asynchroniczny licznik, który może być oczekiwana na

Question

Mam klasy połączenia, które ma kilka async metod, takich jak SendText, SendImage itp

Klasa połączenie ma metodę Disconnect, a gdy to się nazywa muszę być ostrożny aby nie zmieniać stanu wewnętrznego klasy przed zakończeniem wszystkich metod asynchronicznych.

Wierzę, że dobrym sposobem, aby to osiągnąć, jest po prostu utrzymywanie sumy uruchomionych operacji, a kiedy chcę się rozłączyć, mogę po prostu ustawić Disconnecting = true, a następnie poczekać, aż liczba osiągnie 0

myślę o czymś na wzór tej

class ReferenceCounter 
{ 
    void Increment(); 

    void Decrement(); 

    async Task WaitForCounterToReachZero(); 
} 

Następnie po uruchomieniu operacji asynchronicznych mogłem zrobić

refCounter.Increment(); 

kiedy kończy

refCounter.Decrement(); 

i wewnątrz metody Disconnect

disconnecting = true; 
taskCancellationSource.Cancel(); 
await refCounter.WaitForCounterToReachZero(); 
Cleanup(); 

Czy istnieje .NET zbudowany w klasach, jak to?

A co ważniejsze dla mnie, czy jest lepszy sposób na zrobienie tego?

Jeśli był kod synchroniczny byłoby tak proste, jak

lock (thisLock) 
{ 
    while (counter > 0) 
     Monitor.Wait(thisLock); 
} 

Właśnie się zbudowany w klasie CountdownEvent który robi to samo, ale nie ma metody asynchronicznej Wait ani nie ma żadnych wydarzeń, więc musiałbym zablokować.

Answer 1

Cóż, zakładając, że nigdy nie zwiększamy ponownie po zrobić to 0, można zrobić coś takiego:

public class Latch 
{ 
    private int count = 0; 
    private readonly TaskCompletionSource<object> tcs = 
     new TaskCompletionSource<object>(); 

    public void Increment() 
    { 
     Interlocked.Increment(ref count); 
    } 

    public void Decrement() 
    { 
     if (Interlocked.Decrement(ref count) == 0) 
     { 
      tcs.TrySetValue(null); 
     } 
    } 

    public Task Task { get { return tcs.Task; } } 
} 

Następnie można oczekiwać someLatch.Task. Alternatywnie, można zrobić zatrzask sama awaitable:

public TaskAwaiter GetAwaiter() 
{ 
    return tcs.Task.GetAwaiter(); 
} 

powinieneś rozważyć, w jaki sposób chcą zabezpieczyć się przed „liczenia wzrasta po dostaniu się do 0” aspekcie thuogh - zastanów się, co chcesz chcesz to zrobić . (W powyższym kodzie, po ustawieniu wartości TCS, dalsze oczekiwanie zakończy się natychmiast.)