From d19e2c6884c2d409fcc828c870f1839ee84f38cb Mon Sep 17 00:00:00 2001 From: Luke Shumaker Date: Tue, 14 Feb 2023 11:44:36 -0700 Subject: reencode: Factor into separate modules --- reencode_nl.go | 39 +++++++++++++++++++++++++++++++++++++++ 1 file changed, 39 insertions(+) create mode 100644 reencode_nl.go (limited to 'reencode_nl.go') 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 +// +// 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) + } +} -- cgit v1.2.3-2-g168b