From 0a296ee85793148d8f9a7cf805e43484c04246dc Mon Sep 17 00:00:00 2001
From: Redkale <22250530@qq.com>
Date: Wed, 9 May 2018 09:23:43 +0800
Subject: [PATCH]

---
 src/org/redkale/net/http/HttpResponse.java | 21 ++++++++++++++++++++-
 1 file changed, 20 insertions(+), 1 deletion(-)

diff --git a/src/org/redkale/net/http/HttpResponse.java b/src/org/redkale/net/http/HttpResponse.java
index 03c69fc57..7a24f4a4e 100644
--- a/src/org/redkale/net/http/HttpResponse.java
+++ b/src/org/redkale/net/http/HttpResponse.java
@@ -611,7 +611,26 @@ public class HttpResponse extends Response<HttpContext, HttpRequest> {
     public void finish(final byte[] bs) {
         if (isClosed()) return; //避免重复关闭
         if (this.context.getBufferCapacity() >= bs.length) {
-            ByteBuffer buffer = this.context.pollBuffer();
+            ByteBuffer buffer = getBodyBufferSupplier().get();
+            buffer.put(bs);
+            buffer.flip();
+            this.finish(false, buffer);
+        } else {
+            this.finish(false, ByteBuffer.wrap(bs));
+        }
+    }
+
+    /**
+     * 将指定byte[]按响应结果输出
+     *
+     * @param contentType ContentType
+     * @param bs          输出内容
+     */
+    public void finish(final String contentType, final byte[] bs) {
+        if (isClosed()) return; //避免重复关闭
+        this.contentType = contentType;
+        if (this.context.getBufferCapacity() >= bs.length) {
+            ByteBuffer buffer = getBodyBufferSupplier().get();
             buffer.put(bs);
             buffer.flip();
             this.finish(false, buffer);