Examples with ActiveMQActivationService org.wildfly.extension.messaging.activemq.ActiveMQActivationService used on opensource projects

Example 1 with ActiveMQActivationService

use of org.wildfly.extension.messaging.activemq.ActiveMQActivationService in project wildfly by wildfly.

the class JMSService method doStart.

private synchronized void doStart(final StartContext context) throws StartException {
    final ServiceContainer serviceContainer = context.getController().getServiceContainer();
    ClassLoader oldTccl = WildFlySecurityManager.setCurrentContextClassLoaderPrivileged(getClass());
    try {
        jmsServer = new JMSServerManagerImpl(activeMQServer.getValue(), new WildFlyBindingRegistry(context.getController().getServiceContainer())) {

            @Override
            public void stop(ActiveMQServer server) {
            // Suppress ARTEMIS-2438
            }
        };
        activeMQServer.getValue().registerActivationFailureListener(e -> {
            StartException se = new StartException(e);
            context.failed(se);
        });
        activeMQServer.getValue().registerActivateCallback(new ActivateCallback() {

            private volatile ServiceController<Void> activeMQActivationController;

            public void preActivate() {
            }

            public void activated() {
                if (overrideInVMSecurity) {
                    activeMQServer.getValue().getRemotingService().allowInvmSecurityOverride(new ActiveMQPrincipal(DefaultCredentials.getUsername(), DefaultCredentials.getPassword()));
                }
                // ActiveMQ backup server is activated during failover while the WildFly server is shutting down.
                if (serviceContainer.isShutdown()) {
                    return;
                }
                if (activeMQActivationController == null) {
                    activeMQActivationController = serviceContainer.addService(ActiveMQActivationService.getServiceName(serverServiceName), new ActiveMQActivationService()).setInitialMode(Mode.ACTIVE).install();
                } else {
                    activeMQActivationController.setMode(ACTIVE);
                }
            }

            @Override
            public void activationComplete() {
            }

            public void deActivate() {
                // and *not* during AS7 service container shutdown or reload (AS7-6840 / AS7-6881)
                if (activeMQActivationController != null && !activeMQActivationController.getState().in(STOPPING, REMOVED)) {
                    // [WFLY-4597] When Artemis is deactivated during failover, we block until its
                    // activation controller is REMOVED before giving back control to Artemis.
                    // This allow to properly stop any service depending on the activation controller
                    // and avoid spurious warning messages because the resources used by the services
                    // are stopped outside the control of the services.
                    final CountDownLatch latch = new CountDownLatch(1);
                    activeMQActivationController.compareAndSetMode(ACTIVE, REMOVE);
                    activeMQActivationController.addListener(new LifecycleListener() {

                        @Override
                        public void handleEvent(ServiceController<?> controller, LifecycleEvent event) {
                            if (event == LifecycleEvent.REMOVED) {
                                latch.countDown();
                            }
                        }
                    });
                    try {
                        latch.await(5, TimeUnit.SECONDS);
                    } catch (InterruptedException e) {
                    }
                    activeMQActivationController = null;
                }
            }
        });
        jmsServer.start();
    } catch (StartException e) {
        throw e;
    } catch (Throwable t) {
        throw new StartException(t);
    } finally {
        WildFlySecurityManager.setCurrentContextClassLoaderPrivileged(oldTccl);
    }
}