38 lines
1.1 KiB
C#
38 lines
1.1 KiB
C#
|
#pragma warning disable CS1591 // Missing XML comment for publicly visible type or member
|
||
|
|
|
||
|
|
using System;
|
||
|
|
using System.Threading;
|
||
|
|
using System.Threading.Tasks;
|
||
|
|
using RethinkDb.Driver.Net;
|
||
|
|
|
||
|
|
namespace RethinkDb.Driver.Utils
|
||
|
|
{
|
||
|
|
public class CancellableTask : TaskCompletionSource<Response>, IDisposable
|
||
|
|
{
|
||
|
|
private readonly CancellationToken cancelToken;
|
||
|
|
private CancellationTokenRegistration registration;
|
||
|
|
|
||
|
|
|
||
|
|
public CancellableTask(CancellationToken cancelToken)
|
||
|
|
{
|
||
|
|
this.cancelToken = cancelToken;
|
||
|
|
this.registration = this.cancelToken.Register(OnCancellation, false);
|
||
|
|
}
|
||
|
|
|
||
|
|
private void OnCancellation()
|
||
|
|
{
|
||
|
|
this.TrySetCanceled();
|
||
|
|
|
||
|
|
//if the user successfully signaled they want to
|
||
|
|
//cancel, remove the registration because
|
||
|
|
//the task status = canceled has been set.
|
||
|
|
//don't need the registration any more.
|
||
|
|
this.Dispose();
|
||
|
|
}
|
||
|
|
|
||
|
|
public void Dispose()
|
||
|
|
{
|
||
|
|
this.registration.Dispose();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|