IdentityCacheObject.cs

Go to the documentation of this file.

using System;
using System.Threading;
using System.Threading.Tasks;
 
using Tgstation.Server.Host.Utils;
 
namespace Tgstation.Server.Host.Security
{
    sealed class IdentityCacheObject : IDisposable
    {
        public ISystemIdentity SystemIdentity { get; }
 
        readonly CancellationTokenSource cancellationTokenSource;
 
        readonly Task task;
 
        public IdentityCacheObject(ISystemIdentity systemIdentity, IAsyncDelayer asyncDelayer, Action onExpiry, DateTimeOffset expiry)
        {
            SystemIdentity = systemIdentity ?? throw new ArgumentNullException(nameof(systemIdentity));
 
            ArgumentNullException.ThrowIfNull(asyncDelayer);
 
            ArgumentNullException.ThrowIfNull(onExpiry);
            var now = DateTimeOffset.UtcNow;
            if (expiry < now)
                throw new ArgumentOutOfRangeException(nameof(expiry), expiry, "expiry must be greater than DateTimeOffset.UtcNow!");
 
            cancellationTokenSource = new CancellationTokenSource();
 
            async Task DisposeOnExipiry(CancellationToken cancellationToken)
            {
                using (SystemIdentity)
                    try
                    {
                        await asyncDelayer.Delay(expiry - now, cancellationToken);
                    }
                    finally
                    {
                        onExpiry();
                    }
            }
 
            task = DisposeOnExipiry(cancellationTokenSource.Token);
        }
 
        public void Dispose()
        {
            cancellationTokenSource.Cancel();
            try
            {
                task.GetAwaiter().GetResult();
            }
            catch (OperationCanceledException)
            {
            }
 
            cancellationTokenSource.Dispose();
        }
    }
}