Merge pull request #192 from goccy/feature/fix-read-length

Reset read length at finished reading
goccy · Jan 29, 2021 · f6ab5cc · f6ab5cc
2 parents 7c371c2 + 8b20767
commit f6ab5cc
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/ast/ast.go b/ast/ast.go
@@ -129,6 +129,8 @@ type Node interface {
 	readLen() int
 	// append read length
 	addReadLen(int)
+	// clean read length
+	clearLen()
 }
 
 // ScalarNode type for scalar node
@@ -146,6 +148,10 @@ func (n *BaseNode) readLen() int {
 	return n.read
 }
 
+func (n *BaseNode) clearLen() {
+	n.read = 0
+}
+
 func (n *BaseNode) addReadLen(len int) {
 	n.read += len
 }
@@ -176,6 +182,7 @@ func readNode(p []byte, node Node) (int, error) {
 	readLen := node.readLen()
 	remain := len(s) - readLen
 	if remain == 0 {
+		node.clearLen()
 		return 0, io.EOF
 	}
 	size := min(remain, len(p))