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/serviceContractType.go |   16 ++++++++++++----
 1 files changed, 12 insertions(+), 4 deletions(-)

diff --git a/model/serviceContractType.go b/model/serviceContractType.go
index 3524737..4b64b32 100644
--- a/model/serviceContractType.go
+++ b/model/serviceContractType.go
@@ -3,6 +3,7 @@
 import (
 	"aps_crm/pkg/mysqlx"
 	"gorm.io/gorm"
+	"sync"
 )
 
 type (
@@ -90,16 +91,18 @@
 }
 
 // InitDefaultData 鍒濆鍖栨暟鎹�
-func (slf *ServiceContractTypeSearch) InitDefaultData() error {
+func (slf *ServiceContractTypeSearch) 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 := []*ServiceContractType{
 		{1, "鎸夊勾"},
@@ -107,5 +110,10 @@
 		{3, "鍏朵粬"},
 		{4, "铏氭嫙鍚堝悓"},
 	}
-	return slf.CreateBatch(records)
+
+	err := slf.CreateBatch(records)
+	if err != nil {
+		errCh <- err
+		return
+	}
 }

--
Gitblit v1.8.0