Examples with ServiceController org.jboss.msc.service.ServiceController used on opensource projects

Example 46 with ServiceController

use of org.jboss.msc.service.ServiceController in project wildfly by wildfly.

the class RaOperationUtil method restartIfPresent.

public static ServiceName restartIfPresent(OperationContext context, final String raName, final String id) throws OperationFailedException {
    final ServiceName raDeploymentServiceName = ConnectorServices.getDeploymentServiceName(raName, id);
    final ServiceRegistry registry = context.getServiceRegistry(true);
    ServiceController raServiceController = registry.getService(raDeploymentServiceName);
    if (raServiceController != null) {
        final org.jboss.msc.service.ServiceController.Mode originalMode = raServiceController.getMode();
        raServiceController.addListener(new AbstractServiceListener() {

            @Override
            public void transition(ServiceController controller, ServiceController.Transition transition) {
                switch(transition) {
                    case STOPPING_to_DOWN:
                        try {
                            final ServiceController<?> RaxmlController = registry.getService(ServiceName.of(ConnectorServices.RA_SERVICE, id));
                            Activation raxml = (Activation) RaxmlController.getValue();
                            ((ResourceAdapterXmlDeploymentService) controller.getService()).setRaxml(raxml);
                            controller.compareAndSetMode(ServiceController.Mode.NEVER, originalMode);
                        } finally {
                            controller.removeListener(this);
                        }
                }
            }

            @Override
            public void listenerAdded(ServiceController controller) {
                controller.setMode(ServiceController.Mode.NEVER);
            }
        });
        return raDeploymentServiceName;
    } else {
        return null;
    }
}

Example 47 with ServiceController

use of org.jboss.msc.service.ServiceController in project wildfly by wildfly.

the class RaRemove method execute.

public void execute(OperationContext context, ModelNode operation) throws OperationFailedException {
    final ModelNode opAddr = operation.require(OP_ADDR);
    final String idName = PathAddress.pathAddress(opAddr).getLastElement().getValue();
    final boolean isModule;
    // Compensating is add
    final ModelNode model = context.readResource(PathAddress.EMPTY_ADDRESS, false).getModel();
    final String archiveOrModuleName;
    if (!model.hasDefined(ARCHIVE.getName()) && !model.hasDefined(MODULE.getName())) {
        throw ConnectorLogger.ROOT_LOGGER.archiveOrModuleRequired();
    }
    if (model.get(ARCHIVE.getName()).isDefined()) {
        isModule = false;
        archiveOrModuleName = model.get(ARCHIVE.getName()).asString();
    } else {
        isModule = true;
        archiveOrModuleName = model.get(MODULE.getName()).asString();
    }
    final ModelNode compensating = Util.getEmptyOperation(ADD, opAddr);
    if (model.hasDefined(RESOURCEADAPTERS_NAME)) {
        for (ModelNode raNode : model.get(RESOURCEADAPTERS_NAME).asList()) {
            ModelNode raCompensatingNode = raNode.clone();
            compensating.get(RESOURCEADAPTERS_NAME).add(raCompensatingNode);
        }
    }
    context.removeResource(PathAddress.EMPTY_ADDRESS);
    context.addStep(new OperationStepHandler() {

        public void execute(OperationContext context, ModelNode operation) throws OperationFailedException {
            final boolean wasActive;
            wasActive = RaOperationUtil.removeIfActive(context, archiveOrModuleName, idName);
            if (wasActive) {
                if (!context.isResourceServiceRestartAllowed()) {
                    context.reloadRequired();
                    context.completeStep(new OperationContext.RollbackHandler() {

                        @Override
                        public void handleRollback(OperationContext context, ModelNode operation) {
                            context.revertReloadRequired();
                        }
                    });
                    return;
                }
            }
            ServiceName raServiceName = ServiceName.of(ConnectorServices.RA_SERVICE, idName);
            ServiceController<?> serviceController = context.getServiceRegistry(false).getService(raServiceName);
            final ModifiableResourceAdapter resourceAdapter;
            if (serviceController != null) {
                resourceAdapter = (ModifiableResourceAdapter) serviceController.getValue();
            } else {
                resourceAdapter = null;
            }
            final List<ServiceName> serviceNameList = context.getServiceRegistry(false).getServiceNames();
            for (ServiceName name : serviceNameList) {
                if (raServiceName.isParentOf(name)) {
                    context.removeService(name);
                }
            }
            if (model.get(MODULE.getName()).isDefined()) {
                //ServiceName deploymentServiceName = ConnectorServices.getDeploymentServiceName(model.get(MODULE.getName()).asString(),raId);
                //context.removeService(deploymentServiceName);
                ServiceName deployerServiceName = ConnectorServices.RESOURCE_ADAPTER_DEPLOYER_SERVICE_PREFIX.append(idName);
                context.removeService(deployerServiceName);
                ServiceName inactiveServiceName = ConnectorServices.INACTIVE_RESOURCE_ADAPTER_SERVICE.append(idName);
                context.removeService(inactiveServiceName);
            }
            context.removeService(raServiceName);
            context.completeStep(new OperationContext.RollbackHandler() {

                @Override
                public void handleRollback(OperationContext context, ModelNode operation) {
                    if (resourceAdapter != null) {
                        List<ServiceController<?>> newControllers = new LinkedList<ServiceController<?>>();
                        if (model.get(ARCHIVE.getName()).isDefined()) {
                            RaOperationUtil.installRaServices(context, idName, resourceAdapter, newControllers);
                        } else {
                            try {
                                RaOperationUtil.installRaServicesAndDeployFromModule(context, idName, resourceAdapter, archiveOrModuleName, newControllers);
                            } catch (OperationFailedException e) {
                            }
                        }
                        try {
                            if (wasActive) {
                                RaOperationUtil.activate(context, idName, archiveOrModuleName);
                            }
                        } catch (OperationFailedException e) {
                        }
                    }
                }
            });
        }
    }, OperationContext.Stage.RUNTIME);
    context.stepCompleted();
}

Example 48 with ServiceController

use of org.jboss.msc.service.ServiceController in project wildfly by wildfly.

the class GetInstalledDriverOperationHandler method execute.

@Override
public void execute(final OperationContext context, final ModelNode operation) throws OperationFailedException {
    validator.validate(operation);
    final String name = operation.require(DRIVER_NAME.getName()).asString();
    if (context.isNormalServer()) {
        context.addStep(new OperationStepHandler() {

            @Override
            public void execute(final OperationContext context, final ModelNode operation) throws OperationFailedException {
                ServiceController<?> sc = context.getServiceRegistry(false).getRequiredService(ConnectorServices.JDBC_DRIVER_REGISTRY_SERVICE);
                DriverRegistry driverRegistry = DriverRegistry.class.cast(sc.getValue());
                ModelNode result = new ModelNode();
                InstalledDriver driver = driverRegistry.getInstalledDriver(name);
                ModelNode driverNode = new ModelNode();
                driverNode.get(DRIVER_NAME.getName()).set(driver.getDriverName());
                if (driver.isFromDeployment()) {
                    driverNode.get(DEPLOYMENT_NAME.getName()).set(driver.getDriverName());
                    driverNode.get(DRIVER_MODULE_NAME.getName());
                    driverNode.get(MODULE_SLOT.getName());
                    driverNode.get(DRIVER_XA_DATASOURCE_CLASS_NAME.getName());
                } else {
                    driverNode.get(DEPLOYMENT_NAME.getName());
                    driverNode.get(DRIVER_MODULE_NAME.getName()).set(driver.getModuleName().getName());
                    driverNode.get(MODULE_SLOT.getName()).set(driver.getModuleName() != null ? driver.getModuleName().getSlot() : "");
                    driverNode.get(DRIVER_XA_DATASOURCE_CLASS_NAME.getName()).set(driver.getXaDataSourceClassName());
                }
                driverNode.get(DRIVER_CLASS_NAME.getName()).set(driver.getDriverClassName());
                driverNode.get(DRIVER_MAJOR_VERSION.getName()).set(driver.getMajorVersion());
                driverNode.get(DRIVER_MINOR_VERSION.getName()).set(driver.getMinorVersion());
                driverNode.get(JDBC_COMPLIANT.getName()).set(driver.isJdbcCompliant());
                result.add(driverNode);
                context.getResult().set(result);
            }
        }, OperationContext.Stage.RUNTIME);
    }
}

Example 49 with ServiceController

use of org.jboss.msc.service.ServiceController in project wildfly by wildfly.

the class EnlistmentTraceAttributeWriteHandler method applyUpdateToRuntime.

@Override
protected boolean applyUpdateToRuntime(final OperationContext context, final ModelNode operation, final String parameterName, final ModelNode newValue, final ModelNode currentValue, final HandbackHolder<List<ConnectionManager>> handbackHolder) throws OperationFailedException {
    final String jndiName = context.readResource(PathAddress.EMPTY_ADDRESS).getModel().get(Constants.JNDINAME.getName()).asString();
    final ServiceController<?> managementRepoService = context.getServiceRegistry(false).getService(ConnectorServices.MANAGEMENT_REPOSITORY_SERVICE);
    Boolean boolValue = Constants.ENLISTMENT_TRACE.resolveValue(context, newValue).asBoolean();
    try {
        final ManagementRepository repository = (ManagementRepository) managementRepoService.getValue();
        if (repository.getConnectors() != null) {
            List<ConnectionManager> handback = new LinkedList<>();
            repository.getConnectors().stream().forEach(connector -> connector.getConnectionManagers().stream().filter(cm -> jndiName.equalsIgnoreCase(cm.getUniqueId())).forEach(cm -> {
                cm.setEnlistmentTrace(boolValue);
                handback.add(cm);
            }));
            handbackHolder.setHandback(handback);
        }
    } catch (Exception e) {
        throw new OperationFailedException(ConnectorLogger.ROOT_LOGGER.failedToSetAttribute(e.getLocalizedMessage()));
    }
    return false;
}

Example 50 with ServiceController

use of org.jboss.msc.service.ServiceController in project wildfly by wildfly.

the class InstancesService method removeInstance.

private static void removeInstance(ServiceRegistry registry, BeanState state, final Class<?> clazz, Object bean) {
    if (clazz == null)
        return;
    ServiceController controller = registry.getService(BeanMetaDataConfig.toInstancesName(clazz, state));
    if (controller != null) {
        InstancesService service = (InstancesService) controller.getService();
        synchronized (clazz) {
            service.instances.remove(bean);
            invokeCallbacks(uncallbacks, state, clazz, bean);
            if (service.instances.isEmpty()) {
                beans.remove(new TypeBeanStateKey(clazz, state));
                controller.setMode(ServiceController.Mode.REMOVE);
            }
        }
    }
    removeInstance(registry, state, clazz.getSuperclass(), bean);
    Class<?>[] ifaces = clazz.getInterfaces();
    for (Class<?> iface : ifaces) removeInstance(registry, state, iface, bean);
}