package models import ( "eta_gn/eta_api/global" "eta_gn/eta_api/utils" "fmt" "github.com/rdlucklib/rdluck_tools/paging" "time" ) type Classify struct { Id int `gorm:"column:id;primaryKey" json:"id"` //`orm:"column(id);pk"` ClassifyName string `gorm:"column:classify_name" json:"classify_name"` //`description:"分类名称"` Sort int `gorm:"column:sort" json:"sort"` //`json:"-"` ParentId int `gorm:"column:parent_id" json:"parent_id"` //`description:"父级分类id"` CreateTime time.Time `gorm:"column:create_time" json:"create_time"` //`description:"创建时间"` ModifyTime time.Time `gorm:"column:modify_time" json:"modify_time"` //`description:"修改时间"` Abstract string `gorm:"column:abstract" json:"abstract"` //`description:"栏目简介"` Descript string `gorm:"column:descript" json:"descript"` //`description:"分享描述"` ReportAuthor string `gorm:"column:report_author" json:"report_author"` //`description:"栏目作者"` AuthorDescript string `gorm:"column:author_descript" json:"author_descript"` //`description:"作者简介"` ColumnImgUrl string `gorm:"column:column_img_url" json:"column_img_url"` //`description:"栏目配图"` HeadImgUrl string `gorm:"column:head_img_url" json:"head_img_url"` //`description:"头部banner"` AvatarImgUrl string `gorm:"column:avatar_img_url" json:"avatar_img_url"` //`description:"头像"` ReportImgUrl string `gorm:"column:report_img_url" json:"report_img_url"` //`description:"报告配图"` HomeImgUrl string `gorm:"column:home_img_url" json:"home_img_url"` //`description:"首页配图"` ClassifyLabel string `gorm:"column:classify_label" json:"classify_label"` //`description:"分类标签"` ShowType int `gorm:"column:show_type" json:"show_type"` //`description:"展示类型:1-列表 2-专栏"` HasTeleconference int `gorm:"column:has_teleconference" json:"has_teleconference"` //`description:"是否有电话会:0-否 1-是"` VipTitle string `gorm:"column:vip_title" json:"vip_title"` //`description:"研究员头衔"` IsShow int `gorm:"column:is_show" json:"is_show"` //`description:"是否在小程序显示:1-显示 0-隐藏"` YbFiccSort int `gorm:"column:yb_ficc_sort" json:"yb_ficc_sort"` //`description:"小程序FICC页排序"` YbFiccIcon string `gorm:"column:yb_ficc_icon" json:"yb_ficc_icon"` // `description:"小程序FICC页icon"` YbFiccPcIcon string `gorm:"column:yb_ficc_pc_icon" json:"yb_ficc_pc_icon"` //`description:"小程序PC端FICC页背景图"` YbIconUrl string `gorm:"column:yb_icon_url" json:"yb_icon_url"` //`description:"小程序已购页icon"` YbBgUrl string `gorm:"column:yb_bg_url" json:"yb_bg_url"` //`description:"小程序已购详情背景图"` YbListImg string `gorm:"column:yb_list_img" json:"yb_list_img"` //`description:"小程序研报列表封面图"` YbShareBgImg string `gorm:"column:yb_share_bg_img" json:"yb_share_bg_img"` //`description:"小程序研报详情分享背景图"` YbRightBanner string `gorm:"column:yb_right_banner" json:"yb_right_banner"` //`description:"Pc端详情页,右侧,报告合集背景图"` RelateTel int `gorm:"column:relate_tel" json:"relate_tel"` //`description:"是否在电话会中可选: 0-否; 1-是"` RelateVideo int `gorm:"column:relate_video" json:"relate_video"` //`description:"是否在路演视频中可选: 0-否; 1-是"` IsMassSend int `gorm:"column:is_mass_send" json:"is_mass_send"` //`description:"1:群发,0:非群发"` Enabled int `gorm:"column:enabled" json:"enabled"` //`description:"是否可用,1可用,0禁用"` Level int `gorm:"column:level" json:"level"` //`description:"层级"` HasChild int `gorm:"column:has_child" json:"has_child"` //`description:"是否有子级别,0:下面没有子分类,1:下面有子分类;默认:0"` ReportDetailShowType int `gorm:"column:report_detail_show_type" json:"report_detail_show_type"` //`description:"报告详情的展示类型:1-拼接;2:目录"` } type ClassifyAddReq struct { ClassifyName string `description:"分类名称"` ParentId int `description:"父级分类id,没有父级分类传0"` ChartPermissionIdList []int `description:"权限id数组"` /*Abstract string `description:"栏目简介"` Descript string `description:"分享描述"` ReportAuthor string `description:"栏目作者"` AuthorDescript string `description:"作者简介"` ColumnImgUrl string `description:"栏目配图"` ReportImgUrl string `description:"报告配图"` HeadImgUrl string `description:"头部banner"` AvatarImgUrl string `description:"头像"` HomeImgUrl string `description:"首页配图"` ClassifyLabel string `description:"分类标签"` ShowType int `description:"展示类型:1-列表 2-专栏"` HasTeleconference int `description:"是否有电话会:0-否 1-是"` VipTitle string `description:"研究员头衔"` Sort int `description:"后台排序"` IsShow int `description:"是否在小程序显示:1-显示 0-隐藏"` YbFiccSort int `description:"小程序FICC页排序"` YbFiccIcon string `description:"小程序FICC页icon"` YbFiccPcIcon string `description:"小程序PC端FICC页背景图"` YbIconUrl string `description:"小程序已购页icon"` YbBgUrl string `description:"小程序已购详情背景图"` YbListImg string `description:"小程序研报列表封面图"` YbShareBgImg string `description:"小程序研报详情分享背景图"` YbRightBanner string `description:"Pc端详情页,右侧,报告合集背景图"` MenuList []*ClassifyMenuSaveReq `description:"子目录列表"` ClassifyMenuId int `description:"二级分类-子目录ID"` RelateTel int `description:"是否在电话会中可选: 0-否; 1-是"` RelateVideo int `description:"是否在路演视频中可选: 0-否; 1-是"`*/ } func GetClassifyByName(classifyName string, parentId int) (item *Classify, err error) { //sql := `SELECT * FROM classify WHERE classify_name=? AND parent_id=? ` //o := orm.NewOrmUsingDB("rddp") //err = o.Raw(sql, classifyName, parentId).QueryRow(&item) sql := `SELECT * FROM classify WHERE classify_name=? AND parent_id=? ` err = global.DmSQL["rddp"].Raw(sql, classifyName, parentId).First(&item).Error return } func GetClassifyById(classifyId int) (item *Classify, err error) { //sql := `SELECT * FROM classify WHERE id=? ` //o := orm.NewOrmUsingDB("rddp") //err = o.Raw(sql, classifyId).QueryRow(&item) sql := `SELECT * FROM classify WHERE id=? ` err = global.DmSQL["rddp"].Raw(sql, classifyId).First(&item).Error return } // 添加分类 func AddClassify(item *Classify) (err error) { //o := orm.NewOrmUsingDB("rddp") //id, err := o.Insert(item) //if err != nil { // return //} //item.Id = int(id) err = global.DmSQL["rddp"].Create(item).Error return } func GetReportCountByClassifyId(classifyId int) (count int, err error) { //o := orm.NewOrmUsingDB("rddp") //sql := `SELECT COUNT(1) AS count FROM report WHERE classify_id_second=? ` //err = o.Raw(sql, classifyId).QueryRow(&count) sql := `SELECT COUNT(1) as num FROM report WHERE classify_id_second=? ` err = global.DmSQL["rddp"].Raw(sql, classifyId).First(&count).Error return } func GetClassifySubCountByClassifyId(classifyId int) (count int, err error) { sql := `SELECT COUNT(1) as num FROM classify AS a INNER JOIN report AS b ON a.id=b.classify_id_second WHERE a.parent_id=? ` //o := orm.NewOrmUsingDB("rddp") //err = o.Raw(sql, classifyId).QueryRow(&count) err = global.DmSQL["rddp"].Raw(sql, classifyId).First(&count).Error return } func GetClassifySubCountByParentId(classifyId int) (count int, err error) { sqlCount := ` SELECT COUNT(1) as num FROM classify AS a WHERE a.parent_id=? ` //o := orm.NewOrmUsingDB("rddp") //err = o.Raw(sqlCount, classifyId).QueryRow(&count) err = global.DmSQL["rddp"].Raw(sqlCount, classifyId).Scan(&count).Error return } // 删除分类 func DeleteClassify(classifyId int) (err error) { sql := `DELETE FROM classify WHERE id=? ` //o := orm.NewOrmUsingDB("rddp") //_, err = o.Raw(sql, classifyId).Exec() err = global.DmSQL["rddp"].Exec(sql, classifyId).Error if err != nil { return } deleteImgSql := `DELETE FROM banner WHERE classify_id=? ` //_, err = o.Raw(deleteImgSql, classifyId).Exec() err = global.DmSQL["rddp"].Exec(deleteImgSql, classifyId).Error return } // classifyName, abstract, descript string, parentId, classifyId int // 修改分类 func EditClassify(req *EditClassifyReq) (err error) { //sql := `UPDATE classify SET classify_name = ?,abstract=?, parent_id= ?,descript=?,report_author=?,author_descript=?,column_img_url=?,head_img_url=?,avatar_img_url=?,report_img_url=?,home_img_url=?,classify_label=?,show_type=?,has_teleconference=?,vip_title=?,modify_time= NOW() WHERE id = ? ` //_, err = o.Raw(sql, req.ClassifyName, req.Abstract, req.ParentId, req.Descript, req.ReportAuthor, req.AuthorDescript, req.ColumnImgUrl, req.HeadImgUrl, req.AvatarImgUrl, req.ReportImgUrl, req.HomeImgUrl, req.ClassifyLabel, req.ShowType, req.HasTeleconference, req.VipTitle, req.ClassifyId).Exec() sql := `UPDATE classify SET classify_name = ?,parent_id= ?,modify_time= NOW() WHERE id = ? ` //o := orm.NewOrmUsingDB("rddp") //_, err = o.Raw(sql, req.ClassifyName, req.ParentId, req.ClassifyId).Exec() err = global.DmSQL["rddp"].Exec(sql, req.ClassifyName, req.ParentId, req.ClassifyId).Error return } // ParentClassify // @Description: 获取父级分类 // @author: Roc // @datetime 2024-06-18 15:03:49 // @return items []*Classify // @return err error func ParentClassify() (items []*Classify, err error) { sql := `SELECT * FROM classify WHERE parent_id=0 order by id desc ` //o := orm.NewOrmUsingDB("rddp") //_, err = o.Raw(sql).QueryRows(&items) err = global.DmSQL["rddp"].Raw(sql).Find(&items).Error return } // 根据id获取分类详情 func FindByIdClassify(classifyId int) (item *Classify, err error) { sql := `SELECT * FROM classify WHERE id=? ` //o := orm.NewOrmUsingDB("rddp") //err = o.Raw(sql, classifyId).QueryRow(&item) err = global.DmSQL["rddp"].Raw(sql, classifyId).First(&item).Error return } type ClassifyList struct { Id int `gorm:"column:id" json:"id"` //`orm:"column(id);pk"` ClassifyName string `gorm:"column:classify_name" json:"classify_name"` //`description:"分类名称"` Sort int `gorm:"column:sort" json:"sort"` //`description:"排序"` ParentId int `gorm:"column:parent_id" json:"parent_id"` //`description:"父级分类id"` CreateTime time.Time `gorm:"column:create_time" json:"create_time"` //`description:"创建时间"` ModifyTime time.Time `gorm:"column:modify_time" json:"modify_time"` //`description:"修改时间"` Abstract string `gorm:"column:abstract" json:"abstract"` //`description:"简介"` Descript string `gorm:"column:descript" json:"descript"` //`description:"描述"` ClassifyLabel string `gorm:"column:classify_label" json:"classify_label"` //`description:"分类标签"` ShowType int `gorm:"column:show_type" json:"show_type"` //`description:"展示类型:1-列表 2-专栏"` HasTeleconference int `gorm:"column:has_teleconference" json:"has_teleconference"` //`description:"是否有电话会:0-否 1-是"` IsShow int `gorm:"column:is_show" json:"is_show"` //`description:"是否在小程序显示:1-显示 0-隐藏"` YbFiccSort int `gorm:"column:yb_ficc_sort" json:"yb_ficc_sort"` //`description:"小程序FICC页排序"` YbFiccIcon string `gorm:"column:yb_ficc_icon" json:"yb_ficc_icon"` // `description:"小程序FICC页icon"` YbFiccPcIcon string `gorm:"column:yb_ficc_pc_icon" json:"yb_ficc_pc_icon"` //`description:"小程序PC端FICC页背景图"` YbIconUrl string `gorm:"column:yb_icon_url" json:"yb_icon_url"` //`description:"小程序已购页icon"` YbBgUrl string `gorm:"column:yb_bg_url" json:"yb_bg_url"` //`description:"小程序已购详情背景图"` YbListImg string `gorm:"column:yb_list_img" json:"yb_list_img"` //`description:"小程序研报列表封面图"` YbShareBgImg string `gorm:"column:yb_share_bg_img" json:"yb_share_bg_img"` //`description:"小程序研报详情分享背景图"` YbRightBanner string `gorm:"column:yb_right_banner" json:"yb_right_banner"` //`description:"Pc端详情页,右侧,报告合集背景图"` RelateTel int `gorm:"column:relate_tel" json:"relate_tel"` //`description:"是否在电话会中可选: 0-否; 1-是"` RelateVideo int `gorm:"column:relate_video" json:"relate_video"` //`description:"是否在路演视频中可选: 0-否; 1-是"` Enabled int `gorm:"column:enabled" json:"enabled"` //`description:"是否可用,1可用,0禁用"` Child []*ClassifyList `gorm:"-"` ClassifyMenuId int `gorm:"-"` //`description:"二级分类-子目录ID"` ClassifyMenuList []*ClassifyMenu `gorm:"-"` ChartPermissionIdList []int `gorm:"-"` // `description:"绑定的权限ID"` Level int `gorm:"column:level"` //`description:"层级"` HasChild int `gorm:"column:has_child"` //`description:"是否有子级别,0:下面没有子分类,1:下面有子分类;默认:0"` } type ClassifyItem struct { Classify ClassifyMenuId int `description:"二级分类-子目录ID"` ClassifyMenuList []*ClassifyMenu ChartPermissionIdList []int `description:"绑定的权限ID"` Child []*ClassifyItem } type ClassifyListResp struct { List []*ClassifyList } type ClassifyPermissionListResp struct { List []*ClassifyList Paging *paging.PagingItem `description:"分页数据"` } // 获取分类列表 func GetClassifyList(keyWord string, enabled int) (items []*ClassifyList, err error) { sql := `` companyTypeSqlStr := `` if enabled == 1 { companyTypeSqlStr += ` AND enabled = 1 ` } pars := make([]interface{}, 0) if keyWord != "" { sql = `SELECT * FROM ( SELECT * FROM classify WHERE parent_id=0 ` + companyTypeSqlStr + ` AND classify_name LIKE ? UNION SELECT * FROM classify WHERE id IN( SELECT parent_id FROM classify WHERE parent_id>0 ` + companyTypeSqlStr + ` AND classify_name LIKE ? ) )AS t ORDER BY sort ASC,create_time ASC` pars = utils.GetLikeKeywordPars(pars, keyWord, 2) } else { sql = `SELECT * FROM classify WHERE parent_id=0 ` + companyTypeSqlStr sql += ` ORDER BY sort ASC, create_time ASC` } pars = append(pars) //o := orm.NewOrmUsingDB("rddp") //_, err = o.Raw(sql, pars...).QueryRows(&items) err = global.DmSQL["rddp"].Raw(sql, pars).Find(&items).Error return } func GetClassifyListCount(keyWord, companyType string, hideDayWeek int) (count int, err error) { sqlCount := `` companyTypeSqlStr := `` if companyType == "ficc" { companyTypeSqlStr = " AND id != 40 AND parent_id != 40 " } else if companyType == "权益" { companyTypeSqlStr = " AND (id = 40 or parent_id = 40) " } pars := make([]interface{}, 0) if keyWord != "" { sqlCount = `SELECT COUNT(1) AS count FROM ( SELECT * FROM classify WHERE parent_id=0 ` + companyTypeSqlStr + ` AND classify_name LIKE ? UNION SELECT * FROM classify WHERE id IN(SELECT parent_id FROM classify WHERE parent_id>0 ` + companyTypeSqlStr + ` AND classify_name LIKE ? ) )AS t ` pars = utils.GetLikeKeywordPars(pars, keyWord, 2) } else { sqlCount = `SELECT COUNT(1) AS count FROM classify WHERE parent_id=0 ` + companyTypeSqlStr if hideDayWeek == 1 { sqlCount += ` AND classify_name <> '晨报' AND classify_name <> '周报' ` } } //o := orm.NewOrmUsingDB("rddp") //err = o.Raw(sqlCount, pars...).QueryRow(&count) err = global.DmSQL["rddp"].Raw(sqlCount, pars).Scan(&count).Error return } type CheckDeleteClassifyReq struct { ClassifyId int `description:"分类ID"` } type CheckDeleteClassifyResp struct { Code int `description:"编码:0:检测成功,可进行删除,1:分类不存在,2:该分类有关联报告,不允许删除,3:二级分类有关联报告,不允许删除,4:该分类下有关联分类,是否确认全部删除"` Msg string `description:"描述信息"` } type DeleteClassifyReq struct { ClassifyId int `description:"分类ID"` } type EditClassifyReq struct { ClassifyId int `description:"分类ID"` ClassifyAddReq } type FindByIdClassifyReq struct { ClassifyId int `description:"分类ID"` } func GetClassifyChild(parentId int, keyWord string) (items []*Classify, err error) { sql := `` pars := make([]interface{}, 0) if keyWord != "" { sql = `SELECT * FROM classify WHERE classify_name LIKE ? AND parent_id=? ORDER BY create_time ASC ` pars = append(pars, utils.GetLikeKeyword(keyWord)) } else { sql = `SELECT * FROM classify WHERE parent_id=? ORDER BY create_time ASC ` } pars = append(pars, parentId) //o := orm.NewOrmUsingDB("rddp") //_, err = o.Raw(sql, pars...).QueryRows(&items) err = global.DmSQL["rddp"].Raw(sql, pars).Find(&items).Error return } func GetClassifyChildByParentIds(parentId []int, keyWord string, enabled int) (items []*Classify, err error) { parentIdLen := len(parentId) if parentIdLen == 0 { return } sql := `` pars := make([]interface{}, 0) pars = append(pars, parentId) if keyWord != "" { sql = `SELECT * FROM classify WHERE parent_id IN (` + utils.GetOrmInReplace(parentIdLen) + `) AND classify_name LIKE ? ` pars = append(pars, utils.GetLikeKeyword(keyWord)) } else { sql = `SELECT * FROM classify WHERE parent_id IN (` + utils.GetOrmInReplace(parentIdLen) + `) ` } if enabled == 1 { sql += ` AND enabled=1 ` } sql += ` ORDER BY create_time ASC ` //o := orm.NewOrmUsingDB("rddp") //_, err = o.Raw(sql, pars...).QueryRows(&items) err = global.DmSQL["rddp"].Raw(sql, pars).Find(&items).Error return } // EditClassifyPermissionReq 编辑分类权限请求 type EditClassifyPermissionReq struct { ClassifyId int `description:"分类ID"` ChartPermissionIdList []int `description:"权限id数组"` } // GetAllClassify 获取所有分类 func GetAllClassify() (list []*Classify, err error) { //o := orm.NewOrmUsingDB("rddp") //sql := ` SELECT * FROM classify ` //_, err = o.Raw(sql).QueryRows(&list) sql := ` SELECT * FROM classify ` err = global.DmSQL["rddp"].Raw(sql).Find(&list).Error return } // GetClassifyByKeyword 名称获取分类 func GetClassifyByKeyword(keyword string) (item Classify, err error) { //o := orm.NewOrmUsingDB("rddp") //sql := ` SELECT * FROM classify WHERE classify_name = ? LIMIT 1 ` //err = o.Raw(sql, keyword).QueryRow(&item) sql := ` SELECT * FROM classify WHERE classify_name = ? LIMIT 1 ` err = global.DmSQL["rddp"].Raw(sql, keyword).Find(&item).Error return } // UpdateClassify 更新分类 func (classifyInfo *Classify) UpdateClassify(cols []string) (err error) { //o := orm.NewOrmUsingDB("rddp") //_, err = o.Update(classifyInfo, cols...) err = global.DmSQL["rddp"].Select(cols).Updates(classifyInfo).Error return } // SimpleClassifyList 简版分类列表 type SimpleClassifyList struct { Id int `description:"分类ID"` ClassifyName string `description:"分类名称"` ParentId int `description:"父级ID"` Sort int `description:"排序"` Child []*SimpleClassifyList } // GetClassifyByCondition 获取分类列表 func GetClassifyByCondition(condition, orderRule string, pars []interface{}) (items []*SimpleClassifyList, err error) { sql := `SELECT * FROM classify WHERE 1 = 1 ` if condition != `` { sql += condition } order := `sort ASC, create_time ASC` if orderRule != `` { order = orderRule } sql += ` ORDER BY ` + order //o := orm.NewOrmUsingDB("rddp") //_, err = o.Raw(sql, pars).QueryRows(&items) err = global.DmSQL["rddp"].Raw(sql, pars).Find(&items).Error return } // UpdateChildClassifyRelateSetting 更新子分类关联设置 func UpdateChildClassifyRelateSetting(parentId, relateTel, relateVideo int) (err error) { //o := orm.NewOrmUsingDB("rddp") //sql := `UPDATE classify SET relate_tel = ?, relate_video = ? WHERE parent_id = ?` //_, err = o.Raw(sql, relateTel, relateVideo, parentId).Exec() sql := `UPDATE classify SET relate_tel = ?, relate_video = ? WHERE parent_id = ?` err = global.DmSQL["rddp"].Exec(sql, relateTel, relateVideo, parentId).Error return } // RelateTelSecClassifyWithPermissions 关联了电话会的二级分类及权限 type RelateTelSecClassifyWithPermissions struct { Id int `description:"分类ID"` ClassifyName string `description:"分类名称"` ChartPermissionIds string `description:"权限IDs"` } // UpdateClassifySortByParentId 根据父类id更新排序 func UpdateClassifySortByParentId(parentId, permissionId, nowSort int, updateSort string) (err error) { sql := ` update classify set sort = ` + updateSort + ` WHERE parent_id=? AND sort > ? ` if permissionId > 0 { sql += ` or ( id > ` + fmt.Sprint(permissionId) + ` and sort = ` + fmt.Sprint(nowSort) + `)` } //o := orm.NewOrmUsingDB("rddp") //_, err = o.Raw(sql, parentId, nowSort).Exec() err = global.DmSQL["rddp"].Exec(sql, parentId, nowSort).Error return } // GetMaxSortByParentId 获取最大的排序值 func (classifyInfo *Classify) GetMaxSortByParentId(parentId int) (maxSort int, err error) { //o := orm.NewOrmUsingDB("rddp") //sql := `SELECT max(sort) AS sort FROM classify WHERE parent_id = ? ` //err = o.Raw(sql, parentId).QueryRow(&maxSort) sql := `SELECT max(sort) AS sort FROM classify WHERE parent_id = ? ` err = global.DmSQL["rddp"].Raw(sql, parentId).Scan(&maxSort).Error return } // GetMaxSort 获取最大的排序值 func (classifyInfo *Classify) GetMaxSort() (maxSort int, err error) { //o := orm.NewOrmUsingDB("rddp") //sql := `SELECT max(sort) AS sort FROM classify` //err = o.Raw(sql).QueryRow(&maxSort) sql := `SELECT max(sort) AS sort FROM classify` err = global.DmSQL["rddp"].Raw(sql).Scan(&maxSort).Error return } // GetFirstClassifyByParentId 获取当前父级分类下,且排序数相同 的排序第一条的数据 func (classifyInfo *Classify) GetFirstClassifyByParentId(parentId int) (item *Classify, err error) { //o := orm.NewOrmUsingDB("rddp") //sql := `SELECT * FROM classify WHERE parent_id = ? order by sort asc, id asc limit 1` //err = o.Raw(sql, parentId).QueryRow(&item) sql := `SELECT * FROM classify WHERE parent_id = ? order by sort asc, id asc limit 1` err = global.DmSQL["rddp"].Raw(sql, parentId).First(&item).Error return } type ClassifyMoveReq struct { ClassifyId int `description:"分类ID"` PrevClassifyId int `description:"上一个兄弟节点分类id"` NextClassifyId int `description:"下一个兄弟节点分类id"` } type ClassifySetEnabledReq struct { ClassifyId int `description:"分类ID"` Enabled int `description:"是否可用,1可用,0禁用"` } func (classifyInfo *Classify) SetEnabled(id, enabled int) (err error) { //o := orm.NewOrmUsingDB("rddp") //to, err := o.Begin() to := global.DmSQL["rddp"].Begin() if err != nil { return } defer func() { if err != nil { _ = to.Rollback() } else { _ = to.Commit() } }() sql := ` UPDATE classify SET enabled =? WHERE id = ?` //_, err = to.Raw(sql, enabled, id).Exec() err = to.Exec(sql, enabled, id).Error if err != nil { return } sql = ` UPDATE classify SET enabled =? WHERE parent_id = ?` //_, err = to.Raw(sql, enabled, id).Exec() err = to.Exec(sql, enabled, id).Error if err != nil { return } return } // GetCountClassifyChildByParentId // @Description: 获取父级分类下子分类数量 // @author: Roc // @datetime 2024-06-17 10:58:46 // @param parentId int // @return total int // @return err error func GetCountClassifyChildByParentId(parentId int) (total int, err error) { //o := orm.NewOrmUsingDB("rddp") //sql := `SELECT count(1) AS total FROM classify WHERE parent_id = ? ` //err = o.Raw(sql, parentId).QueryRow(&total) sql := `SELECT count(1) AS total FROM classify WHERE parent_id = ? ` err = global.DmSQL["rddp"].Raw(sql, parentId).Scan(&total).Error return } // GetClassifyListByKeyword // @Description: 获取分类列表 // @author: Roc // @datetime 2024-06-19 09:49:33 // @param keyWord string // @param enabled int // @return items []*ClassifyList // @return err error func GetClassifyListByKeyword(keyWord string, enabled int) (items []*ClassifyList, err error) { sql := `` pars := make([]interface{}, 0) sql = `SELECT * FROM classify WHERE 1=1 ` if enabled == 1 { sql += ` AND enabled = 1 ` } if keyWord != `` { sql += ` AND classify_name LIKE ? ` pars = utils.GetLikeKeywordPars(pars, keyWord, 1) } sql += ` ORDER BY sort ASC, create_time ASC` //o := orm.NewOrmUsingDB("rddp") //_, err = o.Raw(sql, pars).QueryRows(&items) err = global.DmSQL["rddp"].Raw(sql, pars).Find(&items).Error return } // GetClassifyListByParentIdList // @Description: 获取分类列表 // @author: Roc // @datetime 2024-06-19 09:49:33 // @param keyWord string // @param enabled int // @return items []*ClassifyList // @return err error func GetClassifyListByParentIdList(parentClassifyIdList []int) (items []*ClassifyList, err error) { num := len(parentClassifyIdList) if num <= 0 { return } sql := `SELECT * FROM classify WHERE id in (` + utils.GetOrmInReplace(num) + `) ORDER BY sort ASC, create_time ASC` //o := orm.NewOrmUsingDB("rddp") //_, err = o.Raw(sql, parentClassifyIdList).QueryRows(&items) err = global.DmSQL["rddp"].Raw(sql, parentClassifyIdList).Find(&items).Error return } // GetClassifyListByIdList // @Description: 根据指标ID列表,获取分类列表 // @author: Roc // @datetime 2024-06-27 15:23:57 // @param classifyIdList []int // @return items []*Classify // @return err error func GetClassifyListByIdList(classifyIdList []int) (items []*Classify, err error) { num := len(classifyIdList) if num <= 0 { return } sql := `SELECT * FROM classify WHERE id IN (` + utils.GetOrmInReplace(num) + `) ` //o := orm.NewOrmUsingDB("rddp") //_, err = o.Raw(sql, classifyIdList).QueryRows(&items) err = global.DmSQL["rddp"].Raw(sql, classifyIdList).Find(&items).Error return }