From d50d503f4519f223b2f1960f7ac1d955be0db13d Mon Sep 17 00:00:00 2001
From: luantranminh <tranminhluan1206@gmail.com>
Date: Sat, 20 Apr 2024 14:49:54 +0700
Subject: [PATCH] fix: use callback instead of struct for params

---
 README.md          |  6 +++---
 gormschema/gorm.go | 25 +++++++------------------
 2 files changed, 10 insertions(+), 21 deletions(-)

diff --git a/README.md b/README.md
index ffc3b21..b3669df 100644
--- a/README.md
+++ b/README.md
@@ -210,9 +210,9 @@ The provider supports the following databases:
   you should use the following code:
   ```go
   stmts, err := gormschema.New("mysql")
-  .WithJoinTables(
-		gormschema.SetupJoinTableParams{Model: &models.Person{}, Field: "Addresses", JoinTable: &models.PersonAddress{}},
-	)
+  .WithJoinTables( func(db *gorm.DB) {
+		db.SetupJoinTable(&customjointable.Person{}, "Addresses", &customjointable.PersonAddress{})
+	})
   .Load(&models.Person{}, &models.Address{})
   ```
 
diff --git a/gormschema/gorm.go b/gormschema/gorm.go
index fa7c3c6..bcd9934 100644
--- a/gormschema/gorm.go
+++ b/gormschema/gorm.go
@@ -27,9 +27,9 @@ func New(dialect string, opts ...Option) *Loader {
 type (
 	// Loader is a Loader for gorm schema.
 	Loader struct {
-		dialect            string
-		config             *gorm.Config
-		joinTableCallbacks []func(*gorm.DB)
+		dialect           string
+		config            *gorm.Config
+		beforeAutoMigrate func(*gorm.DB)
 	}
 	// Option configures the Loader.
 	Option func(*Loader)
@@ -86,8 +86,8 @@ func (l *Loader) Load(models ...any) (string, error) {
 		db.Config.DisableForeignKeyConstraintWhenMigrating = true
 	}
 
-	for _, setupJoinTable := range l.joinTableCallbacks {
-		setupJoinTable(db)
+	if l.beforeAutoMigrate != nil {
+		l.beforeAutoMigrate(db)
 	}
 
 	if err = db.AutoMigrate(models...); err != nil {
@@ -167,18 +167,7 @@ func (m *migrator) CreateConstraints(models []any) error {
 	return nil
 }
 
-type SetupJoinTableParams struct {
-	Model     any
-	Field     string
-	JoinTable any
-}
-
-func (l *Loader) WithJoinTables(params ...SetupJoinTableParams) *Loader {
-	for _, p := range params {
-		l.joinTableCallbacks = append(l.joinTableCallbacks, func(db *gorm.DB) {
-			db.SetupJoinTable(p.Model, p.Field, p.JoinTable)
-		})
-	}
-
+func (l *Loader) WithJoinTables(cb func(*gorm.DB)) *Loader {
+	l.beforeAutoMigrate = cb
 	return l
 }