cs-codex-dist-tests/Tools/CodexNetDeployer/CodexNodeStarter.cs

138 lines
5.0 KiB
C#

using CodexContractsPlugin;
using CodexPlugin;
using Core;
using GethPlugin;
using Utils;
namespace CodexNetDeployer
{
public class CodexNodeStarter
{
private readonly Configuration config;
private readonly CoreInterface ci;
private readonly IGethNode gethNode;
private readonly ICodexContracts contracts;
private ICodexNode? bootstrapNode = null;
private int validatorsLeft;
public CodexNodeStarter(Configuration config, CoreInterface ci, IGethNode gethNode, ICodexContracts contracts, int numberOfValidators)
{
this.config = config;
this.ci = ci;
this.gethNode = gethNode;
this.contracts = contracts;
validatorsLeft = numberOfValidators;
}
public CodexNodeStartResult? Start(int i)
{
var name = GetCodexContainerName(i);
Console.Write($" - {i} ({name})\t");
Console.CursorLeft = 30;
ICodexNode? codexNode = null;
try
{
codexNode = ci.StartCodexNode(s =>
{
s.WithName(name);
s.WithLogLevel(config.CodexLogLevel, new CodexLogCustomTopics(config.Discv5LogLevel, config.Libp2pLogLevel));
s.WithStorageQuota(config.StorageQuota!.Value.MB());
if (config.ShouldMakeStorageAvailable)
{
s.EnableMarketplace(gethNode, contracts, m =>
{
m.WithInitial(100.Eth(), config.InitialTestTokens.TestTokens());
if (validatorsLeft > 0) m.AsValidator();
if (config.ShouldMakeStorageAvailable) m.AsStorageNode();
});
}
if (bootstrapNode != null) s.WithBootstrapNode(bootstrapNode);
if (config.MetricsEndpoints) s.EnableMetrics();
if (config.BlockTTL != Configuration.SecondsIn1Day) s.WithBlockTTL(TimeSpan.FromSeconds(config.BlockTTL));
if (config.BlockMI != Configuration.TenMinutes) s.WithBlockMaintenanceInterval(TimeSpan.FromSeconds(config.BlockMI));
if (config.BlockMN != 1000) s.WithBlockMaintenanceNumber(config.BlockMN);
if (config.IsPublicTestNet)
{
s.AsPublicTestNet(CreatePublicTestNetConfig(i));
}
});
var debugInfo = codexNode.GetDebugInfo();
if (!string.IsNullOrWhiteSpace(debugInfo.Spr))
{
Console.Write("Online\t");
if (config.ShouldMakeStorageAvailable)
{
var availability = new StorageAvailability(
totalSpace: config.StorageSell!.Value.MB(),
maxDuration: TimeSpan.FromSeconds(config.MaxDuration),
minPriceForTotalSpace: config.MinPrice.TestTokens(),
maxCollateral: config.MaxCollateral.TestTokens()
);
var response = codexNode.Marketplace.MakeStorageAvailable(availability);
if (!string.IsNullOrEmpty(response))
{
Console.Write("Storage available\t");
}
else throw new Exception("Failed to make storage available.");
}
Console.Write("OK" + Environment.NewLine);
validatorsLeft--;
if (bootstrapNode == null) bootstrapNode = codexNode;
return new CodexNodeStartResult(codexNode);
}
}
catch (Exception ex)
{
Console.WriteLine("Exception:" + ex.ToString());
}
Console.WriteLine("Unknown failure.");
if (codexNode != null)
{
Console.WriteLine("Downloading container log.");
ci.DownloadLog(codexNode);
}
return null;
}
private CodexTestNetConfig CreatePublicTestNetConfig(int i)
{
var discPort = config.PublicDiscPorts.Split(",")[i];
var listenPort = config.PublicListenPorts.Split(",")[i];
return new CodexTestNetConfig
{
PublicDiscoveryPort = Convert.ToInt32(discPort),
PublicListenPort = Convert.ToInt32(listenPort)
};
}
private string GetCodexContainerName(int i)
{
if (i == 0) return "BOOTSTRAP";
return "CODEX" + i;
}
}
public class CodexNodeStartResult
{
public CodexNodeStartResult(ICodexNode codexNode)
{
CodexNode = codexNode;
}
public ICodexNode CodexNode { get; }
}
}