Examples with ShutdownManager org.forgerock.util.thread.listener.ShutdownManager used on opensource projects

Example 6 with ShutdownManager

use of org.forgerock.util.thread.listener.ShutdownManager in project OpenAM by OpenRock.

the class AMSetupDSConfig method getLDAPConnection.

/**
     * Helper method to return Ldap connection 
     *
     * @param ssl <code>true</code> if directory server is running SSL.
     * @return Ldap connection 
     */
private synchronized Connection getLDAPConnection(boolean ssl) {
    try {
        if (ld == null) {
            ShutdownManager shutdownMan = com.sun.identity.common.ShutdownManager.getInstance();
            // All connections will use authentication
            SimpleBindRequest request = LDAPRequests.newSimpleBindRequest(dsManager, dsAdminPwd.toCharArray());
            Options options = Options.defaultOptions().set(REQUEST_TIMEOUT, new Duration((long) 3, TimeUnit.SECONDS)).set(AUTHN_BIND_REQUEST, request);
            if (ssl) {
                options = options.set(SSL_CONTEXT, new SSLContextBuilder().getSSLContext());
            }
            ld = new LDAPConnectionFactory(dsHostName, getPort(), options);
            shutdownMan.addShutdownListener(new ShutdownListener() {

                public void shutdown() {
                    disconnectDServer();
                }
            });
        }
        return ld.getConnection();
    } catch (LdapException e) {
        disconnectDServer();
        dsConfigInstance = null;
        ld = null;
    } catch (Exception e) {
        dsConfigInstance = null;
        ld = null;
    }
    return null;
}

Example 7 with ShutdownManager

use of org.forgerock.util.thread.listener.ShutdownManager in project OpenAM by OpenRock.

the class ShutdownMonitorTest method setup.

@BeforeTest
public void setup() {
    ShutdownManager shutdownManagerWrapper = mock(ShutdownManager.class);
    ArgumentCaptor<ShutdownListener> captor = ArgumentCaptor.forClass(ShutdownListener.class);
    monitor = new ShutdownMonitor(shutdownManagerWrapper);
    verify(shutdownManagerWrapper).addShutdownListener(captor.capture());
    listener = captor.getValue();
}

Example 8 with ShutdownManager

use of org.forgerock.util.thread.listener.ShutdownManager in project OpenAM by OpenRock.

the class ServerConfigMgr method main.

public static void main(String[] args) {
    try {
        Bootstrap.load();
        if (VersionCheck.isVersionValid() == 1) {
            System.exit(1);
        }
        debug = Debug.getInstance(IUMSConstants.UMS_DEBUG);
        validateArguments(args);
        boolean proceeded = printHelpMessage(args) || encryptPassword(args) || changePassword(args);
    } catch (ConfiguratorException ex) {
        System.err.println(ex.getL10NMessage(Locale.getDefault()));
        System.exit(1);
    } catch (Exception ex) {
        System.err.println(ex.getMessage());
        System.exit(1);
    } finally {
        ShutdownManager shutdownMan = com.sun.identity.common.ShutdownManager.getInstance();
        shutdownMan.shutdown();
    }
}

Example 9 with ShutdownManager

use of org.forgerock.util.thread.listener.ShutdownManager in project OpenAM by OpenRock.

the class PolicyEvaluator method init.

/**
     * Initializes an instance of client policy evaluator object
     *
     * @param serviceName name of the service for which to create 
     *        policy evaluator
     * @param appSSOTokenProvider an object where application single sign on
     *        token can be obtained.
     *
     * @throws PolicyException if required properties cannot be retrieved.
     * @throws SSOException if application single sign on token is invalid.
     *
     */
private void init(final String serviceName, AppSSOTokenProvider appSSOTokenProvider) throws PolicyException, SSOException {
    this.ssoTokenManager = SSOTokenManager.getInstance();
    this.serviceName = serviceName;
    this.appSSOTokenProvider = appSSOTokenProvider;
    this.policyProperties = new PolicyProperties();
    this.logActions = policyProperties.getLogActions();
    this.resourceResultCache = ResourceResultCache.getInstance(policyProperties);
    appSSOToken = getNewAppSSOToken();
    if (PolicyProperties.previouslyNotificationEnabled()) {
        if (policyProperties.useRESTProtocol()) {
            resourceResultCache.removeRESTRemotePolicyListener(appSSOToken, serviceName, PolicyProperties.getPreviousNotificationURL());
        } else {
            resourceResultCache.removeRemotePolicyListener(appSSOToken, serviceName, PolicyProperties.getPreviousNotificationURL());
        }
    }
    if (policyProperties.notificationEnabled()) {
        // register remote policy listener policy service
        if (debug.messageEnabled()) {
            debug.message("PolicyEvaluator.init():" + "adding remote policy listener with policy " + "service " + serviceName);
        }
        if (policyProperties.useRESTProtocol()) {
            resourceResultCache.addRESTRemotePolicyListener(appSSOToken, serviceName, policyProperties.getRESTNotificationURL());
        } else {
            resourceResultCache.addRemotePolicyListener(appSSOToken, serviceName, policyProperties.getNotificationURL());
        }
        // Add a hook to remove our listener on shutdown.
        ShutdownManager shutdownMan = com.sun.identity.common.ShutdownManager.getInstance();
        shutdownMan.addShutdownListener(new ShutdownListener() {

            @Override
            public void shutdown() {
                if (policyProperties.useRESTProtocol()) {
                    resourceResultCache.removeRESTRemotePolicyListener(appSSOToken, serviceName, policyProperties.getRESTNotificationURL());
                    if (debug.messageEnabled()) {
                        debug.message("PolicyEvaluator: called removeRESTRemotePolicyListener, service " + serviceName + ", URL " + policyProperties.getRESTNotificationURL());
                    }
                } else {
                    resourceResultCache.removeRemotePolicyListener(appSSOToken, serviceName, policyProperties.getNotificationURL());
                    if (debug.messageEnabled()) {
                        debug.message("PolicyEvaluator: called removeRemotePolicyListener, service " + serviceName + ", URL " + policyProperties.getNotificationURL());
                    }
                }
            }
        });
    }
    ActionDecision.setClientClockSkew(policyProperties.getClientClockSkew());
    if (debug.messageEnabled()) {
        debug.message("PolicyEvaluator:" + "initialized PolicyEvaluator");
    }
}

Example 10 with ShutdownManager

use of org.forgerock.util.thread.listener.ShutdownManager in project OpenAM by OpenRock.

the class SessionPollerPool method initPollerPool.

/**
     * Configures the threadpool, and registers the threadpool with the shutdown manager so that it correctly
     * shuts down when the server is brought down.
     */
private synchronized void initPollerPool() {
    if (!pollerPoolInitialized) {
        if (pollingEnabled) {
            int poolSize = SystemProperties.getAsInt(Constants.POLLING_THREADPOOL_SIZE, DEFAULT_POOL_SIZE);
            int threshold = SystemProperties.getAsInt(Constants.POLLING_THREADPOOL_THRESHOLD, DEFAULT_THRESHOLD);
            final ShutdownManager shutdownMan = com.sun.identity.common.ShutdownManager.getInstance();
            threadPool = new ThreadPool("amSessionPoller", poolSize, threshold, true, debug);
            shutdownMan.addShutdownListener(new ShutdownListener() {

                public void shutdown() {
                    threadPool.shutdown();
                    threadPool = null;
                    pollerPoolInitialized = false;
                }
            });
            pollerPoolInitialized = true;
        } else {
            if (debug.messageEnabled()) {
                debug.message("Session Cache cleanup is set to " + sessionCleanupEnabled);
            }
        }
    }
}