操作日志

internal/service/iot_card_import/service.go

@@ -21,6 +21,7 @@ type Service struct {
 	importTaskStore *postgres.IotCardImportTaskStore
 	carrierStore    carrierGetter
 	queueClient     *queue.Client
+	assetAudit      AssetAuditService
 }
 
 type carrierGetter interface {
@@ -43,12 +44,18 @@ func (s *CarrierStore) GetByID(ctx context.Context, id uint) (*model.Carrier, er
 	return &carrier, nil
 }
 
-func New(db *gorm.DB, importTaskStore *postgres.IotCardImportTaskStore, queueClient *queue.Client) *Service {
+func New(
+	db *gorm.DB,
+	importTaskStore *postgres.IotCardImportTaskStore,
+	queueClient *queue.Client,
+	assetAudit AssetAuditService,
+) *Service {
 	return &Service{
 		db:              db,
 		importTaskStore: importTaskStore,
 		carrierStore:    NewCarrierStore(db),
 		queueClient:     queueClient,
+		assetAudit:      assetAudit,
 	}
 }
 
@@ -59,12 +66,16 @@ type IotCardImportPayload struct {
 func (s *Service) CreateImportTask(ctx context.Context, req *dto.ImportIotCardRequest) (*dto.ImportIotCardResponse, error) {
 	userID := middleware.GetUserIDFromContext(ctx)
 	if userID == 0 {
-		return nil, errors.New(errors.CodeUnauthorized, "未授权访问")
+		appErr := errors.New(errors.CodeUnauthorized, "未授权访问")
+		s.logIotCardImportAudit(ctx, newIotCardImportAuditParams(0, "", req, constants.AssetAuditResultDenied, appErr))
+		return nil, appErr
 	}
 
 	carrier, err := s.carrierStore.GetByID(ctx, req.CarrierID)
 	if err != nil {
-		return nil, errors.New(errors.CodeInvalidParam, "运营商不存在")
+		appErr := errors.New(errors.CodeInvalidParam, "运营商不存在")
+		s.logIotCardImportAudit(ctx, newIotCardImportAuditParams(0, "", req, constants.AssetAuditResultDenied, appErr))
+		return nil, appErr
 	}
 
 	taskNo := s.importTaskStore.GenerateTaskNo(ctx)
@@ -91,16 +102,22 @@ func (s *Service) CreateImportTask(ctx context.Context, req *dto.ImportIotCardRe
 	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.logIotCardImportAudit(ctx, newIotCardImportAuditParams(0, taskNo, req, constants.AssetAuditResultFailed, appErr))
+		return nil, appErr
 	}
 
 	payload := IotCardImportPayload{TaskID: task.ID}
 	err = s.queueClient.EnqueueTask(ctx, constants.TaskTypeIotCardImport, 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.logIotCardImportAudit(ctx, newIotCardImportAuditParams(task.ID, taskNo, req, constants.AssetAuditResultFailed, appErr))
+		return nil, appErr
 	}
 
+	s.logIotCardImportAudit(ctx, newIotCardImportAuditParams(task.ID, taskNo, req, constants.AssetAuditResultSuccess, nil))
+
 	return &dto.ImportIotCardResponse{
 		TaskID:  task.ID,
 		TaskNo:  taskNo,