From a8f27af5fbfc255199d6065d26c321aac70b3417 Mon Sep 17 00:00:00 2001
From: Arjen Poutsma <apoutsma@pivotal.io>
Date: Thu, 31 Mar 2016 11:52:03 +0200
Subject: [PATCH] Polishing.

---
 .../core/codec/support/ByteBufferEncoder.java             | 8 ++------
 1 file changed, 2 insertions(+), 6 deletions(-)

diff --git a/spring-web-reactive/src/main/java/org/springframework/core/codec/support/ByteBufferEncoder.java b/spring-web-reactive/src/main/java/org/springframework/core/codec/support/ByteBufferEncoder.java
index 4e05e1d62b..3d006f2c99 100644
--- a/spring-web-reactive/src/main/java/org/springframework/core/codec/support/ByteBufferEncoder.java
+++ b/spring-web-reactive/src/main/java/org/springframework/core/codec/support/ByteBufferEncoder.java
@@ -47,12 +47,8 @@ public class ByteBufferEncoder extends AbstractEncoder<ByteBuffer> {
 	public Flux<DataBuffer> encode(Publisher<? extends ByteBuffer> inputStream,
 			DataBufferAllocator allocator, ResolvableType type, MimeType mimeType,
 			Object... hints) {
-		//noinspection unchecked
-		return Flux.from(inputStream).map(byteBuffer -> {
-			DataBuffer dataBuffer = allocator.allocateBuffer(byteBuffer.remaining());
-			dataBuffer.write(byteBuffer);
-			return dataBuffer;
-		});
+
+		return Flux.from(inputStream).map(allocator::wrap);
 	}
 
 }
\ No newline at end of file