-
Notifications
You must be signed in to change notification settings - Fork 4
/
result.go
48 lines (38 loc) · 873 Bytes
/
result.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
// Copyright 2022 lesismal. All rights reserved.
// Use of this source code is governed by an MIT-style
// license that can be found in the LICENSE file.
package sqlw
import (
"database/sql"
"fmt"
)
type sqlResult struct {
sql.Result
query string
args []interface{}
}
type Result = *sqlResult
func (r *sqlResult) LastInsertId() (int64, error) {
if r.Result != nil {
return r.Result.LastInsertId()
}
return 0, nil
}
func (r *sqlResult) RowsAffected() (int64, error) {
if r.Result != nil {
return r.Result.RowsAffected()
}
return 0, nil
}
func (r *sqlResult) Sql() string {
return fmt.Sprintf(`[%s], %v`, r.query, r.args)
}
func (r *sqlResult) Query() string {
return r.query
}
func (r *sqlResult) Args() []interface{} {
return r.args
}
func newResult(r sql.Result, query string, args []interface{}) Result {
return &sqlResult{r, query, args}
}