Adds waiting for debug-mode start message

This commit is contained in:
Ben 2024-05-21 12:58:17 +02:00
parent d6f7e225be
commit fa1b560a91
No known key found for this signature in database
GPG Key ID: 541B9D8C9F1426A1
1 changed files with 44 additions and 1 deletions

View File

@ -1,11 +1,13 @@
using Core; using Core;
using KubernetesWorkflow; using KubernetesWorkflow;
using KubernetesWorkflow.Types; using KubernetesWorkflow.Types;
using Utils;
namespace CodexDiscordBotPlugin namespace CodexDiscordBotPlugin
{ {
public class CodexDiscordBotPlugin : IProjectPlugin, IHasLogPrefix, IHasMetadata public class CodexDiscordBotPlugin : IProjectPlugin, IHasLogPrefix, IHasMetadata
{ {
private const string ExpectedStartupMessage = "Debug option is set. Discord connection disabled!";
private readonly IPluginTools tools; private readonly IPluginTools tools;
public CodexDiscordBotPlugin(IPluginTools tools) public CodexDiscordBotPlugin(IPluginTools tools)
@ -46,7 +48,9 @@ namespace CodexDiscordBotPlugin
var startupConfig = new StartupConfig(); var startupConfig = new StartupConfig();
startupConfig.NameOverride = config.Name; startupConfig.NameOverride = config.Name;
startupConfig.Add(config); startupConfig.Add(config);
return workflow.Start(1, new DiscordBotContainerRecipe(), startupConfig).WaitForOnline(); var pod = workflow.Start(1, new DiscordBotContainerRecipe(), startupConfig).WaitForOnline();
WaitForStartupMessage(workflow, pod);
return pod;
} }
private RunningPod StartRewarderContainer(IStartupWorkflow workflow, RewarderBotStartupConfig config) private RunningPod StartRewarderContainer(IStartupWorkflow workflow, RewarderBotStartupConfig config)
@ -55,5 +59,44 @@ namespace CodexDiscordBotPlugin
startupConfig.Add(config); startupConfig.Add(config);
return workflow.Start(1, new RewarderBotContainerRecipe(), startupConfig).WaitForOnline(); return workflow.Start(1, new RewarderBotContainerRecipe(), startupConfig).WaitForOnline();
} }
private void WaitForStartupMessage(IStartupWorkflow workflow, RunningPod pod)
{
var finder = new LogLineFinder(ExpectedStartupMessage, workflow);
Time.WaitUntil(() =>
{
finder.FindLine(pod);
return finder.Found;
}, nameof(WaitForStartupMessage));
}
public class LogLineFinder : LogHandler
{
private readonly string message;
private readonly IStartupWorkflow workflow;
public LogLineFinder(string message, IStartupWorkflow workflow)
{
this.message = message;
this.workflow = workflow;
}
public void FindLine(RunningPod pod)
{
Found = false;
foreach (var c in pod.Containers)
{
workflow.DownloadContainerLog(c, this);
if (Found) return;
}
}
public bool Found { get; private set; }
protected override void ProcessLine(string line)
{
if (!Found && line.Contains(message)) Found = true;
}
}
} }
} }