61 lines
1.5 KiB
Go
61 lines
1.5 KiB
Go
|
// VulcanizeDB
|
||
|
|
// Copyright © 2021 Vulcanize
|
||
|
|
|
||
|
|
// This program is free software: you can redistribute it and/or modify
|
||
|
|
// it under the terms of the GNU Affero General Public License as published by
|
||
|
|
// the Free Software Foundation, either version 3 of the License, or
|
||
|
|
// (at your option) any later version.
|
||
|
|
|
||
|
|
// This program is distributed in the hope that it will be useful,
|
||
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||
|
|
// GNU Affero General Public License for more details.
|
||
|
|
|
||
|
|
// You should have received a copy of the GNU Affero General Public License
|
||
|
|
// along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||
|
|
|
||
|
|
package file
|
||
|
|
|
||
|
|
import "bytes"
|
||
|
|
|
||
|
|
// formatPostgresStringArray parses an array of strings into the proper Postgres string representation of that array
|
||
|
|
func formatPostgresStringArray(a []string) string {
|
||
|
|
if a == nil {
|
||
|
|
return ""
|
||
|
|
}
|
||
|
|
|
||
|
|
if n := len(a); n > 0 {
|
||
|
|
// There will be at least two curly brackets, 2*N bytes of quotes,
|
||
|
|
// and N-1 bytes of delimiters.
|
||
|
|
b := make([]byte, 1, 1+3*n)
|
||
|
|
b[0] = '{'
|
||
|
|
|
||
|
|
b = appendArrayQuotedBytes(b, []byte(a[0]))
|
||
|
|
for i := 1; i < n; i++ {
|
||
|
|
b = append(b, ',')
|
||
|
|
b = appendArrayQuotedBytes(b, []byte(a[i]))
|
||
|
|
}
|
||
|
|
|
||
|
|
return string(append(b, '}'))
|
||
|
|
}
|
||
|
|
|
||
|
|
return "{}"
|
||
|
|
}
|
||
|
|
|
||
|
|
func appendArrayQuotedBytes(b, v []byte) []byte {
|
||
|
|
b = append(b, '"')
|
||
|
|
for {
|
||
|
|
i := bytes.IndexAny(v, `"\`)
|
||
|
|
if i < 0 {
|
||
|
|
b = append(b, v...)
|
||
|
|
break
|
||
|
|
}
|
||
|
|
if i > 0 {
|
||
|
|
b = append(b, v[:i]...)
|
||
|
|
}
|
||
|
|
b = append(b, '\\', v[i])
|
||
|
|
v = v[i+1:]
|
||
|
|
}
|
||
|
|
return append(b, '"')
|
||
|
|
}
|