操作日志

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,