From bd7e42ec651f66539009371675bff38645b9b6b8 Mon Sep 17 00:00:00 2001
From: Cr <631807682@qq.com>
Date: Wed, 27 Apr 2022 21:13:48 +0800
Subject: [PATCH] fix: AutoMigrate with special table name (#5301)

* fix: AutoMigrate with special table name

* test: migrate with special table name
---
 migrator/migrator.go  |  3 ++-
 tests/migrate_test.go | 11 +++++++++++
 2 files changed, 13 insertions(+), 1 deletion(-)

diff --git a/migrator/migrator.go b/migrator/migrator.go
index 93f4c5d0..d4989410 100644
--- a/migrator/migrator.go
+++ b/migrator/migrator.go
@@ -759,7 +759,8 @@ func (m Migrator) ReorderModels(values []interface{}, autoAdd bool) (results []i
 			Statement: &gorm.Statement{DB: m.DB, Dest: value},
 		}
 		beDependedOn := map[*schema.Schema]bool{}
-		if err := dep.Parse(value); err != nil {
+		// support for special table name
+		if err := dep.ParseWithSpecialTableName(value, m.DB.Statement.Table); err != nil {
 			m.DB.Logger.Error(context.Background(), "failed to parse value %#v, got error %v", value, err)
 		}
 		if _, ok := parsedSchemas[dep.Statement.Schema]; ok {
diff --git a/tests/migrate_test.go b/tests/migrate_test.go
index d6a6c4db..6576a2bd 100644
--- a/tests/migrate_test.go
+++ b/tests/migrate_test.go
@@ -636,3 +636,14 @@ func TestMigrateSerialColumn(t *testing.T) {
 		AssertEqual(t, v.ID, v.UID)
 	}
 }
+
+// https://github.com/go-gorm/gorm/issues/5300
+func TestMigrateWithSpecialName(t *testing.T) {
+	DB.AutoMigrate(&Coupon{})
+	DB.Table("coupon_product_1").AutoMigrate(&CouponProduct{})
+	DB.Table("coupon_product_2").AutoMigrate(&CouponProduct{})
+
+	AssertEqual(t, true, DB.Migrator().HasTable("coupons"))
+	AssertEqual(t, true, DB.Migrator().HasTable("coupon_product_1"))
+	AssertEqual(t, true, DB.Migrator().HasTable("coupon_product_2"))
+}