操作日志

internal/service/device_import/audit.go

@@ -0,0 +1,56 @@
+package device_import
+
+import (
+	"context"
+
+	"github.com/break/junhong_cmp_fiber/internal/model"
+	"github.com/break/junhong_cmp_fiber/internal/model/dto"
+	assetAuditSvc "github.com/break/junhong_cmp_fiber/internal/service/asset_audit"
+	"github.com/break/junhong_cmp_fiber/pkg/constants"
+)
+
+// AssetAuditService 资产审计服务接口。
+type AssetAuditService interface {
+	LogOperation(ctx context.Context, log *model.AssetOperationLog)
+}
+
+func (s *Service) logDeviceImportAudit(ctx context.Context, p assetAuditSvc.BuildLogParams) {
+	if s == nil || s.assetAudit == nil {
+		return
+	}
+	if p.Operator.Type == "" {
+		p.Operator = assetAuditSvc.OperatorFromContext(ctx)
+	}
+	if p.OperationType == "" {
+		p.OperationType = constants.AssetAuditOpDeviceImportTaskCreate
+	}
+	if p.AssetType == "" {
+		p.AssetType = constants.AssetTypeDevice
+	}
+	s.assetAudit.LogOperation(ctx, assetAuditSvc.BuildLog(ctx, p))
+}
+
+func newDeviceImportAuditParams(
+	taskID uint,
+	taskNo string,
+	req *dto.ImportDeviceRequest,
+	resultStatus string,
+	err error,
+) assetAuditSvc.BuildLogParams {
+	afterData := map[string]any{}
+	if req != nil {
+		afterData["batch_no"] = req.BatchNo
+		afterData["file_key"] = req.FileKey
+		afterData["realname_policy"] = req.RealnamePolicy
+	}
+	errorCode, errorMsg := assetAuditSvc.BuildErrorInfo(err)
+	return assetAuditSvc.BuildLogParams{
+		AssetID:         taskID,
+		AssetIdentifier: taskNo,
+		OperationDesc:   "创建设备导入任务",
+		ResultStatus:    resultStatus,
+		ErrorCode:       errorCode,
+		ErrorMsg:        errorMsg,
+		AfterData:       afterData,
+	}
+}

internal/service/device_import/service.go

@@ -20,24 +20,33 @@ type Service struct {
 	db              *gorm.DB
 	importTaskStore *postgres.DeviceImportTaskStore
 	queueClient     *queue.Client
+	assetAudit      AssetAuditService
 }
 
 type DeviceImportPayload struct {
 	TaskID uint `json:"task_id"`
 }
 
-func New(db *gorm.DB, importTaskStore *postgres.DeviceImportTaskStore, queueClient *queue.Client) *Service {
+func New(
+	db *gorm.DB,
+	importTaskStore *postgres.DeviceImportTaskStore,
+	queueClient *queue.Client,
+	assetAudit AssetAuditService,
+) *Service {
 	return &Service{
 		db:              db,
 		importTaskStore: importTaskStore,
 		queueClient:     queueClient,
+		assetAudit:      assetAudit,
 	}
 }
 
 func (s *Service) CreateImportTask(ctx context.Context, req *dto.ImportDeviceRequest) (*dto.ImportDeviceResponse, error) {
 	userID := middleware.GetUserIDFromContext(ctx)
 	if userID == 0 {
-		return nil, errors.New(errors.CodeUnauthorized, "未授权访问")
+		appErr := errors.New(errors.CodeUnauthorized, "未授权访问")
+		s.logDeviceImportAudit(ctx, newDeviceImportAuditParams(0, "", req, constants.AssetAuditResultDenied, appErr))
+		return nil, appErr
 	}
 
 	taskNo := s.importTaskStore.GenerateTaskNo(ctx)
@@ -55,16 +64,22 @@ func (s *Service) CreateImportTask(ctx context.Context, req *dto.ImportDeviceReq
 	task.Updater = userID
 
 	if err := s.importTaskStore.Create(ctx, task); err != nil {
-		return nil, errors.Wrap(errors.CodeInternalError, err, "创建导入任务失败")
+		appErr := errors.Wrap(errors.CodeInternalError, err, "创建导入任务失败")
+		s.logDeviceImportAudit(ctx, newDeviceImportAuditParams(0, taskNo, req, constants.AssetAuditResultFailed, appErr))
+		return nil, appErr
 	}
 
 	payload := DeviceImportPayload{TaskID: task.ID}
 	err := s.queueClient.EnqueueTask(ctx, constants.TaskTypeDeviceImport, payload)
 	if err != nil {
 		s.importTaskStore.UpdateStatus(ctx, task.ID, model.ImportTaskStatusFailed, "任务入队失败: "+err.Error())
-		return nil, errors.Wrap(errors.CodeInternalError, err, "任务入队失败")
+		appErr := errors.Wrap(errors.CodeInternalError, err, "任务入队失败")
+		s.logDeviceImportAudit(ctx, newDeviceImportAuditParams(task.ID, taskNo, req, constants.AssetAuditResultFailed, appErr))
+		return nil, appErr
 	}
 
+	s.logDeviceImportAudit(ctx, newDeviceImportAuditParams(task.ID, taskNo, req, constants.AssetAuditResultSuccess, nil))
+
 	return &dto.ImportDeviceResponse{
 		TaskID:  task.ID,
 		TaskNo:  taskNo,