Merge pull request #1726 from actiontech/fix/workflow-export

Fix/workflow export
actiontech · Aug 7, 2023 · aeccd36 · aeccd36
2 parents 33f7cfc + a6af9f4
commit aeccd36
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/sqle/model/workflow.go b/sqle/model/workflow.go
@@ -10,6 +10,8 @@ import (
 	"github.com/jinzhu/gorm"
 )
 
+var UnScopedFunc = func(db *gorm.DB) *gorm.DB { return db.Unscoped() }
+
 type WorkflowTemplate struct {
 	Model
 	Name                          string
@@ -810,7 +812,7 @@ func (s *Storage) GetWorkflowDetailById(id string) (*Workflow, bool, error) {
 
 func (s *Storage) GetWorkflowExportById(id string) (*Workflow, bool, error) {
 	w := new(Workflow)
-	err := s.db.Preload("CreateUser", func(db *gorm.DB) *gorm.DB { return db.Unscoped() }).
+	err := s.db.Preload("CreateUser", UnScopedFunc).
 		Preload("Record").Where("id = ?", id).First(&w).Error
 	if err == gorm.ErrRecordNotFound {
 		return nil, false, nil
@@ -824,7 +826,8 @@ func (s *Storage) GetWorkflowExportById(id string) (*Workflow, bool, error) {
 	}
 
 	instanceRecordList := make([]*WorkflowInstanceRecord, 0)
-	err = s.db.Preload("Instance").Preload("Task").Preload("User").
+	err = s.db.Preload("Instance", UnScopedFunc).Preload("Task").
+		Preload("User", UnScopedFunc).
 		Where("workflow_record_id = ?", w.Record.ID).
 		Find(&instanceRecordList).Error
 	if err != nil {
@@ -842,7 +845,7 @@ func (s *Storage) GetWorkflowExportById(id string) (*Workflow, bool, error) {
 
 	steps := make([]*WorkflowStep, 0)
 	err = s.db.Where("workflow_record_id = ?", w.Record.ID).
-		Preload("OperationUser").
+		Preload("OperationUser", UnScopedFunc).
 		Find(&steps).Error
 	if err != nil {
 		return nil, false, errors.New(errors.ConnectStorageError, err)