daog-ext/log.go at main · darwinOrg/daog-ext · GitHub

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
package daogext

import (
	"context"

	dgctx "github.com/darwinOrg/go-common/context"
	dglogger "github.com/darwinOrg/go-logger"
	"github.com/rolandhe/daog"
)

func init() {
	daog.GLogger = &daogLogger{}
}

type daogLogger struct {
}

func (dl *daogLogger) Error(ctx context.Context, err error) {
	alarmDatabaseError(getDgContext(ctx), err)
}

func (dl *daogLogger) Info(ctx context.Context, content string) {
	dglogger.Infof(getDgContext(ctx), "[daog] content: %s", content)
}

func (dl *daogLogger) ExecSQLBefore(ctx context.Context, sql string, argsJson []byte, sqlMd5 string) {
	dglogger.Infof(getDgContext(ctx), "[daog] [Trace SQL] sqlMd5=%s, sql: %s, args:%s", sqlMd5, sql, argsJson)
}

func (dl *daogLogger) ExecSQLAfter(ctx context.Context, sqlMd5 string, cost int64) {
	dglogger.Infof(getDgContext(ctx), "[daog] [Trace SQL] sqlMd5=%s, cost %d ms", sqlMd5, cost)
}

func (dl *daogLogger) SimpleLogError(err error) {
	alarmDatabaseError(dgctx.SimpleDgContext(), err)
}

var OnlyErrorLogger = &onlyErrorDaogLogger{}

type onlyErrorDaogLogger struct {
}

func (dl *onlyErrorDaogLogger) Error(ctx context.Context, err error) {
	alarmDatabaseError(getDgContext(ctx), err)
}

func (dl *onlyErrorDaogLogger) Info(ctx context.Context, content string) {
}

func (dl *onlyErrorDaogLogger) ExecSQLBefore(ctx context.Context, sql string, argsJson []byte, sqlMd5 string) {
}

func (dl *onlyErrorDaogLogger) ExecSQLAfter(ctx context.Context, sqlMd5 string, cost int64) {
}

func (dl *onlyErrorDaogLogger) SimpleLogError(err error) {
	alarmDatabaseError(dgctx.SimpleDgContext(), err)
}

func getDgContext(ctx context.Context) *dgctx.DgContext {
	return &dgctx.DgContext{TraceId: daog.GetTraceIdFromContext(ctx), GoId: daog.GetGoroutineIdFromContext(ctx)}
}

func alarmDatabaseError(ctx *dgctx.DgContext, err error) {
	if errorProcessor != nil {
		errorProcessor(ctx, err)
	} else {
		dglogger.Errorf(ctx, "[daog] err: %v", err)
	}
}