mirror of
https://github.com/tiger1103/gfast.git
synced 2026-09-21 18:20:47 +00:00
192 lines
7.2 KiB
Go
192 lines
7.2 KiB
Go
// ============================================================================
|
|
// This is auto-generated by gf cli tool only once. Fill this file as you wish.
|
|
// ============================================================================
|
|
|
|
package gen_table
|
|
|
|
import (
|
|
"gfast/app/model/admin/gen_table_column"
|
|
"gfast/library/service"
|
|
"github.com/gogf/gf/database/gdb"
|
|
"github.com/gogf/gf/errors/gerror"
|
|
"github.com/gogf/gf/frame/g"
|
|
"github.com/gogf/gf/os/gtime"
|
|
"github.com/gogf/gf/text/gstr"
|
|
"github.com/gogf/gf/util/gconv"
|
|
)
|
|
|
|
//分页请求参数
|
|
type SelectPageReq struct {
|
|
TableName string `p:"tableName"` //表名称
|
|
TableComment string `p:"tableComment"` //表描述
|
|
BeginTime string `p:"beginTime"` //开始时间
|
|
EndTime string `p:"endTime"` //结束时间
|
|
PageNum int `p:"pageNum"` //当前页码
|
|
PageSize int `p:"pageSize"` //每页数
|
|
}
|
|
|
|
//实体扩展
|
|
type EntityExtend struct {
|
|
TableId int64 `orm:"table_id,primary" json:"table_id"` // 编号
|
|
TableName string `orm:"table_name" json:"table_name"` // 表名称
|
|
TableComment string `orm:"table_comment" json:"table_comment"` // 表描述
|
|
ClassName string `orm:"class_name" json:"class_name"` // 实体类名称
|
|
TplCategory string `orm:"tpl_category" json:"tpl_category"` // 使用的模板（crud单表操作 tree树表操作）
|
|
PackageName string `orm:"package_name" json:"package_name"` // 生成包路径
|
|
ModuleName string `orm:"module_name" json:"module_name"` // 生成模块名
|
|
BusinessName string `orm:"business_name" json:"business_name"` // 生成业务名
|
|
FunctionName string `orm:"function_name" json:"function_name"` // 生成功能名
|
|
FunctionAuthor string `orm:"function_author" json:"function_author"` // 生成功能作者
|
|
Options string `orm:"options" json:"options"` // 其它生成选项
|
|
CreateBy string `orm:"create_by" json:"create_by"` // 创建者
|
|
CreateTime *gtime.Time `orm:"create_time" json:"create_time"` // 创建时间
|
|
UpdateBy string `orm:"update_by" json:"update_by"` // 更新者
|
|
UpdateTime *gtime.Time `orm:"update_time" json:"update_time"` // 更新时间
|
|
Remark string `orm:"remark" json:"remark"` // 备注
|
|
TreeCode string `json:"tree_code"` // 树编码字段
|
|
TreeParentCode string `json:"tree_parent_code"` // 树父编码字段
|
|
TreeName string `json:"tree_name"` // 树名称字段
|
|
Columns []*gen_table_column.Entity `json:"columns"` // 表列信息
|
|
PkColumn *gen_table_column.Entity `json:"pkColumn"` // 表列信息
|
|
}
|
|
|
|
//查询据库列表
|
|
func SelectDbTableList(param *SelectPageReq) (total int, list []*Entity, err error) {
|
|
db := g.DB()
|
|
sql := " from information_schema.tables where table_schema = (select database())" +
|
|
" and table_name NOT LIKE 'qrtz_%' AND table_name NOT LIKE 'gen_%' and table_name NOT IN (select table_name from gen_table) "
|
|
if param != nil {
|
|
if param.TableName != "" {
|
|
sql += gdb.FormatSqlWithArgs(" and lower(table_name) like lower(?)", []interface{}{"%" + param.TableName + "%"})
|
|
}
|
|
|
|
if param.TableComment != "" {
|
|
sql += gdb.FormatSqlWithArgs(" and lower(table_comment) like lower(?)", []interface{}{"%" + param.TableComment + "%"})
|
|
}
|
|
|
|
if param.BeginTime != "" {
|
|
sql += gdb.FormatSqlWithArgs(" and date_format(create_time,'%y%m%d') >= date_format(?,'%y%m%d') ", []interface{}{param.BeginTime})
|
|
}
|
|
|
|
if param.EndTime != "" {
|
|
sql += gdb.FormatSqlWithArgs(" and date_format(create_time,'%y%m%d') <= date_format(?,'%y%m%d') ", []interface{}{param.EndTime})
|
|
}
|
|
}
|
|
countSql := "select count(1) " + sql
|
|
total, err = db.GetCount(countSql)
|
|
if err != nil {
|
|
g.Log().Error(err)
|
|
err = gerror.New("读取总表数失败")
|
|
return
|
|
}
|
|
sql = "table_name, table_comment, create_time, update_time " + sql
|
|
|
|
if param.PageNum == 0 {
|
|
param.PageNum = 1
|
|
}
|
|
|
|
if param.PageSize == 0 {
|
|
param.PageSize = service.AdminPageNum
|
|
}
|
|
page := (param.PageNum - 1) * param.PageSize
|
|
sql += " order by create_time desc,table_name asc limit " + gconv.String(page) + "," + gconv.String(param.PageSize)
|
|
var res gdb.Result
|
|
res, err = db.GetAll("select " + sql)
|
|
res.Structs(&list)
|
|
if err != nil {
|
|
g.Log().Error(err)
|
|
err = gerror.New("读取数据失败")
|
|
}
|
|
return
|
|
}
|
|
|
|
//根据条件分页查询数据
|
|
func SelectListByPage(param *SelectPageReq) (total int, list []*Entity, err error) {
|
|
model := Model
|
|
if param != nil {
|
|
if param.TableName != "" {
|
|
model = model.Where(Columns.TableName+" like ?", "%"+param.TableName+"%")
|
|
}
|
|
if param.TableComment != "" {
|
|
model = model.Where(Columns.TableComment+"like ?", "%"+param.TableComment+"%")
|
|
}
|
|
if param.BeginTime != "" {
|
|
model = model.Where(Columns.CreateTime+" >= ", param.BeginTime)
|
|
}
|
|
if param.EndTime != "" {
|
|
model = model.Where(Columns.CreateTime+" <= ", param.EndTime)
|
|
}
|
|
total, err = model.Count()
|
|
if err != nil {
|
|
g.Log().Error(err)
|
|
err = gerror.New("获取总行数失败")
|
|
return
|
|
}
|
|
if param.PageNum == 0 {
|
|
param.PageNum = 1
|
|
}
|
|
if param.PageSize == 0 {
|
|
param.PageSize = service.AdminPageNum
|
|
}
|
|
list, err = model.Page(param.PageNum, param.PageSize).Order(Columns.TableId + " asc").All()
|
|
if err != nil {
|
|
g.Log().Error(err)
|
|
err = gerror.New("获取数据失败")
|
|
return
|
|
}
|
|
}
|
|
return
|
|
}
|
|
|
|
//查询据库列表
|
|
func SelectDbTableListByNames(tableNames []string) ([]*Entity, error) {
|
|
db := g.DB()
|
|
sql := "select * from information_schema.tables where table_name NOT LIKE 'qrtz_%' and table_name NOT LIKE 'gen_%' " +
|
|
" and table_schema = (select database()) "
|
|
if len(tableNames) > 0 {
|
|
in := gstr.TrimRight(gstr.Repeat("?,", len(tableNames)), ",")
|
|
sql += " and " + gdb.FormatSqlWithArgs("table_name in ("+in+")", gconv.SliceAny(tableNames))
|
|
}
|
|
var result []*Entity
|
|
res, err := db.GetAll(sql)
|
|
if err != nil {
|
|
g.Log().Error(err)
|
|
return nil, gerror.New("获取表格信息失败")
|
|
}
|
|
err = res.Structs(&result)
|
|
if err != nil {
|
|
g.Log().Error(err)
|
|
return nil, gerror.New("表格信息转换失败")
|
|
}
|
|
return result, err
|
|
}
|
|
|
|
//通过表格ID获取表格信息
|
|
func GetInfoById(tableId int64) (entity *Entity, err error) {
|
|
entity, err = Model.FindOne(tableId)
|
|
if err != nil {
|
|
g.Log().Error(err)
|
|
err = gerror.New("获取表格信息出错")
|
|
}
|
|
return
|
|
}
|
|
|
|
func SelectRecordById(tableId int64) (entityExtend *EntityExtend, err error) {
|
|
var entity *Entity
|
|
entity, err = GetInfoById(tableId)
|
|
if err != nil {
|
|
return
|
|
}
|
|
m := gconv.Map(entity)
|
|
gconv.Struct(m, &entityExtend)
|
|
|
|
//表字段数据
|
|
var columns []*gen_table_column.Entity
|
|
columns, err = gen_table_column.SelectGenTableColumnListByTableId(tableId)
|
|
if err != nil {
|
|
return
|
|
}
|
|
entityExtend.Columns = columns
|
|
return
|
|
}
|