IntenseFinance/nosqlbench
3
0
Fork 0
mirror of https://github.com/nosqlbench/nosqlbench.git synced 2025-02-25 18:55:28 -06:00
Code Issues Packages Projects Releases Wiki Activity

expose scenario state

Browse Source
This commit is contained in:
Jonathan Shook 2020-08-28 12:33:38 -05:00
parent 6db06818fd
commit 4c940ec73b
1 changed files with 24 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
engine-core/src/main/java/io/nosqlbench/engine/core/script/Scenario.java
View File

@ -48,15 +48,24 @@ public class Scenario implements Callable<ScenarioResult> {
private static final Logger logger = (Logger) LoggerFactory.getLogger(Scenario.class);
private State state = State.Scheduled;
public enum State {
Scheduled,
Running,
Errored,
Finished
}
private static final ScriptEngineManager engineManager = new ScriptEngineManager();
private final List<String> scripts = new ArrayList<>();
private ScriptEngine scriptEngine;
private ScenarioController scenarioController;
private ProgressIndicator progressIndicator;
private ActivityProgressIndicator activityProgressIndicator;
private String progressInterval = "console:1m";
private boolean wantsGraaljsCompatMode;
private ScenarioContext scriptEnv;
private String scenarioName;
private final String scenarioName;
private ScenarioLogger scenarioLogger;
private ScriptParams scenarioScriptParams;
private String scriptfile;
@ -109,7 +118,7 @@ public class Scenario implements Callable<ScenarioResult> {
e.printStackTrace();
}
ByteBuffer bb = ByteBuffer.wrap(bytes);
Charset utf8 = Charset.forName("UTF8");
Charset utf8 = StandardCharsets.UTF_8;
String scriptData = utf8.decode(bb).toString();
addScriptText(scriptData);
}
@ -160,7 +169,7 @@ public class Scenario implements Callable<ScenarioResult> {
scenarioController = new ScenarioController();
if (!progressInterval.equals("disabled")) {
progressIndicator = new ProgressIndicator(scenarioController, progressInterval);
activityProgressIndicator = new ActivityProgressIndicator(scenarioController, progressInterval);
}
scriptEnv = new ScenarioContext(scenarioController);
@ -209,10 +218,10 @@ public class Scenario implements Callable<ScenarioResult> {
}
public void run() {
state=State.Running;
startedAtMillis=System.currentTimeMillis();
init();
logger.debug("Running control script for " + getScenarioName() + ".");
for (String script : scripts) {
try {
@ -253,9 +262,13 @@ public class Scenario implements Callable<ScenarioResult> {
System.err.flush();
System.out.flush();
} catch (Exception e) {
this.scenarioController.forceStopScenario(5000);
this.state = State.Errored;
this.scenarioController.forceStopScenario(5000,false);
throw new RuntimeException(e);
} finally {
if (this.state==State.Running) {
this.state = State.Finished;
}
System.out.flush();
System.err.flush();
endedAtMillis=System.currentTimeMillis();
@ -331,6 +344,10 @@ public class Scenario implements Callable<ScenarioResult> {
}});
}
public State getScenarioState() {
return state;
}
public void enableCharting() {
MetricRegistry metricRegistry = ActivityMetrics.getMetricRegistry();
}