src/com/wentch/redkale/convert/json/JsonByteBufferWriter.java

@@ -71,7 +71,7 @@ public final class JsonByteBufferWriter extends JsonWriter {
         return len;
     }
 
-    private void expand(final int byteLength) {
+    private int expand(final int byteLength) {
         if (this.buffers == null) {
             this.index = 0;
             this.buffers = new ByteBuffer[]{supplier.get()};
@@ -86,8 +86,8 @@ public final class JsonByteBufferWriter extends JsonWriter {
             this.buffers = bufs;
             this.index++;
         }
-        if (buffer.remaining() >= byteLength) return;
         int len = buffer.remaining();
+        int size = 0;
         while (len < byteLength) {
             buffer = supplier.get();
             ByteBuffer[] bufs = new ByteBuffer[this.buffers.length + 1];
@@ -95,7 +95,9 @@ public final class JsonByteBufferWriter extends JsonWriter {
             bufs[this.buffers.length] = buffer;
             this.buffers = bufs;
             len += buffer.remaining();
+            size++;
         }
+        return size;
     }
 
     @Override
@@ -107,10 +109,10 @@ public final class JsonByteBufferWriter extends JsonWriter {
 
     @Override
     public void writeTo(final char[] chs, final int start, final int len) {
-        writeTo(false, chs, start, len);
+        writeTo(-1, false, chs, start, len);
     }
 
-    private void writeTo(final boolean quote, final char[] chs, final int start, final int len) {
+    private void writeTo(int expandsize, final boolean quote, final char[] chs, final int start, final int len) {
         int byteLength = quote ? 2 : 0;
         ByteBuffer bb = null;
         if (charset == null) {
@@ -119,7 +121,33 @@ public final class JsonByteBufferWriter extends JsonWriter {
             bb = charset.encode(CharBuffer.wrap(chs, start, len));
             byteLength += bb.remaining();
         }
-        expand(byteLength);
+        if (expandsize < 0) expandsize = expand(byteLength);
+        if (expandsize == 0) { // 只需要一个buffer 
+            final ByteBuffer buffer = this.buffers[index];
+            if (quote) buffer.put((byte) '"');
+
+            if (charset == null) { //UTF-8
+                final int limit = start + len;
+                for (int i = start; i < limit; i++) {
+                    char c = chs[i];
+                    if (c < 0x80) {
+                        buffer.put((byte) c);
+                    } else if (c < 0x800) {
+                        buffer.put((byte) (0xc0 | (c >> 6)));
+                        buffer.put((byte) (0x80 | (c & 0x3f)));
+                    } else {
+                        buffer.put((byte) (0xe0 | ((c >> 12))));
+                        buffer.put((byte) (0x80 | ((c >> 6) & 0x3f)));
+                        buffer.put((byte) (0x80 | (c & 0x3f)));
+                    }
+                }
+            } else {
+                buffer.put(bb);
+            }
+
+            if (quote) buffer.put((byte) '"');
+            return;
+        }
         ByteBuffer buffer = this.buffers[index];
         if (quote) {
             if (!buffer.hasRemaining()) buffer = nextByteBuffer();
@@ -132,8 +160,8 @@ public final class JsonByteBufferWriter extends JsonWriter {
             }
         } else {
             while (bb.hasRemaining()) {
-                buffer.put(bb.get());
                 if (!buffer.hasRemaining()) buffer = nextByteBuffer();
+                buffer.put(bb.get());
             }
         }
         if (quote) {
@@ -179,6 +207,32 @@ public final class JsonByteBufferWriter extends JsonWriter {
         return size;
     }
 
+    private static int encodeEscapeUTF8Length(final char[] text, final int start, final int len) {
+        char c;
+        int size = 0;
+        final char[] chars = text;
+        final int limit = start + len;
+        for (int i = start; i < limit; i++) {
+            c = chars[i];
+            switch (c) {
+                case '\n': size += 2;
+                    break;
+                case '\r': size += 2;
+                    break;
+                case '\t': size += 2;
+                    break;
+                case '\\': size += 2;
+                    break;
+                case '"': size += 2;
+                    break;
+                default:
+                    size += (c < 0x80 ? 1 : (c < 0x800 ? 2 : 3));
+                    break;
+            }
+        }
+        return size;
+    }
+
     /**
      * <b>注意：</b> 该String值不能为null且不会进行转义， 只用于不含需要转义字符的字符串，例如enum、double、BigInteger转换的String
      *
@@ -188,7 +242,7 @@ public final class JsonByteBufferWriter extends JsonWriter {
     @Override
     public void writeTo(final boolean quote, final String value) {
         char[] chs = Utility.charArray(value);
-        writeTo(quote, chs, 0, chs.length);
+        writeTo(-1, quote, chs, 0, chs.length);
     }
 
     @Override
@@ -216,28 +270,65 @@ public final class JsonByteBufferWriter extends JsonWriter {
             }
         }
         if (len == chs.length) {
-            writeTo(true, chs, 0, len);
+            writeTo(-1, true, chs, 0, len);
-        } else {
+            return;
-            StringBuilder sb = new StringBuilder(len);
-            for (char ch : chs) {
-                switch (ch) {
-                    case '\n': sb.append("\\n");
-                        break;
-                    case '\r': sb.append("\\r");
-                        break;
-                    case '\t': sb.append("\\t");
-                        break;
-                    case '\\': sb.append("\\\\");
-                        break;
-                    case '"': sb.append("\\\"");
-                        break;
-                    default: sb.append(ch);
-                        break;
-                }
-            }
-            char[] cs = Utility.charArray(sb);
-            writeTo(true, cs, 0, sb.length());
         }
+        int expandsize = -1;
+        if (this.charset == null) { //UTF-8
+            final int byteLength = 2 + encodeEscapeUTF8Length(chs, 0, chs.length);
+            expandsize = expand(byteLength);
+            if (expandsize == 0) { // 只需要一个buffer 
+                final ByteBuffer buffer = this.buffers[index];
+                buffer.put((byte) '"');
+                for (char c : chs) {
+                    switch (c) {
+                        case '\n': buffer.put((byte) '\\').put((byte) 'n');
+                            break;
+                        case '\r': buffer.put((byte) '\\').put((byte) 'r');
+                            break;
+                        case '\t': buffer.put((byte) '\\').put((byte) 't');
+                            break;
+                        case '\\': buffer.put((byte) '\\').put((byte) '\\');
+                            break;
+                        case '"': buffer.put((byte) '\\').put((byte) '"');
+                            break;
+                        default:
+                            if (c < 0x80) {
+                                buffer.put((byte) c);
+                            } else if (c < 0x800) {
+                                buffer.put((byte) (0xc0 | (c >> 6)));
+                                buffer.put((byte) (0x80 | (c & 0x3f)));
+                            } else {
+                                buffer.put((byte) (0xe0 | ((c >> 12))));
+                                buffer.put((byte) (0x80 | ((c >> 6) & 0x3f)));
+                                buffer.put((byte) (0x80 | (c & 0x3f)));
+                            }
+                            break;
+                    }
+                }
+                buffer.put((byte) '"');
+                return;
+            }
+        }
+        StringBuilder sb = new StringBuilder(len);
+        for (char ch : chs) {
+            switch (ch) {
+                case '\n': sb.append("\\n");
+                    break;
+                case '\r': sb.append("\\r");
+                    break;
+                case '\t': sb.append("\\t");
+                    break;
+                case '\\': sb.append("\\\\");
+                    break;
+                case '"': sb.append("\\\"");
+                    break;
+                default: sb.append(ch);
+                    break;
+            }
+        }
+        char[] cs = Utility.charArray(sb);
+        writeTo(expandsize, true, cs, 0, sb.length());
     }
 
     @Override