com.rabbitmq.client.amqp.impl.AmqpConnectionBuilder Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of amqp-client Show documentation
The RabbitMQ AMQP 1.0 Java client library defines an API to access RabbitMQ with the AMQP 1.0 protocol.
The newest version!
// Copyright (c) 2024 Broadcom. All Rights Reserved.
// The term "Broadcom" refers to Broadcom Inc. and/or its subsidiaries.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//
// If you have any questions regarding licensing, please contact us at
// [email protected].
package com.rabbitmq.client.amqp.impl;

import com.rabbitmq.client.amqp.*;
import java.time.Duration;
import java.util.ArrayList;
import java.util.List;

class AmqpConnectionBuilder implements ConnectionBuilder {

  private final AmqpEnvironment environment;
  private final AmqpRecoveryConfiguration recoveryConfiguration =
      new AmqpRecoveryConfiguration(this);
  private final DefaultConnectionSettings connectionSettings =
      new AmqpConnectionBuilderConnectionSettings(this);
  private final List listeners = new ArrayList<>();
  private String name;
  private TopologyListener topologyListener;
  private boolean isolateResources = false;

  AmqpConnectionBuilder(AmqpEnvironment environment) {
    this.environment = environment;
    this.environment.connectionSettings().copyTo(this.connectionSettings);
  }

  @Override
  public ConnectionBuilder uri(String uri) {
    return this.connectionSettings.uri(uri);
  }

  @Override
  public ConnectionBuilder uris(String... uris) {
    return this.connectionSettings.uris(uris);
  }

  @Override
  public ConnectionBuilder username(String username) {
    return this.connectionSettings.username(username);
  }

  @Override
  public ConnectionBuilder password(String password) {
    return this.connectionSettings.password(password);
  }

  @Override
  public ConnectionBuilder host(String host) {
    return this.connectionSettings.host(host);
  }

  @Override
  public ConnectionBuilder port(int port) {
    return this.connectionSettings.port(port);
  }

  @Override
  public ConnectionBuilder virtualHost(String virtualHost) {
    return this.connectionSettings.virtualHost(virtualHost);
  }

  @Override
  public ConnectionBuilder credentialsProvider(CredentialsProvider credentialsProvider) {
    return this.connectionSettings.credentialsProvider(credentialsProvider);
  }

  @Override
  public ConnectionBuilder idleTimeout(Duration idleTimeout) {
    return this.connectionSettings.idleTimeout(idleTimeout);
  }

  @Override
  public ConnectionBuilder addressSelector(AddressSelector selector) {
    return this.connectionSettings.addressSelector(selector);
  }

  @Override
  public ConnectionBuilder saslMechanism(String mechanism) {
    return this.connectionSettings.saslMechanism(mechanism);
  }

  @Override
  public TlsSettings tls() {
    return this.connectionSettings.tls();
  }

  @Override
  public DefaultConnectionSettings.DefaultAffinity affinity() {
    return this.connectionSettings.affinity();
  }

  @Override
  public ConnectionBuilder listeners(Resource.StateListener... listeners) {
    if (listeners == null || listeners.length == 0) {
      this.listeners.clear();
    } else {
      this.listeners.addAll(List.of(listeners));
    }
    return this;
  }

  @Override
  public RecoveryConfiguration recovery() {
    this.recoveryConfiguration.activated(true);
    return this.recoveryConfiguration;
  }

  AmqpConnectionBuilder isolateResources(boolean isolateResources) {
    this.isolateResources = isolateResources;
    return this;
  }

  boolean isolateResources() {
    return isolateResources;
  }

  @Override
  public Connection build() {
    return this.environment.connection(this);
  }

  void copyTo(AmqpConnectionBuilder copy) {
    this.connectionSettings.copyTo(copy.connectionSettings);
    this.recoveryConfiguration.copyTo(copy.recoveryConfiguration);
    copy.listeners(this.listeners.toArray(new Resource.StateListener[0]));
    copy.name(this.name);
    copy.topologyListener(this.topologyListener);
    copy.isolateResources(this.isolateResources);
  }

  AmqpConnectionBuilder name(String name) {
    this.name = name;
    return this;
  }

  AmqpConnectionBuilder topologyListener(TopologyListener topologyListener) {
    this.topologyListener = topologyListener;
    return this;
  }

  AmqpEnvironment environment() {
    return environment;
  }

  AmqpRecoveryConfiguration recoveryConfiguration() {
    return recoveryConfiguration;
  }

  String name() {
    return name;
  }

  TopologyListener topologyListener() {
    return this.topologyListener;
  }

  List listeners() {
    return listeners;
  }

  DefaultConnectionSettings connectionSettings() {
    return this.connectionSettings;
  }

  static class AmqpRecoveryConfiguration implements RecoveryConfiguration {

    private final AmqpConnectionBuilder connectionBuilder;
    private boolean activated = true;
    private boolean topology = true;
    private BackOffDelayPolicy backOffDelayPolicy = BackOffDelayPolicy.fixed(Duration.ofSeconds(5));

    AmqpRecoveryConfiguration(AmqpConnectionBuilder connectionBuilder) {
      this.connectionBuilder = connectionBuilder;
    }

    @Override
    public AmqpRecoveryConfiguration activated(boolean activated) {
      this.activated = activated;
      return this;
    }

    @Override
    public AmqpRecoveryConfiguration backOffDelayPolicy(BackOffDelayPolicy backOffDelayPolicy) {
      this.backOffDelayPolicy = backOffDelayPolicy;
      return this;
    }

    @Override
    public RecoveryConfiguration topology(boolean activated) {
      this.topology = activated;
      return this;
    }

    @Override
    public ConnectionBuilder connectionBuilder() {
      return this.connectionBuilder;
    }

    boolean activated() {
      return this.activated;
    }

    boolean topology() {
      return this.topology;
    }

    BackOffDelayPolicy backOffDelayPolicy() {
      return this.backOffDelayPolicy;
    }

    void copyTo(RecoveryConfiguration copy) {
      copy.activated(this.activated);
      copy.topology(this.topology);
      copy.backOffDelayPolicy(this.backOffDelayPolicy);
    }
  }

  static class AmqpConnectionBuilderConnectionSettings
      extends DefaultConnectionSettings {

    private final AmqpConnectionBuilder builder;

    private AmqpConnectionBuilderConnectionSettings(AmqpConnectionBuilder builder) {
      this.builder = builder;
    }

    @Override
    AmqpConnectionBuilder toReturn() {
      return this.builder;
    }

    @Override
    public TlsSettings tls() {
      return super.tls();
    }
  }
}