public static LoadTestOptions Parse(string[] args)
{
var builder = new ConfigurationBuilder();
builder.AddCommandLine(args);
var config = builder.Build();
var configFileName = config.GetValue <string>(ConfigurationFileKey, null);
if (configFileName != null)
{
builder = new ConfigurationBuilder();
builder.AddJsonFile(configFileName, false);
builder.AddCommandLine(args);
}
config = builder.Build();
var options = new LoadTestOptions();
config.Bind(options);
return(options);
}
public static async Task Run(LoadTestOptions options)
{
if (options == null)
{
throw new ArgumentNullException(nameof(options));
}
Exception finalException = null;
var loggerConfiguration = new LoggerConfiguration();
if (options.LogToConsole)
{
loggerConfiguration.WriteTo
.LiterateConsole(options.MinimumConsoleLogLevel);
}
if (!string.IsNullOrEmpty(options.LogFileName))
{
var formatter = new JsonFormatter(renderMessage: true, closingDelimiter: $",{Environment.NewLine}");
var logFilePath = Environment.ExpandEnvironmentVariables(options.LogFileName);
loggerConfiguration.WriteTo.File(formatter, logFilePath, buffered: true);
}
if (!string.IsNullOrEmpty(options.ElasticSearchUri))
{
var elasticOptions = new ElasticsearchSinkOptions(new Uri(options.ElasticSearchUri))
{
AutoRegisterTemplate = true,
BufferBaseFilename = "logs/elastic-buffer",
BatchPostingLimit = 100,
BufferLogShippingInterval = options.ElasticSearchFlushInterval
};
loggerConfiguration.WriteTo.Elasticsearch(elasticOptions);
}
Log.Logger = loggerConfiguration
.MinimumLevel.Is(options.MinimumLogLevel)
.Enrich.FromLogContext()
.CreateLogger()
.ForContext("ApplicationId", "Conreign.LoadTest")
.ForContext("InstanceId", options.InstanceId);
SelfLog.Enable(msg => Trace.WriteLine(msg));
var logger = Log.Logger;
try
{
var botOptions = options.BotOptions;
logger.Information("Initialized with the following configuration: {@Configuration}", options);
ServicePointManager.DefaultConnectionLimit = botOptions.RoomsCount * botOptions.BotsPerRoomCount * 2;
var signalrOptions = new SignalRClientOptions(options.ConnectionUri);
var client = new SignalRClient(signalrOptions);
var factory = new LoadTestBotFactory(botOptions);
var farm = new BotFarm(options.InstanceId, client, factory, new BotFarmOptions
{
GracefulStopPeriod = options.GracefulStopPeriod
});
var cts = new CancellationTokenSource();
cts.CancelAfter(options.Timeout);
await farm.Run(cts.Token);
logger.Information("Load test complete.");
}
catch (Exception ex)
{
logger.Fatal(ex, "Load test failed: {ErrorMessage}.", ex.Message);
finalException = ex;
}
try
{
if (!string.IsNullOrEmpty(options.LogFileName) && options.ZipLogFile)
{
logger.Information("Waiting a bit for Serilog to flush.");
// ReSharper disable once MethodSupportsCancellation
// HACK: magic delay here, how to explicitely flush Serilog without closing?
await Task.Delay(TimeSpan.FromSeconds(5));
var zipPath = ArchiveLogFile(options.LogFileName);
logger.Information("Zipped log file {LogFilePath} to {ZippedLogFilePath}", options.LogFileName,
zipPath);
}
}
catch (Exception ex)
{
logger.Error(ex, "Failed to archive log file: {ErrorMessage}", ex.Message);
if (finalException == null)
{
throw;
}
throw new AggregateException(ex, finalException);
}
finally
{
try
{
var cts = new CancellationTokenSource();
cts.CancelAfter(options.LogFlushPeriod);
await Task.Run(() => Log.CloseAndFlush(), cts.Token);
}
catch (TaskCanceledException)
{
logger.Warning("Log flush timeout.");
}
}
if (finalException != null)
{
throw finalException;
}
}
public static Task Run(string[] args)
{
var options = LoadTestOptions.Parse(args);
return(Run(options));
}
