From 066a30751bdc20f9e83b34539de71ae392783e1b Mon Sep 17 00:00:00 2001 From: zhangqian <zhangqian@123.com> Date: 星期六, 19 八月 2023 15:58:09 +0800 Subject: [PATCH] 网络配置获取和删除,品牌增删改查 --- model/index.go | 40 +++++++++++++++++++++++++++++++++++++++- 1 files changed, 39 insertions(+), 1 deletions(-) diff --git a/model/index.go b/model/index.go index 21552fb..12e013e 100644 --- a/model/index.go +++ b/model/index.go @@ -4,6 +4,7 @@ "apsClient/conf" "apsClient/pkg/logx" "apsClient/pkg/sqlitex" + "sync" ) func Init() error { @@ -14,11 +15,48 @@ if err := RegisterTables(); err != nil { return err } + InsertDefaultData() return nil } func RegisterTables() error { db := sqlitex.GetDB() - err := db.AutoMigrate(Order{}, Procedures{}) + err := db.AutoMigrate( + Order{}, + Procedures{}, + NetConfig{}, + PlcBrand{}, + ) return err } + +type InitDefaultData interface { + InitDefaultData(errCh chan<- error, wg *sync.WaitGroup) +} + +func InsertDefaultData() { + errCh := make(chan error, 2) + var wg sync.WaitGroup + + models := []interface{}{ + NewNetConfigSearch(nil), + } + + for _, model := range models { + if id, ok := model.(InitDefaultData); ok { + wg.Add(1) + go id.InitDefaultData(errCh, &wg) + } + } + + go func() { + wg.Wait() // 绛夊緟鎵�鏈塯oroutine瀹屾垚 + close(errCh) // 鍏抽棴閿欒閫氶亾 + }() + + for err := range errCh { + if err != nil { + logx.Errorf("InitDefaultData err: %v", err.Error()) + } + } +} -- Gitblit v1.8.0