1 files changed, 38 insertions, 0 deletions
diff --git a/infra/it/memory-benchmark-scripts/src/main/groovy/io/fd/honeycomb/benchmark/script/MemoryBenchmarkStartupScriptGenerator.groovy b/infra/it/memory-benchmark-scripts/src/main/groovy/io/fd/honeycomb/benchmark/script/MemoryBenchmarkStartupScriptGenerator.groovy
new file mode 100644
index 000000000..1db656aca
--- /dev/null
+++ b/infra/it/memory-benchmark-scripts/src/main/groovy/io/fd/honeycomb/benchmark/script/MemoryBenchmarkStartupScriptGenerator.groovy
@@ -0,0 +1,38 @@
+package io.fd.honeycomb.benchmark.script
+
+import groovy.text.SimpleTemplateEngine
+
+import java.nio.file.Files
+import java.nio.file.Paths
+
+class MemoryBenchmarkStartupScriptGenerator {
+
+    static final def STARTUP_SCRIPT_ON_REST_TEMPLATE = MemoryBenchmarkStartupScriptGenerator.getResource("/memoryBenchmarkScript")
+    static final def STARTUP_SCRIPT_NAME_BASE = "honeycomb-memory-footprint-benchmark-"
+
+    static final def FOOTPRINT_TEST_CLASS = "io.fd.honeycomb.benchmark.memory.HoneycombFootprintTest"
+
+    static final def OUTPUT_PATH_PARAM = "-DoutPath=\$(dirname \$0)/"
+    static final def SAMPLE_SIZE_PARAM = "-DsampleSize="
+
+    /**
+     * Generate script to run io.fd.honeycomb.benchmark.memory.HoneycombWithDataTest with provided params
+     * */
+    public static void generateWithDataScript(project, log, String outputFileName, dataSampleSize) {
+        log.info "Binding execution script for with-data benchmark[output=${outputFileName},sampleSize=${dataSampleSize}]"
+        def scriptContent = new SimpleTemplateEngine().createTemplate(STARTUP_SCRIPT_ON_REST_TEMPLATE).make(
+                [
+                        "testParams"             : "${OUTPUT_PATH_PARAM}${outputFileName}-${dataSampleSize} ${SAMPLE_SIZE_PARAM}${dataSampleSize}",
+                        "testClass"              : FOOTPRINT_TEST_CLASS
+                ]).toString()
+        flushScript(Paths.get(project.build.directory, "${STARTUP_SCRIPT_NAME_BASE}${dataSampleSize}"), scriptContent, log)
+    }
+
+    private static flushScript(filePath, content,log) {
+        log.info "Saving script to path ${filePath}"
+        def file = Files.createFile(filePath).toFile()
+
+        file.text = content
+        file.setExecutable(true)
+    }
+}