ServerService.cs

Go to the documentation of this file.

using Microsoft.Extensions.Logging;
using Microsoft.Extensions.Logging.EventLog;
using System;
using System.Diagnostics;
using System.Globalization;
using System.ServiceProcess;
using System.Threading;
using System.Threading.Tasks;
using Tgstation.Server.Host.Watchdog;

namespace Tgstation.Server.Host.Service
{
    sealed class ServerService : ServiceBase
    {
        public const string Name = "tgstation-server-4";

        readonly IWatchdog watchdog;

        Task watchdogTask;

        CancellationTokenSource cancellationTokenSource;

        public ServerService(IWatchdogFactory watchdogFactory, ILoggerFactory loggerFactory, LogLevel minumumLogLevel)
        {
            if (watchdogFactory == null)
                throw new ArgumentNullException(nameof(watchdogFactory));
            if (loggerFactory == null)
                throw new ArgumentNullException(nameof(loggerFactory));

#pragma warning disable CS0618 // Type or member is obsolete
            loggerFactory.AddEventLog(new EventLogSettings
            {
                LogName = EventLog.Log,
                MachineName = EventLog.MachineName,
                SourceName = EventLog.Source,
                Filter = (message, logLevel) => logLevel >= minumumLogLevel
            });
#pragma warning restore CS0618 // Type or member is obsolete

            ServiceName = Name;
            watchdog = watchdogFactory.CreateWatchdog(loggerFactory);
        }

        async Task RunWatchdog(string[] args, CancellationToken cancellationToken)
        {
            await watchdog.RunAsync(false, args, cancellationTokenSource.Token).ConfigureAwait(false);

            void StopServiceAsync()
            {
                try
                {
                    Task.Run(Stop, cancellationToken);
                }
                catch (OperationCanceledException) { }
                catch (Exception e)
                {
                    EventLog.WriteEntry(String.Format(CultureInfo.InvariantCulture, "Error stopping service! Exception: {0}", e));
                }
            }

            StopServiceAsync();
        }

        protected override void Dispose(bool disposing)
        {
            cancellationTokenSource?.Dispose();
            base.Dispose(disposing);
        }

        protected override void OnStart(string[] args)
        {
            cancellationTokenSource?.Dispose();
            cancellationTokenSource = new CancellationTokenSource();
            watchdogTask = RunWatchdog(args, cancellationTokenSource.Token);
        }

        protected override void OnStop()
        {
            cancellationTokenSource.Cancel();
            watchdogTask.GetAwaiter().GetResult();
        }
    }
}