lc_veterans/app/device/dao/common.go

package dao

import (
	"fmt"
	_ "github.com/go-sql-driver/mysql"
	"github.com/jinzhu/gorm"
	"iot_manager_service/app/system/dao"
)

var Db *gorm.DB

func InitDB(db *gorm.DB) {
	Db = db
	err := Db.AutoMigrate(
		&dao.User{},
		&Dict{},
		&LampPoleGroup{},
		&LampPole{},
		&Gateway{},
		&GatewayRelation{},
		&Transformer{},
		&LightControl{},
		&Garbage{},
		&GarbageWayGroup{},
		&OnDemandGroup{},
		&OnDemandSensor{},
		&Zigbee{},
		&CameraDevice{},
		&InfoBoard{},
		&Alarm{},
		&CaptureUint{},
		&CheckPoint{},
		&AlarmTerminal{},
		&Bridge{},
		&BridgeSensor{},
		&IpBroadcast{},
		&ManholeCover{},
		&OptoSensor{},
		&IntelligentLight{},
		&TimeCondition{},
		&LightStrategy{},
		&LightCondition{},
		&DeviceVendor{},
	).Error
	if err != nil {
		panic(fmt.Sprintf("AutoMigrate err : %v", err))
	}
}