summaryrefslogtreecommitdiff
path: root/opendc-harness/src/main/kotlin/org/opendc/harness/engine/ExperimentEngine.kt
diff options
context:
space:
mode:
authorFabian Mastenbroek <mail.fabianm@gmail.com>2021-05-05 12:23:25 +0200
committerGitHub <noreply@github.com>2021-05-05 12:23:25 +0200
commite1d892bcbaa7e8361c150f684ca6a0dc5e036a87 (patch)
tree99742afbc3cd762ec527f5a6c7e31ee245ee0385 /opendc-harness/src/main/kotlin/org/opendc/harness/engine/ExperimentEngine.kt
parente021d46ab1601d0c3a00724358164483608f6297 (diff)
parent7fdfe48439f33abec8aa63fc4ab36fd687d59a43 (diff)
harness: Improve OpenDC Experiment Harness (v1)
This pull request contains several improvements to the OpenDC Experiment Harness, which aims to improve user-experience of users when re-running experiments in the repository. * Split the harness into separate modules * Add support for adding additional classpath entries when running experiments via the console * Initial support for configuration of experiments using HOCON.
Diffstat (limited to 'opendc-harness/src/main/kotlin/org/opendc/harness/engine/ExperimentEngine.kt')
-rw-r--r--opendc-harness/src/main/kotlin/org/opendc/harness/engine/ExperimentEngine.kt104
1 files changed, 0 insertions, 104 deletions
diff --git a/opendc-harness/src/main/kotlin/org/opendc/harness/engine/ExperimentEngine.kt b/opendc-harness/src/main/kotlin/org/opendc/harness/engine/ExperimentEngine.kt
deleted file mode 100644
index a36f1f9b..00000000
--- a/opendc-harness/src/main/kotlin/org/opendc/harness/engine/ExperimentEngine.kt
+++ /dev/null
@@ -1,104 +0,0 @@
-/*
- * Copyright (c) 2021 AtLarge Research
- *
- * Permission is hereby granted, free of charge, to any person obtaining a copy
- * of this software and associated documentation files (the "Software"), to deal
- * in the Software without restriction, including without limitation the rights
- * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
- * copies of the Software, and to permit persons to whom the Software is
- * furnished to do so, subject to the following conditions:
- *
- * The above copyright notice and this permission notice shall be included in all
- * copies or substantial portions of the Software.
- *
- * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
- * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
- * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
- * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
- * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
- * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
- * SOFTWARE.
- */
-
-package org.opendc.harness.engine
-
-import kotlinx.coroutines.*
-import kotlinx.coroutines.flow.asFlow
-import kotlinx.coroutines.flow.buffer
-import kotlinx.coroutines.flow.collect
-import kotlinx.coroutines.flow.map
-import org.opendc.harness.api.ExperimentDefinition
-import org.opendc.harness.api.Trial
-import org.opendc.harness.engine.scheduler.ExperimentScheduler
-import org.opendc.harness.engine.strategy.ExperimentStrategy
-
-/**
- * The [ExperimentEngine] orchestrates the execution of experiments.
- *
- * @property strategy The [ExperimentStrategy] used to explore the experiment design space.
- * @property scheduler The [ExperimentScheduler] to schedule the trials over compute resources.
- * @property listener The [ExperimentExecutionListener] to observe the progress.
- * @property repeats The number of repeats to perform.
- */
-public class ExperimentEngine(
- private val strategy: ExperimentStrategy,
- private val scheduler: ExperimentScheduler,
- private val listener: ExperimentExecutionListener,
- private val repeats: Int
-) {
- /**
- * Execute the specified [experiment][root].
- *
- * @param root The experiment to execute.
- */
- public suspend fun execute(root: ExperimentDefinition) {
- listener.experimentStarted(root)
-
- try {
- supervisorScope {
- strategy.generate(root)
- .asFlow()
- .map { scenario ->
- listener.scenarioStarted(scenario)
- scenario
- }
- .buffer(100)
- .collect { scenario ->
- launch {
- val jobs = (0 until repeats).map { repeat ->
- val worker = scheduler.allocate()
- launch {
- val trial = Trial(scenario, repeat)
- try {
- listener.trialStarted(trial)
- worker.dispatch(trial)
- listener.trialFinished(trial, null)
- } catch (e: Throwable) {
- listener.trialFinished(trial, e)
- throw e
- }
- }
- }
-
- try {
- jobs.joinAll()
- listener.scenarioFinished(scenario, null)
- } catch (e: CancellationException) {
- listener.scenarioFinished(scenario, null)
- throw e
- } catch (e: Throwable) {
- listener.scenarioFinished(scenario, e)
- }
- }
- }
- }
-
- listener.experimentFinished(root, null)
- } catch (e: Throwable) {
- listener.experimentFinished(root, e)
- throw e
- }
- }
-
- override fun toString(): String = "ExperimentEngine"
-}