添加page 开关

This commit is contained in:
xxj
2022-01-04 18:16:08 +08:00
parent 3c8e9fbe13
commit 1278831995
3 changed files with 19 additions and 12 deletions

View File

@@ -32,6 +32,7 @@ type Config struct {
TableNames string `yaml:"table_names"` // 表名(多个表名用","隔开)
IsColumnName bool `yaml:"is_column_name"` //是否输出列名
IsOutFileByTableName bool `yaml:"is_out_file_by_table_name"` //是否根据表名生成文件(多个表名生成多个文件)
IsOutPage bool `yaml:"is_out_page"` // 是否输出分页支持
}
// DBInfo mysql database information. mysql 数据库信息
@@ -316,3 +317,8 @@ func SetIsColumnName(isColumnName bool) {
func GetIsOutFileByTableName() bool {
return _map.IsOutFileByTableName
}
// GetIsOutPage 是否输出分页函数支持
func GetIsOutPage() bool {
return _map.IsOutPage
}

View File

@@ -324,21 +324,21 @@ func (m *_Model) generateFunc() (genOut []GenOutInfo) {
// -------end------
// gen page 分页查询的基础
genPage, err := template.New("gen_page").Parse(genfunc.GetGenPageTemp())
if err != nil {
panic(err)
if config.GetIsOutPage() {
genPage, err := template.New("gen_page").Parse(genfunc.GetGenPageTemp())
if err != nil {
panic(err)
}
var bufPage bytes.Buffer
genPage.Execute(&bufPage, m.info)
genOut = append(genOut, GenOutInfo{
FileName: "gen.page.go",
FileCtx: bufPage.String(),
})
}
var bufPage bytes.Buffer
genPage.Execute(&bufPage, m.info)
genOut = append(genOut, GenOutInfo{
FileName: "gen.page.go",
FileCtx: bufPage.String(),
})
// -------end------
for _, tab := range m.info.TabList {
var pkg genstruct.GenPackage
pkg.SetPackage(m.info.PackageName) //package name