From 02a84fb6fb2a39bfe7fc5cf6c0137bbf231b17fe Mon Sep 17 00:00:00 2001 From: zhangqian <zhangqian@123.com> Date: 星期四, 09 十一月 2023 14:14:17 +0800 Subject: [PATCH] 设备列表返回是否设置工艺参数字段 --- model/index.go | 58 +++++++++++++++++++++++++++++++++++++++++++++++++++------- 1 files changed, 51 insertions(+), 7 deletions(-) diff --git a/model/index.go b/model/index.go index 21552fb..f109db2 100644 --- a/model/index.go +++ b/model/index.go @@ -4,21 +4,65 @@ "apsClient/conf" "apsClient/pkg/logx" "apsClient/pkg/sqlitex" + "sync" ) func Init() error { - if err := sqlitex.Init(&conf.Conf.Sqlite, logx.GetLogger()); err != nil { + if err := sqlitex.Init(&conf.Conf.Sqlite); err != nil { return err } - if err := RegisterTables(); err != nil { - return err - } + RegisterTables() + InsertDefaultData() return nil } -func RegisterTables() error { +func RegisterTables() { db := sqlitex.GetDB() - err := db.AutoMigrate(Order{}, Procedures{}) - return err + db.AutoMigrate( + Order{}, + Procedures{}, + NetConfig{}, + PlcBrand{}, + DevicePlc{}, + ProcessModel{}, + ProductionProgress{}, + Device{}, + TaskStatusSync{}, + SystemStatus{}, + ProcessModelPlcAddress{}, + ) +} + +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), + NewPlcBrandSearch(), + NewSystemStatusSearch(), + } + + 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