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

pb

Browse Source
This commit is contained in:
redkale
2024-10-20 02:12:30 +08:00
parent 59b9c695ec
commit 7405ce4dab
1 changed files with 18 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
src/main/java/org/redkale/convert/pb/ProtobufBytesWriter.java
View File

@@ -60,7 +60,7 @@ public class ProtobufBytesWriter extends ProtobufWriter { // 存在child情况
s = p; s = p;
p = p.parent; p = p.parent;
offerPool(s); offerPool(s);
} while (p != this && p != null); } while (p != this);
} }
this.delegate = null; this.delegate = null;
if (this.content.length > RESET_MAX_SIZE) { if (this.content.length > RESET_MAX_SIZE) {
@@ -79,6 +79,23 @@ public class ProtobufBytesWriter extends ProtobufWriter { // 存在child情况
@Override @Override
public ProtobufWriter pollChild() { public ProtobufWriter pollChild() {
Queue<ProtobufBytesWriter> queue = this.pool; Queue<ProtobufBytesWriter> queue = this.pool;
if (queue == null) {
// 必须要使用根节点的pool
ProtobufBytesWriter root = null;
if (this.parent != null) {
ProtobufWriter p = this;
while ((p = p.parent) instanceof ProtobufBytesWriter) {
root = (ProtobufBytesWriter) p;
}
}
if (root != null) {
queue = root.pool;
if (queue == null) {
root.pool = new ArrayDeque<>(CHILD_SIZE);
queue = root.pool;
}
}
}
if (queue == null) { if (queue == null) {
this.pool = new ArrayDeque<>(CHILD_SIZE); this.pool = new ArrayDeque<>(CHILD_SIZE);
queue = this.pool; queue = this.pool;