From b71f143dfd622e9e3b7742e032ce982ec00a94b2 Mon Sep 17 00:00:00 2001
From: Arjen Poutsma <apoutsma@pivotal.io>
Date: Tue, 3 May 2016 16:33:56 +0200
Subject: [PATCH] Polishing

---
 .../io/buffer/support/DataBufferUtils.java    | 42 ++++++++-----------
 1 file changed, 18 insertions(+), 24 deletions(-)

diff --git a/spring-web-reactive/src/main/java/org/springframework/core/io/buffer/support/DataBufferUtils.java b/spring-web-reactive/src/main/java/org/springframework/core/io/buffer/support/DataBufferUtils.java
index ad50c7b3efa..28ce6781c09 100644
--- a/spring-web-reactive/src/main/java/org/springframework/core/io/buffer/support/DataBufferUtils.java
+++ b/spring-web-reactive/src/main/java/org/springframework/core/io/buffer/support/DataBufferUtils.java
@@ -16,7 +16,6 @@
 
 package org.springframework.core.io.buffer.support;
 
-import java.io.Closeable;
 import java.io.IOException;
 import java.io.InputStream;
 import java.nio.ByteBuffer;
@@ -44,33 +43,16 @@ import org.springframework.util.Assert;
  */
 public abstract class DataBufferUtils {
 
-	private static final Consumer<? extends Closeable> CLOSE_CONSUMER = closeable -> {
+	private static final Consumer<ReadableByteChannel> CLOSE_CONSUMER = channel -> {
 		try {
-			if (closeable != null) {
-				closeable.close();
+			if (channel != null) {
+				channel.close();
 			}
 		}
 		catch (IOException ignored) {
 		}
 	};
 
-	/**
-	 * Reads the given {@code ReadableByteChannel} into a {@code Flux} of
-	 * {@code DataBuffer}s. Closes the channel when the flux is terminated.
-	 * @param channel the channel to read from
-	 * @param allocator the allocator to create data buffers with
-	 * @param bufferSize the maximum size of the data buffers
-	 * @return a flux of data buffers read from the given channel
-	 */
-	public static Flux<DataBuffer> read(ReadableByteChannel channel,
-			DataBufferAllocator allocator, int bufferSize) {
-		Assert.notNull(channel, "'channel' must not be null");
-		Assert.notNull(allocator, "'allocator' must not be null");
-
-		return Flux.create(new ReadableByteChannelConsumer(allocator, bufferSize),
-				subscriber -> channel, closeConsumer());
-	}
-
 	/**
 	 * Reads the given {@code InputStream} into a {@code Flux} of
 	 * {@code DataBuffer}s. Closes the stream when the flux inputStream terminated.
@@ -88,9 +70,21 @@ public abstract class DataBufferUtils {
 		return read(channel, allocator, bufferSize);
 	}
 
-	@SuppressWarnings("unchecked")
-	private static <T extends Closeable> Consumer<T> closeConsumer() {
-		return (Consumer<T>) CLOSE_CONSUMER;
+	/**
+	 * Reads the given {@code ReadableByteChannel} into a {@code Flux} of
+	 * {@code DataBuffer}s. Closes the channel when the flux is terminated.
+	 * @param channel the channel to read from
+	 * @param allocator the allocator to create data buffers with
+	 * @param bufferSize the maximum size of the data buffers
+	 * @return a flux of data buffers read from the given channel
+	 */
+	public static Flux<DataBuffer> read(ReadableByteChannel channel,
+			DataBufferAllocator allocator, int bufferSize) {
+		Assert.notNull(channel, "'channel' must not be null");
+		Assert.notNull(allocator, "'allocator' must not be null");
+
+		return Flux.create(new ReadableByteChannelConsumer(allocator, bufferSize),
+				subscriber -> channel, CLOSE_CONSUMER);
 	}
 
 	/**