package main

import (
	log "github.com/sirupsen/logrus"
	"github.com/spf13/viper"
	"myschools.me/wyh/token.git/model"
	"myschools.me/wyh/token.git/mysql"
)

//initMysql 初始化表
func initMysql() {
	if !viper.GetBool("mysql.inittable") {
		return
	}

	db := mysql.New()
	// if !db.Migrator().HasTable(&model.Device{}) {
	// 	if err := db.Set("gorm:table_options", "ENGINE=memory").Migrator().CreateTable(&model.Device{}); err != nil {
	// 		log.WithFields(log.Fields{
	// 			"func": "initMysql",
	// 		}).Fatalf("device table: %s", err.Error())
	// 	}
	// }
	// if err := db.AutoMigrate(&model.Device{}, &model.Event{}, &model.EventMedia{}, &model.EventObject{}, &model.Server{}); err != nil {
	// 	log.WithFields(log.Fields{
	// 		"func": "initMysql",
	// 	}).Fatalf("%s", err.Error())
	// }
	if err := db.AutoMigrate(&model.User{}); err != nil {
		log.WithFields(log.Fields{
			"func": "initMysql",
		}).Fatalf("%s", err.Error())
	}
}