discord4j.connect.rabbitmq.ConnectRabbitMQSettings Maven / Gradle / Ivy
package discord4j.connect.rabbitmq;
import com.rabbitmq.client.Address;
import reactor.core.scheduler.Scheduler;
import reactor.core.scheduler.Schedulers;
import reactor.rabbitmq.ReceiverOptions;
import reactor.rabbitmq.SenderOptions;
/**
* A set of commonly used options for connecting to a RabbitMQ broker.
*/
public class ConnectRabbitMQSettings {
/**
* Creates a new RabbitMQ settings object.
*
* @return settings object
*/
public static ConnectRabbitMQSettings create() {
return new ConnectRabbitMQSettings();
}
private final SenderOptions senderOptions;
private final ReceiverOptions receiverOptions;
/**
* Private constructor, use {@link ConnectRabbitMQSettings#create()} instead
* Defaults the schedulers to use {@link Schedulers#boundedElastic()}
*/
private ConnectRabbitMQSettings() {
this.senderOptions = new SenderOptions();
this.receiverOptions = new ReceiverOptions();
withSubscriptionScheduler(Schedulers.boundedElastic());
withResourceManagementScheduler(Schedulers.boundedElastic());
}
/**
* Set the RabbitMQs host-and-port list as {@link Address} array.
*
* @param addresses array of {@link Address} to which RabbitMQ nodes the client should connect
* @return this builder
*/
public ConnectRabbitMQSettings withAddresses(final Address... addresses) {
senderOptions.connectionSupplier(connectionFactory -> connectionFactory.newConnection(addresses));
receiverOptions.connectionSupplier(connectionFactory -> connectionFactory.newConnection(addresses));
return this;
}
/**
* Sets the RabbitMQs host list with default ports.
*
* @param hosts array of hostnames to which RabbitMQ nodes the client should connect on default port
* @return this builder
*/
public ConnectRabbitMQSettings withAddresses(final String... hosts) {
final Address[] addresses = new Address[hosts.length];
for (int i = 0; i < hosts.length; i++) {
addresses[i] = new Address(hosts[i]);
}
return withAddresses(addresses);
}
/**
* Sets the RabbitMQs host and port.
*
* @param host hostname to connect to
* @param port port to connect to
* @return this builder
*/
public ConnectRabbitMQSettings withAddress(final String host, final int port) {
return this.withAddresses(new Address(host, port));
}
/**
* Sets the RabbitMQs host. This method will use the default port.
*
* @param host hostname to connect to
* @return this builder
*/
public ConnectRabbitMQSettings withAddress(final String host) {
return this.withAddresses(new Address(host));
}
/**
* The username to use for authorization.
*
* @param user username for authorization
* @return this builder
*/
public ConnectRabbitMQSettings withUser(final String user) {
senderOptions.getConnectionFactory().setUsername(user);
receiverOptions.getConnectionFactory().setUsername(user);
return this;
}
/**
* The password to use for authorization.
*
* @param password password for authorization
* @return this builder
*/
public ConnectRabbitMQSettings withPassword(final String password) {
senderOptions.getConnectionFactory().setPassword(password);
receiverOptions.getConnectionFactory().setPassword(password);
return this;
}
/**
* Uses a specific scheduler for connection subscriptions.
*
* Set the scheduler to {@code null} to create one scheduler per instance
*
* @param scheduler the scheduler to use
* @return this builder
*/
public ConnectRabbitMQSettings withSubscriptionScheduler(final Scheduler scheduler) {
senderOptions.connectionSubscriptionScheduler(scheduler);
receiverOptions.connectionSubscriptionScheduler(scheduler);
return this;
}
/**
* Uses a specific scheduler for resource management.
*
* Set the scheduler to {@code null} to create one scheduler per instance.
*
* @param scheduler the scheduler to use
* @return this builder
*/
public ConnectRabbitMQSettings withResourceManagementScheduler(final Scheduler scheduler) {
senderOptions.resourceManagementScheduler(scheduler);
return this;
}
/**
* Internal method to retrieve the whole settings object for the RabbitMQ sender.
*
* @return SenderOptions
*/
SenderOptions getSenderOptions() {
return senderOptions;
}
/**
* Internal method to retrieve the whole settings object for the RabbitMQ receiver.
*
* @return ReceiverOptions
*/
ReceiverOptions getReceiverOptions() {
return receiverOptions;
}
}