/*
* JBoss, Home of Professional Open Source.
* Copyright 2006, Red Hat Middleware LLC, and individual contributors
* as indicated by the @author tags. See the copyright.txt file in the
* distribution for a full listing of individual contributors.
*
* This is free software; you can redistribute it and/or modify it
* under the terms of the GNU Lesser General Public License as
* published by the Free Software Foundation; either version 2.1 of
* the License, or (at your option) any later version.
*
* This software is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this software; if not, write to the Free
* Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
* 02110-1301 USA, or see the FSF site: http://www.fsf.org.
*/
package org.jboss.as.controller.remote;
import static java.security.AccessController.doPrivileged;
import static org.jboss.as.controller.logging.ControllerLogger.MGMT_OP_LOGGER;
import java.security.PrivilegedAction;
import java.util.concurrent.BlockingQueue;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.LinkedBlockingQueue;
import java.util.concurrent.RejectedExecutionException;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.ThreadFactory;
import java.util.concurrent.ThreadPoolExecutor;
import java.util.concurrent.TimeUnit;
import org.jboss.as.controller.ModelController;
import org.jboss.as.protocol.mgmt.support.ManagementChannelInitialization;
import org.jboss.msc.service.Service;
import org.jboss.msc.service.ServiceName;
import org.jboss.msc.service.StartContext;
import org.jboss.msc.service.StartException;
import org.jboss.msc.service.StopContext;
import org.jboss.msc.value.InjectedValue;
import org.jboss.threads.JBossThreadFactory;
/**
* Service used to create operation handlers per incoming channel
*
* @author <a href="kabir.khan@jboss.com">Kabir Khan</a>
*/
public abstract class AbstractModelControllerOperationHandlerFactoryService implements Service<AbstractModelControllerOperationHandlerFactoryService>, ManagementChannelInitialization {
public static final ServiceName OPERATION_HANDLER_NAME_SUFFIX = ServiceName.of("operation", "handler");
// The defaults if no executor was defined
private static final int WORK_QUEUE_SIZE = 512;
private static final int POOL_CORE_SIZE = 4;
private static final int POOL_MAX_SIZE = 4;
private final InjectedValue<ModelController> modelControllerValue = new InjectedValue<ModelController>();
private final InjectedValue<ExecutorService> executor = new InjectedValue<ExecutorService>();
private final InjectedValue<ScheduledExecutorService> scheduledExecutor = new InjectedValue<>();
private ResponseAttachmentInputStreamSupport responseAttachmentSupport;
private ExecutorService clientRequestExecutor;
/**
* Use to inject the model controller that will be the target of the operations
*
* @return the injected value holder
*/
public InjectedValue<ModelController> getModelControllerInjector() {
return modelControllerValue;
}
public InjectedValue<ExecutorService> getExecutorInjector() {
return executor;
}
public InjectedValue<ScheduledExecutorService> getScheduledExecutorInjector() {
return scheduledExecutor;
}
/** {@inheritDoc} */
@Override
public synchronized void start(StartContext context) throws StartException {
MGMT_OP_LOGGER.debugf("Starting operation handler service %s", context.getController().getName());
responseAttachmentSupport = new ResponseAttachmentInputStreamSupport(scheduledExecutor.getValue());
final BlockingQueue<Runnable> workQueue = new LinkedBlockingQueue<Runnable>(WORK_QUEUE_SIZE);
final ThreadFactory threadFactory = doPrivileged(new PrivilegedAction<JBossThreadFactory>() {
public JBossThreadFactory run() {
return new JBossThreadFactory(new ThreadGroup("management-handler-thread"), Boolean.FALSE, null, "%G - %t", null, null);
}
});
ThreadPoolExecutor executor = new ThreadPoolExecutor(POOL_CORE_SIZE, POOL_MAX_SIZE,
60L, TimeUnit.SECONDS, workQueue,
threadFactory);
// Allow the core threads to time out as well
executor.allowCoreThreadTimeOut(true);
this.clientRequestExecutor = executor;
}
/** {@inheritDoc} */
@Override
public synchronized void stop(final StopContext stopContext) {
final ExecutorService executorService = executor.getValue();
final Runnable task = new Runnable() {
@Override
public void run() {
try {
responseAttachmentSupport.shutdown();
// Shut down new requests to the client request executor,
// but don't mess with currently running tasks
clientRequestExecutor.shutdown();
} finally {
stopContext.complete();
}
}
};
try {
executorService.execute(task);
} catch (RejectedExecutionException e) {
task.run();
} finally {
stopContext.asynchronous();
}
}
/** {@inheritDoc} */
@Override
public synchronized AbstractModelControllerOperationHandlerFactoryService getValue() throws IllegalStateException {
return this;
}
protected ModelController getController() {
return modelControllerValue.getValue();
}
protected ExecutorService getExecutor() {
return executor.getValue();
}
protected ResponseAttachmentInputStreamSupport getResponseAttachmentSupport() {
return responseAttachmentSupport;
}
protected final ExecutorService getClientRequestExecutor() {
return clientRequestExecutor;
}
}