-
Notifications
You must be signed in to change notification settings - Fork 0
/
executor.go
98 lines (87 loc) · 2.83 KB
/
executor.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
package support
import (
"github.com/goal-web/contracts"
"github.com/goal-web/database/events"
"github.com/goal-web/supports/exceptions"
"time"
)
type BaseExecutor struct {
executor SqlxExecutor
events contracts.EventDispatcher
wrapper func(sql string) string
}
func NewExecutor(executor SqlxExecutor, dispatcher contracts.EventDispatcher, wrapper func(sql string) string) Executor {
return &BaseExecutor{
executor: executor,
events: dispatcher,
wrapper: wrapper,
}
}
func (base *BaseExecutor) DriverName() string {
return base.executor.DriverName()
}
func (base *BaseExecutor) dispatchEvent(event contracts.Event) {
if base.events != nil {
base.events.Dispatch(event)
}
}
func (base *BaseExecutor) getStatement(sql string) string {
if base.wrapper != nil {
return base.wrapper(sql)
}
return sql
}
func (base *BaseExecutor) Query(query string, args ...any) (results contracts.Collection[contracts.Fields], exception contracts.Exception) {
query = base.getStatement(query)
var timeConsuming time.Duration
var err error
defer func() {
if err == nil {
err = exceptions.WrapException(recover())
}
base.dispatchEvent(&events.QueryExecuted{Sql: query, Bindings: args, Time: timeConsuming, Error: err})
}()
var startAt = time.Now()
rows, err := base.executor.Queryx(query, args...)
timeConsuming = time.Now().Sub(startAt)
if err != nil {
return nil, exceptions.WithError(err)
}
list, parseErr := ParseRowsToCollection(rows)
return list, exceptions.WithError(parseErr)
}
func (base *BaseExecutor) Get(dest any, query string, args ...any) (err contracts.Exception) {
query = base.getStatement(query)
var startAt = time.Now()
defer func() {
if err == nil {
err = exceptions.WrapException(recover())
}
base.dispatchEvent(&events.QueryExecuted{Sql: query, Bindings: args, Time: time.Now().Sub(startAt), Error: err})
}()
return exceptions.WithError(base.executor.Get(dest, query, args...))
}
func (base *BaseExecutor) Select(dest any, query string, args ...any) (err contracts.Exception) {
query = base.getStatement(query)
var startAt = time.Now()
defer func() {
if err == nil {
err = exceptions.WrapException(recover())
}
base.dispatchEvent(&events.QueryExecuted{Sql: query, Bindings: args, Time: time.Now().Sub(startAt), Error: err})
}()
return exceptions.WithError(base.executor.Select(dest, query, args...))
}
func (base *BaseExecutor) Exec(query string, args ...any) (result contracts.Result, exception contracts.Exception) {
query = base.getStatement(query)
var startAt = time.Now()
var err error
defer func() {
if err == nil {
exception = exceptions.WrapException(recover())
}
base.dispatchEvent(&events.QueryExecuted{Sql: query, Bindings: args, Time: time.Now().Sub(startAt), Error: exception})
}()
result, err = base.executor.Exec(query, args...)
return result, exceptions.WithError(err)
}