forked from mirror/gorm
118 lines
2.5 KiB
Go
118 lines
2.5 KiB
Go
package gorm
|
|
|
|
import (
|
|
"regexp"
|
|
|
|
"strconv"
|
|
)
|
|
|
|
type search struct {
|
|
db *DB
|
|
whereClause []map[string]interface{}
|
|
orClause []map[string]interface{}
|
|
notClause []map[string]interface{}
|
|
initAttrs []interface{}
|
|
assignAttrs []interface{}
|
|
orders []string
|
|
selectStr string
|
|
offsetStr string
|
|
limitStr string
|
|
tableName string
|
|
unscope bool
|
|
}
|
|
|
|
func (s *search) clone() *search {
|
|
return &search{
|
|
whereClause: s.whereClause,
|
|
orClause: s.orClause,
|
|
notClause: s.notClause,
|
|
initAttrs: s.initAttrs,
|
|
assignAttrs: s.assignAttrs,
|
|
orders: s.orders,
|
|
selectStr: s.selectStr,
|
|
offsetStr: s.offsetStr,
|
|
limitStr: s.limitStr,
|
|
unscope: s.unscope,
|
|
tableName: s.tableName,
|
|
}
|
|
}
|
|
|
|
func (s *search) where(query interface{}, values ...interface{}) *search {
|
|
s.whereClause = append(s.whereClause, map[string]interface{}{"query": query, "args": values})
|
|
return s
|
|
}
|
|
|
|
func (s *search) not(query interface{}, values ...interface{}) *search {
|
|
s.notClause = append(s.notClause, map[string]interface{}{"query": query, "args": values})
|
|
return s
|
|
}
|
|
|
|
func (s *search) or(query interface{}, values ...interface{}) *search {
|
|
s.orClause = append(s.orClause, map[string]interface{}{"query": query, "args": values})
|
|
return s
|
|
}
|
|
|
|
func (s *search) attrs(attrs ...interface{}) *search {
|
|
s.initAttrs = append(s.initAttrs, toSearchableMap(attrs...))
|
|
return s
|
|
}
|
|
|
|
func (s *search) assign(attrs ...interface{}) *search {
|
|
s.assignAttrs = append(s.assignAttrs, toSearchableMap(attrs...))
|
|
return s
|
|
}
|
|
|
|
func (s *search) order(value string, reorder ...bool) *search {
|
|
if len(reorder) > 0 && reorder[0] {
|
|
s.orders = []string{value}
|
|
} else {
|
|
s.orders = append(s.orders, value)
|
|
}
|
|
return s
|
|
}
|
|
|
|
func (s *search) selects(value interface{}) *search {
|
|
s.selectStr = s.getInterfaceAsSql(value)
|
|
return s
|
|
}
|
|
|
|
func (s *search) limit(value interface{}) *search {
|
|
s.limitStr = s.getInterfaceAsSql(value)
|
|
return s
|
|
}
|
|
|
|
func (s *search) offset(value interface{}) *search {
|
|
s.offsetStr = s.getInterfaceAsSql(value)
|
|
return s
|
|
}
|
|
|
|
func (s *search) unscoped() *search {
|
|
s.unscope = true
|
|
return s
|
|
}
|
|
|
|
func (s *search) table(name string) *search {
|
|
s.tableName = name
|
|
return s
|
|
}
|
|
|
|
func (s *search) getInterfaceAsSql(value interface{}) (str string) {
|
|
switch value := value.(type) {
|
|
case string:
|
|
str = value
|
|
case int:
|
|
if value < 0 {
|
|
str = ""
|
|
} else {
|
|
str = strconv.Itoa(value)
|
|
}
|
|
default:
|
|
s.db.err(InvalidSql)
|
|
}
|
|
|
|
if !regexp.MustCompile("^\\s*[\\w\\s,.*()]*\\s*$").MatchString(str) {
|
|
s.db.err(InvalidSql)
|
|
}
|
|
return
|
|
}
|