/** * @Time :2021/11/9 15:17 * @Author :MELF晓宇 * @Email :xyzh.melf@petalmail.com * @FileName:init.go * @Project :gin-start * @Blog :https://blog.csdn.net/qq_29537269 * @Guide :https://guide.melf.space * @Information: * */ package models import ( "errors" conn "gin-start/connection" "github.com/casbin/casbin" "github.com/casbin/gorm-adapter" "log" ) import _ "github.com/go-sql-driver/mysql" // DBAutoMigrate /** * @Description: 自动迁移 * @param DB 数据库连接池 */ func DBAutoMigrate() (err error) { // 自动迁移 err = conn.DB.AutoMigrate( &User{}, // 用户 &Role{}, // 角色 ) if err != nil { return errors.New("自动迁移失败:" + err.Error()) } return nil } // InitModel /** * @Description: 初始化数据表 */ func InitModel() { adapter := gormadapter.NewAdapterByDB(conn.DBForCasbin) Enforcer := casbin.NewEnforcer("config\\rbac_model.conf", adapter) err3 := Enforcer.LoadPolicy() if err3 != nil { log.Fatalln(err3) } }