mirror of https://github.com/OpenIdentityPlatform/OpenDJ.git

Matthew Swift
25.33.2012 263d085885df024dca9250cc03c807912b0a7662
opendj3/opendj-ldap-sdk/src/main/java/org/forgerock/opendj/ldap/LoadBalancer.java
@@ -6,17 +6,16 @@
 * (the "License").  You may not use this file except in compliance
 * with the License.
 *
 * You can obtain a copy of the license at
 * trunk/opendj3/legal-notices/CDDLv1_0.txt
 * You can obtain a copy of the license at legal-notices/CDDLv1_0.txt
 * or http://forgerock.org/license/CDDLv1.0.html.
 * See the License for the specific language governing permissions
 * and limitations under the License.
 *
 * When distributing Covered Code, include this CDDL HEADER in each
 * file and include the License file at
 * trunk/opendj3/legal-notices/CDDLv1_0.txt.  If applicable,
 * add the following below this CDDL HEADER, with the fields enclosed
 * by brackets "[]" replaced with your own identifying information:
 * file and include the License file at legal-notices/CDDLv1_0.txt.
 * If applicable, add the following below this CDDL HEADER, with the
 * fields enclosed by brackets "[]" replaced with your own identifying
 * information:
 *      Portions Copyright [yyyy] [name of copyright owner]
 *
 * CDDL HEADER END
@@ -28,85 +27,63 @@
package org.forgerock.opendj.ldap;
import com.forgerock.opendj.util.CompletedFutureResult;
import com.forgerock.opendj.util.Validator;
/**
 * A load balancing connection factory allocates connections using the provided
 * algorithm.
 */
final class LoadBalancer implements ConnectionFactory
{
  private final LoadBalancingAlgorithm algorithm;
final class LoadBalancer implements ConnectionFactory {
    private final LoadBalancingAlgorithm algorithm;
  /**
   * Creates a new load balancer using the provided algorithm.
   *
   * @param algorithm
   *          The load balancing algorithm which will be used to obtain the next
   *          connection factory.
   */
  public LoadBalancer(final LoadBalancingAlgorithm algorithm)
  {
    Validator.ensureNotNull(algorithm);
    this.algorithm = algorithm;
  }
  /**
   * {@inheritDoc}
   */
  public Connection getConnection() throws ErrorResultException,
      InterruptedException
  {
    return algorithm.getConnectionFactory().getConnection();
  }
  /**
   * {@inheritDoc}
   */
  @Override
  public FutureResult<Connection> getConnectionAsync(
      final ResultHandler<? super Connection> resultHandler)
  {
    final ConnectionFactory factory;
    try
    {
      factory = algorithm.getConnectionFactory();
    }
    catch (final ErrorResultException e)
    {
      if (resultHandler != null)
      {
        resultHandler.handleErrorResult(e);
      }
      return new CompletedFutureResult<Connection>(e);
    /**
     * Creates a new load balancer using the provided algorithm.
     *
     * @param algorithm
     *            The load balancing algorithm which will be used to obtain the
     *            next connection factory.
     */
    public LoadBalancer(final LoadBalancingAlgorithm algorithm) {
        Validator.ensureNotNull(algorithm);
        this.algorithm = algorithm;
    }
    return factory.getConnectionAsync(resultHandler);
  }
    /**
     * {@inheritDoc}
     */
    public Connection getConnection() throws ErrorResultException, InterruptedException {
        return algorithm.getConnectionFactory().getConnection();
    }
    /**
     * {@inheritDoc}
     */
    @Override
    public FutureResult<Connection> getConnectionAsync(
            final ResultHandler<? super Connection> resultHandler) {
        final ConnectionFactory factory;
        try {
            factory = algorithm.getConnectionFactory();
        } catch (final ErrorResultException e) {
            if (resultHandler != null) {
                resultHandler.handleErrorResult(e);
            }
            return new CompletedFutureResult<Connection>(e);
        }
  /**
   * {@inheritDoc}
   */
  public String toString()
  {
    final StringBuilder builder = new StringBuilder();
    builder.append("LoadBalancer(");
    builder.append(String.valueOf(algorithm));
    builder.append(')');
    return builder.toString();
  }
        return factory.getConnectionAsync(resultHandler);
    }
    /**
     * {@inheritDoc}
     */
    public String toString() {
        final StringBuilder builder = new StringBuilder();
        builder.append("LoadBalancer(");
        builder.append(String.valueOf(algorithm));
        builder.append(')');
        return builder.toString();
    }
}