From 8546acdc2a1f06253f30e0543fc202ebf5219bd0 Mon Sep 17 00:00:00 2001
From: wangpengfei <274878379@qq.com>
Date: 星期五, 18 八月 2023 16:31:40 +0800
Subject: [PATCH] fix

---
 model/priorityLevel.go |   16 ++++++++++++----
 1 files changed, 12 insertions(+), 4 deletions(-)

diff --git a/model/priorityLevel.go b/model/priorityLevel.go
index ee2f3fe..b1902f4 100644
--- a/model/priorityLevel.go
+++ b/model/priorityLevel.go
@@ -6,6 +6,7 @@
 	"errors"
 	"fmt"
 	"gorm.io/gorm"
+	"sync"
 )
 
 type (
@@ -119,16 +120,19 @@
 }
 
 // InitDefaultData 鍒濆鍖栨暟鎹�
-func (slf *PriorityLevelSearch) InitDefaultData() error {
+func (slf *PriorityLevelSearch) InitDefaultData(errCh chan<- error, wg *sync.WaitGroup) {
 	var (
 		db          = slf.Orm.Table(slf.TableName())
 		total int64 = 0
 	)
+	defer wg.Done()
+
 	if err := db.Count(&total).Error; err != nil {
-		return err
+		errCh <- err
+		return
 	}
 	if total != 0 {
-		return nil
+		return
 	}
 	records := []*PriorityLevel{
 		{1, "浣�"},
@@ -136,5 +140,9 @@
 		{3, "楂�"},
 		{4, "绱ф��"},
 	}
-	return slf.CreateBatch(records)
+	err := slf.CreateBatch(records)
+	if err != nil {
+		errCh <- err
+		return
+	}
 }

--
Gitblit v1.8.0