From d93d2bc56ed06b3c060ff48f49591e8c5d779230 Mon Sep 17 00:00:00 2001 From: zhangqian <zhangqian@123.com> Date: 星期四, 14 九月 2023 22:33:45 +0800 Subject: [PATCH] 通讯方式字段转换 --- model/index.go | 44 +++++++++++++++++++++++++++++++++++++++++++- 1 files changed, 43 insertions(+), 1 deletions(-) diff --git a/model/index.go b/model/index.go index 21552fb..738e2fc 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,52 @@ 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{}, + DevicePlc{}, + ProcessModel{}, + ProductionProgress{}, + ) 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), + NewPlcBrandSearch(), + } + + 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