re
/
gorm
forked from mirror/gorm
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Overwrite Select conditions

This commit is contained in:
Jinzhu 2015-02-23 08:58:52 +08:00
parent 49a33dd5d0
commit 068e774540
2 changed files with 9 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
scope_private.go
View File

@ -211,14 +211,7 @@ func (scope *Scope) selectSql() string {
if len(scope.Search.Selects) == 0 {
return "*"
}
var selectQueries []string
for _, clause := range scope.Search.Selects {
selectQueries = append(selectQueries, scope.buildSelectQuery(clause))
}
return strings.Join(selectQueries, ", ")
return scope.buildSelectQuery(scope.Search.Selects)
}
func (scope *Scope) orderSql() string {
@ -370,9 +363,7 @@ func (scope *Scope) pluck(column string, value interface{}) *Scope {
}
func (scope *Scope) count(value interface{}) *Scope {
scope.Search = scope.Search.clone()
scope.Search.Selects = []map[string]interface{}{}
scope.Search.selects("count(*)")
scope.Search = scope.Search.clone().selects("count(*)")
scope.Err(scope.row().Scan(value))
return scope
}

14
search.go
View File

@ -7,12 +7,12 @@ type search struct {
WhereConditions []map[string]interface{}
OrConditions []map[string]interface{}
NotConditions []map[string]interface{}
HavingCondition map[string]interface{}
InitAttrs []interface{}
AssignAttrs []interface{}
HavingCondition map[string]interface{}
Selects map[string]interface{}
Orders []string
Joins string
Selects []map[string]interface{}
Preload map[string][]interface{}
Offset string
Limit string
@ -28,17 +28,17 @@ func (s *search) clone() *search {
WhereConditions: s.WhereConditions,
OrConditions: s.OrConditions,
NotConditions: s.NotConditions,
HavingCondition: s.HavingCondition,
InitAttrs: s.InitAttrs,
AssignAttrs: s.AssignAttrs,
HavingCondition: s.HavingCondition,
Orders: s.Orders,
Selects: s.Selects,
Orders: s.Orders,
Joins: s.Joins,
Offset: s.Offset,
Limit: s.Limit,
Unscope: s.Unscope,
Group: s.Group,
Joins: s.Joins,
TableName: s.TableName,
Unscope: s.Unscope,
Raw: s.Raw,
}
}
@ -78,7 +78,7 @@ func (s *search) order(value string, reorder ...bool) *search {
}
func (s *search) selects(query interface{}, args ...interface{}) *search {
s.Selects = append(s.Selects, map[string]interface{}{"query": query, "args": args})
s.Selects = map[string]interface{}{"query": query, "args": args}
return s
}