classify.go 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656
  1. package models
  2. import (
  3. "eta/eta_api/utils"
  4. "fmt"
  5. "time"
  6. "github.com/beego/beego/v2/client/orm"
  7. "github.com/rdlucklib/rdluck_tools/paging"
  8. )
  9. type Classify struct {
  10. Id int `orm:"column(id);pk;auto"`
  11. ClassifyName string `description:"分类名称"`
  12. Sort int `json:"-"`
  13. ParentId int `description:"父级分类id"`
  14. CreateTime time.Time `description:"创建时间"`
  15. ModifyTime time.Time `description:"修改时间"`
  16. Abstract string `description:"栏目简介"`
  17. Descript string `description:"分享描述"`
  18. ReportAuthor string `description:"栏目作者"`
  19. AuthorDescript string `description:"作者简介"`
  20. ColumnImgUrl string `description:"栏目配图"`
  21. HeadImgUrl string `description:"头部banner"`
  22. AvatarImgUrl string `description:"头像"`
  23. ReportImgUrl string `description:"报告配图"`
  24. HomeImgUrl string `description:"首页配图"`
  25. ClassifyLabel string `description:"分类标签"`
  26. ShowType int `description:"展示类型:1-列表 2-专栏"`
  27. HasTeleconference int `description:"是否有电话会:0-否 1-是"`
  28. VipTitle string `description:"研究员头衔"`
  29. IsShow int `description:"是否在小程序显示:1-显示 0-隐藏"`
  30. YbFiccSort int `description:"小程序FICC页排序"`
  31. YbFiccIcon string `description:"小程序FICC页icon"`
  32. YbFiccPcIcon string `description:"小程序PC端FICC页背景图"`
  33. YbIconUrl string `description:"小程序已购页icon"`
  34. YbBgUrl string `description:"小程序已购详情背景图"`
  35. YbListImg string `description:"小程序研报列表封面图"`
  36. YbShareBgImg string `description:"小程序研报详情分享背景图"`
  37. YbRightBanner string `description:"Pc端详情页,右侧,报告合集背景图"`
  38. RelateTel int `description:"是否在电话会中可选: 0-否; 1-是"`
  39. RelateVideo int `description:"是否在路演视频中可选: 0-否; 1-是"`
  40. IsMassSend int `description:"1:群发,0:非群发"`
  41. Enabled int `description:"是否可用,1可用,0禁用"`
  42. Level int `description:"层级"`
  43. HasChild int `description:"是否有子级别,0:下面没有子分类,1:下面有子分类;默认:0"`
  44. ReportDetailShowType int `description:"报告详情的展示类型:1-拼接;2:目录"`
  45. ClassifyType int `description:"分类类型:0研报,1ppt,2研报和ppt"`
  46. }
  47. type ClassifyVO struct {
  48. Id int `orm:"column(id);pk"`
  49. ClassifyName string `description:"分类名称"`
  50. Sort int `json:"-"`
  51. ParentId int `description:"父级分类id"`
  52. ClassifyLabel string `description:"分类标签"`
  53. Enabled int `description:"是否可用,1可用,0禁用"`
  54. Level int `description:"层级"`
  55. Children *[]ClassifyVO
  56. }
  57. type ClassifyAddReq struct {
  58. ClassifyName string `description:"分类名称"`
  59. ParentId int `description:"父级分类id,没有父级分类传0"`
  60. ChartPermissionIdList []int `description:"权限id数组"`
  61. /*Abstract string `description:"栏目简介"`
  62. Descript string `description:"分享描述"`
  63. ReportAuthor string `description:"栏目作者"`
  64. AuthorDescript string `description:"作者简介"`
  65. ColumnImgUrl string `description:"栏目配图"`
  66. ReportImgUrl string `description:"报告配图"`
  67. HeadImgUrl string `description:"头部banner"`
  68. AvatarImgUrl string `description:"头像"`
  69. HomeImgUrl string `description:"首页配图"`
  70. ClassifyLabel string `description:"分类标签"`
  71. ShowType int `description:"展示类型:1-列表 2-专栏"`
  72. HasTeleconference int `description:"是否有电话会:0-否 1-是"`
  73. VipTitle string `description:"研究员头衔"`
  74. Sort int `description:"后台排序"`
  75. IsShow int `description:"是否在小程序显示:1-显示 0-隐藏"`
  76. YbFiccSort int `description:"小程序FICC页排序"`
  77. YbFiccIcon string `description:"小程序FICC页icon"`
  78. YbFiccPcIcon string `description:"小程序PC端FICC页背景图"`
  79. YbIconUrl string `description:"小程序已购页icon"`
  80. YbBgUrl string `description:"小程序已购详情背景图"`
  81. YbListImg string `description:"小程序研报列表封面图"`
  82. YbShareBgImg string `description:"小程序研报详情分享背景图"`
  83. YbRightBanner string `description:"Pc端详情页,右侧,报告合集背景图"`
  84. MenuList []*ClassifyMenuSaveReq `description:"子目录列表"`
  85. ClassifyMenuId int `description:"二级分类-子目录ID"`
  86. RelateTel int `description:"是否在电话会中可选: 0-否; 1-是"`
  87. RelateVideo int `description:"是否在路演视频中可选: 0-否; 1-是"`*/
  88. }
  89. func (c *Classify) Delete(childIds []int) (err error) {
  90. o := orm.NewOrmUsingDB("rddp")
  91. tx, err := o.Begin()
  92. if err != nil {
  93. return
  94. }
  95. defer func() {
  96. if err != nil {
  97. tx.Rollback()
  98. } else {
  99. tx.Commit()
  100. }
  101. }()
  102. // 删除自身
  103. _, err = tx.Delete(c)
  104. if err != nil {
  105. return
  106. }
  107. // 删除子分类
  108. if len(childIds) > 0 {
  109. sql := `DELETE FROM classify WHERE id IN (` + utils.GetOrmInReplace(len(childIds)) + `)`
  110. _, err = tx.Raw(sql, childIds).Exec()
  111. if err != nil {
  112. return
  113. }
  114. }
  115. // 检查是否需要更新父类
  116. if c.ParentId > 0 {
  117. var count int
  118. sql := `SELECT COUNT(*) FROM classify WHERE parent_id=? `
  119. err = tx.Raw(sql, c.ParentId).QueryRow(&count)
  120. if err != nil {
  121. return
  122. }
  123. if count == 0 {
  124. sql = `UPDATE classify SET has_child=0 WHERE id=? `
  125. _, err = tx.Raw(sql, c.ParentId).Exec()
  126. if err != nil {
  127. return
  128. }
  129. }
  130. }
  131. deleteImgSql := `DELETE FROM banner WHERE classify_id=? `
  132. _, err = tx.Raw(deleteImgSql, c.Id).Exec()
  133. return
  134. }
  135. func GetClassifyByName(classifyName string, parentId int) (item *Classify, err error) {
  136. sql := `SELECT * FROM classify WHERE classify_name=? AND parent_id=? `
  137. o := orm.NewOrmUsingDB("rddp")
  138. err = o.Raw(sql, classifyName, parentId).QueryRow(&item)
  139. return
  140. }
  141. func GetClassifyById(classifyId int) (item *Classify, err error) {
  142. sql := `SELECT * FROM classify WHERE id=? `
  143. o := orm.NewOrmUsingDB("rddp")
  144. err = o.Raw(sql, classifyId).QueryRow(&item)
  145. return
  146. }
  147. // 添加分类
  148. func AddClassify(item *Classify) (err error) {
  149. o := orm.NewOrmUsingDB("rddp")
  150. id, err := o.Insert(item)
  151. if err != nil {
  152. return
  153. }
  154. item.Id = int(id)
  155. return
  156. }
  157. func GetReportCountByClassifyId(classifyId int) (count int, err error) {
  158. o := orm.NewOrmUsingDB("rddp")
  159. sql := `SELECT COUNT(1) AS count FROM report WHERE classify_id_second=? `
  160. err = o.Raw(sql, classifyId).QueryRow(&count)
  161. return
  162. }
  163. func GetClassifySubCountByClassifyId(classifyId int) (count int, err error) {
  164. o := orm.NewOrmUsingDB("rddp")
  165. sql := `SELECT COUNT(1) as num FROM classify AS a
  166. INNER JOIN report AS b ON a.id=b.classify_id_second
  167. WHERE a.parent_id=? `
  168. err = o.Raw(sql, classifyId).QueryRow(&count)
  169. return
  170. }
  171. func GetClassifySubCountByParentId(classifyId int) (count int, err error) {
  172. sqlCount := `
  173. SELECT COUNT(1) as num FROM classify AS a
  174. WHERE a.parent_id=? `
  175. o := orm.NewOrmUsingDB("rddp")
  176. err = o.Raw(sqlCount, classifyId).QueryRow(&count)
  177. return
  178. }
  179. // 删除分类
  180. func DeleteClassify(classifyId int) (err error) {
  181. sql := `DELETE FROM classify WHERE id=? `
  182. o := orm.NewOrmUsingDB("rddp")
  183. _, err = o.Raw(sql, classifyId).Exec()
  184. if err != nil {
  185. return
  186. }
  187. deleteImgSql := `DELETE FROM banner WHERE classify_id=? `
  188. _, err = o.Raw(deleteImgSql, classifyId).Exec()
  189. return
  190. }
  191. // classifyName, abstract, descript string, parentId, classifyId int
  192. // 修改分类
  193. func EditClassify(req *EditClassifyReq) (err error) {
  194. o := orm.NewOrmUsingDB("rddp")
  195. //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 = ? `
  196. //_, 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()
  197. sql := `UPDATE classify SET classify_name = ?,parent_id= ?,modify_time= NOW() WHERE id = ? `
  198. _, err = o.Raw(sql, req.ClassifyName, req.ParentId, req.ClassifyId).Exec()
  199. return
  200. }
  201. // ParentClassify
  202. // @Description: 获取父级分类
  203. // @author: Roc
  204. // @datetime 2024-06-18 15:03:49
  205. // @return items []*Classify
  206. // @return err error
  207. func ParentClassify() (items []*Classify, err error) {
  208. sql := `SELECT * FROM classify WHERE parent_id=0 order by id desc `
  209. o := orm.NewOrmUsingDB("rddp")
  210. _, err = o.Raw(sql).QueryRows(&items)
  211. return
  212. }
  213. // 根据id获取分类详情
  214. func FindByIdClassify(classifyId int) (item *Classify, err error) {
  215. sql := `SELECT * FROM classify WHERE id=? `
  216. o := orm.NewOrmUsingDB("rddp")
  217. err = o.Raw(sql, classifyId).QueryRow(&item)
  218. return
  219. }
  220. type ClassifyList struct {
  221. Id int `orm:"column(id);pk"`
  222. ClassifyName string `description:"分类名称"`
  223. Sort int `description:"排序"`
  224. ParentId int `description:"父级分类id"`
  225. CreateTime time.Time `description:"创建时间"`
  226. ModifyTime time.Time `description:"修改时间"`
  227. Abstract string `description:"简介"`
  228. Descript string `description:"描述"`
  229. ClassifyLabel string `description:"分类标签"`
  230. ShowType int `description:"展示类型:1-列表 2-专栏"`
  231. HasTeleconference int `description:"是否有电话会:0-否 1-是"`
  232. IsShow int `description:"是否在小程序显示:1-显示 0-隐藏"`
  233. YbFiccSort int `description:"小程序FICC页排序"`
  234. YbFiccIcon string `description:"小程序FICC页icon"`
  235. YbFiccPcIcon string `description:"小程序PC端FICC页背景图"`
  236. YbIconUrl string `description:"小程序已购页icon"`
  237. YbBgUrl string `description:"小程序已购详情背景图"`
  238. YbListImg string `description:"小程序研报列表封面图"`
  239. YbShareBgImg string `description:"小程序研报详情分享背景图"`
  240. YbRightBanner string `description:"Pc端详情页,右侧,报告合集背景图"`
  241. RelateTel int `description:"是否在电话会中可选: 0-否; 1-是"`
  242. RelateVideo int `description:"是否在路演视频中可选: 0-否; 1-是"`
  243. Enabled int `description:"是否可用,1可用,0禁用"`
  244. Child []*ClassifyList
  245. ClassifyMenuId int `description:"二级分类-子目录ID"`
  246. ClassifyMenuList []*ClassifyMenu
  247. ChartPermissionIdList []int `description:"绑定的权限ID"`
  248. Level int `description:"层级"`
  249. HasChild int `description:"是否有子级别,0:下面没有子分类,1:下面有子分类;默认:0"`
  250. IsEnableDelete int `description:"是否允许删除: 0-否; 1-是"`
  251. }
  252. type ClassifyItem struct {
  253. Classify
  254. ClassifyMenuId int `description:"二级分类-子目录ID"`
  255. ClassifyMenuList []*ClassifyMenu
  256. ChartPermissionIdList []int `description:"绑定的权限ID"`
  257. Child []*ClassifyItem
  258. }
  259. type ClassifyListResp struct {
  260. List []*ClassifyList
  261. }
  262. type ClassifyPermissionListResp struct {
  263. List []*ClassifyList
  264. Paging *paging.PagingItem `description:"分页数据"`
  265. }
  266. // 获取分类列表
  267. func GetClassifyList(keyWord string, enabled int) (items []*ClassifyList, err error) {
  268. sql := ``
  269. companyTypeSqlStr := ``
  270. if enabled == 1 {
  271. companyTypeSqlStr += ` AND enabled = 1 `
  272. }
  273. pars := make([]interface{}, 0)
  274. if keyWord != "" {
  275. sql = `SELECT * FROM (
  276. SELECT * FROM classify
  277. WHERE parent_id=0 ` + companyTypeSqlStr + ` AND classify_name LIKE ?
  278. UNION
  279. SELECT * FROM classify
  280. WHERE id IN( SELECT parent_id FROM classify
  281. WHERE parent_id>0 ` + companyTypeSqlStr + ` AND classify_name LIKE ? )
  282. )AS t
  283. ORDER BY sort ASC,create_time ASC`
  284. pars = utils.GetLikeKeywordPars(pars, keyWord, 2)
  285. } else {
  286. sql = `SELECT * FROM classify WHERE parent_id=0 ` + companyTypeSqlStr
  287. sql += ` ORDER BY sort ASC, create_time ASC`
  288. }
  289. pars = append(pars)
  290. o := orm.NewOrmUsingDB("rddp")
  291. _, err = o.Raw(sql, pars...).QueryRows(&items)
  292. return
  293. }
  294. func GetClassifyListCount(keyWord, companyType string, hideDayWeek int) (count int, err error) {
  295. sqlCount := ``
  296. companyTypeSqlStr := ``
  297. if companyType == "ficc" {
  298. companyTypeSqlStr = " AND id != 40 AND parent_id != 40 "
  299. } else if companyType == "权益" {
  300. companyTypeSqlStr = " AND (id = 40 or parent_id = 40) "
  301. }
  302. pars := make([]interface{}, 0)
  303. if keyWord != "" {
  304. sqlCount = `SELECT COUNT(1) AS count FROM (
  305. SELECT * FROM classify
  306. WHERE parent_id=0 ` + companyTypeSqlStr + ` AND classify_name LIKE ?
  307. UNION
  308. SELECT * FROM classify
  309. WHERE id IN(SELECT parent_id FROM classify
  310. WHERE parent_id>0 ` + companyTypeSqlStr + ` AND classify_name LIKE ? )
  311. )AS t `
  312. pars = utils.GetLikeKeywordPars(pars, keyWord, 2)
  313. } else {
  314. sqlCount = `SELECT COUNT(1) AS count FROM classify WHERE parent_id=0 ` + companyTypeSqlStr
  315. if hideDayWeek == 1 {
  316. sqlCount += ` AND classify_name <> '晨报' AND classify_name <> '周报' `
  317. }
  318. }
  319. o := orm.NewOrmUsingDB("rddp")
  320. err = o.Raw(sqlCount, pars...).QueryRow(&count)
  321. return
  322. }
  323. type CheckDeleteClassifyReq struct {
  324. ClassifyId int `description:"分类ID"`
  325. }
  326. type CheckDeleteClassifyResp struct {
  327. Code int `description:"编码:0:检测成功,可进行删除,1:分类不存在,2:该分类有关联报告,不允许删除,3:二级分类有关联报告,不允许删除,4:该分类下有关联分类,是否确认全部删除"`
  328. Msg string `description:"描述信息"`
  329. }
  330. type DeleteClassifyReq struct {
  331. ClassifyId int `description:"分类ID"`
  332. }
  333. type EditClassifyReq struct {
  334. ClassifyId int `description:"分类ID"`
  335. ClassifyAddReq
  336. }
  337. type FindByIdClassifyReq struct {
  338. ClassifyId int `description:"分类ID"`
  339. }
  340. func GetClassifyChild(parentId int, keyWord string) (items []*Classify, err error) {
  341. o := orm.NewOrmUsingDB("rddp")
  342. sql := ``
  343. pars := make([]interface{}, 0)
  344. if keyWord != "" {
  345. sql = `SELECT * FROM classify WHERE classify_name LIKE ? AND parent_id=? ORDER BY create_time ASC `
  346. pars = append(pars, utils.GetLikeKeyword(keyWord))
  347. } else {
  348. sql = `SELECT * FROM classify WHERE parent_id=? ORDER BY create_time ASC `
  349. }
  350. pars = append(pars, parentId)
  351. _, err = o.Raw(sql, pars...).QueryRows(&items)
  352. return
  353. }
  354. func GetClassifyChildByParentIds(parentId []int, keyWord string, enabled int) (items []*Classify, err error) {
  355. parentIdLen := len(parentId)
  356. if parentIdLen == 0 {
  357. return
  358. }
  359. o := orm.NewOrmUsingDB("rddp")
  360. sql := ``
  361. pars := make([]interface{}, 0)
  362. pars = append(pars, parentId)
  363. if keyWord != "" {
  364. sql = `SELECT * FROM classify WHERE parent_id IN (` + utils.GetOrmInReplace(parentIdLen) + `) AND classify_name LIKE ? `
  365. pars = append(pars, utils.GetLikeKeyword(keyWord))
  366. } else {
  367. sql = `SELECT * FROM classify WHERE parent_id IN (` + utils.GetOrmInReplace(parentIdLen) + `) `
  368. }
  369. if enabled == 1 {
  370. sql += ` AND enabled=1 `
  371. }
  372. sql += ` ORDER BY create_time ASC `
  373. _, err = o.Raw(sql, pars...).QueryRows(&items)
  374. return
  375. }
  376. // EditClassifyPermissionReq 编辑分类权限请求
  377. type EditClassifyPermissionReq struct {
  378. ClassifyId int `description:"分类ID"`
  379. ChartPermissionIdList []int `description:"权限id数组"`
  380. }
  381. // GetAllClassify 获取所有分类
  382. func GetAllClassify() (list []*Classify, err error) {
  383. o := orm.NewOrmUsingDB("rddp")
  384. sql := ` SELECT * FROM classify `
  385. _, err = o.Raw(sql).QueryRows(&list)
  386. return
  387. }
  388. // GetAllClassifyWithDesc 获取所有倒序分类
  389. func GetAllClassifyWithDesc() (list []*Classify, err error) {
  390. o := orm.NewOrmUsingDB("rddp")
  391. sql := ` SELECT * FROM classify ORDER BY id DESC `
  392. _, err = o.Raw(sql).QueryRows(&list)
  393. return
  394. }
  395. // GetClassifyByKeyword 名称获取分类
  396. func GetClassifyByKeyword(keyword string) (item Classify, err error) {
  397. o := orm.NewOrmUsingDB("rddp")
  398. sql := ` SELECT * FROM classify WHERE classify_name = ? LIMIT 1 `
  399. err = o.Raw(sql, keyword).QueryRow(&item)
  400. return
  401. }
  402. // UpdateClassify 更新分类
  403. func (classifyInfo *Classify) UpdateClassify(cols []string) (err error) {
  404. o := orm.NewOrmUsingDB("rddp")
  405. _, err = o.Update(classifyInfo, cols...)
  406. return
  407. }
  408. // SimpleClassifyList 简版分类列表
  409. type SimpleClassifyList struct {
  410. Id int `description:"分类ID"`
  411. ClassifyName string `description:"分类名称"`
  412. ParentId int `description:"父级ID"`
  413. Sort int `description:"排序"`
  414. Child []*SimpleClassifyList
  415. }
  416. // GetClassifyByCondition 获取分类列表
  417. func GetClassifyByCondition(condition, orderRule string, pars []interface{}) (items []*SimpleClassifyList, err error) {
  418. o := orm.NewOrmUsingDB("rddp")
  419. sql := `SELECT * FROM classify WHERE 1 = 1 `
  420. if condition != `` {
  421. sql += condition
  422. }
  423. order := `sort ASC, create_time ASC`
  424. if orderRule != `` {
  425. order = orderRule
  426. }
  427. sql += ` ORDER BY ` + order
  428. _, err = o.Raw(sql, pars).QueryRows(&items)
  429. return
  430. }
  431. // UpdateChildClassifyRelateSetting 更新子分类关联设置
  432. func UpdateChildClassifyRelateSetting(parentId, relateTel, relateVideo int) (err error) {
  433. o := orm.NewOrmUsingDB("rddp")
  434. sql := `UPDATE classify SET relate_tel = ?, relate_video = ? WHERE parent_id = ?`
  435. _, err = o.Raw(sql, relateTel, relateVideo, parentId).Exec()
  436. return
  437. }
  438. // RelateTelSecClassifyWithPermissions 关联了电话会的二级分类及权限
  439. type RelateTelSecClassifyWithPermissions struct {
  440. Id int `description:"分类ID"`
  441. ClassifyName string `description:"分类名称"`
  442. ChartPermissionIds string `description:"权限IDs"`
  443. }
  444. // UpdateClassifySortByParentId 根据父类id更新排序
  445. func UpdateClassifySortByParentId(parentId, permissionId, nowSort int, updateSort string) (err error) {
  446. o := orm.NewOrmUsingDB("rddp")
  447. sql := ` update classify set sort = ` + updateSort + ` WHERE parent_id=? AND sort > ? `
  448. if permissionId > 0 {
  449. sql += ` or ( id > ` + fmt.Sprint(permissionId) + ` and sort = ` + fmt.Sprint(nowSort) + `)`
  450. }
  451. _, err = o.Raw(sql, parentId, nowSort).Exec()
  452. return
  453. }
  454. // GetMaxSortByParentId 获取最大的排序值
  455. func (classifyInfo *Classify) GetMaxSortByParentId(parentId int) (maxSort int, err error) {
  456. o := orm.NewOrmUsingDB("rddp")
  457. sql := `SELECT max(sort) AS sort FROM classify WHERE parent_id = ? `
  458. err = o.Raw(sql, parentId).QueryRow(&maxSort)
  459. return
  460. }
  461. // GetMaxSort 获取最大的排序值
  462. func (classifyInfo *Classify) GetMaxSort() (maxSort int, err error) {
  463. o := orm.NewOrmUsingDB("rddp")
  464. sql := `SELECT max(sort) AS sort FROM classify`
  465. err = o.Raw(sql).QueryRow(&maxSort)
  466. return
  467. }
  468. // GetFirstClassifyByParentId 获取当前父级分类下,且排序数相同 的排序第一条的数据
  469. func (classifyInfo *Classify) GetFirstClassifyByParentId(parentId int) (item *Classify, err error) {
  470. o := orm.NewOrmUsingDB("rddp")
  471. sql := `SELECT * FROM classify WHERE parent_id = ? order by sort asc, id asc limit 1`
  472. err = o.Raw(sql, parentId).QueryRow(&item)
  473. return
  474. }
  475. type ClassifyMoveReq struct {
  476. ClassifyId int `description:"分类ID"`
  477. PrevClassifyId int `description:"上一个兄弟节点分类id"`
  478. NextClassifyId int `description:"下一个兄弟节点分类id"`
  479. }
  480. type ClassifySetEnabledReq struct {
  481. ClassifyId int `description:"分类ID"`
  482. Enabled int `description:"是否可用,1可用,0禁用"`
  483. }
  484. func (classifyInfo *Classify) SetEnabled(id, enabled int) (err error) {
  485. o := orm.NewOrmUsingDB("rddp")
  486. to, err := o.Begin()
  487. if err != nil {
  488. return
  489. }
  490. defer func() {
  491. if err != nil {
  492. _ = to.Rollback()
  493. } else {
  494. _ = to.Commit()
  495. }
  496. }()
  497. sql := ` UPDATE classify SET enabled =? WHERE id = ?`
  498. _, err = to.Raw(sql, enabled, id).Exec()
  499. if err != nil {
  500. return
  501. }
  502. sql = ` UPDATE classify SET enabled =? WHERE parent_id = ?`
  503. _, err = to.Raw(sql, enabled, id).Exec()
  504. if err != nil {
  505. return
  506. }
  507. return
  508. }
  509. // GetCountClassifyChildByParentId
  510. // @Description: 获取父级分类下子分类数量
  511. // @author: Roc
  512. // @datetime 2024-06-17 10:58:46
  513. // @param parentId int
  514. // @return total int
  515. // @return err error
  516. func GetCountClassifyChildByParentId(parentId int) (total int, err error) {
  517. o := orm.NewOrmUsingDB("rddp")
  518. sql := `SELECT count(1) AS total FROM classify WHERE parent_id = ? `
  519. err = o.Raw(sql, parentId).QueryRow(&total)
  520. return
  521. }
  522. // GetClassifyListByKeyword
  523. // @Description: 获取分类列表
  524. // @author: Roc
  525. // @datetime 2024-06-19 09:49:33
  526. // @param keyWord string
  527. // @param enabled int
  528. // @return items []*ClassifyList
  529. // @return err error
  530. func GetClassifyListByKeyword(keyWord string, enabled int) (items []*ClassifyList, err error) {
  531. sql := ``
  532. pars := make([]interface{}, 0)
  533. sql = `SELECT * FROM classify WHERE 1=1 `
  534. if enabled == 1 {
  535. sql += ` AND enabled = 1 `
  536. }
  537. if keyWord != `` {
  538. sql += ` AND classify_name LIKE ? `
  539. pars = utils.GetLikeKeywordPars(pars, keyWord, 1)
  540. }
  541. sql += ` ORDER BY sort ASC, create_time ASC`
  542. o := orm.NewOrmUsingDB("rddp")
  543. _, err = o.Raw(sql, pars...).QueryRows(&items)
  544. return
  545. }
  546. // GetClassifyListByParentIdList
  547. // @Description: 获取分类列表
  548. // @author: Roc
  549. // @datetime 2024-06-19 09:49:33
  550. // @param keyWord string
  551. // @param enabled int
  552. // @return items []*ClassifyList
  553. // @return err error
  554. func GetClassifyListByParentIdList(parentClassifyIdList []int) (items []*ClassifyList, err error) {
  555. num := len(parentClassifyIdList)
  556. if num <= 0 {
  557. return
  558. }
  559. sql := `SELECT * FROM classify WHERE id in (` + utils.GetOrmInReplace(num) + `) ORDER BY sort ASC, create_time ASC`
  560. o := orm.NewOrmUsingDB("rddp")
  561. _, err = o.Raw(sql, parentClassifyIdList).QueryRows(&items)
  562. return
  563. }
  564. // GetClassifyListByIdList
  565. // @Description: 根据指标ID列表,获取分类列表
  566. // @author: Roc
  567. // @datetime 2024-06-27 15:23:57
  568. // @param classifyIdList []int
  569. // @return items []*Classify
  570. // @return err error
  571. func GetClassifyListByIdList(classifyIdList []int) (items []*Classify, err error) {
  572. num := len(classifyIdList)
  573. if num <= 0 {
  574. return
  575. }
  576. sql := `SELECT * FROM classify WHERE id IN (` + utils.GetOrmInReplace(num) + `) `
  577. o := orm.NewOrmUsingDB("rddp")
  578. _, err = o.Raw(sql, classifyIdList).QueryRows(&items)
  579. return
  580. }
  581. // GetClassifyListByParentId 查询父分类下所有的子分类
  582. func GetClassifyListByParentId(parentId int) (items []*Classify, err error) {
  583. o := orm.NewOrmUsingDB("rddp")
  584. sql := `SELECT * FROM classify WHERE parent_id = ? and enabled = 1`
  585. _, err = o.Raw(sql, parentId).QueryRows(&items)
  586. return items, err
  587. }