diff --git a/src/main/java/teetime/framework/AbstractRunnableStage.java b/src/main/java/teetime/framework/AbstractRunnableStage.java index 7e4965e61f87d5a8689209d259a4292846bcbbe9..c5d3eb428f2e4f1e8d6f1c26f25660c69d587232 100644 --- a/src/main/java/teetime/framework/AbstractRunnableStage.java +++ b/src/main/java/teetime/framework/AbstractRunnableStage.java @@ -32,6 +32,8 @@ abstract class AbstractRunnableStage implements Runnable { public AbstractRunnableStage(final Stage stage) { this.stage = stage; this.logger = LoggerFactory.getLogger(stage.getClass()); + + // stage.owningContext.getThreadCounter().inc(); } @Override @@ -70,6 +72,8 @@ abstract class AbstractRunnableStage implements Runnable { throw new IllegalStateException("Terminated by StageExceptionListener"); } + // normal and exceptional termination + // stage.owningContext.getThreadCounter().dec(); } protected abstract void beforeStageExecution() throws InterruptedException;