Merge pull request buger#241 from unxcepted/master

fix EachKey pIdxFlags allocation
autopilot3 · Apr 18, 2022 · 61b32cf · 61b32cf
2 parents 2181e83 + 6fc2e48
commit 61b32cf
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/parser.go b/parser.go
@@ -499,7 +499,12 @@ func EachKey(data []byte, cb func(int, []byte, ValueType, error), paths ...[]str
 		case '[':
 			var ok bool
 			arrIdxFlags := make(map[int]struct{})
-			pIdxFlags := make([]bool, len(paths))
+
+			pIdxFlags := make([]bool, stackArraySize)[:]
+			if len(paths) > cap(pIdxFlags) {
+				pIdxFlags = make([]bool, len(paths))[:]
+			}
+			pIdxFlags = pIdxFlags[0:len(paths)]
 
 			if level < 0 {
 				cb(-1, nil, Unknown, MalformedJsonError)