Skip to content
项目
群组
代码片段
帮助
当前项目
正在载入...
登录 / 注册
切换导航面板
X
xpkg
项目
项目
详情
活动
周期分析
仓库
仓库
文件
提交
分支
标签
贡献者
图表
比较
统计图
议题
0
议题
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
CI / CD
CI / CD
流水线
作业
日程
统计图
Wiki
Wiki
代码片段
代码片段
成员
成员
折叠边栏
关闭边栏
活动
图像
聊天
创建新问题
作业
提交
问题看板
Open sidebar
屠思豪
xpkg
Commits
bbbda169
提交
bbbda169
authored
11月 18, 2025
作者:
mooncake
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
update
上级
f1fb00e9
隐藏空白字符变更
内嵌
并排
正在显示
7 个修改的文件
包含
57 行增加
和
32 行删除
+57
-32
mysql.go
database/mysql.go
+1
-1
httpcli.go
pkg/httpcli/httpcli.go
+1
-1
logger.go
pkg/logger/logger.go
+6
-6
glog.go
pkg/sgorm/glog/glog.go
+30
-16
mysql.go
pkg/sgorm/mysql/mysql.go
+16
-5
ctx.go
xutils/ctxutils/ctx.go
+2
-2
retry_utils.go
xutils/retryutils/retry_utils.go
+1
-1
没有找到文件。
database/mysql.go
浏览文件 @
bbbda169
...
...
@@ -9,7 +9,7 @@ import (
"gitlab.wanzhuangkj.com/tush/xpkg/pkg/logger"
"gitlab.wanzhuangkj.com/tush/xpkg/pkg/sgorm"
"gitlab.wanzhuangkj.com/tush/xpkg/pkg/sgorm/mysql"
"gitlab.wanzhuangkj.com/tush/xpkg/xutils"
utils
"gitlab.wanzhuangkj.com/tush/xpkg/xutils"
)
func
InitMysql
(
dbConfig
*
config
.
Database
)
(
*
sgorm
.
DB
,
error
)
{
...
...
pkg/httpcli/httpcli.go
浏览文件 @
bbbda169
...
...
@@ -22,7 +22,7 @@ import (
retryutils
"gitlab.wanzhuangkj.com/tush/xpkg/xutils/retryutils"
)
const
defaultTimeout
=
3
0
*
time
.
Second
const
defaultTimeout
=
6
0
*
time
.
Second
type
Request
struct
{
cli
*
http
.
Client
...
...
pkg/logger/logger.go
浏览文件 @
bbbda169
...
...
@@ -192,14 +192,14 @@ func checkNil() {
func
InitZapWriter
(
logger
*
zap
.
Logger
)
*
ZapWriter
{
return
&
ZapWriter
{
logger
:
logger
,
level
:
logger
.
Level
(),
L
:
logger
,
level
:
logger
.
Level
(),
}
}
type
ZapWriter
struct
{
logger
*
zap
.
Logger
level
zapcore
.
Level
// 日志级别(如 Info、Error)
L
*
zap
.
Logger
level
zapcore
.
Level
// 日志级别(如 Info、Error)
}
// 实现 io.Writer 接口
...
...
@@ -207,9 +207,9 @@ func (w *ZapWriter) Write(p []byte) (n int, err error) {
msg
:=
strings
.
TrimSuffix
(
string
(
p
),
"
\n
"
)
// 去除 Gin 日志自带的换行符
switch
w
.
level
{
case
zapcore
.
ErrorLevel
:
w
.
logger
.
Error
(
msg
)
w
.
L
.
Error
(
msg
)
default
:
w
.
logger
.
Info
(
msg
)
w
.
L
.
Info
(
msg
)
}
return
len
(
p
),
nil
}
...
...
pkg/sgorm/glog/glog.go
浏览文件 @
bbbda169
...
...
@@ -3,12 +3,14 @@ package glog
import
(
"context"
"errors"
"runtime"
"strconv"
"strings"
"time"
"go.uber.org/zap"
"gorm.io/gorm"
"gorm.io/gorm/logger"
)
...
...
@@ -17,15 +19,16 @@ const (
)
type
gormLogger
struct
{
gLog
*
zap
.
Logger
requestIDKey
string
logLevel
logger
.
LogLevel
gLog
*
zap
.
Logger
requestIDKey
string
IgnoreRecordNotFoundError
bool
logLevel
logger
.
LogLevel
}
// NewCustomGormLogger custom gorm logger
func
NewCustomGormLogger
(
l
*
zap
.
Logger
,
requestIDKey
string
,
logLevel
logger
.
LogLevel
)
logger
.
Interface
{
func
NewCustomGormLogger
(
l
*
zap
.
Logger
,
requestIDKey
string
,
logLevel
logger
.
LogLevel
,
opts
...
zap
.
Option
)
logger
.
Interface
{
if
l
==
nil
{
l
,
_
=
zap
.
NewProduction
()
l
,
_
=
zap
.
NewProduction
(
opts
...
)
}
if
requestIDKey
==
""
{
requestIDKey
=
"X-Request-ID"
...
...
@@ -34,9 +37,10 @@ func NewCustomGormLogger(l *zap.Logger, requestIDKey string, logLevel logger.Log
logLevel
=
logger
.
Info
}
return
&
gormLogger
{
gLog
:
l
,
requestIDKey
:
requestIDKey
,
logLevel
:
logLevel
,
gLog
:
l
,
requestIDKey
:
requestIDKey
,
logLevel
:
logLevel
,
IgnoreRecordNotFoundError
:
true
,
}
}
...
...
@@ -97,14 +101,24 @@ func (l *gormLogger) Trace(ctx context.Context, begin time.Time, fc func() (sql
sql
=
clearSql
(
sql
)
if
err
!=
nil
{
l
.
gLog
.
Warn
(
"[gorm]"
,
zap
.
Error
(
err
),
zap
.
String
(
"sql"
,
sql
),
rowsField
,
zap
.
String
(
"cost"
,
cost
),
fileLineField
,
requestIDField
(
ctx
,
l
.
requestIDKey
),
)
if
errors
.
Is
(
err
,
gorm
.
ErrRecordNotFound
)
&&
l
.
IgnoreRecordNotFoundError
{
l
.
gLog
.
Info
(
"[gorm]"
,
zap
.
String
(
"sql"
,
sql
),
rowsField
,
zap
.
String
(
"cost"
,
cost
),
fileLineField
,
requestIDField
(
ctx
,
l
.
requestIDKey
),
)
}
else
{
l
.
gLog
.
Warn
(
"[gorm]"
,
zap
.
Error
(
err
),
zap
.
String
(
"sql"
,
sql
),
rowsField
,
zap
.
String
(
"cost"
,
cost
),
fileLineField
,
requestIDField
(
ctx
,
l
.
requestIDKey
),
)
}
return
}
...
...
pkg/sgorm/mysql/mysql.go
浏览文件 @
bbbda169
...
...
@@ -5,6 +5,7 @@ import (
"database/sql"
"log"
"os"
"time"
"github.com/uptrace/opentelemetry-go-extra/otelgorm"
mysqlDriver
"gorm.io/driver/mysql"
...
...
@@ -83,12 +84,12 @@ func gormConfig(o *options) *gorm.Config {
// print SQL
if
o
.
isLog
{
if
o
.
gLog
==
nil
{
config
.
Logger
=
logger
.
Default
.
LogMode
(
o
.
logLevel
)
config
.
Logger
=
defaultLogger
()
.
LogMode
(
o
.
logLevel
)
}
else
{
config
.
Logger
=
glog
.
NewCustomGormLogger
(
o
.
gLog
,
o
.
requestIDKey
,
o
.
logLevel
)
}
}
else
{
config
.
Logger
=
logger
.
Default
.
LogMode
(
logger
.
Silent
)
config
.
Logger
=
defaultLogger
()
.
LogMode
(
logger
.
Silent
)
}
// print only slow queries
...
...
@@ -96,9 +97,10 @@ func gormConfig(o *options) *gorm.Config {
config
.
Logger
=
logger
.
New
(
log
.
New
(
os
.
Stdout
,
"
\r\n
"
,
log
.
LstdFlags
),
// use the standard output asWriter
logger
.
Config
{
SlowThreshold
:
o
.
slowThreshold
,
Colorful
:
true
,
LogLevel
:
logger
.
Warn
,
// set the logging level, only above the specified level will output the slow query log
SlowThreshold
:
o
.
slowThreshold
,
Colorful
:
true
,
LogLevel
:
logger
.
Warn
,
// set the logging level, only above the specified level will output the slow query log
IgnoreRecordNotFoundError
:
true
,
},
)
}
...
...
@@ -106,6 +108,15 @@ func gormConfig(o *options) *gorm.Config {
return
config
}
func
defaultLogger
()
logger
.
Interface
{
return
logger
.
New
(
log
.
New
(
os
.
Stdout
,
"
\r\n
"
,
log
.
LstdFlags
),
logger
.
Config
{
SlowThreshold
:
200
*
time
.
Millisecond
,
LogLevel
:
logger
.
Warn
,
IgnoreRecordNotFoundError
:
true
,
Colorful
:
true
,
})
}
func
rwSeparationPlugin
(
o
*
options
)
gorm
.
Plugin
{
slaves
:=
[]
gorm
.
Dialector
{}
for
_
,
dsn
:=
range
o
.
slavesDsn
{
...
...
xutils/ctxutils/ctx.go
浏览文件 @
bbbda169
...
...
@@ -118,11 +118,11 @@ func CtxTraceIDField(c context.Context) zap.Field {
}
func
NewEmptyCtx
()
context
.
Context
{
return
context
.
WithValue
(
context
.
Background
(),
ContextTraceIDKey
,
GenerateTid
())
return
context
.
WithValue
(
context
.
TODO
(),
ContextTraceIDKey
,
GenerateTid
())
}
func
NewCtx
(
ctx
context
.
Context
)
context
.
Context
{
return
context
.
WithValue
(
context
.
Background
(),
ContextTraceIDKey
,
ctx
.
Value
(
ContextTraceIDKey
))
return
context
.
WithValue
(
context
.
TODO
(),
ContextTraceIDKey
,
ctx
.
Value
(
ContextTraceIDKey
))
}
func
GetCtxString
(
c
context
.
Context
,
key
string
)
string
{
...
...
xutils/retryutils/retry_utils.go
浏览文件 @
bbbda169
...
...
@@ -112,7 +112,7 @@ func Retry(retryFunc RetryFunc, opts ...Option) error {
select
{
case
<-
after
:
case
<-
config
.
context
.
Done
()
:
return
errors
.
New
(
"retry is cancelled"
)
return
errors
.
New
(
"retry is cancelled
, because request timeout or cancelled
"
)
}
}
else
{
return
nil
...
...
编写
预览
Markdown
格式
0%
重试
或
添加新文件
添加附件
取消
您添加了
0
人
到此讨论。请谨慎行事。
请先完成此评论的编辑!
取消
请
注册
或者
登录
后发表评论