fix(dec.arr): handle whitespace after comma

go-faster · Nov 2, 2021 · 1571e57 · 1571e57
1 parent 0d696ce
commit 1571e57
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/dec_arr.go b/dec_arr.go
@@ -61,6 +61,11 @@ func (d *Decoder) Arr(f func(d *Decoder) error) error {
 		return xerrors.Errorf("next: %w", err)
 	}
 	for c == ',' {
+		// Skip whitespace before reading element.
+		if c, err = d.next(); err != nil {
+			return xerrors.Errorf("next: %w", err)
+		}
+		d.unread()
 		if err := f(d); err != nil {
 			return xerrors.Errorf("callback: %w", err)
 		}

diff --git a/dec_arr_test.go b/dec_arr_test.go
@@ -35,6 +35,13 @@ func TestDecoder_Arr(t *testing.T) {
 			require.Error(t, d.Arr(nil))
 		}
 	})
+	t.Run("Whitespace", func(t *testing.T) {
+		d := DecodeStr(`[1 , 2,  3 ,45, 6]`)
+		require.NoError(t, d.Arr(func(d *Decoder) error {
+			_, err := d.Int()
+			return err
+		}))
+	})
 	t.Run("Depth", func(t *testing.T) {
 		var data []byte
 		for i := 0; i <= maxDepth; i++ {