forked from diku-dk/EventBenchmark
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Program.cs
119 lines (111 loc) · 4.09 KB
/
Program.cs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
using Common.DataGeneration;
using Common.Experiment;
using Common.Http;
using Common.Infra;
using DuckDB.NET.Data;
using Orleans.Infra;
using Orleans.Workload;
namespace Orleans;
public class Program
{
public static async Task Main(string[] args)
{
Console.WriteLine("Initializing benchmark driver...");
ExperimentConfig config = ConsoleUtility.BuildExperimentConfig(args);
Console.WriteLine("Configuration parsed. Starting program...");
DuckDBConnection connection = null;
try{
while(true){
Console.WriteLine("\n Select an option: \n 1 - Generate Data \n 2 - Ingest Data \n 3 - Run Experiment \n 4 - Ingest and Run (2 and 3) \n 5 - Parse New Configuration \n q - Exit");
string op = Console.ReadLine();
switch (op)
{
case "1":
{
connection = ConsoleUtility.GenerateData(config);
break;
}
case "2":
{
if(connection is null){
if(config.connectionString.SequenceEqual("DataSource=:memory:"))
{
Console.WriteLine("Please generate some data first by selecting option 1.");
break;
}
else
{
connection = new DuckDBConnection(config.connectionString);
connection.Open();
}
}
await CustomIngestionOrchestrator.Run(connection, config.ingestionConfig);
break;
}
case "3":
{
if (connection is null)
{
if (config.connectionString.SequenceEqual("DataSource=:memory:"))
{
Console.WriteLine("Please generate some data first by selecting option 1.");
break;
}
else
{
connection = new DuckDBConnection(config.connectionString);
connection.Open();
}
}
var expManager = ActorExperimentManager.BuildActorExperimentManager(new CustomHttpClientFactory(), config, connection);
await expManager.RunSimpleExperiment(2);
break;
}
case "4":
{
if(connection is null) {
if(config.connectionString.SequenceEqual("DataSource=:memory:"))
{
Console.WriteLine("Please generate some data first by selecting option 1.");
break;
}
else
{
connection = new DuckDBConnection(config.connectionString);
connection.Open();
}
}
// ingest data
await CustomIngestionOrchestrator.Run(connection, config.ingestionConfig);
var expManager = ActorExperimentManager.BuildActorExperimentManager(new CustomHttpClientFactory(), config, connection);
// could be a config param: delay after ingest
Console.WriteLine("Delay after ingest...");
await Task.Delay(10000);
// run
await expManager.RunSimpleExperiment(2);
Console.WriteLine("Experiment finished.");
break;
}
case "5":
{
config = ConsoleUtility.BuildExperimentConfig(args);
Console.WriteLine("Configuration parsed.");
break;
}
case "q":
{
return;
}
default:
{
Console.WriteLine("Input invalid");
break;
}
}
}
} catch(Exception e)
{
Console.WriteLine("Exception catched. Source: {0}; StackTrace: \n {1}", e.Source, e.StackTrace );
}
}
}