From a792fe6e0ac0ff14ebf6253a22beda3aa01602bf Mon Sep 17 00:00:00 2001
From: Jay <me@bsavage.xyz>
Date: Thu, 1 Dec 2022 12:46:42 +0530
Subject: [PATCH] Renamed generated SetSchema to Set<SchemaName>Schema

---
 generator/template/file_templates.go |  2 +-
 generator/template/process.go        | 16 +++++++++++-----
 tests/mysql/generator_test.go        | 22 +++++++++++++++++++++-
 3 files changed, 33 insertions(+), 7 deletions(-)

diff --git a/generator/template/file_templates.go b/generator/template/file_templates.go
index 6ef375d..7eadcd8 100644
--- a/generator/template/file_templates.go
+++ b/generator/template/file_templates.go
@@ -98,7 +98,7 @@ func new{{tableTemplate.TypeName}}Impl(schemaName, tableName, alias string) {{st
 `
 
 var tableSqlBuilderSetSchemaTemplate = `
-func SetSchema(schema string) {
+func {{setSchemaMethodName}}(schema string) {
 {{- range .}}
 	{{ .InstanceName }} = {{ .InstanceName }}.FromSchema(schema)
 {{- end}}
diff --git a/generator/template/process.go b/generator/template/process.go
index 7643aaf..eb10f60 100644
--- a/generator/template/process.go
+++ b/generator/template/process.go
@@ -63,11 +63,11 @@ func processSQLBuilder(dirPath string, dialect jet.Dialect, schemaMetaData metad
 	processTableSQLBuilder("table", sqlBuilderPath, dialect, schemaMetaData, schemaMetaData.TablesMetaData, sqlBuilderTemplate)
 	processTableSQLBuilder("view", sqlBuilderPath, dialect, schemaMetaData, schemaMetaData.ViewsMetaData, sqlBuilderTemplate)
 	processEnumSQLBuilder(sqlBuilderPath, dialect, schemaMetaData.EnumsMetaData, sqlBuilderTemplate)
-	processTableSQLBuilderSetSchema(sqlBuilderPath, schemaMetaData.TablesMetaData, sqlBuilderTemplate)
+	processTableSQLBuilderSetSchema(sqlBuilderPath, schemaMetaData, sqlBuilderTemplate)
 }
 
-func processTableSQLBuilderSetSchema(dirPath string, tablesMetadata []metadata.Table, builderTemplate SQLBuilder) {
-	if len(tablesMetadata) == 0 {
+func processTableSQLBuilderSetSchema(dirPath string, schemaMetadata metadata.Schema, builderTemplate SQLBuilder) {
+	if schemaMetadata.IsEmpty() {
 		return
 	}
 
@@ -77,10 +77,12 @@ func processTableSQLBuilderSetSchema(dirPath string, tablesMetadata []metadata.T
 	throw.OnError(err)
 
 	var builders []TableSQLBuilder
-	for _, tm := range tablesMetadata {
+	for _, tm := range schemaMetadata.TablesMetaData {
 		builders = append(builders, builderTemplate.Table(tm))
 	}
 
+	schemaIdentifier := utils.ToGoIdentifier(schemaMetadata.Name)
+
 	funcPath := path.Join(dirPath, builders[0].Path)
 
 	origText, err := os.ReadFile(path.Join(funcPath, builders[0].FileName+".go"))
@@ -89,7 +91,11 @@ func processTableSQLBuilderSetSchema(dirPath string, tablesMetadata []metadata.T
 	text, err := generateTemplate(
 		tableSqlBuilderSetSchemaTemplate,
 		builders,
-		nil,
+		template.FuncMap{
+			"setSchemaMethodName": func() string {
+				return "Set" + schemaIdentifier + "Schema"
+			},
+		},
 	)
 	throw.OnError(err)
 
diff --git a/tests/mysql/generator_test.go b/tests/mysql/generator_test.go
index acbe04b..5c57be9 100644
--- a/tests/mysql/generator_test.go
+++ b/tests/mysql/generator_test.go
@@ -7,10 +7,11 @@ import (
 	"strconv"
 	"testing"
 
+	"github.com/stretchr/testify/require"
+
 	"github.com/go-jet/jet/v2/generator/mysql"
 	"github.com/go-jet/jet/v2/internal/testutils"
 	"github.com/go-jet/jet/v2/tests/dbconfig"
-	"github.com/stretchr/testify/require"
 )
 
 const genTestDirRoot = "./.gentestdata3"
@@ -307,6 +308,25 @@ func newActorTableImpl(schemaName, tableName, alias string) actorTable {
 		MutableColumns: mutableColumns,
 	}
 }
+
+func SetDvdsSchema(schema string) {
+	Actor = Actor.FromSchema(schema)
+	Address = Address.FromSchema(schema)
+	Category = Category.FromSchema(schema)
+	City = City.FromSchema(schema)
+	Country = Country.FromSchema(schema)
+	Customer = Customer.FromSchema(schema)
+	Film = Film.FromSchema(schema)
+	FilmActor = FilmActor.FromSchema(schema)
+	FilmCategory = FilmCategory.FromSchema(schema)
+	FilmText = FilmText.FromSchema(schema)
+	Inventory = Inventory.FromSchema(schema)
+	Language = Language.FromSchema(schema)
+	Payment = Payment.FromSchema(schema)
+	Rental = Rental.FromSchema(schema)
+	Staff = Staff.FromSchema(schema)
+	Store = Store.FromSchema(schema)
+}
 `
 
 var actorModelFile = `