/
commons.go
60 lines (48 loc) · 1.54 KB
/
commons.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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
package query
import (
"strings"
"gorm.io/gorm/clause"
)
var LocalFilter = "deleted_at is NULL AND agent_id = '00000000-0000-0000-0000-000000000000' OR agent_id IS NULL"
// ParseFilteringQuery parses a filtering query string.
// It returns four slices: 'in', 'notIN', 'prefix', and 'suffix'.
func ParseFilteringQuery(query string) (in []interface{}, notIN []interface{}, prefix, suffix []string) {
items := strings.Split(query, ",")
for _, item := range items {
if strings.HasPrefix(item, "!") {
notIN = append(notIN, strings.TrimPrefix(item, "!"))
} else if strings.HasPrefix(item, "*") {
suffix = append(suffix, strings.TrimPrefix(item, "*"))
} else if strings.HasSuffix(item, "*") {
prefix = append(prefix, strings.TrimSuffix(item, "*"))
} else {
in = append(in, item)
}
}
return
}
func parseAndBuildFilteringQuery(query string, field string) []clause.Expression {
var clauses []clause.Expression
in, notIN, prefixes, suffixes := ParseFilteringQuery(query)
if len(in) > 0 {
clauses = append(clauses, clause.IN{Column: clause.Column{Name: field}, Values: in})
}
if len(notIN) > 0 {
clauses = append(clauses, clause.NotConditions{
Exprs: []clause.Expression{clause.IN{Column: clause.Column{Name: field}, Values: notIN}},
})
}
for _, p := range prefixes {
clauses = append(clauses, clause.Like{
Column: clause.Column{Name: field},
Value: p + "%",
})
}
for _, s := range suffixes {
clauses = append(clauses, clause.Like{
Column: clause.Column{Name: field},
Value: "%" + s,
})
}
return clauses
}