123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238 |
- package builder
- import (
- "fmt"
- "reflect"
- "strings"
- )
- type condIn struct {
- col string
- vals []interface{}
- }
- var _ Cond = condIn{}
- 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
- }
|