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.Events
{
    sealed class EventConsumer : IEventConsumer
    {
        readonly IConfiguration configuration;
 
        IWatchdog? watchdog;
 
        public EventConsumer(IConfiguration configuration)
        {
            this.configuration = configuration ?? throw new ArgumentNullException(nameof(configuration));
        }
 
        public ValueTask? HandleCustomEvent(string eventName, IEnumerable<string?> parameters, CancellationToken cancellationToken)
            => configuration.HandleCustomEvent(eventName, parameters, cancellationToken);
 
        public async ValueTask HandleEvent(EventType eventType, IEnumerable<string?> parameters, bool deploymentPipeline, CancellationToken cancellationToken)
        {
            ArgumentNullException.ThrowIfNull(parameters);
 
            if (watchdog == null)
                throw new InvalidOperationException("EventConsumer used without watchdog set!");
 
            var scriptTask = configuration.HandleEvent(eventType, parameters, deploymentPipeline, cancellationToken);
            await watchdog.HandleEvent(eventType, parameters, deploymentPipeline, cancellationToken);
            await scriptTask;
        }
 
        public void SetWatchdog(IWatchdog watchdog)
        {
            ArgumentNullException.ThrowIfNull(watchdog);
            var oldWatchdog = Interlocked.CompareExchange(ref this.watchdog, watchdog, null);
            if (oldWatchdog != null)
                throw new InvalidOperationException("watchdog already set!");
        }
    }
}