diff --git a/g_appcode.go b/g_appcode.go index b033ab6..9d848d5 100644 --- a/g_appcode.go +++ b/g_appcode.go @@ -761,6 +761,7 @@ func writeModelFiles(tables []*Table, mPath string, selectedTables map[string]bo } fileStr := strings.Replace(template, "{{modelStruct}}", tb.String(), 1) fileStr = strings.Replace(fileStr, "{{modelName}}", camelCase(tb.Name), -1) + fileStr = strings.Replace(fileStr, "{{tableName}}", tb.Name, -1) // if table contains time field, import time.Time package timePkg := "" importTimePkg := "" @@ -1003,6 +1004,10 @@ import ( {{modelStruct}} +func (t *{{modelName}}) TableName() string { + return "{{tableName}}" +} + func init() { orm.RegisterModel(new({{modelName}})) }