/
gorm.go
111 lines (95 loc) · 2.79 KB
/
gorm.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
package gorm
import (
"time"
"github.com/LyricTian/gin-admin/v6/pkg/logger"
"github.com/LyricTian/gin-admin/v6/pkg/util"
"github.com/jinzhu/gorm"
"github.com/sirupsen/logrus"
)
var tableName string
// Config 配置参数
type Config struct {
DBType string
DSN string
MaxLifetime int
MaxOpenConns int
MaxIdleConns int
TableName string
}
// New 创建基于gorm的钩子实例(需要指定表名)
func New(c *Config) *Hook {
tableName = c.TableName
db, err := gorm.Open(c.DBType, c.DSN)
if err != nil {
panic(err)
}
db.DB().SetMaxIdleConns(c.MaxIdleConns)
db.DB().SetMaxOpenConns(c.MaxOpenConns)
db.DB().SetConnMaxLifetime(time.Duration(c.MaxLifetime) * time.Second)
db.AutoMigrate(new(LogItem))
return &Hook{
db: db,
}
}
// Hook gorm日志钩子
type Hook struct {
db *gorm.DB
}
// Exec 执行日志写入
func (h *Hook) Exec(entry *logrus.Entry) error {
item := &LogItem{
Level: entry.Level.String(),
Message: entry.Message,
CreatedAt: entry.Time,
}
data := entry.Data
if v, ok := data[logger.TraceIDKey]; ok {
item.TraceID, _ = v.(string)
delete(data, logger.TraceIDKey)
}
if v, ok := data[logger.UserIDKey]; ok {
item.UserID, _ = v.(string)
delete(data, logger.UserIDKey)
}
if v, ok := data[logger.SpanTitleKey]; ok {
item.SpanTitle, _ = v.(string)
delete(data, logger.SpanTitleKey)
}
if v, ok := data[logger.SpanFunctionKey]; ok {
item.SpanFunction, _ = v.(string)
delete(data, logger.SpanFunctionKey)
}
if v, ok := data[logger.VersionKey]; ok {
item.Version, _ = v.(string)
delete(data, logger.VersionKey)
}
if len(data) > 0 {
item.Data = util.JSONMarshalToString(data)
}
result := h.db.Create(item)
if err := result.Error; err != nil {
return err
}
return nil
}
// Close 关闭钩子
func (h *Hook) Close() error {
return h.db.Close()
}
// LogItem 存储日志项
type LogItem struct {
ID uint `gorm:"column:id;primary_key;auto_increment;"` // id
Level string `gorm:"column:level;size:20;index;"` // 日志级别
Message string `gorm:"column:message;size:1024;"` // 消息
TraceID string `gorm:"column:trace_id;size:128;index;"` // 跟踪ID
UserID string `gorm:"column:user_id;size:36;index;"` // 用户ID
SpanTitle string `gorm:"column:span_title;size:256;"` // 跟踪单元标题
SpanFunction string `gorm:"column:span_function;size:256;"` // 跟踪单元函数名
Data string `gorm:"column:data;type:text;"` // 日志数据(json)
Version string `gorm:"column:version;index;size:32;"` // 服务版本号
CreatedAt time.Time `gorm:"column:created_at;index"` // 创建时间
}
// TableName 表名
func (LogItem) TableName() string {
return tableName
}