mirror of
https://github.com/go-gitea/gitea.git
synced 2024-08-25 13:47:26 +02:00
238 lines
5.9 KiB
Go
238 lines
5.9 KiB
Go
// Copyright 2016 The Xorm Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package builder
|
|
|
|
import (
|
|
"fmt"
|
|
"reflect"
|
|
"strings"
|
|
)
|
|
|
|
type condIn struct {
|
|
col string
|
|
vals []interface{}
|
|
}
|
|
|
|
var _ Cond = condIn{}
|
|
|
|
// In generates IN condition
|
|
func In(col string, values ...interface{}) Cond {
|
|
return condIn{col, values}
|
|
}
|
|
|
|
func (condIn condIn) handleBlank(w Writer) error {
|
|
_, err := fmt.Fprint(w, "0=1")
|
|
return err
|
|
}
|
|
|
|
func (condIn condIn) WriteTo(w Writer) error {
|
|
if len(condIn.vals) <= 0 {
|
|
return condIn.handleBlank(w)
|
|
}
|
|
|
|
switch condIn.vals[0].(type) {
|
|
case []int8:
|
|
vals := condIn.vals[0].([]int8)
|
|
if len(vals) <= 0 {
|
|
return condIn.handleBlank(w)
|
|
}
|
|
questionMark := strings.Repeat("?,", len(vals))
|
|
if _, err := fmt.Fprintf(w, "%s IN (%s)", condIn.col, questionMark[:len(questionMark)-1]); err != nil {
|
|
return err
|
|
}
|
|
for _, val := range vals {
|
|
w.Append(val)
|
|
}
|
|
case []int16:
|
|
vals := condIn.vals[0].([]int16)
|
|
if len(vals) <= 0 {
|
|
return condIn.handleBlank(w)
|
|
}
|
|
questionMark := strings.Repeat("?,", len(vals))
|
|
if _, err := fmt.Fprintf(w, "%s IN (%s)", condIn.col, questionMark[:len(questionMark)-1]); err != nil {
|
|
return err
|
|
}
|
|
for _, val := range vals {
|
|
w.Append(val)
|
|
}
|
|
case []int:
|
|
vals := condIn.vals[0].([]int)
|
|
if len(vals) <= 0 {
|
|
return condIn.handleBlank(w)
|
|
}
|
|
questionMark := strings.Repeat("?,", len(vals))
|
|
if _, err := fmt.Fprintf(w, "%s IN (%s)", condIn.col, questionMark[:len(questionMark)-1]); err != nil {
|
|
return err
|
|
}
|
|
for _, val := range vals {
|
|
w.Append(val)
|
|
}
|
|
case []int32:
|
|
vals := condIn.vals[0].([]int32)
|
|
if len(vals) <= 0 {
|
|
return condIn.handleBlank(w)
|
|
}
|
|
questionMark := strings.Repeat("?,", len(vals))
|
|
if _, err := fmt.Fprintf(w, "%s IN (%s)", condIn.col, questionMark[:len(questionMark)-1]); err != nil {
|
|
return err
|
|
}
|
|
for _, val := range vals {
|
|
w.Append(val)
|
|
}
|
|
case []int64:
|
|
vals := condIn.vals[0].([]int64)
|
|
if len(vals) <= 0 {
|
|
return condIn.handleBlank(w)
|
|
}
|
|
questionMark := strings.Repeat("?,", len(vals))
|
|
if _, err := fmt.Fprintf(w, "%s IN (%s)", condIn.col, questionMark[:len(questionMark)-1]); err != nil {
|
|
return err
|
|
}
|
|
for _, val := range vals {
|
|
w.Append(val)
|
|
}
|
|
case []uint8:
|
|
vals := condIn.vals[0].([]uint8)
|
|
if len(vals) <= 0 {
|
|
return condIn.handleBlank(w)
|
|
}
|
|
questionMark := strings.Repeat("?,", len(vals))
|
|
if _, err := fmt.Fprintf(w, "%s IN (%s)", condIn.col, questionMark[:len(questionMark)-1]); err != nil {
|
|
return err
|
|
}
|
|
for _, val := range vals {
|
|
w.Append(val)
|
|
}
|
|
case []uint16:
|
|
vals := condIn.vals[0].([]uint16)
|
|
if len(vals) <= 0 {
|
|
return condIn.handleBlank(w)
|
|
}
|
|
questionMark := strings.Repeat("?,", len(vals))
|
|
if _, err := fmt.Fprintf(w, "%s IN (%s)", condIn.col, questionMark[:len(questionMark)-1]); err != nil {
|
|
return err
|
|
}
|
|
for _, val := range vals {
|
|
w.Append(val)
|
|
}
|
|
case []uint:
|
|
vals := condIn.vals[0].([]uint)
|
|
if len(vals) <= 0 {
|
|
return condIn.handleBlank(w)
|
|
}
|
|
questionMark := strings.Repeat("?,", len(vals))
|
|
if _, err := fmt.Fprintf(w, "%s IN (%s)", condIn.col, questionMark[:len(questionMark)-1]); err != nil {
|
|
return err
|
|
}
|
|
for _, val := range vals {
|
|
w.Append(val)
|
|
}
|
|
case []uint32:
|
|
vals := condIn.vals[0].([]uint32)
|
|
if len(vals) <= 0 {
|
|
return condIn.handleBlank(w)
|
|
}
|
|
questionMark := strings.Repeat("?,", len(vals))
|
|
if _, err := fmt.Fprintf(w, "%s IN (%s)", condIn.col, questionMark[:len(questionMark)-1]); err != nil {
|
|
return err
|
|
}
|
|
for _, val := range vals {
|
|
w.Append(val)
|
|
}
|
|
case []uint64:
|
|
vals := condIn.vals[0].([]uint64)
|
|
if len(vals) <= 0 {
|
|
return condIn.handleBlank(w)
|
|
}
|
|
questionMark := strings.Repeat("?,", len(vals))
|
|
if _, err := fmt.Fprintf(w, "%s IN (%s)", condIn.col, questionMark[:len(questionMark)-1]); err != nil {
|
|
return err
|
|
}
|
|
for _, val := range vals {
|
|
w.Append(val)
|
|
}
|
|
case []string:
|
|
vals := condIn.vals[0].([]string)
|
|
if len(vals) <= 0 {
|
|
return condIn.handleBlank(w)
|
|
}
|
|
questionMark := strings.Repeat("?,", len(vals))
|
|
if _, err := fmt.Fprintf(w, "%s IN (%s)", condIn.col, questionMark[:len(questionMark)-1]); err != nil {
|
|
return err
|
|
}
|
|
for _, val := range vals {
|
|
w.Append(val)
|
|
}
|
|
case []interface{}:
|
|
vals := condIn.vals[0].([]interface{})
|
|
if len(vals) <= 0 {
|
|
return condIn.handleBlank(w)
|
|
}
|
|
questionMark := strings.Repeat("?,", len(vals))
|
|
if _, err := fmt.Fprintf(w, "%s IN (%s)", condIn.col, questionMark[:len(questionMark)-1]); err != nil {
|
|
return err
|
|
}
|
|
w.Append(vals...)
|
|
case expr:
|
|
val := condIn.vals[0].(expr)
|
|
if _, err := fmt.Fprintf(w, "%s IN (", condIn.col); err != nil {
|
|
return err
|
|
}
|
|
if err := val.WriteTo(w); err != nil {
|
|
return err
|
|
}
|
|
if _, err := fmt.Fprintf(w, ")"); err != nil {
|
|
return err
|
|
}
|
|
case *Builder:
|
|
bd := condIn.vals[0].(*Builder)
|
|
if _, err := fmt.Fprintf(w, "%s IN (", condIn.col); err != nil {
|
|
return err
|
|
}
|
|
if err := bd.WriteTo(w); err != nil {
|
|
return err
|
|
}
|
|
if _, err := fmt.Fprintf(w, ")"); err != nil {
|
|
return err
|
|
}
|
|
default:
|
|
v := reflect.ValueOf(condIn.vals[0])
|
|
if v.Kind() == reflect.Slice {
|
|
l := v.Len()
|
|
if l == 0 {
|
|
return condIn.handleBlank(w)
|
|
}
|
|
|
|
questionMark := strings.Repeat("?,", l)
|
|
if _, err := fmt.Fprintf(w, "%s IN (%s)", condIn.col, questionMark[:len(questionMark)-1]); err != nil {
|
|
return err
|
|
}
|
|
|
|
for i := 0; i < l; i++ {
|
|
w.Append(v.Index(i).Interface())
|
|
}
|
|
} else {
|
|
questionMark := strings.Repeat("?,", len(condIn.vals))
|
|
if _, err := fmt.Fprintf(w, "%s IN (%s)", condIn.col, questionMark[:len(questionMark)-1]); err != nil {
|
|
return err
|
|
}
|
|
w.Append(condIn.vals...)
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (condIn condIn) And(conds ...Cond) Cond {
|
|
return And(condIn, And(conds...))
|
|
}
|
|
|
|
func (condIn condIn) Or(conds ...Cond) Cond {
|
|
return Or(condIn, Or(conds...))
|
|
}
|
|
|
|
func (condIn condIn) IsValid() bool {
|
|
return len(condIn.col) > 0 && len(condIn.vals) > 0
|
|
}
|