Merge pull request #91 from closetool/master

fix: create index when using custom schema
casbin · Mar 23, 2021 · fb0d729 · fb0d729
2 parents ae4ce83 + c1c559e
commit fb0d729
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/adapter.go b/adapter.go
@@ -305,10 +305,11 @@ func (a *Adapter) casbinRuleTable() func(db *gorm.DB) *gorm.DB {
 func (a *Adapter) createTable() error {
 	t := a.db.Statement.Context.Value(customTableKey{})
 
-	if t == nil {
-		t = a.getTableInstance()
+	if t != nil {
+		return a.db.AutoMigrate(t)
 	}
 
+	t = a.getTableInstance()
 	if err := a.db.AutoMigrate(t); err != nil {
 		return err
 	}