← Back to branch summary

~ubuntu-branches/debian/sid/golang-github-go-xorm-builder/sid

« back to all changes in this revision

Viewing changes to cond_in.go

  • Committer: Package Import Robot
  • Author(s): Michael Lustfield
  • Date: 2017-03-12 03:54:03 UTC
  • Revision ID: package-import@ubuntu.com-20170312035403-csl9031z2j2aswfd
Tags: upstream-0.0~git20170224.0.c6e604e
Import upstream version 0.0~git20170224.0.c6e604e

Show diffs side-by-side

added added

removed removed

Lines of Context:
cond_in.go
 
1
// Copyright 2016 The Xorm Authors. All rights reserved.
 
2
// Use of this source code is governed by a BSD-style
 
3
// license that can be found in the LICENSE file.
 
4
 
 
5
package builder
 
6
 
 
7
import (
 
8
        "fmt"
 
9
        "reflect"
 
10
        "strings"
 
11
)
 
12
 
 
13
type condIn struct {
 
14
        col  string
 
15
        vals []interface{}
 
16
}
 
17
 
 
18
var _ Cond = condIn{}
 
19
 
 
20
// In generates IN condition
 
21
func In(col string, values ...interface{}) Cond {
 
22
        return condIn{col, values}
 
23
}
 
24
 
 
25
func (condIn condIn) handleBlank(w Writer) error {
 
26
        if _, err := fmt.Fprintf(w, "%s IN ()", condIn.col); err != nil {
 
27
                return err
 
28
        }
 
29
        return nil
 
30
}
 
31
 
 
32
func (condIn condIn) WriteTo(w Writer) error {
 
33
        if len(condIn.vals) <= 0 {
 
34
                return condIn.handleBlank(w)
 
35
        }
 
36
 
 
37
        switch condIn.vals[0].(type) {
 
38
        case []int8:
 
39
                vals := condIn.vals[0].([]int8)
 
40
                if len(vals) <= 0 {
 
41
                        return condIn.handleBlank(w)
 
42
                }
 
43
                questionMark := strings.Repeat("?,", len(vals))
 
44
                if _, err := fmt.Fprintf(w, "%s IN (%s)", condIn.col, questionMark[:len(questionMark)-1]); err != nil {
 
45
                        return err
 
46
                }
 
47
                for _, val := range vals {
 
48
                        w.Append(val)
 
49
                }
 
50
        case []int16:
 
51
                vals := condIn.vals[0].([]int16)
 
52
                if len(vals) <= 0 {
 
53
                        return condIn.handleBlank(w)
 
54
                }
 
55
                questionMark := strings.Repeat("?,", len(vals))
 
56
                if _, err := fmt.Fprintf(w, "%s IN (%s)", condIn.col, questionMark[:len(questionMark)-1]); err != nil {
 
57
                        return err
 
58
                }
 
59
                for _, val := range vals {
 
60
                        w.Append(val)
 
61
                }
 
62
        case []int:
 
63
                vals := condIn.vals[0].([]int)
 
64
                if len(vals) <= 0 {
 
65
                        return condIn.handleBlank(w)
 
66
                }
 
67
                questionMark := strings.Repeat("?,", len(vals))
 
68
                if _, err := fmt.Fprintf(w, "%s IN (%s)", condIn.col, questionMark[:len(questionMark)-1]); err != nil {
 
69
                        return err
 
70
                }
 
71
                for _, val := range vals {
 
72
                        w.Append(val)
 
73
                }
 
74
        case []int32:
 
75
                vals := condIn.vals[0].([]int32)
 
76
                if len(vals) <= 0 {
 
77
                        return condIn.handleBlank(w)
 
78
                }
 
79
                questionMark := strings.Repeat("?,", len(vals))
 
80
                if _, err := fmt.Fprintf(w, "%s IN (%s)", condIn.col, questionMark[:len(questionMark)-1]); err != nil {
 
81
                        return err
 
82
                }
 
83
                for _, val := range vals {
 
84
                        w.Append(val)
 
85
                }
 
86
        case []int64:
 
87
                vals := condIn.vals[0].([]int64)
 
88
                if len(vals) <= 0 {
 
89
                        return condIn.handleBlank(w)
 
90
                }
 
91
                questionMark := strings.Repeat("?,", len(vals))
 
92
                if _, err := fmt.Fprintf(w, "%s IN (%s)", condIn.col, questionMark[:len(questionMark)-1]); err != nil {
 
93
                        return err
 
94
                }
 
95
                for _, val := range vals {
 
96
                        w.Append(val)
 
97
                }
 
98
        case []uint8:
 
99
                vals := condIn.vals[0].([]uint8)
 
100
                if len(vals) <= 0 {
 
101
                        return condIn.handleBlank(w)
 
102
                }
 
103
                questionMark := strings.Repeat("?,", len(vals))
 
104
                if _, err := fmt.Fprintf(w, "%s IN (%s)", condIn.col, questionMark[:len(questionMark)-1]); err != nil {
 
105
                        return err
 
106
                }
 
107
                for _, val := range vals {
 
108
                        w.Append(val)
 
109
                }
 
110
        case []uint16:
 
111
                vals := condIn.vals[0].([]uint16)
 
112
                if len(vals) <= 0 {
 
113
                        return condIn.handleBlank(w)
 
114
                }
 
115
                questionMark := strings.Repeat("?,", len(vals))
 
116
                if _, err := fmt.Fprintf(w, "%s IN (%s)", condIn.col, questionMark[:len(questionMark)-1]); err != nil {
 
117
                        return err
 
118
                }
 
119
                for _, val := range vals {
 
120
                        w.Append(val)
 
121
                }
 
122
        case []uint:
 
123
                vals := condIn.vals[0].([]uint)
 
124
                if len(vals) <= 0 {
 
125
                        return condIn.handleBlank(w)
 
126
                }
 
127
                questionMark := strings.Repeat("?,", len(vals))
 
128
                if _, err := fmt.Fprintf(w, "%s IN (%s)", condIn.col, questionMark[:len(questionMark)-1]); err != nil {
 
129
                        return err
 
130
                }
 
131
                for _, val := range vals {
 
132
                        w.Append(val)
 
133
                }
 
134
        case []uint32:
 
135
                vals := condIn.vals[0].([]uint32)
 
136
                if len(vals) <= 0 {
 
137
                        return condIn.handleBlank(w)
 
138
                }
 
139
                questionMark := strings.Repeat("?,", len(vals))
 
140
                if _, err := fmt.Fprintf(w, "%s IN (%s)", condIn.col, questionMark[:len(questionMark)-1]); err != nil {
 
141
                        return err
 
142
                }
 
143
                for _, val := range vals {
 
144
                        w.Append(val)
 
145
                }
 
146
        case []uint64:
 
147
                vals := condIn.vals[0].([]uint64)
 
148
                if len(vals) <= 0 {
 
149
                        return condIn.handleBlank(w)
 
150
                }
 
151
                questionMark := strings.Repeat("?,", len(vals))
 
152
                if _, err := fmt.Fprintf(w, "%s IN (%s)", condIn.col, questionMark[:len(questionMark)-1]); err != nil {
 
153
                        return err
 
154
                }
 
155
                for _, val := range vals {
 
156
                        w.Append(val)
 
157
                }
 
158
        case []string:
 
159
                vals := condIn.vals[0].([]string)
 
160
                if len(vals) <= 0 {
 
161
                        return condIn.handleBlank(w)
 
162
                }
 
163
                questionMark := strings.Repeat("?,", len(vals))
 
164
                if _, err := fmt.Fprintf(w, "%s IN (%s)", condIn.col, questionMark[:len(questionMark)-1]); err != nil {
 
165
                        return err
 
166
                }
 
167
                for _, val := range vals {
 
168
                        w.Append(val)
 
169
                }
 
170
        case []interface{}:
 
171
                vals := condIn.vals[0].([]interface{})
 
172
                if len(vals) <= 0 {
 
173
                        return condIn.handleBlank(w)
 
174
                }
 
175
                questionMark := strings.Repeat("?,", len(vals))
 
176
                if _, err := fmt.Fprintf(w, "%s IN (%s)", condIn.col, questionMark[:len(questionMark)-1]); err != nil {
 
177
                        return err
 
178
                }
 
179
                w.Append(vals...)
 
180
        case expr:
 
181
                val := condIn.vals[0].(expr)
 
182
                if _, err := fmt.Fprintf(w, "%s IN (", condIn.col); err != nil {
 
183
                        return err
 
184
                }
 
185
                if err := val.WriteTo(w); err != nil {
 
186
                        return err
 
187
                }
 
188
                if _, err := fmt.Fprintf(w, ")"); err != nil {
 
189
                        return err
 
190
                }
 
191
        case *Builder:
 
192
                bd := condIn.vals[0].(*Builder)
 
193
                if _, err := fmt.Fprintf(w, "%s IN (", condIn.col); err != nil {
 
194
                        return err
 
195
                }
 
196
                if err := bd.WriteTo(w); err != nil {
 
197
                        return err
 
198
                }
 
199
                if _, err := fmt.Fprintf(w, ")"); err != nil {
 
200
                        return err
 
201
                }
 
202
        default:
 
203
                v := reflect.ValueOf(condIn.vals[0])
 
204
                if v.Kind() == reflect.Slice {
 
205
                        l := v.Len()
 
206
                        if l == 0 {
 
207
                                return condIn.handleBlank(w)
 
208
                        }
 
209
 
 
210
                        questionMark := strings.Repeat("?,", l)
 
211
                        if _, err := fmt.Fprintf(w, "%s IN (%s)", condIn.col, questionMark[:len(questionMark)-1]); err != nil {
 
212
                                return err
 
213
                        }
 
214
 
 
215
                        for i := 0; i < l; i++ {
 
216
                                w.Append(v.Index(i).Interface())
 
217
                        }
 
218
                } else {
 
219
                        questionMark := strings.Repeat("?,", len(condIn.vals))
 
220
                        if _, err := fmt.Fprintf(w, "%s IN (%s)", condIn.col, questionMark[:len(questionMark)-1]); err != nil {
 
221
                                return err
 
222
                        }
 
223
                        w.Append(condIn.vals...)
 
224
                }
 
225
        }
 
226
        return nil
 
227
}
 
228
 
 
229
func (condIn condIn) And(conds ...Cond) Cond {
 
230
        return And(condIn, And(conds...))
 
231
}
 
232
 
 
233
func (condIn condIn) Or(conds ...Cond) Cond {
 
234
        return Or(condIn, Or(conds...))
 
235
}
 
236
 
 
237
func (condIn condIn) IsValid() bool {
 
238
        return len(condIn.col) > 0 && len(condIn.vals) > 0
 
239
}