-
Notifications
You must be signed in to change notification settings - Fork 335
/
json.go
42 lines (35 loc) · 858 Bytes
/
json.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
package postgres
import (
"github.com/doug-martin/goqu/v9/exp"
"strconv"
"strings"
)
// DeepIdentJSON constructs expression with chain of JSON operators
// that point value inside JSON document
//
// https://www.postgresql.org/docs/9.3/functions-json.html
func DeepIdentJSON(asJSON bool, jsonDoc exp.Expression, pp ...any) exp.LiteralExpression {
var (
sql strings.Builder
last = len(pp) - 1
)
sql.WriteString("?")
for i, p := range pp {
sql.WriteString("-")
sql.WriteString(">")
if i == last && !asJSON {
sql.WriteString(">")
}
switch path := p.(type) {
case string:
sql.WriteString("'")
sql.WriteString(strings.ReplaceAll(path, "'", `\'`))
sql.WriteString("'")
case int:
sql.WriteString(strconv.Itoa(path))
default:
panic("invalid type")
}
}
return exp.NewLiteralExpression(sql.String(), jsonDoc)
}