org.glassfish.enterprise.concurrent.AbstractManagedExecutorService Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of javax.enterprise.concurrent Show documentation
Show all versions of javax.enterprise.concurrent Show documentation
Reference Implementation for JSR 236 - Concurrency Utilities for Java EE
/*
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
*
* Copyright (c) 2010-2013 Oracle and/or its affiliates. All rights reserved.
*
* The contents of this file are subject to the terms of either the GNU
* General Public License Version 2 only ("GPL") or the Common Development
* and Distribution License("CDDL") (collectively, the "License"). You
* may not use this file except in compliance with the License. You can
* obtain a copy of the License at
* https://glassfish.dev.java.net/public/CDDL+GPL_1_1.html
* or packager/legal/LICENSE.txt. See the License for the specific
* language governing permissions and limitations under the License.
*
* When distributing the software, include this License Header Notice in each
* file and include the License file at packager/legal/LICENSE.txt.
*
* GPL Classpath Exception:
* Oracle designates this particular file as subject to the "Classpath"
* exception as provided by Oracle in the GPL Version 2 section of the License
* file that accompanied this code.
*
* Modifications:
* If applicable, add the following below the License Header, with the fields
* enclosed by brackets [] replaced by your own identifying information:
* "Portions Copyright [year] [name of copyright owner]"
*
* Contributor(s):
* If you wish your version of this file to be governed by only the CDDL or
* only the GPL Version 2, indicate your decision by adding "[Contributor]
* elects to include this software in this distribution under the [CDDL or GPL
* Version 2] license." If you don't indicate a single choice of license, a
* recipient has the option to distribute your version of this file under
* either the CDDL, the GPL Version 2 or to extend the choice of license to
* its licensees as provided above. However, if you add GPL Version 2 code
* and therefore, elected the GPL Version 2 license, then the option applies
* only if the new code is made subject to such option by the copyright
* holder.
*/
package org.glassfish.enterprise.concurrent;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Iterator;
import java.util.List;
import java.util.concurrent.*;
import javax.enterprise.concurrent.ContextService;
import javax.enterprise.concurrent.ManagedExecutorService;
import org.glassfish.enterprise.concurrent.internal.ManagedFutureTask;
import org.glassfish.enterprise.concurrent.spi.ContextSetupProvider;
/**
* Abstract base class for {@code ManagedExecutorService} and
* {@code ManagedScheduledExecutorService}
* implementation classes. Lifecycle operations are available for use by the
* application server. Application components should be handed instances
* that extends from AbstractManagedExecutorServiceAdapter instead, which have
* their lifecycle operations disabled.
* Instances of subclasses of this class could be used by the Java EE
* product provider to control the life cycle.
*/
public abstract class AbstractManagedExecutorService
extends AbstractExecutorService implements ManagedExecutorService {
public enum RejectPolicy {
ABORT, RETRY_ABORT
};
public enum RunLocation {
LOCAL, DISTRIBUTABLE, DISTRIBUTABLE_WITH_AFFINITY
};
protected final String name;
protected final ContextSetupProvider contextSetupProvider;
protected final ContextServiceImpl contextService;
protected final ManagedThreadFactoryImpl managedThreadFactory;
protected RejectPolicy rejectPolicy;
protected RunLocation runLocation;
protected final boolean contextualCallback;
public AbstractManagedExecutorService(String name,
ManagedThreadFactoryImpl managedThreadFactory,
long hungTaskThreshold,
boolean longRunningTasks,
ContextServiceImpl contextService,
ContextSetupProvider contextCallback,
RejectPolicy rejectPolicy,
RunLocation runLocation,
boolean contextualCallback) {
this.name = name;
this.contextSetupProvider = contextCallback;
this.contextService = contextService;
this.rejectPolicy = rejectPolicy;
this.runLocation = runLocation;
this.contextualCallback = contextualCallback;
if (managedThreadFactory == null) {
managedThreadFactory = new ManagedThreadFactoryImpl(
name + "-ManagedThreadFactory",
null,
Thread.NORM_PRIORITY,
longRunningTasks);
}
managedThreadFactory.setHungTaskThreshold(hungTaskThreshold);
this.managedThreadFactory = managedThreadFactory;
}
protected T doInvokeAny(Collection> tasks, boolean timed, long nanos) throws InterruptedException, ExecutionException, TimeoutException {
// adopted from java.util.concurrent.AbstractExecutorService.doInvokeAny()
if (tasks == null) {
throw new NullPointerException();
}
int ntasks = tasks.size();
if (ntasks == 0) {
throw new IllegalArgumentException();
}
List> futures = new ArrayList>(ntasks);
ManagedExecutorCompletionService ecs = new ManagedExecutorCompletionService(this);
// For efficiency, especially in executors with limited
// parallelism, check to see if previously submitted tasks are
// done before submitting more of them. This interleaving
// plus the exception mechanics account for messiness of main
// loop.
try {
// Record exceptions so that if we fail to obtain any
// result, we can throw the last exception we got.
ExecutionException ee = null;
long lastTime = timed ? System.nanoTime() : 0;
Iterator> it = tasks.iterator();
// Start one task for sure; the rest incrementally
futures.add(ecs.submit(it.next()));
--ntasks;
int active = 1;
for (;;) {
Future f = ecs.poll();
if (f == null) {
if (ntasks > 0) {
--ntasks;
futures.add(ecs.submit(it.next()));
++active;
} else if (active == 0) {
break;
} else if (timed) {
f = ecs.poll(nanos, TimeUnit.NANOSECONDS);
if (f == null) {
throw new TimeoutException();
}
long now = System.nanoTime();
nanos -= now - lastTime;
lastTime = now;
} else {
f = ecs.take();
}
}
if (f != null) {
--active;
try {
return f.get();
} catch (ExecutionException eex) {
ee = eex;
} catch (RuntimeException rex) {
ee = new ExecutionException(rex);
}
}
}
if (ee == null) {
ee = new ExecutionException(null);
}
throw ee;
} finally {
for (Future f : futures) {
f.cancel(true);
}
}
}
public ContextSetupProvider getContextSetupProvider() {
return contextSetupProvider;
}
public ContextService getContextService() {
return contextService;
}
public boolean isContextualCallback() {
return contextualCallback;
}
public String[] getHungThreads() {
// TODO: implement me
return null;
}
public ManagedThreadFactoryImpl getManagedThreadFactory() {
return managedThreadFactory;
}
public String getName() {
return name;
}
public RejectPolicy getRejectPolicy() {
return rejectPolicy;
}
public RunLocation getRunLocation() {
return runLocation;
}
// managed objects methods
public String getObjectName() {
return null;
}
public String[] getThreads() {
return null;
}
@Override
public List> invokeAll(Collection> tasks) throws InterruptedException {
// code adopted from java.util.concurrent.AbstractExecutorService
if (tasks == null) {
throw new NullPointerException();
}
List> futures = new ArrayList>(tasks.size());
boolean done = false;
try {
for (Callable t : tasks) {
ManagedFutureTask f = getNewTaskFor(t);
futures.add(f);
executeManagedFutureTask(f);
}
for (Future f : futures) {
if (!f.isDone()) {
try {
f.get();
} catch (CancellationException ignore) {
} catch (ExecutionException ignore) {
}
}
}
done = true;
return futures;
} finally {
if (!done) {
for (Future f : futures) {
f.cancel(true);
}
}
}
}
@Override
public List> invokeAll(Collection> tasks, long timeout, TimeUnit unit) throws InterruptedException {
// code adopted from java.util.concurrent.AbstractExecutorService
if (tasks == null || unit == null) {
throw new NullPointerException();
}
long nanos = unit.toNanos(timeout);
List> futures = new ArrayList>(tasks.size());
boolean done = false;
try {
for (Callable t : tasks) {
futures.add(getNewTaskFor(t));
}
long lastTime = System.nanoTime();
// Interleave time checks and calls to execute in case
// executor doesn't have any/much parallelism.
Iterator> it = futures.iterator();
while (it.hasNext()) {
executeManagedFutureTask((ManagedFutureTask) (it.next()));
long now = System.nanoTime();
nanos -= now - lastTime;
lastTime = now;
if (nanos <= 0) {
return futures;
}
}
for (Future f : futures) {
if (!f.isDone()) {
if (nanos <= 0) {
return futures;
}
try {
f.get(nanos, TimeUnit.NANOSECONDS);
} catch (CancellationException ignore) {
} catch (ExecutionException ignore) {
} catch (TimeoutException toe) {
return futures;
}
long now = System.nanoTime();
nanos -= now - lastTime;
lastTime = now;
}
}
done = true;
return futures;
} finally {
if (!done) {
for (Future f : futures) {
f.cancel(true);
}
}
}
}
@Override
public T invokeAny(Collection> tasks) throws InterruptedException, ExecutionException {
try {
return doInvokeAny(tasks, false, 0);
} catch (TimeoutException cannotHappen) {
assert false;
return null;
}
}
@Override
public T invokeAny(Collection> tasks, long timeout, TimeUnit unit) throws InterruptedException, ExecutionException, TimeoutException {
return doInvokeAny(tasks, true, unit.toNanos(timeout));
}
public boolean isEventProvider() {
return true;
}
public boolean isStateManageable() {
return true;
}
public boolean isStatisticsProvider() {
return true;
}
/**
* @throws RejectedExecutionException {@inheritDoc}
* @throws NullPointerException {@inheritDoc}
*/
@Override
public Future submit(Runnable task) {
if (task == null) {
throw new NullPointerException();
}
ManagedFutureTask ftask = getNewTaskFor(task, null);
executeManagedFutureTask(ftask);
return ftask;
}
/**
* @throws RejectedExecutionException {@inheritDoc}
* @throws NullPointerException {@inheritDoc}
*/
@Override
public Future submit(Runnable task, T result) {
if (task == null) {
throw new NullPointerException();
}
ManagedFutureTask ftask = getNewTaskFor(task, result);
executeManagedFutureTask(ftask);
return ftask;
}
/**
* @throws RejectedExecutionException {@inheritDoc}
* @throws NullPointerException {@inheritDoc}
*/
@Override
public Future submit(Callable task) {
if (task == null) {
throw new NullPointerException();
}
ManagedFutureTask ftask = getNewTaskFor(task);
executeManagedFutureTask(ftask);
return ftask;
}
protected abstract ExecutorService getThreadPoolExecutor();
@Override
public abstract void execute(Runnable command);
/**
* Executes a ManagedFutureTask created by getNewTaskFor()
*
* @param task The ManagedFutureTask to be run
*/
protected void executeManagedFutureTask(ManagedFutureTask task) {
task.submitted();
getThreadPoolExecutor().execute(task);
}
@Override
public boolean awaitTermination(long timeout, TimeUnit unit) throws InterruptedException {
return getThreadPoolExecutor().awaitTermination(timeout, unit);
}
@Override
public boolean isShutdown() {
return getThreadPoolExecutor().isShutdown();
}
@Override
public boolean isTerminated() {
return getThreadPoolExecutor().isTerminated();
}
@Override
public void shutdown() {
getThreadPoolExecutor().shutdown();
}
@Override
public List shutdownNow() {
List runnable = getThreadPoolExecutor().shutdownNow();
// TODO: unwrap runnable from FutureTasks
// TODO: unwrap runnable from FutureTasks
return runnable;
}
/**
* newTaskFor from super class java.util.concurrent.AbstractExecutorService
* is protected access. This method allows newTaskFor to be called by
* other methods in this package that is not a subclass of AbstractExecutorService,
* and to return Object of type ManagedFutureTask, which extends from
* the return type of newTaskFor of FutureRunnable.
*/
protected abstract ManagedFutureTask getNewTaskFor(Runnable r, V result);
/**
* newTaskFor from super class java.util.concurrent.AbstractExecutorService
* is protected access. This method allows newTaskFor to be called by
* other methods in this package that is not a subclass of AbstractExecutorService,
* and to return Object of type ManagedFutureTask, which extends from
* the return type of newTaskFor of FutureRunnable.
*/
protected abstract ManagedFutureTask getNewTaskFor(Callable callable);
@Override
protected RunnableFuture newTaskFor(Runnable r, T result) {
return getNewTaskFor(r, result);
}
@Override
protected RunnableFuture newTaskFor(Callable callable) {
return getNewTaskFor(callable);
}
/**
* Returns the ManagedExecutorService instance that is used for passing
* as the executor argument of ManagedTaskListener calls
*
* @return the ManagedExecutorService instance that is used for passing
* as the executor argument of ManagedTaskListener calls
*
*/
public abstract ManagedExecutorService getExecutorForTaskListener();
}