forked from gobuffalo/pop
-
Notifications
You must be signed in to change notification settings - Fork 0
/
finders.go
137 lines (123 loc) · 3.54 KB
/
finders.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
package pop
import (
"reflect"
"strconv"
"github.com/satori/go.uuid"
)
// Find the first record of the model in the database with a particular id.
//
// c.Find(&User{}, 1)
func (c *Connection) Find(model interface{}, id interface{}) error {
return Q(c).Find(model, id)
}
// Find the first record of the model in the database with a particular id.
//
// q.Find(&User{}, 1)
func (q *Query) Find(model interface{}, id interface{}) error {
var idi int
switch t := id.(type) {
case int:
idi = t
case uuid.UUID:
return q.Where("id = ?", t.String()).First(model)
case string:
var err error
idi, err = strconv.Atoi(t)
if err != nil {
return q.Where("id = ?", t).First(model)
}
}
return q.Where("id = ?", idi).First(model)
}
// First record of the model in the database that matches the query.
//
// c.First(&User{})
func (c *Connection) First(model interface{}) error {
return Q(c).First(model)
}
// First record of the model in the database that matches the query.
//
// q.Where("name = ?", "mark").First(&User{})
func (q *Query) First(model interface{}) error {
return q.Connection.timeFunc("First", func() error {
q.Limit(1)
m := &Model{Value: model}
return q.Connection.Dialect.SelectOne(q.Connection.Store, m, *q)
})
}
// Last record of the model in the database that matches the query.
//
// c.Last(&User{})
func (c *Connection) Last(model interface{}) error {
return Q(c).Last(model)
}
// Last record of the model in the database that matches the query.
//
// q.Where("name = ?", "mark").Last(&User{})
func (q *Query) Last(model interface{}) error {
return q.Connection.timeFunc("Last", func() error {
q.Limit(1)
q.Order("id desc")
m := &Model{Value: model}
return q.Connection.Dialect.SelectOne(q.Connection.Store, m, *q)
})
}
// All retrieves all of the records in the database that match the query.
//
// c.All(&[]User{})
func (c *Connection) All(models interface{}) error {
return Q(c).All(models)
}
// All retrieves all of the records in the database that match the query.
//
// q.Where("name = ?", "mark").All(&[]User{})
func (q *Query) All(models interface{}) error {
return q.Connection.timeFunc("All", func() error {
m := &Model{Value: models}
err := q.Connection.Dialect.SelectMany(q.Connection.Store, m, *q)
if err == nil && q.Paginator != nil {
ct, err := q.Count(models)
if err == nil {
q.Paginator.TotalEntriesSize = ct
st := reflect.ValueOf(models).Elem()
q.Paginator.CurrentEntriesSize = st.Len()
q.Paginator.TotalPages = (q.Paginator.TotalEntriesSize / q.Paginator.PerPage)
if q.Paginator.TotalEntriesSize%q.Paginator.PerPage > 0 {
q.Paginator.TotalPages = q.Paginator.TotalPages + 1
}
}
}
return err
})
}
// Exists returns true/false if a record exists in the database that matches
// the query.
//
// q.Where("name = ?", "mark").Exists(&User{})
func (q *Query) Exists(model interface{}) (bool, error) {
i, err := q.Count(model)
return i != 0, err
}
// Count the number of records in the database.
//
// c.Count(&User{})
func (c *Connection) Count(model interface{}) (int, error) {
return Q(c).Count(model)
}
// Count the number of records in the database.
//
// q.Where("name = ?", "mark").Count(&User{})
func (q Query) Count(model interface{}) (int, error) {
res := &rowCount{}
err := q.Connection.timeFunc("Count", func() error {
q.Paginator = nil
col := "count(*) as row_count"
query, args := q.ToSQL(&Model{Value: model}, col)
Log(query, args...)
return q.Connection.Store.Get(res, query, args...)
})
return res.Count, err
}
type rowCount struct {
Count int `db:"row_count"`
}