com.rabbitmq.client.amqp.impl.AmqpBindingManagement 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.Management;
import java.util.LinkedHashMap;
import java.util.Map;
import java.util.function.BiConsumer;

abstract class AmqpBindingManagement {

  private AmqpBindingManagement() {}

  static class BindingState {

    private final AmqpManagement managememt;

    private String source, destination, key;
    private final Map arguments = new LinkedHashMap<>();
    private boolean toQueue = true;

    BindingState(AmqpManagement managememt) {
      this.managememt = managememt;
    }

    String source() {
      return this.source;
    }

    String destination() {
      return this.destination;
    }

    String key() {
      return this.key;
    }

    boolean toQueue() {
      return this.toQueue;
    }

    void arguments(BiConsumer consumer) {
      this.arguments.forEach(consumer);
    }
  }

  static class AmqpBindingSpecification implements Management.BindingSpecification {

    private final BindingState state;

    AmqpBindingSpecification(AmqpManagement management) {
      this.state = new BindingState(management);
    }

    @Override
    public Management.BindingSpecification sourceExchange(String source) {
      this.state.source = source;
      return this;
    }

    @Override
    public Management.BindingSpecification destinationQueue(String queue) {
      this.state.toQueue = true;
      this.state.destination = queue;
      return this;
    }

    @Override
    public Management.BindingSpecification destinationExchange(String exchange) {
      this.state.toQueue = false;
      this.state.destination = exchange;
      return this;
    }

    @Override
    public Management.BindingSpecification key(String key) {
      this.state.key = key;
      return this;
    }

    @Override
    public Management.BindingSpecification argument(String key, Object value) {
      this.state.arguments.put(key, value);
      return this;
    }

    @Override
    public Management.BindingSpecification arguments(Map arguments) {
      this.state.arguments.clear();
      this.state.arguments.putAll(arguments);
      return this;
    }

    @Override
    public void bind() {
      Map body = new LinkedHashMap<>();
      body.put("source", this.state.source);
      body.put("binding_key", this.state.key == null ? "" : this.state.key);
      body.put("arguments", this.state.arguments);
      if (this.state.toQueue) {
        body.put("destination_queue", this.state.destination);
        this.state.managememt.bind(body);
      } else {
        body.put("destination_exchange", this.state.destination);
        this.state.managememt.bind(body);
      }
      this.state.managememt.recovery().bindingDeclared(this);
    }

    BindingState state() {
      return this.state;
    }
  }

  static class AmqpUnbindSpecification implements Management.UnbindSpecification {

    private final BindingState state;

    AmqpUnbindSpecification(AmqpManagement management) {
      this.state = new BindingState(management);
    }

    @Override
    public Management.UnbindSpecification sourceExchange(String exchange) {
      this.state.source = exchange;
      return this;
    }

    @Override
    public Management.UnbindSpecification destinationQueue(String queue) {
      this.state.toQueue = true;
      this.state.destination = queue;
      return this;
    }

    @Override
    public Management.UnbindSpecification destinationExchange(String exchange) {
      this.state.toQueue = false;
      this.state.destination = exchange;
      return this;
    }

    @Override
    public Management.UnbindSpecification key(String key) {
      this.state.key = key;
      return this;
    }

    @Override
    public Management.UnbindSpecification argument(String key, Object value) {
      this.state.arguments.put(key, value);
      return this;
    }

    @Override
    public Management.UnbindSpecification arguments(Map arguments) {
      this.state.arguments.clear();
      this.state.arguments.putAll(arguments);
      return this;
    }

    @Override
    public void unbind() {
      String destinationCharacter = this.state.toQueue ? "dstq" : "dste";
      this.state.managememt.unbind(
          destinationCharacter,
          this.state.source,
          this.state.destination,
          this.state.key == null ? "" : this.state.key,
          this.state.arguments);
      this.state.managememt.recovery().bindingDeleted(this);
    }

    BindingState state() {
      return this.state;
    }
  }
}