feat: MigrateTenants for Database interface

internal/db/ch/clickhouse.go

@@ -68,7 +68,7 @@ func (p *Clickhouse) GetDB() *gorm.DB {
 }
 
 // WithTransaction executes a function within a transaction
-func (p *Clickhouse) WithTransaction(fn func(tx *gorm.DB) error) error {
+func (p *Clickhouse) WithTransaction(_ func(tx *gorm.DB) error) error {
 	return fmt.Errorf("transactions are not supported in ClickHouse")
 }
 
@@ -97,3 +97,7 @@ func (p *Clickhouse) SwitchSchema(schema string) *gorm.DB {
 	}
 	return p.db
 }
+
+func (p *Clickhouse) MigrateTenants(_ []string, _ []any) error {
+	panic(fmt.Errorf("not implemented yet"))
+}

internal/db/database.go

@@ -18,6 +18,7 @@ type Database interface {
 	WithTransaction(func(tx *gorm.DB) error) error
 	SeedData(data []any) error
 	SwitchSchema(schema string) *gorm.DB
+	MigrateTenants(schemas []string, autoMigrateEntities []any) error
 }
 
 // DatabaseType defines the type of databases supported

internal/db/pg/pgsql.go

@@ -106,3 +106,19 @@ func (p *Postgres) SwitchSchema(schema string) *gorm.DB {
 	}
 	return p.db
 }
+
+func (p *Postgres) MigrateTenants(schemas []string, autoMigrateEntities []any) error {
+	for _, schema := range schemas {
+		if err := p.SwitchSchema(schema).Error; err != nil {
+			return fmt.Errorf("failed to switch to schema %s: %w", schema, err)
+		}
+
+		for _, entity := range autoMigrateEntities {
+			if err := p.db.AutoMigrate(entity); err != nil {
+				return fmt.Errorf("failed to auto-migrate entity for schema %s: %w", schema, err)
+			}
+		}
+	}
+
+	return nil
+}