From d1afbd5ab3666b1df66137d773bb39f39ecce286 Mon Sep 17 00:00:00 2001 From: Administrator Date: Thu, 4 Mar 2021 23:27:29 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8Dbug?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- data/view/genfunc/def.go | 6 +++--- data/view/model/common.go | 2 +- data/view/model/model.go | 11 ++++++++--- 3 files changed, 12 insertions(+), 7 deletions(-) diff --git a/data/view/genfunc/def.go b/data/view/genfunc/def.go index f96790d..5c5d089 100644 --- a/data/view/genfunc/def.go +++ b/data/view/genfunc/def.go @@ -184,20 +184,20 @@ func (obj *_{{$obj.StructName}}Mgr) GetByOptions(opts ...Option) (results []*{{$ {{range $oem := $obj.Em}} // GetFrom{{$oem.ColStructName}} 通过{{$oem.ColName}}获取内容 {{$oem.Notes}} {{if $oem.IsMulti}} func (obj *_{{$obj.StructName}}Mgr) GetFrom{{$oem.ColStructName}}({{CapLowercase $oem.ColStructName}} {{$oem.Type}}) (results []*{{$obj.StructName}}, err error) { - err = obj.DB.WithContext(obj.ctx).Table(obj.GetTableName()).Where("{{$oem.ColNameEx}} = ?", {{CapLowercase $oem.ColStructName}}).Find(&results).Error + err = obj.DB.WithContext(obj.ctx).Table(obj.GetTableName()).Where("{{$oem.ColName}} = ?", {{CapLowercase $oem.ColStructName}}).Find(&results).Error {{GenPreloadList $obj.PreloadList true}} return } {{else}} func (obj *_{{$obj.StructName}}Mgr) GetFrom{{$oem.ColStructName}}({{CapLowercase $oem.ColStructName}} {{$oem.Type}}) (result {{$obj.StructName}}, err error) { - err = obj.DB.WithContext(obj.ctx).Table(obj.GetTableName()).Where("{{$oem.ColNameEx}} = ?", {{CapLowercase $oem.ColStructName}}).Find(&result).Error + err = obj.DB.WithContext(obj.ctx).Table(obj.GetTableName()).Where("{{$oem.ColName}} = ?", {{CapLowercase $oem.ColStructName}}).Find(&result).Error {{GenPreloadList $obj.PreloadList false}} return } {{end}} // GetBatchFrom{{$oem.ColStructName}} 批量唯一主键查找 {{$oem.Notes}} func (obj *_{{$obj.StructName}}Mgr) GetBatchFrom{{$oem.ColStructName}}({{CapLowercase $oem.ColStructName}}s []{{$oem.Type}}) (results []*{{$obj.StructName}}, err error) { - err = obj.DB.WithContext(obj.ctx).Table(obj.GetTableName()).Where("{{$oem.ColNameEx}} IN (?)", {{CapLowercase $oem.ColStructName}}s).Find(&results).Error + err = obj.DB.WithContext(obj.ctx).Table(obj.GetTableName()).Where("{{$oem.ColName}} IN (?)", {{CapLowercase $oem.ColStructName}}s).Find(&results).Error {{GenPreloadList $obj.PreloadList true}} return } diff --git a/data/view/model/common.go b/data/view/model/common.go index 812d99c..6765844 100644 --- a/data/view/model/common.go +++ b/data/view/model/common.go @@ -189,7 +189,7 @@ func GenFListIndex(info FList, status int) string { { var strs []string for _, v := range info.Kem { - strs = append(strs, fmt.Sprintf("`%v` = ?", v.ColName)) + strs = append(strs, fmt.Sprintf("%v = ?", v.ColName)) //wxw 2021.2.26 16:53 } return strings.Join(strs, " AND ") } diff --git a/data/view/model/model.go b/data/view/model/model.go index 194169b..66c5a15 100644 --- a/data/view/model/model.go +++ b/data/view/model/model.go @@ -254,10 +254,15 @@ func (m *_Model) generateFunc() (genOut []GenOutInfo) { pkg.AddImport(`"context"`) // 添加import信息 pkg.AddImport(cnf.EImportsHead["gorm.Model"]) - data := funDef{ - StructName: getCamelName(tab.Name), - TableName: tab.Name, + // wxw 2021.2.26 17:17 + var data funDef + data.TableName = tab.Name + tablePrefix := config.GetTablePrefix() + //如果设置了表前缀 + if tablePrefix != "" { + tab.Name = strings.TrimLeft(tab.Name, tablePrefix) } + data.StructName = getCamelName(tab.Name) var primary, unique, uniqueIndex, index []FList for _, el := range tab.Em {