gfast/app/model/admin/gen_table/gen_table.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
}