lxy/redkale
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Browse Source
This commit is contained in:
kamhung
2015-11-10 11:03:58 +08:00
parent e2812ba11d
commit 04dbcc664e
1 changed files with 119 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

147
src/com/wentch/redkale/convert/json/JsonByteBufferWriter.java
View File

@@ -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);
} else {
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());
writeTo(-1, true, chs, 0, len);
return;
}
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