forked from go-pg/pg
-
Notifications
You must be signed in to change notification settings - Fork 0
/
struct_filter.go
53 lines (41 loc) · 924 Bytes
/
struct_filter.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
package orm
import (
"reflect"
"sync"
"github.com/go-pg/pg/internal/struct_filter"
)
type structFilter struct {
value reflect.Value // reflect.Struct
strctOnce sync.Once
strct *struct_filter.Struct // lazy
}
var _ sepFormatAppender = (*structFilter)(nil)
func newStructFilter(v interface{}) *structFilter {
if v, ok := v.(*structFilter); ok {
return v
}
return &structFilter{
value: reflect.Indirect(reflect.ValueOf(v)),
}
}
func (sf *structFilter) AppendSep(b []byte) []byte {
return append(b, " AND "...)
}
func (sf *structFilter) AppendFormat(b []byte, f QueryFormatter) []byte {
const and = " AND "
sf.strctOnce.Do(func() {
sf.strct = struct_filter.GetStruct(sf.value.Type())
})
before := len(b)
for _, f := range sf.strct.Fields {
fv := f.Value(sf.value)
if f.Omit(fv) {
continue
}
if len(b) != before {
b = append(b, and...)
}
b = f.Append(b, fv)
}
return b
}