-
Notifications
You must be signed in to change notification settings - Fork 6
/
Copy pathput.go
37 lines (33 loc) · 784 Bytes
/
put.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
package y
import sq "github.com/Masterminds/squirrel"
// Put inserts new objects
func Put(db sq.BaseRunner, p *Proxy) (int64, error) {
return p.v.put(db, p.schema)
}
func (v plural) put(db sq.BaseRunner, s schema) (int64, error) {
l := v.size()
if l == 0 {
return 0, nil
}
q := builder{s}.forInsert()
for i := 0; i < l; i++ {
ptrs := s.ptrs()
s.set(ptrs, v.index(i))
q = q.Values(ptrs...)
}
result, err := q.RunWith(db).Exec()
if err != nil {
return 0, err
}
return result.RowsAffected()
}
func (v singular) put(db sq.BaseRunner, s schema) (int64, error) {
set := sq.Eq{}
for name, f := range s.fields {
if !f.autoincr {
set[name] = v.field(f.Name).Interface()
}
}
b := builder{s}
return b.grabInsertID(v, b.forInsert().SetMap(set).RunWith(db))
}