Examples with TimerManager org.jbpm.process.instance.timer.TimerManager used on opensource projects

Example 11 with TimerManager

use of org.jbpm.process.instance.timer.TimerManager in project jbpm by kiegroup.

the class MigrationManager method rescheduleTimersAfterMigration.

protected void rescheduleTimersAfterMigration(RuntimeManager manager, Map<Long, List<TimerInstance>> timerMigrated) {
    RuntimeEngine engine = manager.getRuntimeEngine(ProcessInstanceIdContext.get(migrationSpec.getProcessInstanceId()));
    try {
        engine.getKieSession().execute(new ExecutableCommand<Void>() {

            private static final long serialVersionUID = 7144657913971146080L;

            @Override
            public Void execute(Context context) {
                KieSession kieSession = ((RegistryContext) context).lookup(KieSession.class);
                TimerManager timerManager = getTimerManager(kieSession);
                WorkflowProcessInstanceImpl processInstance = (WorkflowProcessInstanceImpl) kieSession.getProcessInstance(migrationSpec.getProcessInstanceId());
                for (Entry<Long, List<TimerInstance>> entry : timerMigrated.entrySet()) {
                    org.jbpm.workflow.instance.NodeInstance active = processInstance.getNodeInstance(entry.getKey(), false);
                    if (active instanceof TimerNodeInstance) {
                        TimerInstance timerInstance = entry.getValue().get(0);
                        long delay = timerInstance.getDelay() - (System.currentTimeMillis() - timerInstance.getActivated().getTime());
                        timerInstance.setDelay(delay);
                        timerManager.registerTimer(timerInstance, processInstance);
                        ((TimerNodeInstance) active).internalSetTimerId(timerInstance.getId());
                    } else if (active instanceof StateBasedNodeInstance) {
                        List<TimerInstance> timerInstances = entry.getValue();
                        List<Long> timers = new ArrayList<>();
                        for (TimerInstance timerInstance : timerInstances) {
                            long delay = timerInstance.getDelay() - (System.currentTimeMillis() - timerInstance.getActivated().getTime());
                            timerInstance.setDelay(delay);
                            timerManager.registerTimer(timerInstance, processInstance);
                            timers.add(timerInstance.getId());
                        }
                        ((StateBasedNodeInstance) active).internalSetTimerInstances(timers);
                    }
                }
                return null;
            }
        });
    } finally {
        manager.disposeRuntimeEngine(engine);
    }
}