87 lines
2.8 KiB
C#
87 lines
2.8 KiB
C#
namespace CodexDistTestCore
|
|
{
|
|
public interface IK8sManager
|
|
{
|
|
ICodexNodeGroup BringOnline(OfflineCodexNodes node);
|
|
IOfflineCodexNodes BringOffline(ICodexNodeGroup node);
|
|
}
|
|
|
|
public class K8sManager : IK8sManager
|
|
{
|
|
private readonly CodexGroupNumberSource codexGroupNumberSource = new CodexGroupNumberSource();
|
|
private readonly List<CodexNodeGroup> onlineCodexNodes = new List<CodexNodeGroup>();
|
|
private readonly KnownK8sPods knownPods = new KnownK8sPods();
|
|
private readonly TestLog log;
|
|
private readonly IFileManager fileManager;
|
|
|
|
public K8sManager(TestLog log, IFileManager fileManager)
|
|
{
|
|
this.log = log;
|
|
this.fileManager = fileManager;
|
|
}
|
|
|
|
public ICodexNodeGroup BringOnline(OfflineCodexNodes offline)
|
|
{
|
|
var online = CreateOnlineCodexNodes(offline);
|
|
|
|
K8s(k => k.BringOnline(online, offline));
|
|
|
|
log.Log($"{online.Describe()} online.");
|
|
|
|
return online;
|
|
}
|
|
|
|
public IOfflineCodexNodes BringOffline(ICodexNodeGroup node)
|
|
{
|
|
var online = GetAndRemoveActiveNodeFor(node);
|
|
|
|
K8s(k => k.BringOffline(online));
|
|
|
|
log.Log($"{online.Describe()} offline.");
|
|
|
|
return online.Origin;
|
|
}
|
|
|
|
public void DeleteAllResources()
|
|
{
|
|
K8s(k => k.DeleteAllResources());
|
|
}
|
|
|
|
public void FetchAllPodsLogs(IPodLogsHandler logHandler)
|
|
{
|
|
K8s(k => k.FetchAllPodsLogs(onlineCodexNodes.ToArray(), logHandler));
|
|
}
|
|
|
|
private CodexNodeGroup CreateOnlineCodexNodes(OfflineCodexNodes offline)
|
|
{
|
|
var containers = CreateContainers(offline.NumberOfNodes);
|
|
var online = containers.Select(c => new OnlineCodexNode(log, fileManager, c)).ToArray();
|
|
var result = new CodexNodeGroup(codexGroupNumberSource.GetNextCodexNodeGroupNumber(), offline, this, online);
|
|
onlineCodexNodes.Add(result);
|
|
return result;
|
|
}
|
|
|
|
private CodexNodeContainer[] CreateContainers(int number)
|
|
{
|
|
var factory = new CodexNodeContainerFactory(codexGroupNumberSource);
|
|
var containers = new List<CodexNodeContainer>();
|
|
for (var i = 0; i < number; i++) containers.Add(factory.CreateNext());
|
|
return containers.ToArray();
|
|
}
|
|
|
|
private CodexNodeGroup GetAndRemoveActiveNodeFor(ICodexNodeGroup node)
|
|
{
|
|
var n = (CodexNodeGroup)node;
|
|
onlineCodexNodes.Remove(n);
|
|
return n;
|
|
}
|
|
|
|
private void K8s(Action<K8sOperations> action)
|
|
{
|
|
var k8s = new K8sOperations(knownPods);
|
|
action(k8s);
|
|
k8s.Close();
|
|
}
|
|
}
|
|
}
|