EventConsumer.cs

Go to the documentation of this file.

using System;
using System.Collections.Generic;
using System.Threading;
using System.Threading.Tasks;
using Tgstation.Server.Host.Components.StaticFiles;
using Tgstation.Server.Host.Components.Watchdog;

namespace Tgstation.Server.Host.Components
{
    sealed class EventConsumer : IEventConsumer
    {
        readonly IConfiguration configuration;

        IWatchdog watchdog;

        public EventConsumer(IConfiguration configuration)
        {
            this.configuration = configuration ?? throw new ArgumentNullException(nameof(configuration));
        }

        public async Task<bool> HandleEvent(EventType eventType, IEnumerable<string> parameters, CancellationToken cancellationToken)
        {
            if (watchdog == null)
                throw new InvalidOperationException("EventConsumer used without watchdog set!");

            if (!await configuration.HandleEvent(eventType, parameters, cancellationToken).ConfigureAwait(false))
                return false;
            return await watchdog.HandleEvent(eventType, parameters, cancellationToken).ConfigureAwait(false);
        }

        public void SetWatchdog(IWatchdog watchdog)
        {
            if (watchdog == null)
                throw new ArgumentNullException(nameof(watchdog));
            if (this.watchdog != null)
                throw new InvalidOperationException("watchdog already set!");
            this.watchdog = watchdog;
        }
    }
}