defaultconnectionpool.java

来自「一个java方面的消息订阅发送的源码」· Java 代码 · 共 644 行 · 第 1/2 页

/**
 * Redistribution and use of this software and associated documentation
 * ("Software"), with or without modification, are permitted provided
 * that the following conditions are met:
 *
 * 1. Redistributions of source code must retain copyright
 *    statements and notices.  Redistributions must also contain a
 *    copy of this document.
 *
 * 2. Redistributions in binary form must reproduce the
 *    above copyright notice, this list of conditions and the
 *    following disclaimer in the documentation and/or other
 *    materials provided with the distribution.
 *
 * 3. The name "Exolab" must not be used to endorse or promote
 *    products derived from this Software without prior written
 *    permission of Exoffice Technologies.  For written permission,
 *    please contact [e-mail].
 *
 * 4. Products derived from this Software may not be called "Exolab"
 *    nor may "Exolab" appear in their names without prior written
 *    permission of Exoffice Technologies. Exolab is a registered
 *    trademark of Exoffice Technologies.
 *
 * 5. Due credit should be given to the Exolab Project
 *    (http://www.exolab.org/).
 *
 * THIS SOFTWARE IS PROVIDED BY EXOFFICE TECHNOLOGIES AND CONTRIBUTORS
 * ``AS IS'' AND ANY EXPRESSED OR IMPLIED WARRANTIES, INCLUDING, BUT
 * NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND
 * FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL
 * EXOFFICE TECHNOLOGIES OR ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
 * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
 * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
 * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
 * OF THE POSSIBILITY OF SUCH DAMAGE.
 *
 * Copyright 2005 (C) Exoffice Technologies Inc. All Rights Reserved.
 *
 * $Id: DefaultConnectionPool.java,v 1.5 2005/06/05 13:56:49 tanderson Exp $
 */
package org.exolab.jms.net.connector;

import java.security.Principal;
import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import EDU.oswego.cs.dl.util.concurrent.ClockDaemon;

import org.exolab.jms.net.uri.URI;
import org.exolab.jms.net.util.ThreadFactory;
import org.exolab.jms.net.util.Properties;


/**
 * Manages a pool of {@link ManagedConnection} instances, for a particular
 * {@link ManagedConnectionFactory}.
 *
 * @author <a href="mailto:[e-mail]">Tim Anderson</a>
 * @version $Revision: 1.5 $ $Date: 2005/06/05 13:56:49 $
 * @see AbstractConnectionManager
 */
class DefaultConnectionPool
        implements ManagedConnectionAcceptorListener,
        ManagedConnectionListener, ConnectionPool {

    /**
     * The connection factory.
     */
    private final ManagedConnectionFactory _factory;

    /**
     * Invocation handler to assign to each new connection.
     */
    private final InvocationHandler _handler;

    /**
     * The connection factory for resolving connections via their URI.
     */
    private final ConnectionFactory _resolver;

    /**
     * The set of allocated connections.
     */
    private List _connections = Collections.synchronizedList(new ArrayList());

    /**
     * A map of ManagedConnection -> ManagedConnectionHandle. The handles are
     * used to reap idle connections.
     */
    private Map _handles = Collections.synchronizedMap(new HashMap());

    /**
     * The set of connection acceptors.
     */
    private List _acceptors = Collections.synchronizedList(new ArrayList());

    /**
     * The set of accepted connections.
     */
    private List _accepted = Collections.synchronizedList(new ArrayList());

    /**
     * The set of all connections, as a map of ManagedConnection -> PoolEntry
     * instances.
     */
    private Map _entries = Collections.synchronizedMap(new HashMap());

    /**
     * Clock daemon for periodically running the reaper.
     */
    private ClockDaemon _daemon;

    /**
     * Interval between reaping dead/idle connections, in milliseconds.
     * If <code>0</code> indicates not to reap connections.
     */
    private final long _reapInterval;

    /**
     * The caller event listener.
     */
    private volatile CallerListener _listener;

    /**
     * Property name prefix for pool configuration items.
     */
    private static final String POOL_PREFIX = "org.exolab.jms.net.pool.";

    /**
     * Configuration property to indicate the reap interval.
     */
    private static final String REAP_INTERVAL = "reapInterval";

    /**
     * The logger.
     */
    private static final Log _log
            = LogFactory.getLog(DefaultConnectionPool.class);


    /**
     * Construct a new <code>DefaultConnectionPool</code>.
     *
     * @param factory  the managed connection factory
     * @param handler  the invocation handler, assigned to each new managed
     *                 connection
     * @param resolver the connection factory for resolving connections via
     *                 their URI
     * @param properties configuration properties. May be <code>null</code>
     * @throws ResourceException if any configuration property is invalid
     */
    public DefaultConnectionPool(ManagedConnectionFactory factory,
                                 InvocationHandler handler,
                                 ConnectionFactory resolver,
                                 Map properties) throws ResourceException {
        if (factory == null) {
            throw new IllegalArgumentException("Argument 'factory' is null");
        }
        if (handler == null) {
            throw new IllegalArgumentException("Argument 'handler' is null");
        }
        if (resolver == null) {
            throw new IllegalArgumentException("Argument 'resolver' is null");
        }
        _factory = factory;
        _handler = handler;
        _resolver = resolver;

        Properties config = new Properties(properties, POOL_PREFIX);
        int seconds = config.getInt(REAP_INTERVAL, 15);
        if (seconds < 0) {
            seconds = 0;
        }
        _reapInterval = seconds * 1000;

    }

    /**
     * Creates a new connection.
     *
     * @param principal the security principal
     * @param info      the connection request info
     * @return a new connection
     * @throws ResourceException if a connection cannot be established
     */
    public ManagedConnection createManagedConnection(Principal principal,
                                                     ConnectionRequestInfo info)
            throws ResourceException {
        ManagedConnection connection = _factory.createManagedConnection(
                principal, info);
        return add(connection, false);
    }

    /**
     * Creates an acceptor for connections.
     *
     * @param authenticator authenticates incoming connections
     * @param info          the connection request info
     * @return a new connection acceptor
     * @throws ResourceException if an acceptor cannot be created
     */
    public ManagedConnectionAcceptor createManagedConnectionAcceptor(
            Authenticator authenticator, ConnectionRequestInfo info)
            throws ResourceException {

        ManagedConnectionAcceptor acceptor;

        acceptor = _factory.createManagedConnectionAcceptor(authenticator,
                                                            info);
        _acceptors.add(acceptor);
        return acceptor;
    }

    /**
     * Returns a matched connection from the set of pooled connections.
     *
     * @param principal the security principal
     * @param info      the connection request info
     * @return the first acceptable match, or <code>null</code> if none is
     *         found
     * @throws ResourceException for any error
     */
    public ManagedConnection matchManagedConnections(Principal principal,
                                                     ConnectionRequestInfo info)
            throws ResourceException {

        ManagedConnection result = null;
        result = _factory.matchManagedConnections(_connections, principal,
                                                  info);
        if (result != null) {
            // return the handle corresponding to the connection
            result = (ManagedConnection) _handles.get(result);
        } else {
            result = _factory.matchManagedConnections(_accepted, principal,
                                                      info);
        }
        return result;
    }

    /**
     * Returns a matched acceptor from the set of pooled connections.
     *
     * @param info the connection request info
     * @return the first acceptable match, or <code>null</code> if none is
     *         found
     * @throws ResourceException for any error
     */
    public ManagedConnectionAcceptor matchManagedConnectionAcceptors(
            ConnectionRequestInfo info) throws ResourceException {

        return _factory.matchManagedConnectionAcceptors(_acceptors, info);
    }

    /**
     * Returns a listener for handling accepted connections.
     *
     * @return a listener for handling accepted connections
     */
    public ManagedConnectionAcceptorListener
            getManagedConnectionAcceptorListener() {
            return this;
    }

    /**
     * Invoked when a new connection is accepted.
     *
     * @param acceptor   the acceptor which created the connection
     * @param connection the accepted connection
     */
    public void accepted(ManagedConnectionAcceptor acceptor,
                         ManagedConnection connection) {
        try {
            add(connection, true);
        } catch (ResourceException exception) {
            _log.debug("Failed to accept connection", exception);
        }
    }

    /**
     * Notifies closure of a connection. The <code>ManagedConnection</code>
     * instance invokes this to notify its registered listeners when the peer
     * closes the connection.
     *
     * @param source the managed connection that is the source of the event
     */
    public void closed(ManagedConnection source) {
        if (_log.isDebugEnabled()) {
            _log.debug("Connection " + source + " closed by peer, destroying");
        }
        remove(source);
    }

    /**
     * Notifies a connection related error. The <code>ManagedConnection</code>
     * instance invokes this to notify of the occurrence of a physical
     * connection-related error.
     *
     * @param source    the managed connection that is the source of the event
     * @param throwable the error
     */
    public void error(ManagedConnection source, Throwable throwable) {
        if (_log.isDebugEnabled()) {
            _log.debug("Error on connection " + source + ", destroying",
                       throwable);
        }
        remove(source);
    }

    /**
     * Closes this connection pool, cleaning up any allocated resources.
     *
     * @throws ResourceException for any error
     */