ipbl/vendor/xorm.io/builder/cond_notin.go

324 lines
9.5 KiB
Go
Raw Normal View History

2021-11-07 20:44:48 +01:00
// 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 condNotIn condIn
var _ Cond = condNotIn{}
// NotIn generate NOT IN condition
func NotIn(col string, values ...interface{}) Cond {
return condNotIn{col, values}
}
func (condNotIn condNotIn) handleBlank(w Writer) error {
_, err := fmt.Fprint(w, "0=0")
return err
}
func (condNotIn condNotIn) WriteTo(w Writer) error {
if len(condNotIn.vals) <= 0 {
return condNotIn.handleBlank(w)
}
switch condNotIn.vals[0].(type) {
case []int8:
vals := condNotIn.vals[0].([]int8)
if len(vals) <= 0 {
return condNotIn.handleBlank(w)
}
2022-07-16 11:42:03 +02:00
// We're using this map to track if a parameter was already added to the condition to not add the same multiple times.
trackMap := make(map[int8]bool, len(vals))
2021-11-07 20:44:48 +01:00
for _, val := range vals {
2022-07-16 11:42:03 +02:00
if _, exists := trackMap[val]; exists {
continue
}
2021-11-07 20:44:48 +01:00
w.Append(val)
2022-07-16 11:42:03 +02:00
trackMap[val] = true
}
questionMark := strings.Repeat("?,", len(trackMap))
if _, err := fmt.Fprintf(w, "%s NOT IN (%s)", condNotIn.col, questionMark[:len(questionMark)-1]); err != nil {
return err
2021-11-07 20:44:48 +01:00
}
case []int16:
vals := condNotIn.vals[0].([]int16)
if len(vals) <= 0 {
return condNotIn.handleBlank(w)
}
2022-07-16 11:42:03 +02:00
// We're using this map to track if a parameter was already added to the condition to not add the same multiple times.
trackMap := make(map[int16]bool, len(vals))
2021-11-07 20:44:48 +01:00
for _, val := range vals {
2022-07-16 11:42:03 +02:00
if _, exists := trackMap[val]; exists {
continue
}
2021-11-07 20:44:48 +01:00
w.Append(val)
2022-07-16 11:42:03 +02:00
trackMap[val] = true
}
questionMark := strings.Repeat("?,", len(trackMap))
if _, err := fmt.Fprintf(w, "%s NOT IN (%s)", condNotIn.col, questionMark[:len(questionMark)-1]); err != nil {
return err
2021-11-07 20:44:48 +01:00
}
case []int:
vals := condNotIn.vals[0].([]int)
if len(vals) <= 0 {
return condNotIn.handleBlank(w)
}
2022-07-16 11:42:03 +02:00
// We're using this map to track if a parameter was already added to the condition to not add the same multiple times.
trackMap := make(map[int]bool, len(vals))
2021-11-07 20:44:48 +01:00
for _, val := range vals {
2022-07-16 11:42:03 +02:00
if _, exists := trackMap[val]; exists {
continue
}
2021-11-07 20:44:48 +01:00
w.Append(val)
2022-07-16 11:42:03 +02:00
trackMap[val] = true
}
questionMark := strings.Repeat("?,", len(trackMap))
if _, err := fmt.Fprintf(w, "%s NOT IN (%s)", condNotIn.col, questionMark[:len(questionMark)-1]); err != nil {
return err
2021-11-07 20:44:48 +01:00
}
case []int32:
vals := condNotIn.vals[0].([]int32)
if len(vals) <= 0 {
return condNotIn.handleBlank(w)
}
2022-07-16 11:42:03 +02:00
// We're using this map to track if a parameter was already added to the condition to not add the same multiple times.
trackMap := make(map[int32]bool, len(vals))
2021-11-07 20:44:48 +01:00
for _, val := range vals {
2022-07-16 11:42:03 +02:00
if _, exists := trackMap[val]; exists {
continue
}
2021-11-07 20:44:48 +01:00
w.Append(val)
2022-07-16 11:42:03 +02:00
trackMap[val] = true
}
questionMark := strings.Repeat("?,", len(trackMap))
if _, err := fmt.Fprintf(w, "%s NOT IN (%s)", condNotIn.col, questionMark[:len(questionMark)-1]); err != nil {
return err
2021-11-07 20:44:48 +01:00
}
case []int64:
vals := condNotIn.vals[0].([]int64)
if len(vals) <= 0 {
return condNotIn.handleBlank(w)
}
2022-07-16 11:42:03 +02:00
// We're using this map to track if a parameter was already added to the condition to not add the same multiple times.
trackMap := make(map[int64]bool, len(vals))
2021-11-07 20:44:48 +01:00
for _, val := range vals {
2022-07-16 11:42:03 +02:00
if _, exists := trackMap[val]; exists {
continue
}
2021-11-07 20:44:48 +01:00
w.Append(val)
2022-07-16 11:42:03 +02:00
trackMap[val] = true
}
questionMark := strings.Repeat("?,", len(trackMap))
if _, err := fmt.Fprintf(w, "%s NOT IN (%s)", condNotIn.col, questionMark[:len(questionMark)-1]); err != nil {
return err
2021-11-07 20:44:48 +01:00
}
case []uint8:
vals := condNotIn.vals[0].([]uint8)
if len(vals) <= 0 {
return condNotIn.handleBlank(w)
}
2022-07-16 11:42:03 +02:00
// We're using this map to track if a parameter was already added to the condition to not add the same multiple times.
trackMap := make(map[uint8]bool, len(vals))
2021-11-07 20:44:48 +01:00
for _, val := range vals {
2022-07-16 11:42:03 +02:00
if _, exists := trackMap[val]; exists {
continue
}
2021-11-07 20:44:48 +01:00
w.Append(val)
2022-07-16 11:42:03 +02:00
trackMap[val] = true
}
questionMark := strings.Repeat("?,", len(trackMap))
if _, err := fmt.Fprintf(w, "%s NOT IN (%s)", condNotIn.col, questionMark[:len(questionMark)-1]); err != nil {
return err
2021-11-07 20:44:48 +01:00
}
case []uint16:
vals := condNotIn.vals[0].([]uint16)
if len(vals) <= 0 {
return condNotIn.handleBlank(w)
}
2022-07-16 11:42:03 +02:00
// We're using this map to track if a parameter was already added to the condition to not add the same multiple times.
trackMap := make(map[uint16]bool, len(vals))
2021-11-07 20:44:48 +01:00
for _, val := range vals {
2022-07-16 11:42:03 +02:00
if _, exists := trackMap[val]; exists {
continue
}
2021-11-07 20:44:48 +01:00
w.Append(val)
2022-07-16 11:42:03 +02:00
trackMap[val] = true
}
questionMark := strings.Repeat("?,", len(trackMap))
if _, err := fmt.Fprintf(w, "%s NOT IN (%s)", condNotIn.col, questionMark[:len(questionMark)-1]); err != nil {
return err
2021-11-07 20:44:48 +01:00
}
case []uint:
vals := condNotIn.vals[0].([]uint)
if len(vals) <= 0 {
return condNotIn.handleBlank(w)
}
2022-07-16 11:42:03 +02:00
// We're using this map to track if a parameter was already added to the condition to not add the same multiple times.
trackMap := make(map[uint]bool, len(vals))
2021-11-07 20:44:48 +01:00
for _, val := range vals {
2022-07-16 11:42:03 +02:00
if _, exists := trackMap[val]; exists {
continue
}
2021-11-07 20:44:48 +01:00
w.Append(val)
2022-07-16 11:42:03 +02:00
trackMap[val] = true
}
questionMark := strings.Repeat("?,", len(trackMap))
if _, err := fmt.Fprintf(w, "%s NOT IN (%s)", condNotIn.col, questionMark[:len(questionMark)-1]); err != nil {
return err
2021-11-07 20:44:48 +01:00
}
case []uint32:
vals := condNotIn.vals[0].([]uint32)
if len(vals) <= 0 {
return condNotIn.handleBlank(w)
}
2022-07-16 11:42:03 +02:00
// We're using this map to track if a parameter was already added to the condition to not add the same multiple times.
trackMap := make(map[uint32]bool, len(vals))
2021-11-07 20:44:48 +01:00
for _, val := range vals {
2022-07-16 11:42:03 +02:00
if _, exists := trackMap[val]; exists {
continue
}
2021-11-07 20:44:48 +01:00
w.Append(val)
2022-07-16 11:42:03 +02:00
trackMap[val] = true
}
questionMark := strings.Repeat("?,", len(trackMap))
if _, err := fmt.Fprintf(w, "%s NOT IN (%s)", condNotIn.col, questionMark[:len(questionMark)-1]); err != nil {
return err
2021-11-07 20:44:48 +01:00
}
case []uint64:
vals := condNotIn.vals[0].([]uint64)
if len(vals) <= 0 {
return condNotIn.handleBlank(w)
}
2022-07-16 11:42:03 +02:00
// We're using this map to track if a parameter was already added to the condition to not add the same multiple times.
trackMap := make(map[uint64]bool, len(vals))
2021-11-07 20:44:48 +01:00
for _, val := range vals {
2022-07-16 11:42:03 +02:00
if _, exists := trackMap[val]; exists {
continue
}
2021-11-07 20:44:48 +01:00
w.Append(val)
2022-07-16 11:42:03 +02:00
trackMap[val] = true
}
questionMark := strings.Repeat("?,", len(trackMap))
if _, err := fmt.Fprintf(w, "%s NOT IN (%s)", condNotIn.col, questionMark[:len(questionMark)-1]); err != nil {
return err
2021-11-07 20:44:48 +01:00
}
case []string:
vals := condNotIn.vals[0].([]string)
if len(vals) <= 0 {
return condNotIn.handleBlank(w)
}
2022-07-16 11:42:03 +02:00
// We're using this map to track if a parameter was already added to the condition to not add the same multiple times.
trackMap := make(map[string]bool, len(vals))
2021-11-07 20:44:48 +01:00
for _, val := range vals {
2022-07-16 11:42:03 +02:00
if _, exists := trackMap[val]; exists {
continue
}
2021-11-07 20:44:48 +01:00
w.Append(val)
2022-07-16 11:42:03 +02:00
trackMap[val] = true
}
questionMark := strings.Repeat("?,", len(trackMap))
if _, err := fmt.Fprintf(w, "%s NOT IN (%s)", condNotIn.col, questionMark[:len(questionMark)-1]); err != nil {
return err
2021-11-07 20:44:48 +01:00
}
case []interface{}:
vals := condNotIn.vals[0].([]interface{})
if len(vals) <= 0 {
return condNotIn.handleBlank(w)
}
questionMark := strings.Repeat("?,", len(vals))
if _, err := fmt.Fprintf(w, "%s NOT IN (%s)", condNotIn.col, questionMark[:len(questionMark)-1]); err != nil {
return err
}
w.Append(vals...)
2022-06-05 23:07:32 +02:00
case *Expression:
val := condNotIn.vals[0].(*Expression)
2021-11-07 20:44:48 +01:00
if _, err := fmt.Fprintf(w, "%s NOT IN (", condNotIn.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:
val := condNotIn.vals[0].(*Builder)
if _, err := fmt.Fprintf(w, "%s NOT IN (", condNotIn.col); err != nil {
return err
}
if err := val.WriteTo(w); err != nil {
return err
}
if _, err := fmt.Fprintf(w, ")"); err != nil {
return err
}
default:
v := reflect.ValueOf(condNotIn.vals[0])
if v.Kind() == reflect.Slice {
l := v.Len()
if l == 0 {
return condNotIn.handleBlank(w)
}
2022-07-16 11:42:03 +02:00
trackMap := make(map[interface{}]bool, l)
for i := 0; i < l; i++ {
val := v.Index(i).Interface()
if _, exists := trackMap[val]; exists {
continue
}
w.Append(val)
trackMap[val] = true
}
questionMark := strings.Repeat("?,", len(trackMap))
2021-11-07 20:44:48 +01:00
if _, err := fmt.Fprintf(w, "%s NOT IN (%s)", condNotIn.col, questionMark[:len(questionMark)-1]); err != nil {
return err
}
2022-07-16 11:42:03 +02:00
} else {
// Using a map for better efficiency
trackMap := make(map[interface{}]bool, len(condNotIn.vals))
2021-11-07 20:44:48 +01:00
2022-07-16 11:42:03 +02:00
i := 0
for in, val := range condNotIn.vals {
if _, exists := trackMap[val]; exists {
// This sets empty values to nil, they get sliced off later.
condNotIn.vals[in] = nil
continue
}
trackMap[val] = true
condNotIn.vals[i] = val
i++
2021-11-07 20:44:48 +01:00
}
2022-07-16 11:42:03 +02:00
// Here we slice the slice to only contain those values we defined as correct.
condNotIn.vals = condNotIn.vals[:i]
2021-11-07 20:44:48 +01:00
questionMark := strings.Repeat("?,", len(condNotIn.vals))
if _, err := fmt.Fprintf(w, "%s NOT IN (%s)", condNotIn.col, questionMark[:len(questionMark)-1]); err != nil {
return err
}
w.Append(condNotIn.vals...)
}
}
return nil
}
func (condNotIn condNotIn) And(conds ...Cond) Cond {
return And(condNotIn, And(conds...))
}
func (condNotIn condNotIn) Or(conds ...Cond) Cond {
return Or(condNotIn, Or(conds...))
}
func (condNotIn condNotIn) IsValid() bool {
return len(condNotIn.col) > 0 && len(condNotIn.vals) > 0
}