adding cleanup for sysbench

Author: nmalkapuram

src/VirtualClient/VirtualClient.Actions/Sysbench/SysbenchClientExecutor.cs

@@ -182,6 +182,10 @@ private Task ExecuteWorkloadAsync(EventContext telemetryContext, CancellationTok
                         {
                             await this.PrepareOLTPMySQLDatabase(telemetryContext, cancellationToken);
                         }
+                        else if (this.Action == ClientAction.Cleanup)
+                        {
+                            await this.CleanUpDatabase(telemetryContext, cancellationToken);
+                        }
                         else
                         {
                             await this.RunOLTPWorkloadAsync(telemetryContext, cancellationToken);
@@ -229,6 +233,35 @@ private async Task RunOLTPWorkloadAsync(EventContext telemetryContext, Cancellat
             }
         }
 
+        private async Task CleanUpDatabase(EventContext telemetryContext, CancellationToken cancellationToken)
+        {
+            int tableCount = GetTableCount(this.DatabaseScenario, this.TableCount, this.Workload);
+
+            string serverIp = (this.GetLayoutClientInstances(ClientRole.Server, false) ?? Enumerable.Empty<ClientInstance>())
+                                    .FirstOrDefault()?.IPAddress
+                                    ?? "localhost";
+
+            string sysbenchCleanupArguments = $"--dbName {this.DatabaseName} --databaseSystem {this.DatabaseSystem} --benchmark {this.Benchmark} --tableCount {tableCount}  --hostIpAddress {serverIp}";
+
+            string script = $"{this.SysbenchPackagePath}/cleanup-database.py ";
+
+            using (IProcessProxy process = await this.ExecuteCommandAsync(
+                SysbenchExecutor.PythonCommand,
+                script + this.sysbenchExecutionArguments,
+                this.SysbenchPackagePath,
+                telemetryContext,
+                cancellationToken))
+            {
+                if (!cancellationToken.IsCancellationRequested)
+                {
+                    await this.LogProcessDetailsAsync(process, telemetryContext, "Sysbench", logToFile: true);
+                    process.ThrowIfErrored<WorkloadException>(process.StandardError.ToString(), ErrorReason.WorkloadFailed);
+
+                    this.CaptureMetrics(process, telemetryContext, cancellationToken);
+                }
+            }
+        }
+
         private async Task RunTPCCWorkloadAsync(EventContext telemetryContext, CancellationToken cancellationToken)
         {
             int tableCount = GetTableCount(this.Scenario, this.TableCount, this.Workload);
@@ -288,7 +321,7 @@ private async Task PrepareOLTPMySQLDatabase(EventContext telemetryContext, Cance
             int threadCount = GetThreadCount(this.SystemManager, this.DatabaseScenario, this.Threads);
             int recordCount = GetRecordCount(this.SystemManager, this.DatabaseScenario, this.RecordCount);
 
-            this.sysbenchLoggingArguments = $"--dbName {this.DatabaseName} --databaseSystem {this.DatabaseSystem} --benchmark {this.Benchmark} --tableCount {tableCount} --recordCount {recordCount} --threadCount {threadCount}";
+            this.sysbenchLoggingArguments = $"--dbName {this.DatabaseName} --databaseSystem {this.DatabaseSystem} --benchmark {this.Benchmark} --threadCount {threadCount} --tableCount {tableCount} --recordCount {recordCount}";
             this.sysbenchPrepareArguments = $"{this.sysbenchLoggingArguments} --password {this.SuperUserPassword}";
 
             string serverIp = (this.GetLayoutClientInstances(ClientRole.Server, false) ?? Enumerable.Empty<ClientInstance>())

src/VirtualClient/VirtualClient.Actions/Sysbench/SysbenchExecutor.cs

@@ -417,7 +417,7 @@ protected void AddMetric(string arguments, IProcessProxy process, EventContext t
 
                 List<Metric> metrics = new List<Metric>();
                 double duration = (process.ExitTime - process.StartTime).TotalMinutes;
-                metrics.Add(new Metric("PopulateDatabaseTime_Minutes ", duration, "minutes", MetricRelativity.LowerIsBetter));
+                metrics.Add(new Metric("PopulateDatabaseTime_Minutes", duration, "minutes", MetricRelativity.LowerIsBetter));
 
                 this.Logger.LogMetrics(
                     toolName: "Sysbench",
@@ -532,9 +532,14 @@ internal class ClientAction
             public const string TruncateDatabase = nameof(TruncateDatabase);
 
             /// <summary>
-            /// Truncates all tables existing in database
+            /// Runs specified workload.
             /// </summary>
             public const string RunWorkload = nameof(RunWorkload);
+
+            /// <summary>
+            /// Runs sysbench cleanup action.
+            /// </summary>
+            public const string Cleanup = nameof(Cleanup);
         }
     }
 }