gopkg/pgx
2
0
Fork 0
Code Activity

Fix: also consider \r, \f, \t as whitespace (jackc/pgtype#86)

Browse Source
This commit is contained in:
Stephane Martin
2021-01-10 01:05:56 +01:00
committed by Jack Christensen
parent 59b79a2e49
commit 6830cc0984
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
array.go
+6 -1
View File
@@ -347,8 +347,13 @@ func quoteArrayElement(src string) string {
return `"` + quoteArrayReplacer.Replace(src) + `"`
}
func isSpace(ch byte) bool {
// see https://github.com/postgres/postgres/blob/REL_12_STABLE/src/backend/parser/scansup.c#L224
return ch == ' ' || ch == '\t' || ch == '\n' || ch == '\r' || ch == '\f'
}
func QuoteArrayElementIfNeeded(src string) string {
if src == "" || (len(src) == 4 && strings.ToLower(src) == "null") || src[0] == ' ' || src[0] == '\n' || src[len(src)-1] == ' ' || src[len(src)-1] == '\n' || strings.ContainsAny(src, `{},"\`) {
if src == "" || (len(src) == 4 && strings.ToLower(src) == "null") || isSpace(src[0]) || isSpace(src[len(src)-1]) || strings.ContainsAny(src, `{},"\`) {
return quoteArrayElement(src)
}
return src