From 1b3bc9d5f81659e8cff88e4b586ba06c66032c3d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Arnaud=20Cogolu=C3=A8gnes?= <acogoluegnes@gmail.com>
Date: Mon, 28 Nov 2016 04:46:41 -0500
Subject: [PATCH] AMQP-677: RabbitConnectionFactory and 4.0 client

JIRA: https://jira.spring.io/browse/AMQP-677

Add setter for NIO flag, NioParams, and MetricsCollector.

Polish

(cherry picked from commit bd4397a)
---
 .../RabbitConnectionFactoryBean.java          | 34 ++++++++++++++++++-
 1 file changed, 33 insertions(+), 1 deletion(-)

diff --git a/spring-rabbit/src/main/java/org/springframework/amqp/rabbit/connection/RabbitConnectionFactoryBean.java b/spring-rabbit/src/main/java/org/springframework/amqp/rabbit/connection/RabbitConnectionFactoryBean.java
index 6d8e424f..a3a47523 100644
--- a/spring-rabbit/src/main/java/org/springframework/amqp/rabbit/connection/RabbitConnectionFactoryBean.java
+++ b/spring-rabbit/src/main/java/org/springframework/amqp/rabbit/connection/RabbitConnectionFactoryBean.java
@@ -45,9 +45,10 @@ import org.springframework.util.StringUtils;
 
 import com.rabbitmq.client.ConnectionFactory;
 import com.rabbitmq.client.ExceptionHandler;
+import com.rabbitmq.client.MetricsCollector;
 import com.rabbitmq.client.SaslConfig;
 import com.rabbitmq.client.SocketConfigurator;
-
+import com.rabbitmq.client.impl.nio.NioParams;
 
 /**
  * Factory bean to create a RabbitMQ ConnectionFactory, delegating most
@@ -64,6 +65,7 @@ import com.rabbitmq.client.SocketConfigurator;
  *
  * @author Gary Russell
  * @author Heath Abelson
+ * @author Arnaud Cogoluègnes
  *
  * @since 1.4
  */
@@ -506,6 +508,36 @@ public class RabbitConnectionFactoryBean extends AbstractFactoryBean<ConnectionF
 		this.connectionFactory.setExceptionHandler(exceptionHandler);
 	}
 
+	/**
+	 * Whether or not the factory should be configured to use Java NIO.
+	 * @param useNio true to use Java NIO, false to use blocking IO
+	 * @see com.rabbitmq.client.ConnectionFactory#useNio()
+	 */
+	public void setUseNio(boolean useNio) {
+		if (useNio) {
+			this.connectionFactory.useNio();
+		}
+		else {
+			this.connectionFactory.useBlockingIo();
+		}
+	}
+
+	/**
+	 * @param nioParams the NIO parameters
+	 * @see com.rabbitmq.client.ConnectionFactory#setNioParams(com.rabbitmq.client.impl.nio.NioParams)
+	 */
+	public void setNioParams(NioParams nioParams) {
+		this.connectionFactory.setNioParams(nioParams);
+	}
+
+	/**
+	 * @param metricsCollector the metrics collector instance
+	 * @see com.rabbitmq.client.ConnectionFactory#setMetricsCollector(MetricsCollector)
+	 */
+	public void setMetricsCollector(MetricsCollector metricsCollector) {
+		this.connectionFactory.setMetricsCollector(metricsCollector);
+	}
+
 	@Override
 	public Class<?> getObjectType() {
 		return ConnectionFactory.class;