break
c64f3d8b80
All checks were successful
构建并部署到测试环境（无 SSH） / build-and-deploy (push) Successful in 8m31s
77 lines
3.4 KiB
Go
77 lines
3.4 KiB
Go
package device_import
|
|
|
|
import (
|
|
"context"
|
|
"strconv"
|
|
"time"
|
|
|
|
"gorm.io/gorm"
|
|
|
|
infraAudit "github.com/break/junhong_cmp_fiber/internal/infrastructure/audit"
|
|
"github.com/break/junhong_cmp_fiber/internal/model"
|
|
"github.com/break/junhong_cmp_fiber/pkg/auditfailure"
|
|
"github.com/break/junhong_cmp_fiber/pkg/constants"
|
|
"github.com/break/junhong_cmp_fiber/pkg/errors"
|
|
"github.com/break/junhong_cmp_fiber/pkg/middleware"
|
|
)
|
|
|
|
func (s *Service) writeDeviceImportTaskAudit(ctx context.Context, tx *gorm.DB, task *model.DeviceImportTask, before, after map[string]any, result, phase, errorCode, errorSummary string) error {
|
|
scopeType, scopeID := constants.AuditScopePlatform, ""
|
|
if task.OperatorShopID != nil {
|
|
scopeType, scopeID = constants.AuditScopeShop, strconv.FormatUint(uint64(*task.OperatorShopID), 10)
|
|
}
|
|
return s.auditWriter.WriteTask(ctx, tx, infraAudit.TaskInput{
|
|
EventID: infraAudit.TaskEventID(constants.AuditResourceDeviceImportTask, task.ID, phase),
|
|
ActionCode: constants.AuditActionDeviceImportTaskCreated, Summary: "创建设备导入任务",
|
|
TaskID: task.ID, TaskNo: task.TaskNo,
|
|
Actor: infraAudit.ActorInput{
|
|
Kind: constants.AuditActorAccount, ID: strconv.FormatUint(uint64(middleware.GetUserIDFromContext(ctx)), 10),
|
|
Name: middleware.GetUsernameFromContext(ctx), ShopID: task.OperatorShopID,
|
|
},
|
|
Source: constants.AuditSourceAdminAPI, ScopeType: scopeType, ScopeID: scopeID,
|
|
Result: result, ErrorCode: errorCode, ErrorSummary: errorSummary,
|
|
IdentitySnapshot: map[string]any{
|
|
"id": task.ID, "task_no": task.TaskNo, "file_name": task.FileName,
|
|
"operation_type": task.OperationType, "target_id": task.TargetID,
|
|
"batch_no": task.BatchNo, "realname_policy": task.RealnamePolicy,
|
|
},
|
|
BeforeData: before, AfterData: after,
|
|
})
|
|
}
|
|
|
|
func (s *Service) recordDeviceImportTaskAudit(ctx context.Context, task *model.DeviceImportTask, before, after map[string]any, result, phase string, errorCode int, summary string) {
|
|
if s == nil || s.db == nil || s.auditWriter == nil || task == nil || task.TaskNo == "" {
|
|
return
|
|
}
|
|
code := strconv.Itoa(errorCode)
|
|
if err := s.db.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
|
|
return s.writeDeviceImportTaskAudit(ctx, tx, task, before, after, result, phase, code, summary)
|
|
}); err != nil {
|
|
auditfailure.RecordSecondaryWriteFailure(constants.AuditActionDeviceImportTaskCreated, task.TaskNo, "", task.TaskNo, code, err)
|
|
}
|
|
}
|
|
|
|
func (s *Service) failEnqueueWithAudit(ctx context.Context, task *model.DeviceImportTask, summary string) error {
|
|
before := deviceImportTaskState(task)
|
|
return s.db.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
|
|
now := time.Now()
|
|
if err := tx.WithContext(ctx).Model(&model.DeviceImportTask{}).Where("id = ?", task.ID).Updates(map[string]any{
|
|
"status": model.ImportTaskStatusFailed, "error_message": summary, "completed_at": now, "updated_at": now,
|
|
}).Error; err != nil {
|
|
return err
|
|
}
|
|
task.Status, task.ErrorMessage = model.ImportTaskStatusFailed, summary
|
|
return s.writeDeviceImportTaskAudit(ctx, tx, task, before, deviceImportTaskState(task), constants.AuditResultFailed, "enqueue_failed", strconv.Itoa(errors.CodeTaskQueueError), summary)
|
|
})
|
|
}
|
|
|
|
func deviceImportTaskState(task *model.DeviceImportTask) map[string]any {
|
|
if task == nil {
|
|
return nil
|
|
}
|
|
return map[string]any{
|
|
"status": task.Status, "total_count": task.TotalCount, "success_count": task.SuccessCount,
|
|
"skip_count": task.SkipCount, "fail_count": task.FailCount, "warning_count": task.WarningCount,
|
|
}
|
|
}
|