org.jbpm.process.instance.ProcessRuntimeImpl Maven / Gradle / Ivy
package org.jbpm.process.instance;
import java.io.IOException;
import java.io.ObjectInput;
import java.io.ObjectOutput;
import java.util.Collection;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import org.drools.core.SessionConfiguration;
import org.drools.core.common.InternalKnowledgeRuntime;
import org.drools.core.common.InternalWorkingMemory;
import org.drools.core.common.WorkingMemoryAction;
import org.drools.core.definitions.rule.impl.RuleImpl;
import org.drools.core.event.ProcessEventSupport;
import org.drools.core.impl.InternalKnowledgeBase;
import org.drools.core.marshalling.impl.MarshallerReaderContext;
import org.drools.core.marshalling.impl.MarshallerWriteContext;
import org.drools.core.marshalling.impl.ProtobufMessages.ActionQueue.Action;
import org.drools.core.time.AcceptsTimerJobFactoryManager;
import org.drools.core.time.TimeUtils;
import org.drools.core.time.impl.CronExpression;
import org.drools.core.time.impl.DefaultTimerJobFactoryManager;
import org.drools.core.time.impl.TrackableTimeJobFactoryManager;
import org.jbpm.process.core.event.EventFilter;
import org.jbpm.process.core.event.EventTypeFilter;
import org.jbpm.process.core.timer.BusinessCalendar;
import org.jbpm.process.core.timer.DateTimeUtils;
import org.jbpm.process.core.timer.Timer;
import org.jbpm.process.instance.event.SignalManager;
import org.jbpm.process.instance.event.SignalManagerFactory;
import org.jbpm.process.instance.timer.TimerInstance;
import org.jbpm.process.instance.timer.TimerManager;
import org.jbpm.ruleflow.core.RuleFlowProcess;
import org.jbpm.workflow.core.node.EventTrigger;
import org.jbpm.workflow.core.node.StartNode;
import org.jbpm.workflow.core.node.Trigger;
import org.kie.api.KieBase;
import org.kie.api.definition.process.Node;
import org.kie.api.definition.process.Process;
import org.kie.api.event.kiebase.AfterProcessAddedEvent;
import org.kie.api.event.kiebase.AfterProcessRemovedEvent;
import org.kie.api.event.kiebase.DefaultKieBaseEventListener;
import org.kie.api.event.process.ProcessEventListener;
import org.kie.api.event.rule.DefaultAgendaEventListener;
import org.kie.api.event.rule.MatchCreatedEvent;
import org.kie.api.event.rule.RuleFlowGroupDeactivatedEvent;
import org.kie.api.runtime.process.EventListener;
import org.kie.api.runtime.process.ProcessInstance;
import org.kie.api.runtime.process.WorkItemManager;
import org.kie.internal.process.CorrelationKey;
import org.kie.internal.runtime.StatefulKnowledgeSession;
import org.kie.internal.utils.CompositeClassLoader;
public class ProcessRuntimeImpl implements InternalProcessRuntime {
private InternalKnowledgeRuntime kruntime;
private ProcessInstanceManager processInstanceManager;
private SignalManager signalManager;
private TimerManager timerManager;
private ProcessEventSupport processEventSupport;
private DefaultKieBaseEventListener knowledgeBaseListener;
public ProcessRuntimeImpl(InternalKnowledgeRuntime kruntime) {
this.kruntime = kruntime;
AcceptsTimerJobFactoryManager jfm = ( AcceptsTimerJobFactoryManager ) kruntime.getTimerService();
if ( jfm.getTimerJobFactoryManager() instanceof DefaultTimerJobFactoryManager ) {
jfm.setTimerJobFactoryManager( new TrackableTimeJobFactoryManager() );
}
((AcceptsTimerJobFactoryManager)kruntime.getTimerService()).setTimerJobFactoryManager( new TrackableTimeJobFactoryManager() );
((CompositeClassLoader) getRootClassLoader()).addClassLoader( getClass().getClassLoader() );
initProcessInstanceManager();
initSignalManager();
timerManager = new TimerManager(kruntime, kruntime.getTimerService());
processEventSupport = new ProcessEventSupport();
initProcessEventListeners();
initProcessActivationListener();
initStartTimers();
}
private void initStartTimers() {
KieBase kbase = kruntime.getKieBase();
Collection processes = kbase.getProcesses();
for (Process process : processes) {
RuleFlowProcess p = (RuleFlowProcess) process;
List startNodes = p.getTimerStart();
if (startNodes != null && !startNodes.isEmpty()) {
kruntime.queueWorkingMemoryAction(new RegisterStartTimerAction(p.getId(), startNodes, this.timerManager));
kruntime.executeQueuedActions();
}
}
}
public ProcessRuntimeImpl(InternalWorkingMemory workingMemory) {
AcceptsTimerJobFactoryManager jfm = ( AcceptsTimerJobFactoryManager ) workingMemory.getTimerService();
if ( jfm.getTimerJobFactoryManager() instanceof DefaultTimerJobFactoryManager ) {
jfm.setTimerJobFactoryManager( new TrackableTimeJobFactoryManager() );
}
this.kruntime = (InternalKnowledgeRuntime) workingMemory.getKnowledgeRuntime();
initProcessInstanceManager();
initSignalManager();
timerManager = new TimerManager(kruntime, kruntime.getTimerService());
processEventSupport = new ProcessEventSupport();
initProcessEventListeners();
initProcessActivationListener();
initStartTimers();
}
private void initProcessInstanceManager() {
String processInstanceManagerClass = ((SessionConfiguration) kruntime.getSessionConfiguration()).getProcessInstanceManagerFactory();
try {
processInstanceManager =
((ProcessInstanceManagerFactory) loadClass(processInstanceManagerClass).newInstance())
.createProcessInstanceManager(kruntime);
} catch (InstantiationException e) {
throw new RuntimeException(e);
} catch (IllegalAccessException e) {
throw new RuntimeException(e);
}
}
private void initSignalManager() {
String signalManagerClass = ((SessionConfiguration) kruntime.getSessionConfiguration()).getSignalManagerFactory();
try {
signalManager = ((SignalManagerFactory) loadClass(signalManagerClass).newInstance())
.createSignalManager(kruntime);
} catch (InstantiationException e) {
throw new RuntimeException(e);
} catch (IllegalAccessException e) {
throw new RuntimeException(e);
}
}
private Class loadClass(String className) {
try {
return getRootClassLoader().loadClass(className);
} catch (ClassNotFoundException e) {
throw new RuntimeException(e);
}
}
private ClassLoader getRootClassLoader() {
KieBase kbase = ((InternalKnowledgeBase) kruntime.getKieBase());
if (kbase != null) {
return ((InternalKnowledgeBase) kbase).getRootClassLoader();
}
CompositeClassLoader result = new CompositeClassLoader();
result.addClassLoader(Thread.currentThread().getContextClassLoader());
return result;
}
public ProcessInstance startProcess(final String processId) {
return startProcess(processId, null);
}
public ProcessInstance startProcess(String processId,
Map parameters) {
return startProcess(processId, parameters, null);
}
public ProcessInstance startProcess(String processId,
Map parameters, String trigger) {
ProcessInstance processInstance = createProcessInstance(processId, parameters);
if ( processInstance != null ) {
// start process instance
return startProcessInstance(processInstance.getId(), trigger);
}
return null;
}
public ProcessInstance createProcessInstance(String processId,
Map parameters) {
return createProcessInstance(processId, null, parameters);
}
public ProcessInstance startProcessInstance(long processInstanceId, String trigger) {
try {
kruntime.startOperation();
if ( !kruntime.getActionQueue().isEmpty() ) {
kruntime.executeQueuedActions();
}
ProcessInstance processInstance = getProcessInstance(processInstanceId);
getProcessEventSupport().fireBeforeProcessStarted( processInstance, kruntime );
((org.jbpm.process.instance.ProcessInstance) processInstance).start(trigger);
getProcessEventSupport().fireAfterProcessStarted( processInstance, kruntime );
return processInstance;
} finally {
kruntime.endOperation();
}
}
public ProcessInstance startProcessInstance(long processInstanceId) {
return startProcessInstance(processInstanceId, null);
}
@Override
public ProcessInstance startProcess(String processId,
CorrelationKey correlationKey, Map parameters) {
ProcessInstance processInstance = createProcessInstance(processId, correlationKey, parameters);
if ( processInstance != null ) {
return startProcessInstance(processInstance.getId());
}
return null;
}
@Override
public ProcessInstance createProcessInstance(String processId,
CorrelationKey correlationKey, Map parameters) {
try {
kruntime.startOperation();
if ( !kruntime.getActionQueue().isEmpty() ) {
kruntime.executeQueuedActions();
}
final Process process = kruntime.getKieBase().getProcess( processId );
if ( process == null ) {
throw new IllegalArgumentException( "Unknown process ID: " + processId );
}
return startProcess( process, correlationKey, parameters );
} finally {
kruntime.endOperation();
}
}
@Override
public ProcessInstance getProcessInstance(CorrelationKey correlationKey) {
return processInstanceManager.getProcessInstance(correlationKey);
}
private org.jbpm.process.instance.ProcessInstance startProcess(final Process process, CorrelationKey correlationKey,
Map parameters) {
ProcessInstanceFactory conf = ProcessInstanceFactoryRegistry.INSTANCE.getProcessInstanceFactory( process );
if ( conf == null ) {
throw new IllegalArgumentException( "Illegal process type: " + process.getClass() );
}
return conf.createProcessInstance( process,
correlationKey,
kruntime,
parameters );
}
public ProcessInstanceManager getProcessInstanceManager() {
return processInstanceManager;
}
public TimerManager getTimerManager() {
return timerManager;
}
public SignalManager getSignalManager() {
return signalManager;
}
public Collection getProcessInstances() {
return processInstanceManager.getProcessInstances();
}
public ProcessInstance getProcessInstance(long id) {
return getProcessInstance( id, false );
}
public ProcessInstance getProcessInstance(long id, boolean readOnly) {
return processInstanceManager.getProcessInstance( id, readOnly );
}
public void removeProcessInstance(ProcessInstance processInstance) {
processInstanceManager.removeProcessInstance( processInstance );
}
private void initProcessEventListeners() {
for ( Process process : kruntime.getKieBase().getProcesses() ) {
initProcessEventListener(process);
}
knowledgeBaseListener = new DefaultKieBaseEventListener() {
@Override
public void afterProcessAdded(AfterProcessAddedEvent event) {
initProcessEventListener(event.getProcess());
}
@Override
public void afterProcessRemoved(AfterProcessRemovedEvent event) {
if (event.getProcess() instanceof RuleFlowProcess) {
String type = (String)
((RuleFlowProcess) event.getProcess()).getRuntimeMetaData().get("StartProcessEventType");
StartProcessEventListener listener = (StartProcessEventListener)
((RuleFlowProcess) event.getProcess()).getRuntimeMetaData().get("StartProcessEventListener");
if (type != null && listener != null) {
signalManager.removeEventListener(type, listener);
}
}
}
};
kruntime.getKieBase().addEventListener(knowledgeBaseListener);
}
private void initProcessEventListener(Process process) {
if ( process instanceof RuleFlowProcess ) {
for (Node node : ((RuleFlowProcess) process).getNodes()) {
if (node instanceof StartNode) {
StartNode startNode = (StartNode) node;
if (startNode != null) {
List triggers = startNode.getTriggers();
if ( triggers != null ) {
for ( Trigger trigger : triggers ) {
if ( trigger instanceof EventTrigger ) {
final List filters = ((EventTrigger) trigger).getEventFilters();
String type = null;
for ( EventFilter filter : filters ) {
if ( filter instanceof EventTypeFilter ) {
type = ((EventTypeFilter) filter).getType();
}
}
StartProcessEventListener listener = new StartProcessEventListener( process.getId(),
filters,
trigger.getInMappings() );
signalManager.addEventListener( type,
listener );
((RuleFlowProcess) process).getRuntimeMetaData().put("StartProcessEventType", type);
((RuleFlowProcess) process).getRuntimeMetaData().put("StartProcessEventListener", listener);
}
}
}
}
}
}
}
}
public ProcessEventSupport getProcessEventSupport() {
return processEventSupport;
}
public void addEventListener(final ProcessEventListener listener) {
this.processEventSupport.addEventListener( listener );
}
public void removeEventListener(final ProcessEventListener listener) {
this.processEventSupport.removeEventListener( listener );
}
public List getProcessEventListeners() {
return processEventSupport.getEventListeners();
}
private class StartProcessEventListener implements EventListener {
private String processId;
private List eventFilters;
private Map inMappings;
public StartProcessEventListener(String processId,
List eventFilters,
Map inMappings) {
this.processId = processId;
this.eventFilters = eventFilters;
this.inMappings = inMappings;
}
public String[] getEventTypes() {
return null;
}
public void signalEvent(String type,
Object event) {
for ( EventFilter filter : eventFilters ) {
if ( !filter.acceptsEvent( type,
event ) ) {
return;
}
}
Map params = null;
if ( inMappings != null && !inMappings.isEmpty() ) {
params = new HashMap();
for ( Map.Entry entry : inMappings.entrySet() ) {
if ( "event".equals( entry.getValue() ) ) {
params.put( entry.getKey(),
event );
} else {
params.put( entry.getKey(),
entry.getValue() );
}
}
}
startProcess( processId,
params, type );
}
}
private void initProcessActivationListener() {
kruntime.addEventListener(new DefaultAgendaEventListener() {
public void matchCreated(MatchCreatedEvent event) {
String ruleFlowGroup = ((RuleImpl) event.getMatch().getRule()).getRuleFlowGroup();
if ( "DROOLS_SYSTEM".equals( ruleFlowGroup ) ) {
// new activations of the rule associate with a state node
// signal process instances of that state node
String ruleName = event.getMatch().getRule().getName();
if ( ruleName.startsWith( "RuleFlowStateNode-" )) {
int index = ruleName.indexOf( "-",
18 );
index = ruleName.indexOf( "-",
index + 1 );
String eventType = ruleName.substring( 0,
index );
signalManager.signalEvent( eventType,
event );
} else if (ruleName.startsWith( "RuleFlowStateEventSubProcess-" ) || ruleName.startsWith( "RuleFlowStateEvent-" )) {
signalManager.signalEvent( ruleName, event );
}
}
}
});
kruntime.addEventListener(new DefaultAgendaEventListener() {
public void afterRuleFlowGroupDeactivated(final RuleFlowGroupDeactivatedEvent event) {
if (kruntime instanceof StatefulKnowledgeSession) {
signalManager.signalEvent( "RuleFlowGroup_" + event.getRuleFlowGroup().getName() + "_" + ((StatefulKnowledgeSession) kruntime).getId(),
null );
} else {
signalManager.signalEvent( "RuleFlowGroup_" + event.getRuleFlowGroup().getName(),
null );
}
}
} );
}
public void abortProcessInstance(long processInstanceId) {
ProcessInstance processInstance = getProcessInstance(processInstanceId);
if ( processInstance == null ) {
throw new IllegalArgumentException( "Could not find process instance for id " + processInstanceId );
}
((org.jbpm.process.instance.ProcessInstance) processInstance).setState( ProcessInstance.STATE_ABORTED );
}
public WorkItemManager getWorkItemManager() {
return kruntime.getWorkItemManager();
}
public void signalEvent(String type, Object event) {
signalManager.signalEvent(type, event);
}
public void signalEvent(String type, Object event, long processInstanceId) {
signalManager.signalEvent(processInstanceId, type, event);
}
public void setProcessEventSupport(ProcessEventSupport processEventSupport) {
this.processEventSupport = processEventSupport;
}
public void dispose() {
this.processEventSupport.reset();
this.timerManager.dispose();
if( kruntime != null ) {
kruntime.getKieBase().removeEventListener(knowledgeBaseListener);
kruntime = null;
}
}
public void clearProcessInstances() {
this.processInstanceManager.clearProcessInstances();
}
public void clearProcessInstancesState() {
this.processInstanceManager.clearProcessInstancesState();
}
public static class RegisterStartTimerAction implements WorkingMemoryAction {
private List startNodes;
private String processId;
private TimerManager timerManager;
public RegisterStartTimerAction(String processId, List startNodes, TimerManager timerManager) {
this.processId = processId;
this.startNodes = startNodes;
this.timerManager = timerManager;
}
public RegisterStartTimerAction(MarshallerReaderContext context) {
}
@Override
public void readExternal(ObjectInput in) throws IOException,
ClassNotFoundException {
}
@Override
public void writeExternal(ObjectOutput out) throws IOException {
}
@Override
public void execute(InternalWorkingMemory workingMemory) {
initTimer(workingMemory.getKnowledgeRuntime());
}
@Override
public void execute(InternalKnowledgeRuntime kruntime) {
initTimer(kruntime);
}
@Override
public void write(MarshallerWriteContext context) throws IOException {
}
@Override
public Action serialize(MarshallerWriteContext context)
throws IOException {
return null;
}
private void initTimer(InternalKnowledgeRuntime kruntime) {
for (StartNode startNode : startNodes) {
if (startNode != null && startNode.getTimer() != null) {
TimerInstance timerInstance = null;
if (CronExpression.isValidExpression(startNode.getTimer().getDelay())) {
timerInstance = new TimerInstance();
timerInstance.setCronExpression(startNode.getTimer().getDelay());
} else {
timerInstance = createTimerInstance(startNode.getTimer(), kruntime);
}
timerManager.registerTimer(timerInstance, processId, null);
}
}
}
protected TimerInstance createTimerInstance(Timer timer, InternalKnowledgeRuntime kruntime) {
TimerInstance timerInstance = new TimerInstance();
if (kruntime != null && kruntime.getEnvironment().get("jbpm.business.calendar") != null){
BusinessCalendar businessCalendar = (BusinessCalendar) kruntime.getEnvironment().get("jbpm.business.calendar");
String delay = timer.getDelay();
timerInstance.setDelay(businessCalendar.calculateBusinessTimeAsDuration(delay));
if (timer.getPeriod() == null) {
timerInstance.setPeriod(0);
} else {
String period = timer.getPeriod();
timerInstance.setPeriod(businessCalendar.calculateBusinessTimeAsDuration(period));
}
} else {
configureTimerInstance(timer, timerInstance);
}
timerInstance.setTimerId(timer.getId());
return timerInstance;
}
private void configureTimerInstance(Timer timer, TimerInstance timerInstance) {
String s = null;
long duration = -1;
switch (timer.getTimeType()) {
case Timer.TIME_CYCLE:
// when using ISO date/time period is not set
long[] repeatValues = DateTimeUtils.parseRepeatableDateTime(timer.getDelay());
if (repeatValues.length == 3) {
int parsedReapedCount = (int)repeatValues[0];
if (parsedReapedCount > -1) {
timerInstance.setRepeatLimit(parsedReapedCount+1);
}
timerInstance.setDelay(repeatValues[1]);
timerInstance.setPeriod(repeatValues[2]);
} else {
timerInstance.setDelay(repeatValues[0]);
try {
long period = DateTimeUtils.parseTimeString(timer.getPeriod());
timerInstance.setPeriod(period);
} catch (RuntimeException e) {
timerInstance.setPeriod(repeatValues[0]);
}
}
break;
case Timer.TIME_DURATION:
duration = DateTimeUtils.parseDuration(timer.getDelay());
timerInstance.setDelay(duration);
timerInstance.setPeriod(0);
break;
case Timer.TIME_DATE:
duration = DateTimeUtils.parseDateAsDuration(timer.getDate());
timerInstance.setDelay(duration);
timerInstance.setPeriod(0);
break;
default:
break;
}
}
private long resolveValue(String s) {
return TimeUtils.parseTimeString(s);
}
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy