cs-codex-dist-tests/CodexPlugin/CodexStarter.cs

using DistTestCore;
using KubernetesWorkflow;
using Logging;

namespace CodexPlugin
{
    public class CodexStarter
    {
        private readonly IPluginActions pluginActions;

        //public CodexStarter(TestLifecycle lifecycle)
        //    : base(lifecycle)
        //{
        //}

        public CodexStarter(IPluginActions pluginActions)
        {
            this.pluginActions = pluginActions;
        }

        public RunningContainers[] BringOnline(CodexSetup codexSetup)
        {
            //LogSeparator();
            //LogStart($"Starting {codexSetup.Describe()}...");
            //var gethStartResult = lifecycle.GethStarter.BringOnlineMarketplaceFor(codexSetup);

            var startupConfig = CreateStartupConfig(/*gethStartResult,*/ codexSetup);

            return StartCodexContainers(startupConfig, codexSetup.NumberOfNodes, codexSetup.Location);

            //var metricAccessFactory = CollectMetrics(codexSetup, containers);

            //var codexNodeFactory = new CodexNodeFactory(lifecycle, metricAccessFactory, gethStartResult.MarketplaceAccessFactory);

            //var group = CreateCodexGroup(codexSetup, containers, codexNodeFactory);
            //lifecycle.SetCodexVersion(group.Version);

            //var nl = Environment.NewLine;
            //var podInfos = string.Join(nl, containers.Containers().Select(c => $"Container: '{c.Name}' runs at '{c.Pod.PodInfo.K8SNodeName}'={c.Pod.PodInfo.Ip}"));
            //LogEnd($"Started {codexSetup.NumberOfNodes} nodes " +
            //    $"of image '{containers.Containers().First().Recipe.Image}' " +
            //    $"and version '{group.Version}'{nl}" +
            //    podInfos);
            //LogSeparator();

            //return group;
        }

        public ICodexNodeGroup WrapCodexContainers(RunningContainers[] containers)
        {
            //var metricAccessFactory = CollectMetrics(codexSetup, containers);

            var codexNodeFactory = new CodexNodeFactory();// (lifecycle, metricAccessFactory, gethStartResult.MarketplaceAccessFactory);

            return CreateCodexGroup(/*codexSetup,*/ containers, codexNodeFactory);
            //lifecycle.SetCodexVersion(group.Version);

            //var nl = Environment.NewLine;
            //var podInfos = string.Join(nl, containers.Containers().Select(c => $"Container: '{c.Name}' runs at '{c.Pod.PodInfo.K8SNodeName}'={c.Pod.PodInfo.Ip}"));
            //LogEnd($"Started {codexSetup.NumberOfNodes} nodes " +
            //    $"of image '{containers.Containers().First().Recipe.Image}' " +
            //    $"and version '{group.Version}'{nl}" +
            //    podInfos);
            //LogSeparator();

            //return group;
        }

        public void BringOffline(CodexNodeGroup group)
        {
            //LogStart($"Stopping {group.Describe()}...");
            //var workflow = CreateWorkflow();
            //foreach (var c in group.Containers)
            //{
            //    StopCrashWatcher(c);
            //    workflow.Stop(c);
            //}
            //LogEnd("Stopped.");
        }

        public void DeleteAllResources()
        {
            //var workflow = CreateWorkflow();
            //workflow.DeleteTestResources();
        }

        public void DownloadLog(RunningContainer container, ILogHandler logHandler, int? tailLines)
        {
            //var workflow = CreateWorkflow();
            //workflow.DownloadContainerLog(container, logHandler, tailLines);
        }

        //private IMetricsAccessFactory CollectMetrics(CodexSetup codexSetup, RunningContainers[] containers)
        //{
        //    if (codexSetup.MetricsMode == MetricsMode.None) return new MetricsUnavailableAccessFactory();

        //    var runningContainers = lifecycle.PrometheusStarter.CollectMetricsFor(containers);

        //    if (codexSetup.MetricsMode == MetricsMode.Dashboard)
        //    {
        //        lifecycle.GrafanaStarter.StartDashboard(runningContainers.Containers.First(), codexSetup);
        //    }

        //    return new CodexNodeMetricsAccessFactory(lifecycle, runningContainers);
        //}

        private StartupConfig CreateStartupConfig(/*GethStartResult gethStartResult, */ CodexSetup codexSetup)
        {
            var startupConfig = new StartupConfig();
            startupConfig.NameOverride = codexSetup.NameOverride;
            startupConfig.CreateCrashWatcher = true;
            startupConfig.Add(codexSetup);
            //startupConfig.Add(gethStartResult);
            return startupConfig;
        }

        private RunningContainers[] StartCodexContainers(StartupConfig startupConfig, int numberOfNodes, Location location)
        {
            var result = new List<RunningContainers>();
            var recipe = new CodexContainerRecipe();
            for (var i = 0; i < numberOfNodes; i++)
            {
                var workflow = pluginActions.CreateWorkflow();
                result.Add(workflow.Start(1, location, recipe, startupConfig));
            }
            return result.ToArray();
        }

        private CodexNodeGroup CreateCodexGroup(/*CodexSetup codexSetup, */RunningContainers[] runningContainers, CodexNodeFactory codexNodeFactory)
        {
            var group = new CodexNodeGroup(pluginActions.GetLog(), pluginActions.GetTimeSet(), /*lifecycle, codexSetup,*/ runningContainers, codexNodeFactory);

            try
            {
                Stopwatch.Measure(pluginActions.GetLog(), "EnsureOnline", group.EnsureOnline, debug: true);
            }
            catch
            {
                CodexNodesNotOnline(runningContainers);
                throw;
            }

            return group;
        }

        private void CodexNodesNotOnline(RunningContainers[] runningContainers)
        {
            pluginActions.GetLog().Log("Codex nodes failed to start");
            // todo:
            //foreach (var container in runningContainers.Containers()) lifecycle.DownloadLog(container);
        }

        //private StartupWorkflow CreateWorkflow()
        //{
        //    return lifecycle.WorkflowCreator.CreateWorkflow();
        //}

        //private void LogSeparator()
        //{
        //    Log("----------------------------------------------------------------------------");
        //}

        //private void StopCrashWatcher(RunningContainers containers)
        //{
        //    foreach (var c in containers.Containers)
        //    {
        //        c.CrashWatcher?.Stop();
        //    }
        //}
    }
}