113 lines
3.2 KiB
Go
113 lines
3.2 KiB
Go
package service
|
|
|
|
import (
|
|
"context"
|
|
|
|
"gitee.com/red-future---jilin-g/common/log/consts"
|
|
"gitee.com/red-future---jilin-g/common/log/dao"
|
|
"gitee.com/red-future---jilin-g/common/log/model/dto"
|
|
logEntity "gitee.com/red-future---jilin-g/common/log/model/entity"
|
|
"gitee.com/red-future---jilin-g/common/utils"
|
|
"github.com/gogf/gf/v2/frame/g"
|
|
"github.com/gogf/gf/v2/util/gconv"
|
|
)
|
|
|
|
type operationLog struct{}
|
|
|
|
// OperationLog 操作日志服务
|
|
var OperationLog = &operationLog{}
|
|
|
|
// RecordCreate 记录创建操作
|
|
func (s *operationLog) RecordCreate(ctx context.Context, serviceName, collection, collectionID string, data map[string]interface{}) error {
|
|
return s.record(ctx, serviceName, collection, collectionID, string(consts.OperationCreate), data)
|
|
}
|
|
|
|
// RecordUpdate 记录更新操作
|
|
func (s *operationLog) RecordUpdate(ctx context.Context, serviceName, collection, collectionID string, data map[string]interface{}) error {
|
|
return s.record(ctx, serviceName, collection, collectionID, string(consts.OperationUpdate), data)
|
|
}
|
|
|
|
// RecordDelete 记录删除操作
|
|
func (s *operationLog) RecordDelete(ctx context.Context, serviceName, collection, collectionID string, data map[string]interface{}) error {
|
|
return s.record(ctx, serviceName, collection, collectionID, string(consts.OperationDelete), data)
|
|
}
|
|
|
|
// BatchRecordCreate 批量记录创建操作
|
|
func (s *operationLog) BatchRecordCreate(ctx context.Context, logs []*logEntity.OperationLog) error {
|
|
return dao.Log.CreateBatch(ctx, logs)
|
|
}
|
|
|
|
// GetByID 根据ID获取操作日志
|
|
func (s *operationLog) GetByID(ctx context.Context, id string) (*dto.OperationLogInfo, error) {
|
|
log, err := dao.Log.GetByID(ctx, id)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var logInfo dto.OperationLogInfo
|
|
if err := gconv.Struct(log, &logInfo); err != nil {
|
|
return nil, err
|
|
}
|
|
logInfo.ID = log.Id.Hex()
|
|
return &logInfo, nil
|
|
}
|
|
|
|
// List 查询操作日志列表
|
|
func (s *operationLog) List(ctx context.Context, req *dto.ListLogsReq) ([]dto.OperationLogInfo, int64, error) {
|
|
logs, total, err := dao.Log.List(ctx, req)
|
|
if err != nil {
|
|
return nil, 0, err
|
|
}
|
|
|
|
var logInfos []dto.OperationLogInfo
|
|
err = gconv.Structs(logs, &logInfos)
|
|
if err != nil {
|
|
return nil, 0, err
|
|
}
|
|
|
|
// 处理特殊字段
|
|
for i, log := range logs {
|
|
logInfos[i].ID = log.Id.Hex()
|
|
}
|
|
|
|
return logInfos, total, nil
|
|
}
|
|
|
|
// record 记录操作日志的通用方法
|
|
func (s *operationLog) record(ctx context.Context, serviceName, collection, collectionID, operation string, data map[string]interface{}) error {
|
|
// 获取用户信息
|
|
user, err := utils.GetUserInfo(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// 获取请求信息
|
|
ipAddress := getHTTPRequestInfo(ctx)
|
|
|
|
var userName string
|
|
if user.UserName != nil {
|
|
userName = gconv.String(user.UserName)
|
|
}
|
|
|
|
log := &logEntity.OperationLog{
|
|
ServiceName: serviceName,
|
|
Collection: collection,
|
|
CollectionID: collectionID,
|
|
Operation: operation,
|
|
UserName: userName,
|
|
IPAddress: ipAddress,
|
|
Data: data,
|
|
}
|
|
|
|
return dao.Log.Create(ctx, log)
|
|
}
|
|
|
|
// getHTTPRequestInfo 从上下文中获取HTTP请求信息
|
|
func getHTTPRequestInfo(ctx context.Context) string {
|
|
request := g.RequestFromCtx(ctx)
|
|
if request != nil {
|
|
return request.GetClientIp()
|
|
}
|
|
return ""
|
|
}
|