blob: 1db656aca48b9ac4ab4617d2588443eda69ec0f0 (
plain)
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
|
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)
}
}
|
