From 538b7486c29a752bf628bd8b0bed21992bf9e9d6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Bj=C3=B6rn=20Vonheiden?= <bjoern.vonheiden@hotmail.de>
Date: Tue, 31 Aug 2021 16:15:49 +0200
Subject: [PATCH] Add a Patcher to handle ospbench data volume settings
This handler takes the total defined load and compute the
needed load generator instances and set the volume that each generator
should generate
---
.../DataVolumeLoadGeneratorReplicaPatcher.kt | 31 +++++++++++++++++++
.../theodolite/patcher/EnvVarPatcher.kt | 2 +-
.../theodolite/patcher/PatcherFactory.kt | 6 ++++
3 files changed, 38 insertions(+), 1 deletion(-)
create mode 100644 theodolite/src/main/kotlin/theodolite/patcher/DataVolumeLoadGeneratorReplicaPatcher.kt
diff --git a/theodolite/src/main/kotlin/theodolite/patcher/DataVolumeLoadGeneratorReplicaPatcher.kt b/theodolite/src/main/kotlin/theodolite/patcher/DataVolumeLoadGeneratorReplicaPatcher.kt
new file mode 100644
index 000000000..f797fc870
--- /dev/null
+++ b/theodolite/src/main/kotlin/theodolite/patcher/DataVolumeLoadGeneratorReplicaPatcher.kt
@@ -0,0 +1,31 @@
+package theodolite.patcher
+
+import io.fabric8.kubernetes.api.model.KubernetesResource
+import io.fabric8.kubernetes.api.model.apps.Deployment
+
+/**
+ * The DataVolumeLoadGeneratorReplicaPatcher allows to modify the value of an environment variable
+ *
+ * @property k8sResource Kubernetes resource to be patched.
+ * @property container Container to be patched.
+ * @property variableName Name of the environment variable to be patched.
+ */
+class DataVolumeLoadGeneratorReplicaPatcher(
+ private val k8sResource: KubernetesResource,
+ private val maxVolume: String,
+ container: String,
+ variableName: String
+) : EnvVarPatcher(k8sResource, container, variableName) {
+
+ override fun <String> patch(value: String) {
+ if (k8sResource is Deployment) {
+ if (value is kotlin.String) {
+ val load = value.toInt()
+ val loadGenInstances = (load + maxVolume.toInt() - 1) / maxVolume.toInt()
+ this.k8sResource.spec.replicas = loadGenInstances
+ val loadPerInstance = load / loadGenInstances
+ super.patch(loadPerInstance.toString())
+ }
+ }
+ }
+}
diff --git a/theodolite/src/main/kotlin/theodolite/patcher/EnvVarPatcher.kt b/theodolite/src/main/kotlin/theodolite/patcher/EnvVarPatcher.kt
index 416aec74a..080afb055 100644
--- a/theodolite/src/main/kotlin/theodolite/patcher/EnvVarPatcher.kt
+++ b/theodolite/src/main/kotlin/theodolite/patcher/EnvVarPatcher.kt
@@ -12,7 +12,7 @@ import io.fabric8.kubernetes.api.model.apps.Deployment
* @property container Container to be patched.
* @property variableName Name of the environment variable to be patched.
*/
-class EnvVarPatcher(
+open class EnvVarPatcher(
private val k8sResource: KubernetesResource,
private val container: String,
private val variableName: String
diff --git a/theodolite/src/main/kotlin/theodolite/patcher/PatcherFactory.kt b/theodolite/src/main/kotlin/theodolite/patcher/PatcherFactory.kt
index 29723355c..01a104cbd 100644
--- a/theodolite/src/main/kotlin/theodolite/patcher/PatcherFactory.kt
+++ b/theodolite/src/main/kotlin/theodolite/patcher/PatcherFactory.kt
@@ -48,6 +48,12 @@ class PatcherFactory {
k8sResource = resource,
loadGenMaxRecords = patcherDefinition.properties["loadGenMaxRecords"]!!
)
+ "DataVolumeLoadGeneratorReplicaPatcher" -> DataVolumeLoadGeneratorReplicaPatcher(
+ k8sResource = resource,
+ maxVolume = patcherDefinition.properties["maxVolume"]!!,
+ container = patcherDefinition.properties["container"]!!,
+ variableName = patcherDefinition.properties["variableName"]!!
+ )
"EnvVarPatcher" -> EnvVarPatcher(
k8sResource = resource,
container = patcherDefinition.properties["container"]!!,
--
GitLab