添加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

@@ -19,6 +19,7 @@ is_null_to_point : false # 数据库默认 'DEFAULT NULL' 时设置结构为指
table_prefix : "" # 表前缀, 如果有则使用, 没有留空(如果表前缀以"-"开头则表示去掉该前缀struct、文件名都会去掉该前缀) table_prefix : "" # 表前缀, 如果有则使用, 没有留空(如果表前缀以"-"开头则表示去掉该前缀struct、文件名都会去掉该前缀)
table_names: "" # 指定表生成,多个表用,隔开 table_names: "" # 指定表生成,多个表用,隔开
is_out_file_by_table_name: false # 是否根据表名生成多个model is_out_file_by_table_name: false # 是否根据表名生成多个model
is_out_page: true # 是否输出分页函数
db_info: db_info:
host : 127.0.0.1 # type=1的时候host为yml文件全路径 host : 127.0.0.1 # type=1的时候host为yml文件全路径

View File

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

View File

@@ -324,21 +324,21 @@ func (m *_Model) generateFunc() (genOut []GenOutInfo) {
// -------end------ // -------end------
// gen page 分页查询的基础 // gen page 分页查询的基础
genPage, err := template.New("gen_page").Parse(genfunc.GetGenPageTemp()) if config.GetIsOutPage() {
if err != nil { genPage, err := template.New("gen_page").Parse(genfunc.GetGenPageTemp())
panic(err) 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------ // -------end------
for _, tab := range m.info.TabList { for _, tab := range m.info.TabList {
var pkg genstruct.GenPackage var pkg genstruct.GenPackage
pkg.SetPackage(m.info.PackageName) //package name pkg.SetPackage(m.info.PackageName) //package name