diff options
author | Luke Shumaker <lukeshu@lukeshu.com> | 2023-02-16 22:56:37 -0700 |
---|---|---|
committer | Luke Shumaker <lukeshu@lukeshu.com> | 2023-02-16 22:56:37 -0700 |
commit | a0113140d447e59ce02d131499861aeafb02d328 (patch) | |
tree | 3a61b0c070a5db186e2c49fe70dff6f40431124e /reencode_nl.go | |
parent | 6f8e7db1ac5ddd21b8e3fcc39a1e30fde9b62c3a (diff) | |
parent | d19e2c6884c2d409fcc828c870f1839ee84f38cb (diff) |
Merge branch 'lukeshu/reencode-refactor'
Diffstat (limited to 'reencode_nl.go')
-rw-r--r-- | reencode_nl.go | 39 |
1 files changed, 39 insertions, 0 deletions
diff --git a/reencode_nl.go b/reencode_nl.go new file mode 100644 index 0000000..b7a3cd9 --- /dev/null +++ b/reencode_nl.go @@ -0,0 +1,39 @@ +// Copyright (C) 2022-2023 Luke Shumaker <lukeshu@lukeshu.com> +// +// SPDX-License-Identifier: GPL-2.0-or-later + +package lowmemjson + +import ( + "git.lukeshu.com/go/lowmemjson/internal/jsonparse" +) + +type reEncodeForceNL struct { + out reEncoderModule + + // state + skipNL bool +} + +var _ reEncoderModule = (*reEncodeForceNL)(nil) + +func (enc *reEncodeForceNL) PopWriteBarrier() { + enc.out.PopWriteBarrier() +} + +func (enc *reEncodeForceNL) HandleRune(c rune, t jsonparse.RuneType, escape BackslashEscapeMode, stackSize int) error { + switch { + case t == jsonparse.RuneTypeEOF: + if err := enc.out.HandleRune('\n', jsonparse.RuneTypeSpace, 0, stackSize); err != nil { + return err + } + enc.skipNL = true + return enc.out.HandleRune(c, t, escape, stackSize) + case c == '\n' && t == jsonparse.RuneTypeSpace && enc.skipNL: + enc.skipNL = false + return nil + default: + enc.skipNL = false + return enc.out.HandleRune(c, t, escape, stackSize) + } +} |