59 lines
1.5 KiB
Go
59 lines
1.5 KiB
Go
package middleware
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"net/http"
|
|
"time"
|
|
|
|
systemmodel "management/internal/erpserver/model/system"
|
|
v1 "management/internal/erpserver/service/v1"
|
|
"management/internal/pkg/know"
|
|
"management/internal/pkg/session"
|
|
|
|
"github.com/drhin/logger"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
func Audit(sess session.Manager, auditLogService v1.AuditLogService, log *logger.Logger) func(http.Handler) http.Handler {
|
|
return func(next http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
start := time.Now()
|
|
|
|
// 提前获取用户信息（同步操作）
|
|
user, err := sess.GetUser(r.Context(), know.StoreName)
|
|
if err != nil {
|
|
log.Error("获取用户会话失败", err)
|
|
next.ServeHTTP(w, r) // 继续处理请求
|
|
return
|
|
}
|
|
|
|
defer func() {
|
|
go func() {
|
|
if user.ID == 0 {
|
|
log.Error("用户信息为空", errors.New("scs get user is empty"))
|
|
return
|
|
}
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
|
|
defer cancel()
|
|
|
|
al := systemmodel.NewAuditLog(r, user.Email, user.OS, user.Browser, start, time.Now())
|
|
if err := auditLogService.Create(ctx, al); err != nil {
|
|
log.Error(err.Error(), err,
|
|
zap.Int32("user_id", user.ID),
|
|
zap.String("user", user.Email),
|
|
zap.String("ip", al.Ip),
|
|
zap.String("os", al.Os),
|
|
zap.String("method", al.Method),
|
|
zap.String("path", al.Url),
|
|
)
|
|
}
|
|
}()
|
|
}()
|
|
|
|
next.ServeHTTP(w, r)
|
|
})
|
|
}
|
|
}
|