Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance. Project price only 1 $
You can buy this project and download/modify it how often you want.
/*
* Copyright (c) 2010, 2018 Oracle and/or its affiliates. All rights reserved.
* Copyright (c) 2018 [Payara Foundation and/or its affiliates].
*
* This program and the accompanying materials are made available under the
* terms of the Eclipse Public License v. 2.0, which is available at
* http://www.eclipse.org/legal/epl-2.0.
*
* This Source Code may also be made available under the following Secondary
* Licenses when the conditions for such availability set forth in the
* Eclipse Public License v. 2.0 are satisfied: GNU General Public License,
* version 2 with the GNU Classpath Exception, which is available at
* https://www.gnu.org/software/classpath/license.html.
*
* SPDX-License-Identifier: EPL-2.0 OR GPL-2.0 WITH Classpath-exception-2.0
*/
package org.glassfish.enterprise.concurrent.internal;
import java.util.Map;
import java.util.concurrent.Callable;
import java.util.concurrent.CancellationException;
import java.util.concurrent.Future;
import java.util.concurrent.FutureTask;
import jakarta.enterprise.concurrent.AbortedException;
import jakarta.enterprise.concurrent.ContextService;
import jakarta.enterprise.concurrent.ManagedTask;
import jakarta.enterprise.concurrent.ManagedTaskListener;
import org.glassfish.enterprise.concurrent.AbstractManagedExecutorService;
import org.glassfish.enterprise.concurrent.spi.ContextHandle;
import org.glassfish.enterprise.concurrent.spi.ContextSetupProvider;
/**
* Future implementation to be returned by methods in ManagedExecutorSerivceImpl.
* This class is responsible for saving and restoring thread context, as well
* as invoking ManagedTaskListener methods.
*/
public class ManagedFutureTask extends FutureTask implements Future {
final protected AbstractManagedExecutorService executor;
final protected ManagedTaskListener taskListener;
protected ContextHandle contextHandleForSetup = null;
protected ContextHandle contextHandleForReset = null;
protected Object task;
protected Throwable taskRunThrowable;
protected TaskDoneCallback taskDoneCallback;
final boolean isContextualCallback;
IllegalStateException contextSetupException = null;
public ManagedFutureTask(
AbstractManagedExecutorService executor,
Runnable runnable,
V result) {
super(runnable, result);
this.task = runnable;
this.executor = executor;
this.taskListener = getManagedTaskListener(task);
this.isContextualCallback = isTaskContextualCallback(task) || executor.isContextualCallback();
captureContext(executor);
}
public ManagedFutureTask(
AbstractManagedExecutorService executor,
Callable callable) {
super(callable);
this.task = callable;
this.executor = executor;
this.taskListener = getManagedTaskListener(task);
this.isContextualCallback = isTaskContextualCallback(task) || executor.isContextualCallback();
captureContext (executor);
}
private ManagedTaskListener getManagedTaskListener(Object task) {
if (task instanceof ManagedTask) {
return ((ManagedTask) task).getManagedTaskListener();
}
return null;
}
private boolean isTaskContextualCallback(Object task) {
// Contextual callback no longer specified through execution properties
return false;
}
protected final void captureContext(AbstractManagedExecutorService executor) {
ContextSetupProvider contextSetupProvider = executor.getContextSetupProvider();
ContextService contextService = executor.getContextService();
if (contextService != null && contextSetupProvider != null) {
contextHandleForSetup = contextSetupProvider.saveContext(contextService);
}
}
public void setupContext() {
ContextSetupProvider contextSetupProvider = executor.getContextSetupProvider();
if (contextSetupProvider != null) {
try {
contextHandleForReset = contextSetupProvider.setup(contextHandleForSetup);
} catch (IllegalStateException ex) {
// context handle not in valid state. Do not run the task.
contextSetupException = ex;
}
}
}
public void resetContext() {
if (contextSetupException == null) {
// only call reset() if setupContext() was called successfully
ContextSetupProvider contextSetupProvider = executor.getContextSetupProvider();
if (contextSetupProvider != null) {
contextSetupProvider.reset(contextHandleForReset);
}
}
}
@Override
public void run() {
if (contextSetupException == null) {
super.run();
}
else {
abort();
}
}
@Override
public boolean runAndReset() {
if (contextSetupException == null) {
return super.runAndReset();
}
else {
abort();
}
return false;
}
@Override
public boolean cancel(boolean mayInterruptIfRunning) {
boolean result = super.cancel(mayInterruptIfRunning);
if (result && taskListener != null) {
try {
if (isContextualCallback) {
setupContext();
}
taskListener.taskAborted(this,
executor.getExecutorForTaskListener(),
task,
new CancellationException());
}
finally {
if (isContextualCallback) {
resetContext();
}
}
}
return result;
}
public void submitted() {
if (taskListener != null) {
try {
if (isContextualCallback) {
setupContext();
}
taskListener.taskSubmitted(this,
executor.getExecutorForTaskListener(),
task);
}
finally {
if (isContextualCallback) {
resetContext();
}
}
}
}
@Override
protected void done() {
// for calling taskDone for cancelled tasks
super.done();
if (taskDoneCallback != null) {
taskDoneCallback.taskDone(this);
}
if (taskListener != null && isCancelled()) {
try {
if (isContextualCallback) {
setupContext();
}
taskListener.taskDone(this,
executor.getExecutorForTaskListener(),
task,
new CancellationException());
}
finally {
if (isContextualCallback) {
resetContext();
}
}
}
}
@Override
protected void setException(Throwable t) {
super.setException(t);
taskRunThrowable = t;
}
public void starting(Thread t) {
if (executor.getManagedThreadFactory() != null) {
executor.getManagedThreadFactory().taskStarting(t, this);
}
if (taskListener != null) {
taskListener.taskStarting(this,
executor.getExecutorForTaskListener(),
task);
}
}
/**
* Call by ThreadPoolExecutor after a task is done execution.
* This is called on the thread where the task is run, so there is no
* need to set up thread context before calling the ManagedTaskListener
* callback method.
*
* @param t any runtime exception encountered during executing the task. But
* any Throwable thrown during executing of running the task would have
* been caught by FutureTask and would have been set by setException(). So
* t is ignored here.
*/
public void done(Throwable t) {
if (executor.getManagedThreadFactory() != null) {
executor.getManagedThreadFactory().taskDone(Thread.currentThread());
}
if (taskListener != null) {
taskListener.taskDone(this,
executor.getExecutorForTaskListener(),
task,
t != null? t: taskRunThrowable);
}
}
public void setTaskDoneCallback(TaskDoneCallback taskDoneCallback) {
this.taskDoneCallback = taskDoneCallback;
}
public String getTaskIdentityName() {
if (task instanceof ManagedTask) {
Map executionProperties = ((ManagedTask)task).getExecutionProperties();
if (executionProperties != null) {
String taskName = executionProperties.get(ManagedTask.IDENTITY_NAME);
if (taskName != null) {
return taskName;
}
}
}
// if a name is not provided for the task, use toString() as the name
return task.toString();
}
private void abort() {
// context handle not in valid state, throws AbortedException and
// do not run the task
AbortedException ex = new AbortedException(contextSetupException.getMessage());
setException(ex);
if (taskListener != null) {
// notify listener. No need to set context here as it wouldn't work
// anyway
taskListener.taskAborted(this,
executor.getExecutorForTaskListener(),
task,
ex);
}
}
}