ProgramShutdownTokenSource.cs

Go to the documentation of this file.

using System;
using System.Threading;
 
namespace Tgstation.Server.Host.System
{
    sealed class ProgramShutdownTokenSource : IDisposable
    {
        readonly object tokenSourceAccessLock;
 
        CancellationTokenSource? cancellationTokenSource;
 
        public CancellationToken Token => cancellationTokenSource?.Token ?? default;
 
        public ProgramShutdownTokenSource()
        {
            tokenSourceAccessLock = new object();
            cancellationTokenSource = new CancellationTokenSource();
 
            AppDomain.CurrentDomain.ProcessExit += (sender, args) =>
            {
                lock (tokenSourceAccessLock)
                    cancellationTokenSource?.Cancel();
            };
 
            Console.CancelKeyPress += (sender, args) =>
            {
                args.Cancel = true;
                lock (tokenSourceAccessLock)
                    cancellationTokenSource?.Cancel();
            };
        }
 
        public void Dispose()
        {
            lock (tokenSourceAccessLock)
            {
                cancellationTokenSource?.Dispose();
                cancellationTokenSource = null;
            }
        }
    }
}