lc_veterans/app/warn/dao/common.go

package dao

import (
	"fmt"
	"gorm.io/gorm"
	"iot_manager_service/util/common"
)

var Db *gorm.DB

func InitDB(db *gorm.DB) {
	Db = db
	models := []common.TableModelAuto{
		{&NoticeSet{}, "告警通知设置"},
		{&NoticeRecord{}, "告警通知记录"},
		{&PlatformAlarm{}, "平台所有告警"},
	}
	for _, val := range models {
		//fmt.Println(val.Model)
		err := Db.Set("gorm:table_options", "comment '"+val.Comment+"'").AutoMigrate(val.Model)
		if err != nil {
			panic(fmt.Sprintf("AutoMigrate err : %v", err))
		}
	}
}