package org.jbpm.simulation.impl.simulators;
import java.util.Map;
import java.util.concurrent.TimeUnit;
import org.jbpm.simulation.ActivitySimulator;
import org.jbpm.simulation.SimulationContext;
import org.jbpm.simulation.SimulationDataProvider;
import org.jbpm.simulation.SimulationEvent;
import org.jbpm.simulation.TimeGenerator;
import org.jbpm.simulation.TimeGeneratorFactory;
import org.jbpm.simulation.impl.events.ActivitySimulationEvent;
import org.kie.api.definition.process.Node;
import org.kie.api.runtime.process.NodeInstance;
import org.kie.api.runtime.process.ProcessInstance;
public class StateBasedActivitySimulator implements ActivitySimulator {
public SimulationEvent simulate(Object activity, SimulationContext context) {
NodeInstance stateNode = (NodeInstance) activity;
long startTime = context.getClock().getCurrentTime();
Map<String, Object> metadata = stateNode.getNode().getMetaData();
ProcessInstance pi = stateNode.getProcessInstance();
Node node = stateNode.getNode();
String bpmn2NodeId = (String) metadata.get("UniqueId");
SimulationDataProvider provider = context.getDataProvider();
TimeGenerator timeGenerator=TimeGeneratorFactory.newTimeGenerator(provider.getSimulationDataForNode(node));
long duration = timeGenerator.generateTime();
context.getClock().advanceTime(duration, TimeUnit.MILLISECONDS);
// set end time for processinstance end time
context.setMaxEndTime(context.getClock().getCurrentTime());
String type = (String) provider.getProcessDataForNode(node).get("node.type");
return new ActivitySimulationEvent(pi.getProcessId(), context.getProcessInstanceId(), node.getName(), bpmn2NodeId, duration,
startTime, context.getClock().getCurrentTime(), type);
}
}