astra/plc-mirror
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Properly escape null character in the consumer too

Browse source
This commit is contained in:
Max Ignatenko 2024-03-24 12:43:52 +00:00
parent 693ae1ba0a
commit ffa2faa420
4 changed files with 20 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

14
util/fix/postgres_json.go Normal file
View file

@ -0,0 +1,14 @@
package fix
import (
"bytes"
"regexp"
)
var postgresFixRegexp = regexp.MustCompile(`([^\\](\\\\)*)(\\u0000)+`)
func EscapeNullCharForPostgres(b []byte) []byte {
return postgresFixRegexp.ReplaceAllFunc(b, func(b []byte) []byte {
return bytes.ReplaceAll(b, []byte(`\u0000`), []byte(`<0x00>`))
})
}

26
util/fix/postgres_json_test.go Normal file
View file

@ -0,0 +1,26 @@
package fix
import (
"testing"
)
func TestPostgresFix(t *testing.T) {
type testCase struct{ input, want string }
cases := []testCase{
{`"a"`, `"a"`},
{`"\u0000"`, `"<0x00>"`},
{`"description":"\u0000"`, `"description":"<0x00>"`},
{`"\\u0000"`, `"\\u0000"`},
{`"\\\u0000"`, `"\\<0x00>"`},
{`\n\n\u0000\u0000 \u0000\u0000\u0000\u0000 \u0000\u0000\u0000\u0000\u0000`,
`\n\n<0x00><0x00> <0x00><0x00><0x00><0x00> <0x00><0x00><0x00><0x00><0x00>`},
}
for _, tc := range cases {
got := EscapeNullCharForPostgres([]byte(tc.input))
if string(got) != tc.want {
t.Errorf("escapeNullCharForPostgres(%s) = %s, want %s", tc.input, string(got), tc.want)
}
}
}