private static async Task _ExecuteBackgroundServiceAsync(
Type serviceType,
GatewayServiceResolver gatewayServiceResolver,
CancellationToken ct)
{
while (!ct.IsCancellationRequested)
{
ServiceLog(serviceType, "Instantiating new instance");
var service = (IBackgroundService)Activator.CreateInstance(serviceType);
try
{
ServiceLog(serviceType, "Initializing");
await service.InitializeAsync(ct);
ServiceLog(serviceType, "Initialization success");
await service.ExecuteBackgroundAsync(gatewayServiceResolver, ct);
ct.ThrowIfCancellationRequested();
ServiceLog(serviceType, "Background service function returned without cancellation requested, this is not expected behavior");
}
catch (Exception ex)
{
ServiceLog(serviceType, "Failure");
ServiceLog(serviceType, ex.ToString());
}
finally
{
(service as IDisposable)?.Dispose();
}
await Task.Delay(TimeSpan.FromSeconds(10), ct);
}
}
private static async Task _ExecuteGatewayServiceAsync(
Type serviceType,
GatewayServiceResolver gatewayServiceResolver,
IEnumerable <ChannelWriter <IGatewayMessage> > messageChannelWriters,
CancellationToken ct)
{
while (!ct.IsCancellationRequested)
{
ServiceLog(serviceType, "Instantiating new instance");
var service = (IGatewayService)Activator.CreateInstance(serviceType);
try
{
ServiceLog(serviceType, "Initializing");
await service.InitializeAsync(ct);
ServiceLog(serviceType, "Initialization success");
gatewayServiceResolver.RegisterGateway(serviceType, service);
await foreach (var message in service.ExecuteGatewayAsync(ct))
{
foreach (var channelWriter in messageChannelWriters)
{
await channelWriter.WriteAsync(message);
}
}
}
catch (Exception ex)
{
ServiceLog(serviceType, "Failure");
ServiceLog(serviceType, ex.ToString());
}
finally
{
gatewayServiceResolver.UnregisterGateway(serviceType);
(service as IDisposable)?.Dispose();
}
await Task.Delay(TimeSpan.FromSeconds(60), ct);
}
}
public static async Task Main(string[] args)
{
var rootCommand = new RootCommand
{
new Option <DirectoryInfo>(
"--config-dir",
getDefaultValue: () => new DirectoryInfo("./Configurations"),
description: "A path to a directory containing applebot configuration files as defined in the default configurations directory"),
new Option <DirectoryInfo>(
"--runtime-data-dir",
getDefaultValue: () => new DirectoryInfo("./RuntimeData"),
description: "A path to a directory containing applebot runtime data directories")
};
rootCommand.Handler = CommandHandler.Create <DirectoryInfo, DirectoryInfo>((configDir, runtimeDataDir) =>
{
if (!configDir.Exists)
{
throw new Exception($"Specified configuration directory \"{configDir.FullName}\" does not exist");
}
ResourceResolver.ConfigurationsDirectory = configDir;
if (!runtimeDataDir.Exists)
{
throw new Exception($"Specified runtime data directory \"{runtimeDataDir.FullName}\" does not exist");
}
ResourceResolver.RuntimeDataDirectory = runtimeDataDir;
});
await rootCommand.InvokeAsync(args);
Console.WriteLine("Applebot! 🍎🍎🍎");
Console.WriteLine($"Using configrations directory: \"{ResourceResolver.ConfigurationsDirectory}\"");
Console.WriteLine($"Using runtime data directory: \"{ResourceResolver.RuntimeDataDirectory}\"");
// TODO: this is all pretty jank, ideally we will integrate this with the Microsoft.Extensions.DependencyInjection
// platform and that will give us a centeralised way to deal with logging and configuration. for now though since
// its not fully obvious how this is all gonna work we are doing everything ad-hoc, which is very little really
var programConfig = await ResourceResolver.LoadConfigurationAsync <Program, ProgramSettings>();
IEnumerable <Type> FindServiceTypes <T>(IEnumerable <string> serviceTypeNames)
where T : IApplebotService
{
var baseType = typeof(T);
foreach (var typeName in serviceTypeNames)
{
var foundType = Assembly.GetExecutingAssembly().GetType(typeName);
if (foundType == null)
{
throw new Exception($"Specified service type [{typeName}] is not defined in this assembly");
}
if (!foundType.IsAssignableTo(baseType))
{
throw new Exception($"Specified service type [{typeName}] is not derived from [{baseType.Name}] as required");
}
if (foundType.GetConstructor(Type.EmptyTypes) == null)
{
throw new Exception($"Specified service type [{typeName}] does not implement a public parameterless constructor");
}
yield return(foundType);
}
}
var gatewayServiceTypes = FindServiceTypes <IGatewayService>(programConfig.GatewayServices.Distinct()).ToArray();
var gatewayConsumerServiceTypes = FindServiceTypes <IGatewayConsumerService>(programConfig.GatewayConsumerServices.Distinct()).ToArray();
var backgroundServiceTypes = FindServiceTypes <IBackgroundService>(programConfig.BackgroundServices.Distinct()).ToArray();
if (gatewayServiceTypes.Length == 0)
{
Console.WriteLine("No gateway services defined in program configuration. Application exiting");
return;
}
using var tokenSource = new CancellationTokenSource();
var gatewayServiceResolver = new GatewayServiceResolver();
// we force a semantic here with channels that each gateway consumer service should only ever be handling a
// single gateway message at a time. unsure if this is really going to solve any problems
// but its makes things easier to think about with the amount of async stuff going on
var gatewayConsumerServices = gatewayConsumerServiceTypes
.Select(st =>
{
var channel = Channel.CreateUnbounded <IGatewayMessage>();
var gatewayConsumerTask = _ExecuteGatewayConsumerServiceAsync(st, channel.Reader, tokenSource.Token);
return(gatewayConsumerTask, channel);
})
.ToArray <(Task Task, Channel <IGatewayMessage> Channel)>();
var gatewayConsumerServiceMessageChannels = gatewayConsumerServices
.Select(gcs => gcs.Channel.Writer)
.ToArray();
var gatewayServiceTasks = gatewayServiceTypes
.Select(st => _ExecuteGatewayServiceAsync(
st,
gatewayServiceResolver,
gatewayConsumerServiceMessageChannels,
tokenSource.Token))
.ToArray();
var backgroundServiceTasks = backgroundServiceTypes
.Select(st => _ExecuteBackgroundServiceAsync(
st,
gatewayServiceResolver,
tokenSource.Token))
.ToArray();
var serviceTasks = Enumerable.Empty <Task>()
.Concat(gatewayServiceTasks)
.Concat(backgroundServiceTasks)
.Concat(gatewayConsumerServices.Select(gcs => gcs.Task))
.ToArray();
await Task.WhenAny(serviceTasks);
// we only get to this point if something in the application logic itself crashes, ideally the
// services are taken care of and restart on their own when something goes wrong
// this is just me trying to get some sort of info about what broke
Console.WriteLine("Applebot encountered a fatal error and is terminating");
foreach (var task in serviceTasks)
{
if (task.IsFaulted && task.Exception is not null)
{
Console.WriteLine(task.Exception.ToString());
}
}
// currently the cancellation of the application will not work correctly because the twitch
// stream reader will not listen to cancellation requests, as such we just terminate the applicaiton
// but ideally we can fix that and get some better shutdown / error reporting logic in place
//tokenSource.Cancel();
//await Task.WhenAll(serviceTasks);
}
