/*
* JBoss, Home of Professional Open Source.
* Copyright 2013, Red Hat, Inc., 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.wildfly.extension.batch;
import static org.jboss.as.controller.descriptions.ModelDescriptionConstants.SUBSYSTEM;
import javax.xml.stream.XMLStreamException;
import javax.xml.stream.XMLStreamWriter;
import org.jberet.spi.ContextClassLoaderJobOperatorContextSelector;
import org.jberet.spi.JobExecutor;
import org.jberet.spi.JobOperatorContext;
import org.jboss.as.controller.AbstractBoottimeAddStepHandler;
import org.jboss.as.controller.AttributeDefinition;
import org.jboss.as.controller.DefaultAttributeMarshaller;
import org.jboss.as.controller.ModelVersion;
import org.jboss.as.controller.OperationContext;
import org.jboss.as.controller.OperationFailedException;
import org.jboss.as.controller.PathAddress;
import org.jboss.as.controller.PathElement;
import org.jboss.as.controller.ReloadRequiredRemoveStepHandler;
import org.jboss.as.controller.ReloadRequiredWriteAttributeHandler;
import org.jboss.as.controller.SimpleAttributeDefinition;
import org.jboss.as.controller.SimpleAttributeDefinitionBuilder;
import org.jboss.as.controller.SimpleResourceDefinition;
import org.jboss.as.controller.descriptions.ModelDescriptionConstants;
import org.jboss.as.controller.operations.common.GenericSubsystemDescribeHandler;
import org.jboss.as.controller.operations.validation.EnumValidator;
import org.jboss.as.controller.registry.ManagementResourceRegistration;
import org.jboss.as.server.AbstractDeploymentChainStep;
import org.jboss.as.server.DeploymentProcessorTarget;
import org.jboss.as.server.deployment.Phase;
import org.jboss.as.threads.ManagedJBossThreadPoolExecutorService;
import org.jboss.as.threads.ThreadFactoryResolver;
import org.jboss.as.threads.ThreadFactoryResourceDefinition;
import org.jboss.as.threads.ThreadsServices;
import org.jboss.as.threads.UnboundedQueueThreadPoolResourceDefinition;
import org.jboss.dmr.ModelNode;
import org.jboss.dmr.ModelType;
import org.jboss.msc.service.ServiceController;
import org.jboss.msc.service.ServiceTarget;
import org.wildfly.extension.batch._private.Capabilities;
import org.wildfly.extension.batch.deployment.BatchEnvironmentProcessor;
import org.wildfly.extension.batch.jberet.BatchConfiguration;
import org.wildfly.extension.batch.jberet.DefaultBatchEnvironment;
import org.wildfly.extension.batch.jberet.deployment.BatchDependencyProcessor;
import org.wildfly.extension.batch.jberet.deployment.BatchDeploymentResourceProcessor;
import org.wildfly.extension.batch.jberet.thread.pool.JobExecutorService;
import org.wildfly.extension.batch.job.repository.JobRepositoryFactory;
import org.wildfly.extension.batch.job.repository.JobRepositoryType;
import org.wildfly.extension.requestcontroller.RequestControllerExtension;
public class BatchSubsystemDefinition extends SimpleResourceDefinition {
/**
* The name of our subsystem within the model.
*/
public static final String NAME = "batch";
public static final PathElement SUBSYSTEM_PATH = PathElement.pathElement(SUBSYSTEM, NAME);
static final PathElement THREAD_POOL_PATH = PathElement.pathElement(BatchConstants.THREAD_POOL, BatchConstants.THREAD_POOL_NAME);
@Deprecated
static final SimpleAttributeDefinition JOB_REPOSITORY_TYPE = SimpleAttributeDefinitionBuilder.create("job-repository-type", ModelType.STRING, true)
.setAllowExpression(false)
.setAttributeMarshaller(new DefaultAttributeMarshaller() {
@Override
public void marshallAsElement(final AttributeDefinition attribute, final ModelNode resourceModel, final XMLStreamWriter writer) throws XMLStreamException {
marshallAsElement(attribute, resourceModel, true, writer);
}
@Override
public void marshallAsElement(final AttributeDefinition attribute, final ModelNode resourceModel, final boolean marshallDefault, final XMLStreamWriter writer) throws XMLStreamException {
// Write the job-repository
writer.writeStartElement(Element.JOB_REPOSITORY.getLocalName());
// The value is the job repository type
final String value = resourceModel.get(JOB_REPOSITORY_TYPE.getName()).asString();
if (JobRepositoryType.JDBC.toString().equals(value)) {
writer.writeStartElement(Element.JDBC.getLocalName());
final PathElement jdbcPath = JobRepositoryDefinition.JDBC.getPathElement();
final ModelNode jdbcModel = resourceModel.clone().get(jdbcPath.getKey(), jdbcPath.getValue());
if (jdbcModel.isDefined()) {
JobRepositoryDefinition.JNDI_NAME.marshallAsAttribute(jdbcModel, false, writer);
}
writer.writeEndElement();
} else {
// Write in-memory by default
writer.writeStartElement(Element.IN_MEMORY.getLocalName());
writer.writeEndElement();
}
writer.writeEndElement();
}
})
.setDefaultValue(new ModelNode(JobRepositoryType.IN_MEMORY.toString()))
.setValidator(new EnumValidator<>(JobRepositoryType.class, true, true))
.setRestartJVM()
.setDeprecated(ModelVersion.create(1, 0, 0), false)
.build();
public static final BatchSubsystemDefinition INSTANCE = new BatchSubsystemDefinition();
private BatchSubsystemDefinition() {
super(SUBSYSTEM_PATH, BatchResourceDescriptionResolver.getResourceDescriptionResolver(), BatchSubsystemAdd.INSTANCE,
ReloadRequiredRemoveStepHandler.INSTANCE);
}
@Override
public void registerOperations(final ManagementResourceRegistration resourceRegistration) {
super.registerOperations(resourceRegistration);
resourceRegistration.registerOperationHandler(GenericSubsystemDescribeHandler.DEFINITION, GenericSubsystemDescribeHandler.INSTANCE);
}
@Override
public void registerChildren(final ManagementResourceRegistration resourceRegistration) {
super.registerChildren(resourceRegistration);
// thread-pool resource
final UnboundedQueueThreadPoolResourceDefinition threadPoolResource = UnboundedQueueThreadPoolResourceDefinition.create(THREAD_POOL_PATH,
BatchThreadFactoryResolver.INSTANCE, BatchServiceNames.BASE_BATCH_THREAD_POOL_NAME, false);
resourceRegistration.registerSubModel(threadPoolResource);
// thread-factory resource
final ThreadFactoryResourceDefinition threadFactoryResource = new ThreadFactoryResourceDefinition();
resourceRegistration.registerSubModel(threadFactoryResource);
resourceRegistration.registerSubModel(JobRepositoryDefinition.JDBC);
}
@Override
public void registerAttributes(ManagementResourceRegistration resourceRegistration) {
resourceRegistration.registerReadWriteAttribute(JOB_REPOSITORY_TYPE, null, new ReloadRequiredWriteAttributeHandler(JOB_REPOSITORY_TYPE));
}
private static class BatchThreadFactoryResolver extends ThreadFactoryResolver.SimpleResolver {
static final BatchThreadFactoryResolver INSTANCE = new BatchThreadFactoryResolver();
private BatchThreadFactoryResolver() {
super(ThreadsServices.FACTORY);
}
@Override
protected String getThreadGroupName(String threadPoolName) {
return "Batch Thread";
}
}
/**
* Handler responsible for adding the subsystem resource to the model.
*/
static class BatchSubsystemAdd extends AbstractBoottimeAddStepHandler {
static final BatchSubsystemAdd INSTANCE = new BatchSubsystemAdd();
private final ContextClassLoaderJobOperatorContextSelector selector;
private BatchSubsystemAdd() {
super(Capabilities.BATCH_CONFIGURATION_CAPABILITY);
selector = new ContextClassLoaderJobOperatorContextSelector(() -> JobOperatorContext.create(DefaultBatchEnvironment.INSTANCE));
JobOperatorContext.setJobOperatorContextSelector(selector);
}
@Override
protected void populateModel(final ModelNode operation, final ModelNode model) throws OperationFailedException {
model.setEmptyObject();
JOB_REPOSITORY_TYPE.validateAndSet(operation, model);
}
@Override
protected void performBoottime(final OperationContext context, final ModelNode operation, final ModelNode model)
throws OperationFailedException {
// Check if the request-controller subsystem exists
final boolean rcPresent = context.readResourceFromRoot(PathAddress.EMPTY_ADDRESS).hasChild(PathElement.pathElement(ModelDescriptionConstants.SUBSYSTEM, RequestControllerExtension.SUBSYSTEM_NAME));
context.addStep(new AbstractDeploymentChainStep() {
public void execute(DeploymentProcessorTarget processorTarget) {
processorTarget.addDeploymentProcessor(BatchSubsystemDefinition.NAME,
Phase.DEPENDENCIES, Phase.DEPENDENCIES_BATCH, new BatchDependencyProcessor());
processorTarget.addDeploymentProcessor(BatchSubsystemDefinition.NAME,
Phase.POST_MODULE, Phase.POST_MODULE_BATCH_ENVIRONMENT, new BatchEnvironmentProcessor(rcPresent, selector));
processorTarget.addDeploymentProcessor(BatchSubsystemDefinition.NAME,
Phase.INSTALL, Phase.INSTALL_BATCH_RESOURCES, new BatchDeploymentResourceProcessor(NAME));
}
}, OperationContext.Stage.RUNTIME);
final ServiceTarget target = context.getServiceTarget();
final JobExecutorService service = new JobExecutorService();
target.addService(BatchServiceNames.BATCH_JOB_EXECUTOR_NAME, service)
.addDependency(BatchServiceNames.BATCH_THREAD_POOL_NAME,
ManagedJBossThreadPoolExecutorService.class,
service.getThreadPoolInjector()
)
// Only start this service if there are deployments present, allow it to be stopped as deployments
// are removed.
.setInitialMode(ServiceController.Mode.ON_DEMAND)
.install();
// Determine the repository type
final String repositoryType = JOB_REPOSITORY_TYPE.resolveModelAttribute(context, model).asString();
JobRepositoryFactory.getInstance().setJobRepositoryType(repositoryType);
final DefaultConfigurationService configurationService = new DefaultConfigurationService();
target.addService(context.getCapabilityServiceName(Capabilities.BATCH_CONFIGURATION_CAPABILITY.getName(), BatchConfiguration.class), configurationService)
.addDependency(BatchServiceNames.BATCH_JOB_EXECUTOR_NAME, JobExecutor.class, configurationService.getJobExecutorInjector())
// Only start this service if there are deployments present, allow it to be stopped as deployments
// are removed.
.setInitialMode(ServiceController.Mode.ON_DEMAND)
.install();
}
}
}