Renamed generated schema func to UseSchema.

This commit is contained in:
Jay 2022-12-05 17:26:52 +05:30
parent 04c1a51ba7
commit 1bf48d640a
6 changed files with 46 additions and 41 deletions

View file

@ -99,10 +99,10 @@ func new{{tableTemplate.TypeName}}Impl(schemaName, tableName, alias string) {{st
var tableSqlBuilderSetSchemaTemplate = `package {{package}}
// {{schemaMethodName}} changes all global tables/views with the value returned
// UseSchema changes all global tables/views with the value returned
// returned by calling FromSchema on them. Passing an empty string to this function
// will cause queries to be generated without any table/view alias.
func {{schemaMethodName}}(schema string) {
func UseSchema(schema string) {
{{- range .}}
{{ .InstanceName }} = {{ .InstanceName }}.FromSchema(schema)
{{- end}}

View file

@ -177,25 +177,20 @@ func processTableSQLBuilder(fileTypes, dirPath string,
}
if len(generatedBuilders) > 0 {
generateUseSchemaFunc(dirPath, fileTypes, schemaMetaData, generatedBuilders)
generateUseSchemaFunc(dirPath, fileTypes, generatedBuilders)
}
}
func generateUseSchemaFunc(dirPath, fileTypes string, schemaMetadata metadata.Schema, builders []TableSQLBuilder) {
func generateUseSchemaFunc(dirPath, fileTypes string, builders []TableSQLBuilder) {
basePath := path.Join(dirPath, builders[0].Path)
schemaIdentifier := utils.ToGoIdentifier(schemaMetadata.Name)
methodName := fmt.Sprintf("Use%sSchema", schemaIdentifier)
fmt.Printf("Generating global `%s` method for %s\n", methodName, fileTypes)
fmt.Printf("Generating global `UseSchema` method for %s\n", fileTypes)
text, err := generateTemplate(
autoGenWarningTemplate+tableSqlBuilderSetSchemaTemplate,
builders,
template.FuncMap{
"package": func() string { return builders[0].PackageName() },
"schemaMethodName": func() string { return methodName },
"package": func() string { return builders[0].PackageName() },
},
)
throw.OnError(err)