[JBPM-10209] Disposable handling

jbpm-flow/src/main/java/org/jbpm/process/core/DisposableRuntimeEngine.java

@@ -0,0 +1,24 @@
+/*
+ * Copyright 2023 Red Hat, Inc. and/or its affiliates.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.jbpm.process.core;
+
+import org.kie.internal.runtime.manager.Disposable;
+import org.kie.internal.runtime.manager.InternalRuntimeEngine;
+
+public interface DisposableRuntimeEngine extends InternalRuntimeEngine, Disposable {
+ boolean isDisposed();
+}

jbpm-flow/src/main/java/org/jbpm/process/core/timer/impl/GlobalTimerService.java

@@ -37,6 +37,7 @@
 import org.drools.core.time.impl.DefaultJobHandle;
 import org.drools.core.time.impl.TimerJobFactoryManager;
 import org.drools.core.time.impl.TimerJobInstance;
+import org.jbpm.process.core.DisposableRuntimeEngine;
 import org.jbpm.process.core.timer.GlobalSchedulerService;
 import org.jbpm.process.core.timer.NamedJobContext;
 import org.jbpm.process.instance.timer.TimerManager.ProcessJobContext;
@@ -420,6 +421,14 @@ public Environment getEnvironment() {
 
  return runtime.getKieSession().getEnvironment();
  }
+
+ public RuntimeEngine getEngine() {
+ return runtime;
+ }
+
+ public boolean isDisposed() {
+ return runtime instanceof DisposableRuntimeEngine && ((DisposableRuntimeEngine)runtime).isDisposed();
+ }
 
  }
 

jbpm-persistence/jbpm-persistence-jpa/src/main/java/org/jbpm/persistence/timer/GlobalJpaTimerJobInstance.java

@@ -65,7 +65,7 @@ public GlobalJpaTimerJobInstance(Job job, JobContext ctx, Trigger trigger,
  @Override
  public Void call() throws Exception {
  AsyncExecutionMarker.markAsync();
- ExecutableRunner runner = null;
+ ExecutableRunner<?> runner = null;
  TransactionManager jtaTm = null;
  boolean success = false;
  try { 
@@ -88,16 +88,16 @@ public Void call() throws Exception {
  success = true;
  return null;
  } catch( Exception e ) { 
-  e.printStackTrace();
+  logger.error("Exception executing timer", e);
  success = false;
  throw e;
  } finally {
  AsyncExecutionMarker.reset();
- if (runner != null && runner instanceof DisposableCommandService) {
-  if (allowedToDispose(((DisposableCommandService) runner).getEnvironment())) {
-  logger.debug("Allowed to dispose command service from global timer job instance");
-  ((DisposableCommandService) runner).dispose();
-  }
+ if (runner != null && runner instanceof DisposableCommandService) { 
+  if (allowedToDispose(((DisposableCommandService) runner))) {
+  logger.debug("Allowed to dispose command service from global timer job instance");
+  ((DisposableCommandService) runner).dispose();
+  }
  }
  closeTansactionIfNeeded(jtaTm, success);
  }
@@ -123,7 +123,11 @@ public String toString() {
  return "GlobalJpaTimerJobInstance [timerServiceId=" + timerServiceId
  + ", getJobHandle()=" + getJobHandle() + "]";
  }
-
+
+ private boolean allowedToDispose(DisposableCommandService disposableCommandService) {
+ return !disposableCommandService.isDisposed() && allowedToDispose (disposableCommandService.getEnvironment());
+ }
+
  protected boolean allowedToDispose(Environment environment) {
  if (hasEnvironmentEntry(environment, "IS_JTA_TRANSACTION", false)) {
  return true;

jbpm-runtime-manager/src/main/java/org/jbpm/runtime/manager/impl/RuntimeEngineImpl.java

@@ -19,14 +19,13 @@
 import java.util.concurrent.CopyOnWriteArrayList;
 
 import org.jbpm.process.audit.JPAAuditLogService;
+import org.jbpm.process.core.DisposableRuntimeEngine;
 import org.kie.api.runtime.KieSession;
 import org.kie.api.runtime.manager.Context;
 import org.kie.api.runtime.manager.RuntimeManager;
 import org.kie.api.runtime.manager.audit.AuditService;
 import org.kie.api.task.TaskService;
-import org.kie.internal.runtime.manager.Disposable;
 import org.kie.internal.runtime.manager.DisposeListener;
-import org.kie.internal.runtime.manager.InternalRuntimeEngine;
 import org.kie.internal.runtime.manager.InternalRuntimeManager;
 import org.kie.internal.runtime.manager.SessionNotFoundException;
 
@@ -36,7 +35,7 @@
  * and work item handlers might be interested in receiving notification when the runtime engine is disposed of,
  * in order deactivate themselves too and not receive any other events.
  */
-public class RuntimeEngineImpl implements InternalRuntimeEngine, Disposable {
+public class RuntimeEngineImpl implements DisposableRuntimeEngine {
 
  private RuntimeEngineInitlializer initializer;
  private Context<?> context;
@@ -143,6 +142,7 @@ public void setManager(RuntimeManager manager) {
  this.manager = manager;
  }
 
+ @Override
  public boolean isDisposed() {
  return disposed;
  }