From 28654b4fd4318c2913a33afb5860374dda601afd Mon Sep 17 00:00:00 2001 From: "stu126940@mail.uni-kiel.de" <stu126940@mail.uni-kiel.de> Date: Thu, 22 Apr 2021 16:39:57 +0200 Subject: [PATCH] Add delay before start the load generator --- .../config/example-execution-yaml-resource.yaml | 1 + .../config/example-operator-execution.yaml | 1 + .../kotlin/theodolite/benchmark/Benchmark.kt | 3 ++- .../theodolite/benchmark/BenchmarkExecution.kt | 1 + .../theodolite/benchmark/KubernetesBenchmark.kt | 17 +++++++++++------ .../benchmark/KubernetesBenchmarkDeployment.kt | 15 ++++++++++----- .../theodolite/execution/BenchmarkExecutor.kt | 5 +++-- .../execution/BenchmarkExecutorImpl.kt | 9 +++++---- .../kotlin/theodolite/execution/Shutdown.kt | 3 ++- .../theodolite/execution/TheodoliteExecutor.kt | 3 ++- .../example-execution-k8s-resource.yaml | 1 + .../kotlin/theodolite/CompositeStrategyTest.kt | 6 +++--- .../src/test/kotlin/theodolite/TestBenchmark.kt | 3 ++- .../theodolite/TestBenchmarkExecutorImpl.kt | 6 ++++-- 14 files changed, 48 insertions(+), 26 deletions(-) diff --git a/theodolite-quarkus/config/example-execution-yaml-resource.yaml b/theodolite-quarkus/config/example-execution-yaml-resource.yaml index 23c1587ec..ccb2442d3 100644 --- a/theodolite-quarkus/config/example-execution-yaml-resource.yaml +++ b/theodolite-quarkus/config/example-execution-yaml-resource.yaml @@ -17,6 +17,7 @@ execution: strategy: "LinearSearch" duration: 300 # in seconds repetitions: 1 + delay: 30 # in seconds restrictions: - "LowerBound" configOverrides: [] diff --git a/theodolite-quarkus/config/example-operator-execution.yaml b/theodolite-quarkus/config/example-operator-execution.yaml index 3df1a723d..882c38a97 100644 --- a/theodolite-quarkus/config/example-operator-execution.yaml +++ b/theodolite-quarkus/config/example-operator-execution.yaml @@ -21,6 +21,7 @@ execution: strategy: "LinearSearch" duration: 300 # in seconds repetitions: 1 + delay: 30 # in seconds restrictions: - "LowerBound" configOverrides: [] diff --git a/theodolite-quarkus/src/main/kotlin/theodolite/benchmark/Benchmark.kt b/theodolite-quarkus/src/main/kotlin/theodolite/benchmark/Benchmark.kt index db7999b20..95af36d1b 100644 --- a/theodolite-quarkus/src/main/kotlin/theodolite/benchmark/Benchmark.kt +++ b/theodolite-quarkus/src/main/kotlin/theodolite/benchmark/Benchmark.kt @@ -21,6 +21,7 @@ interface Benchmark { fun buildDeployment( load: LoadDimension, res: Resource, - configurationOverrides: List<ConfigurationOverride?> + configurationOverrides: List<ConfigurationOverride?>, + delay: Long ): BenchmarkDeployment } diff --git a/theodolite-quarkus/src/main/kotlin/theodolite/benchmark/BenchmarkExecution.kt b/theodolite-quarkus/src/main/kotlin/theodolite/benchmark/BenchmarkExecution.kt index 2d5d15b33..f3eeaeba1 100644 --- a/theodolite-quarkus/src/main/kotlin/theodolite/benchmark/BenchmarkExecution.kt +++ b/theodolite-quarkus/src/main/kotlin/theodolite/benchmark/BenchmarkExecution.kt @@ -47,6 +47,7 @@ class BenchmarkExecution : CustomResource(), Namespaced { var duration by Delegates.notNull<Long>() var repetitions by Delegates.notNull<Int>() lateinit var restrictions: List<String> + var delay by Delegates.notNull<Long>() } /** diff --git a/theodolite-quarkus/src/main/kotlin/theodolite/benchmark/KubernetesBenchmark.kt b/theodolite-quarkus/src/main/kotlin/theodolite/benchmark/KubernetesBenchmark.kt index 53fcb5793..11cf09066 100644 --- a/theodolite-quarkus/src/main/kotlin/theodolite/benchmark/KubernetesBenchmark.kt +++ b/theodolite-quarkus/src/main/kotlin/theodolite/benchmark/KubernetesBenchmark.kt @@ -70,31 +70,36 @@ class KubernetesBenchmark : Benchmark, CustomResource(), Namespaced { override fun buildDeployment( load: LoadDimension, res: Resource, - configurationOverrides: List<ConfigurationOverride?> + configurationOverrides: List<ConfigurationOverride?>, + delay: Long ): BenchmarkDeployment { logger.info { "Using $namespace as namespace." } logger.info { "Using $path as resource path." } - val resources = loadKubernetesResources(this.appResource + this.loadGenResource) + val appResources = loadKubernetesResources(this.appResource) + val loadGenResources = loadKubernetesResources(this.loadGenResource) + val patcherFactory = PatcherFactory() // patch the load dimension the resources load.getType().forEach { patcherDefinition -> - patcherFactory.createPatcher(patcherDefinition, resources).patch(load.get().toString()) + patcherFactory.createPatcher(patcherDefinition, loadGenResources).patch(load.get().toString()) } res.getType().forEach { patcherDefinition -> - patcherFactory.createPatcher(patcherDefinition, resources).patch(res.get().toString()) + patcherFactory.createPatcher(patcherDefinition, appResources).patch(res.get().toString()) } // Patch the given overrides configurationOverrides.forEach { override -> override?.let { - patcherFactory.createPatcher(it.patcher, resources).patch(override.value) + patcherFactory.createPatcher(it.patcher, appResources + loadGenResources).patch(override.value) } } return KubernetesBenchmarkDeployment( namespace = namespace, - resources = resources.map { r -> r.second }, + appResources = appResources.map { it.second }, + loadGenResources = loadGenResources.map { it.second }, + delay = delay, kafkaConfig = hashMapOf("bootstrap.servers" to kafkaConfig.bootstrapServer), topics = kafkaConfig.topics, client = DefaultKubernetesClient().inNamespace(namespace) diff --git a/theodolite-quarkus/src/main/kotlin/theodolite/benchmark/KubernetesBenchmarkDeployment.kt b/theodolite-quarkus/src/main/kotlin/theodolite/benchmark/KubernetesBenchmarkDeployment.kt index 49662d4bf..82e8adeb1 100644 --- a/theodolite-quarkus/src/main/kotlin/theodolite/benchmark/KubernetesBenchmarkDeployment.kt +++ b/theodolite-quarkus/src/main/kotlin/theodolite/benchmark/KubernetesBenchmarkDeployment.kt @@ -8,6 +8,7 @@ import org.apache.kafka.clients.admin.NewTopic import theodolite.k8s.K8sManager import theodolite.k8s.TopicManager import theodolite.util.KafkaConfig +import java.time.Duration private val logger = KotlinLogging.logger {} @@ -22,7 +23,9 @@ private val logger = KotlinLogging.logger {} @RegisterForReflection class KubernetesBenchmarkDeployment( val namespace: String, - val resources: List<KubernetesResource>, + val appResources: List<KubernetesResource>, + val loadGenResources: List<KubernetesResource>, + private val delay: Long, private val kafkaConfig: HashMap<String, Any>, private val topics: List<KafkaConfig.TopicWrapper>, private val client: NamespacedKubernetesClient @@ -41,7 +44,10 @@ class KubernetesBenchmarkDeployment( val kafkaTopics = this.topics.filter { !it.removeOnly } .map { NewTopic(it.name, it.numPartitions, it.replicationFactor) } kafkaController.createTopics(kafkaTopics) - resources.forEach { kubernetesManager.deploy(it) } + appResources.forEach { kubernetesManager.deploy(it) } + logger.info { "Wait ${this.delay} seconds before starting the workload generator." } + Thread.sleep(Duration.ofSeconds(this.delay).toMillis()) + loadGenResources.forEach { kubernetesManager.deploy(it) } } /** @@ -51,9 +57,8 @@ class KubernetesBenchmarkDeployment( * - Remove the [KubernetesResource]s. */ override fun teardown() { - resources.forEach { - kubernetesManager.remove(it) - } + loadGenResources.forEach { kubernetesManager.remove(it) } + appResources.forEach { kubernetesManager.remove(it) } kafkaController.removeTopics(this.topics.map { topic -> topic.name }) KafkaLagExporterRemover(client).remove(LAG_EXPORTER_POD_LABEL) logger.info { "Teardown complete. Wait $SLEEP_AFTER_TEARDOWN ms to let everything come down." } diff --git a/theodolite-quarkus/src/main/kotlin/theodolite/execution/BenchmarkExecutor.kt b/theodolite-quarkus/src/main/kotlin/theodolite/execution/BenchmarkExecutor.kt index 4fb99fb72..3d362e437 100644 --- a/theodolite-quarkus/src/main/kotlin/theodolite/execution/BenchmarkExecutor.kt +++ b/theodolite-quarkus/src/main/kotlin/theodolite/execution/BenchmarkExecutor.kt @@ -24,9 +24,10 @@ abstract class BenchmarkExecutor( val benchmark: Benchmark, val results: Results, val executionDuration: Duration, - configurationOverrides: List<ConfigurationOverride?>, + val configurationOverrides: List<ConfigurationOverride?>, val slo: BenchmarkExecution.Slo, - val executionId: Int + val executionId: Int, + val delay: Long ) { var run: AtomicBoolean = AtomicBoolean(true) diff --git a/theodolite-quarkus/src/main/kotlin/theodolite/execution/BenchmarkExecutorImpl.kt b/theodolite-quarkus/src/main/kotlin/theodolite/execution/BenchmarkExecutorImpl.kt index cd85c143e..c70ab2037 100644 --- a/theodolite-quarkus/src/main/kotlin/theodolite/execution/BenchmarkExecutorImpl.kt +++ b/theodolite-quarkus/src/main/kotlin/theodolite/execution/BenchmarkExecutorImpl.kt @@ -18,13 +18,14 @@ class BenchmarkExecutorImpl( benchmark: Benchmark, results: Results, executionDuration: Duration, - private val configurationOverrides: List<ConfigurationOverride?>, + configurationOverrides: List<ConfigurationOverride?>, slo: BenchmarkExecution.Slo, - executionId: Int -) : BenchmarkExecutor(benchmark, results, executionDuration, configurationOverrides, slo, executionId) { + executionId: Int, + delay: Long +) : BenchmarkExecutor(benchmark, results, executionDuration, configurationOverrides, slo, executionId, delay) { override fun runExperiment(load: LoadDimension, res: Resource): Boolean { var result = false - val benchmarkDeployment = benchmark.buildDeployment(load, res, this.configurationOverrides) + val benchmarkDeployment = benchmark.buildDeployment(load, res, configurationOverrides, delay) try { benchmarkDeployment.setup() diff --git a/theodolite-quarkus/src/main/kotlin/theodolite/execution/Shutdown.kt b/theodolite-quarkus/src/main/kotlin/theodolite/execution/Shutdown.kt index a50a38e79..c74824a4d 100644 --- a/theodolite-quarkus/src/main/kotlin/theodolite/execution/Shutdown.kt +++ b/theodolite-quarkus/src/main/kotlin/theodolite/execution/Shutdown.kt @@ -30,7 +30,8 @@ class Shutdown(private val benchmarkExecution: BenchmarkExecution, private val b benchmark.buildDeployment( load = LoadDimension(0, emptyList()), res = Resource(0, emptyList()), - configurationOverrides = benchmarkExecution.configOverrides + configurationOverrides = benchmarkExecution.configOverrides, + delay = 0L ) deployment.teardown() } catch (e: Exception) { diff --git a/theodolite-quarkus/src/main/kotlin/theodolite/execution/TheodoliteExecutor.kt b/theodolite-quarkus/src/main/kotlin/theodolite/execution/TheodoliteExecutor.kt index 01b25e110..00c768b2c 100644 --- a/theodolite-quarkus/src/main/kotlin/theodolite/execution/TheodoliteExecutor.kt +++ b/theodolite-quarkus/src/main/kotlin/theodolite/execution/TheodoliteExecutor.kt @@ -71,7 +71,8 @@ class TheodoliteExecutor( executionDuration = executionDuration, configurationOverrides = config.configOverrides, slo = config.slos[0], - executionId = config.executionId + executionId = config.executionId, + delay = config.execution.delay ) return Config( diff --git a/theodolite-quarkus/src/main/resources/operator/example-execution-k8s-resource.yaml b/theodolite-quarkus/src/main/resources/operator/example-execution-k8s-resource.yaml index ef625dfe6..7f76b1bca 100644 --- a/theodolite-quarkus/src/main/resources/operator/example-execution-k8s-resource.yaml +++ b/theodolite-quarkus/src/main/resources/operator/example-execution-k8s-resource.yaml @@ -22,6 +22,7 @@ execution: strategy: "LinearSearch" duration: 60 repetitions: 1 + delay: 30 # in seconds restrictions: - "LowerBound" configOverrides: diff --git a/theodolite-quarkus/src/test/kotlin/theodolite/CompositeStrategyTest.kt b/theodolite-quarkus/src/test/kotlin/theodolite/CompositeStrategyTest.kt index 7802529bf..726d4490b 100644 --- a/theodolite-quarkus/src/test/kotlin/theodolite/CompositeStrategyTest.kt +++ b/theodolite-quarkus/src/test/kotlin/theodolite/CompositeStrategyTest.kt @@ -31,7 +31,7 @@ class CompositeStrategyTest { val results = Results() val benchmark = TestBenchmark() val sloChecker: BenchmarkExecution.Slo = BenchmarkExecution.Slo() - val benchmarkExecutor = TestBenchmarkExecutorImpl(mockResults, benchmark, results, sloChecker, 0) + val benchmarkExecutor = TestBenchmarkExecutorImpl(mockResults, benchmark, results, sloChecker, 0, 0) val linearSearch = LinearSearch(benchmarkExecutor) val lowerBoundRestriction = LowerBoundRestriction(results) val strategy = @@ -65,7 +65,7 @@ class CompositeStrategyTest { val benchmark = TestBenchmark() val sloChecker: BenchmarkExecution.Slo = BenchmarkExecution.Slo() val benchmarkExecutorImpl = - TestBenchmarkExecutorImpl(mockResults, benchmark, results, sloChecker, 0) + TestBenchmarkExecutorImpl(mockResults, benchmark, results, sloChecker, 0, 0) val binarySearch = BinarySearch(benchmarkExecutorImpl) val lowerBoundRestriction = LowerBoundRestriction(results) val strategy = @@ -98,7 +98,7 @@ class CompositeStrategyTest { val results = Results() val benchmark = TestBenchmark() val sloChecker: BenchmarkExecution.Slo = BenchmarkExecution.Slo() - val benchmarkExecutor = TestBenchmarkExecutorImpl(mockResults, benchmark, results, sloChecker, 0) + val benchmarkExecutor = TestBenchmarkExecutorImpl(mockResults, benchmark, results, sloChecker, 0, 0) val binarySearch = BinarySearch(benchmarkExecutor) val lowerBoundRestriction = LowerBoundRestriction(results) val strategy = diff --git a/theodolite-quarkus/src/test/kotlin/theodolite/TestBenchmark.kt b/theodolite-quarkus/src/test/kotlin/theodolite/TestBenchmark.kt index 6f476278d..2d3dcb261 100644 --- a/theodolite-quarkus/src/test/kotlin/theodolite/TestBenchmark.kt +++ b/theodolite-quarkus/src/test/kotlin/theodolite/TestBenchmark.kt @@ -11,7 +11,8 @@ class TestBenchmark : Benchmark { override fun buildDeployment( load: LoadDimension, res: Resource, - configurationOverrides: List<ConfigurationOverride?> + configurationOverrides: List<ConfigurationOverride?>, + delay: Long ): BenchmarkDeployment { return TestBenchmarkDeployment() } diff --git a/theodolite-quarkus/src/test/kotlin/theodolite/TestBenchmarkExecutorImpl.kt b/theodolite-quarkus/src/test/kotlin/theodolite/TestBenchmarkExecutorImpl.kt index 2bafcb76d..18cdc9aaf 100644 --- a/theodolite-quarkus/src/test/kotlin/theodolite/TestBenchmarkExecutorImpl.kt +++ b/theodolite-quarkus/src/test/kotlin/theodolite/TestBenchmarkExecutorImpl.kt @@ -13,7 +13,8 @@ class TestBenchmarkExecutorImpl( benchmark: Benchmark, results: Results, slo: BenchmarkExecution.Slo, - executionId: Int + executionId: Int, + delay: Long ) : BenchmarkExecutor( benchmark, @@ -21,7 +22,8 @@ class TestBenchmarkExecutorImpl( executionDuration = Duration.ofSeconds(1), configurationOverrides = emptyList(), slo = slo, - executionId = executionId + executionId = executionId, + delay = delay ) { override fun runExperiment(load: LoadDimension, res: Resource): Boolean { -- GitLab