Hein
e0e7b64c69
All checks were successful
CI / Test (1.24) (push) Successful in -27m21s
CI / Test (1.25) (push) Successful in -27m12s
CI / Build (push) Successful in -27m37s
CI / Lint (push) Successful in -27m26s
Release / Build and Release (push) Successful in -27m25s
Integration Tests / Integration Tests (push) Successful in -27m20s
* Implement field name collision resolution in model generation. * Add tests to verify renaming of fields that conflict with generated method names. * Ensure primary key type safety in UpdateID method.
110 lines
2.6 KiB
Go
110 lines
2.6 KiB
Go
package gorm
|
|
|
|
import (
|
|
"bytes"
|
|
"text/template"
|
|
)
|
|
|
|
// modelTemplate defines the template for generating GORM models
|
|
const modelTemplate = `// Code generated by relspecgo. DO NOT EDIT.
|
|
package {{.PackageName}}
|
|
|
|
{{if .Imports -}}
|
|
import (
|
|
{{range .Imports -}}
|
|
{{.}}
|
|
{{end -}}
|
|
)
|
|
{{end}}
|
|
{{range .Models}}
|
|
{{if .Comment}}// {{.Comment}}{{end}}
|
|
type {{.Name}} struct {
|
|
{{- range .Fields}}
|
|
{{.Name}} {{.Type}} ` + "`gorm:\"{{.GormTag}}\" json:\"{{.JSONTag}}\"`" + `{{if .Comment}} // {{.Comment}}{{end}}
|
|
{{- end}}
|
|
}
|
|
{{if .Config.GenerateTableName}}
|
|
// TableName returns the table name for {{.Name}}
|
|
func (m {{.Name}}) TableName() string {
|
|
return "{{.TableName}}"
|
|
}
|
|
{{end}}
|
|
{{if .Config.GenerateTableNameOnly}}
|
|
// TableNameOnly returns the table name without schema for {{.Name}}
|
|
func (m {{.Name}}) TableNameOnly() string {
|
|
return "{{.TableNameOnly}}"
|
|
}
|
|
{{end}}
|
|
{{if .Config.GenerateSchemaName}}
|
|
// SchemaName returns the schema name for {{.Name}}
|
|
func (m {{.Name}}) SchemaName() string {
|
|
return "{{.SchemaName}}"
|
|
}
|
|
{{end}}
|
|
{{if and .Config.GenerateGetID .PrimaryKeyField}}
|
|
// GetID returns the primary key value
|
|
func (m {{.Name}}) GetID() int64 {
|
|
return int64(m.{{.PrimaryKeyField}})
|
|
}
|
|
{{end}}
|
|
{{if and .Config.GenerateGetIDStr .PrimaryKeyField}}
|
|
// GetIDStr returns the primary key as a string
|
|
func (m {{.Name}}) GetIDStr() string {
|
|
return fmt.Sprintf("%d", m.{{.PrimaryKeyField}})
|
|
}
|
|
{{end}}
|
|
{{if and .Config.GenerateSetID .PrimaryKeyField}}
|
|
// SetID sets the primary key value
|
|
func (m {{.Name}}) SetID(newid int64) {
|
|
m.UpdateID(newid)
|
|
}
|
|
{{end}}
|
|
{{if and .Config.GenerateUpdateID .PrimaryKeyField}}
|
|
// UpdateID updates the primary key value
|
|
func (m *{{.Name}}) UpdateID(newid int64) {
|
|
m.{{.PrimaryKeyField}} = {{.PrimaryKeyType}}(newid)
|
|
}
|
|
{{end}}
|
|
{{if and .Config.GenerateGetIDName .IDColumnName}}
|
|
// GetIDName returns the name of the primary key column
|
|
func (m {{.Name}}) GetIDName() string {
|
|
return "{{.IDColumnName}}"
|
|
}
|
|
{{end}}
|
|
{{if .Config.GenerateGetPrefix}}
|
|
// GetPrefix returns the table prefix
|
|
func (m {{.Name}}) GetPrefix() string {
|
|
return "{{.Prefix}}"
|
|
}
|
|
{{end}}
|
|
{{end -}}
|
|
`
|
|
|
|
// Templates holds the parsed templates
|
|
type Templates struct {
|
|
modelTmpl *template.Template
|
|
}
|
|
|
|
// NewTemplates creates and parses the templates
|
|
func NewTemplates() (*Templates, error) {
|
|
modelTmpl, err := template.New("model").Parse(modelTemplate)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &Templates{
|
|
modelTmpl: modelTmpl,
|
|
}, nil
|
|
}
|
|
|
|
// GenerateCode executes the template with the given data
|
|
func (t *Templates) GenerateCode(data *TemplateData) (string, error) {
|
|
var buf bytes.Buffer
|
|
err := t.modelTmpl.Execute(&buf, data)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
return buf.String(), nil
|
|
}
|