-
Notifications
You must be signed in to change notification settings - Fork 7
/
base_query.go
58 lines (46 loc) · 1.02 KB
/
base_query.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
package materialize
import (
"fmt"
"sort"
"strings"
)
type BaseQuery struct {
statement string
customPredicate []string
order string
}
func NewBaseQuery(statement string) *BaseQuery {
return &BaseQuery{statement: statement}
}
func (b *BaseQuery) CustomPredicate(c []string) *BaseQuery {
b.customPredicate = c
return b
}
func (b *BaseQuery) Order(c string) *BaseQuery {
b.order = c
return b
}
func (b *BaseQuery) QueryPredicate(predicate map[string]string) string {
q := strings.Builder{}
q.WriteString(b.statement)
var p []string
// predicate mapping
for k, v := range predicate {
if v != "" {
p = append(p, fmt.Sprintf(`%s = %s`, k, QuoteString(v)))
}
}
// custom predicates
p = append(p, b.customPredicate...)
if len(p) > 0 {
// sort predicateds for testing consistency
sort.Strings(p)
f := strings.Join(p, " AND ")
q.WriteString(fmt.Sprintf(` WHERE %s`, f))
}
if b.order != "" {
q.WriteString(fmt.Sprintf(` ORDER BY %s`, b.order))
}
q.WriteString(";")
return q.String()
}