From 58999ef1a43c5b3f97d5eb5d89b4e1185baf7b62 Mon Sep 17 00:00:00 2001 From: zhangwenjian Date: Sun, 28 Jun 2020 22:49:42 +0800 Subject: [PATCH] =?UTF-8?q?feat=EF=BC=9Aadd=20CustomMigrate?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- models/gorm/gorm.go | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/models/gorm/gorm.go b/models/gorm/gorm.go index 50c878f2..195eb71f 100644 --- a/models/gorm/gorm.go +++ b/models/gorm/gorm.go @@ -8,6 +8,7 @@ import ( func AutoMigrate(db *gorm.DB) error { db.SingularTable(true) + return db.AutoMigrate( new(models.CasbinRule), new(tools.SysTables), @@ -26,3 +27,10 @@ func AutoMigrate(db *gorm.DB) error { new(models.DictType), ).Error } + + +func CustomMigrate(db *gorm.DB,table interface{}) error { + db.SingularTable(true) + return db.AutoMigrate(&table).Error +} +