chart_classify.go 26 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731
  1. package data_manage
  2. import (
  3. "eta/eta_api/utils"
  4. "fmt"
  5. "time"
  6. "database/sql"
  7. "eta/eta_api/global"
  8. )
  9. type ChartClassify struct {
  10. ChartClassifyId int `orm:"column(chart_classify_id);pk" gorm:"primaryKey" `
  11. ChartClassifyName string `description:"分类名称"`
  12. ParentId int `description:"父级id"`
  13. HasData int `description:"是否含有指标数据"`
  14. CreateTime time.Time `description:"创建时间"`
  15. ModifyTime time.Time `description:"修改时间"`
  16. SysUserId int `description:"创建人id"`
  17. SysUserRealName string `description:"创建人姓名"`
  18. Level int `description:"层级"`
  19. UniqueCode string `description:"唯一编码"`
  20. Sort int `description:"排序字段,越小越靠前,默认值:10"`
  21. Source int `description:"1:ETA图库;2:商品价格曲线"`
  22. IsJoinPermission int `description:"是否加入权限管控,0:不加入;1:加入;默认:0"`
  23. ChartClassifyNameEn string `description:"英文分类名称"`
  24. RootId int `description:"顶级ID"`
  25. IsSelected int `description:"是否精选资源,0:否;1:是"`
  26. ResourceStatus int `description:"在ETA投研资源库中的状态,0:初始状态,1上架,2下架"`
  27. LevelPath string `description:"所有的父级分类id"`
  28. }
  29. func AddChartClassify(item *ChartClassify) (lastId int64, err error) {
  30. err = global.DbMap[utils.DbNameIndex].Create(item).Error
  31. lastId = int64(item.ChartClassifyId)
  32. return
  33. }
  34. type AddChartClassifyReq struct {
  35. ChartClassifyName string `description:"分类名称"`
  36. ParentId int `description:"父级id,第一级传0"`
  37. Level int `description:"层级,第一级传0,其余传上一级的层级"`
  38. IsSelected int `description:"是否精选资源,0:否;1:是"`
  39. }
  40. // GetChartClassifyCount
  41. // @Description: 根据图表分类名称获取对应的数量
  42. // @author: Roc
  43. // @datetime 2024-04-16 18:13:40
  44. // @param chartClassifyName string
  45. // @param parentId int
  46. // @param source int
  47. // @return count int
  48. // @return err error
  49. func GetChartClassifyCount(chartClassifyName string, parentId, source int) (count int, err error) {
  50. o := global.DbMap[utils.DbNameIndex]
  51. sql := `SELECT COUNT(1) AS count FROM chart_classify WHERE parent_id=? AND source = ? AND chart_classify_name=? `
  52. err = o.Raw(sql, parentId, source, chartClassifyName).Scan(&count).Error
  53. return
  54. }
  55. // GetChartClassifyEnCount
  56. // @Description: 根据图表分类英文名称获取对应的数量
  57. // @author: Roc
  58. // @datetime 2024-04-16 18:13:24
  59. // @param chartClassifyNameEn string
  60. // @param parentId int
  61. // @param source int
  62. // @return count int
  63. // @return err error
  64. func GetChartClassifyEnCount(chartClassifyNameEn string, parentId, source int) (count int, err error) {
  65. o := global.DbMap[utils.DbNameIndex]
  66. sql := `SELECT COUNT(1) AS count FROM chart_classify WHERE parent_id=? AND source = ? AND chart_classify_name_en = ? `
  67. err = o.Raw(sql, parentId, source, chartClassifyNameEn).Scan(&count).Error
  68. return
  69. }
  70. type EditChartClassifyReq struct {
  71. ChartClassifyName string `description:"分类名称"`
  72. ChartClassifyId int `description:"分类id"`
  73. ParentId int `description:"父级分类id"`
  74. IsSelected int `description:"是否精选资源,0:否;1:是"`
  75. }
  76. func GetChartClassifyById(classifyId int) (item *ChartClassify, err error) {
  77. o := global.DbMap[utils.DbNameIndex]
  78. sql := `SELECT * FROM chart_classify WHERE chart_classify_id=? `
  79. err = o.Raw(sql, classifyId).First(&item).Error
  80. return
  81. }
  82. func GetChartClassifyCountById(classifyId int) (count int, err error) {
  83. o := global.DbMap[utils.DbNameIndex]
  84. sqlStr := `SELECT count(1) AS count FROM chart_classify WHERE chart_classify_id=? `
  85. err = o.Raw(sqlStr, classifyId).Scan(&count).Error
  86. return
  87. }
  88. // EditChartClassify
  89. // @Description: 修改图表分类的名称
  90. // @author: Roc
  91. // @datetime 2024-04-16 18:12:51
  92. // @param classifyId int
  93. // @param source int
  94. // @param chartClassifyName string
  95. // @return err error
  96. func EditChartClassify(classifyId, source int, chartClassifyName string) (err error) {
  97. sql := `UPDATE chart_classify SET chart_classify_name=?,modify_time=NOW() WHERE chart_classify_id=? AND source = ? `
  98. err = global.DbMap[utils.DbNameIndex].Exec(sql, chartClassifyName, classifyId, source).Error
  99. return
  100. }
  101. // EditChartClassifyEn
  102. // @Description: 修改图表分类的英文名称
  103. // @author: Roc
  104. // @datetime 2024-04-16 18:12:40
  105. // @param classifyId int
  106. // @param source int
  107. // @param chartClassifyNameEn string
  108. // @return err error
  109. func EditChartClassifyEn(classifyId, source int, chartClassifyNameEn string) (err error) {
  110. sql := `UPDATE chart_classify SET chart_classify_name_en = ?,modify_time=NOW() WHERE chart_classify_id=? AND source = ? `
  111. err = global.DbMap[utils.DbNameIndex].Exec(sql, chartClassifyNameEn, classifyId, source).Error
  112. return
  113. }
  114. type DeleteChartClassifyReq struct {
  115. ChartClassifyId int `description:"分类id"`
  116. ChartInfoId int `description:"指标id"`
  117. }
  118. func GetChartInfoCountByClassifyId(classifyId int) (count int, err error) {
  119. var pars []interface{}
  120. var sqlStr string
  121. if utils.DbDriverName == utils.DbDriverByDm {
  122. sqlStr = `WITH RECURSIVE chart_classify_cte (chart_classify_id, parent_id) AS (
  123. SELECT chart_classify_id, parent_id
  124. FROM chart_classify
  125. WHERE parent_id = ?
  126. UNION ALL
  127. SELECT c.chart_classify_id, c.parent_id
  128. FROM chart_classify c
  129. INNER JOIN chart_classify_cte ct ON c.parent_id = ct.chart_classify_id
  130. )
  131. SELECT COUNT(1) AS count
  132. FROM chart_info a
  133. WHERE a.chart_classify_id IN (
  134. SELECT chart_classify_id
  135. FROM chart_classify_cte
  136. UNION
  137. SELECT chart_classify_id
  138. FROM chart_classify
  139. WHERE chart_classify_id = ?
  140. )`
  141. pars = append(pars, classifyId, classifyId)
  142. } else {
  143. sqlStr = ` SELECT COUNT(1) AS count FROM chart_info AS a
  144. WHERE a.chart_classify_id IN(
  145. SELECT t.chart_classify_id FROM
  146. (
  147. SELECT rd.*
  148. FROM (SELECT * FROM chart_classify WHERE parent_id IS NOT NULL) rd,
  149. (SELECT @pid := ?) pd
  150. WHERE FIND_IN_SET(parent_id, @pid) > 0
  151. AND @pid := CONCAT(@pid, ',', chart_classify_id)
  152. UNION SELECT * FROM chart_classify WHERE chart_classify_id = @pid
  153. )AS t
  154. ) `
  155. pars = append(pars, classifyId)
  156. }
  157. var totalNull sql.NullInt64
  158. err = global.DbMap[utils.DbNameIndex].Raw(sqlStr, pars...).Scan(&totalNull).Error
  159. if !totalNull.Valid {
  160. count = 0
  161. } else {
  162. count = int(totalNull.Int64)
  163. }
  164. return
  165. }
  166. func DeleteChartClassify(classifyId int) (err error) {
  167. var pars []interface{}
  168. var strSql string
  169. if utils.DbDriverName == utils.DbDriverByDm {
  170. strSql = `WITH RECURSIVE chart_classify_cte (chart_classify_id, parent_id) AS (
  171. SELECT chart_classify_id, parent_id
  172. FROM chart_classify
  173. WHERE parent_id = ?
  174. UNION ALL
  175. SELECT c.chart_classify_id, c.parent_id
  176. FROM chart_classify c
  177. INNER JOIN chart_classify_cte ct ON c.parent_id = ct.chart_classify_id
  178. )
  179. DELETE FROM chart_classify
  180. WHERE chart_classify_id IN (
  181. SELECT chart_classify_id
  182. FROM chart_classify_cte
  183. UNION
  184. SELECT chart_classify_id
  185. FROM chart_classify
  186. WHERE chart_classify_id = ?
  187. )`
  188. pars = append(pars, classifyId, classifyId)
  189. } else {
  190. strSql = ` DELETE FROM chart_classify
  191. WHERE chart_classify_id IN(
  192. SELECT t.chart_classify_id FROM
  193. (
  194. SELECT rd.*
  195. FROM (SELECT * FROM chart_classify WHERE parent_id IS NOT NULL) rd,
  196. (SELECT @pid := ?) pd
  197. WHERE FIND_IN_SET(parent_id, @pid) > 0
  198. AND @pid := CONCAT(@pid, ',', chart_classify_id)
  199. UNION SELECT * FROM chart_classify WHERE chart_classify_id = @pid
  200. )AS t
  201. ) `
  202. pars = append(pars, classifyId)
  203. }
  204. err = global.DbMap[utils.DbNameIndex].Exec(strSql, pars...).Error
  205. return
  206. }
  207. // GetChartClassifyByParentId
  208. // @param source int 1:ETA图库;2:商品价格曲线;3:相关性图表
  209. func GetChartClassifyByParentId(parentId, source int) (items []*ChartClassifyItems, err error) {
  210. sql := ` SELECT * FROM chart_classify WHERE parent_id=? AND source = ? order by sort asc,chart_classify_id asc`
  211. err = global.DbMap[utils.DbNameIndex].Raw(sql, parentId, source).Find(&items).Error
  212. return
  213. }
  214. // GetChartClassifyAll
  215. // @param source int 1:ETA图库;2:商品价格曲线;3:相关性图表
  216. func GetChartClassifyAll(source int) (items []*ChartClassifyItems, err error) {
  217. sql := ` SELECT * FROM chart_classify WHERE parent_id<>0 AND source = ? order by sort asc,chart_classify_id asc`
  218. err = global.DbMap[utils.DbNameIndex].Raw(sql, source).Find(&items).Error
  219. return
  220. }
  221. // GetChartClassifyBySource 获取图表分类列表
  222. func GetChartClassifyBySource(source int) (items []*ChartClassifyItems, err error) {
  223. sql := ` SELECT * FROM chart_classify WHERE source = ? ORDER BY level ASC,chart_classify_id ASC`
  224. err = global.DbMap[utils.DbNameIndex].Raw(sql, source).Find(&items).Error
  225. return
  226. }
  227. type ChartClassifyItems struct {
  228. ChartClassifyId int `description:"分类id"`
  229. ChartInfoId int `description:"指标id"`
  230. ChartClassifyName string
  231. ChartClassifyNameEn string
  232. ParentId int
  233. Level int `description:"层级"`
  234. Sort int `description:"排序字段,越小越靠前,默认值:10"`
  235. UniqueCode string `description:"唯一编码"`
  236. Source int `description:"来源id"`
  237. SourceName string `description:"来源名称"`
  238. SysUserId int `description:"创建人id"`
  239. SysUserRealName string `description:"创建人姓名"`
  240. DateType int `description:"日期类型:1:00年至今,2:10年至今,3:15年至今,4:年初至今,5:自定义时间"`
  241. StartDate string `description:"自定义开始日期"`
  242. EndDate string `description:"自定义结束日期"`
  243. ChartType int `description:"生成样式:1:曲线图,2:季节性图"`
  244. Calendar string `description:"公历/农历"`
  245. SeasonStartDate string `description:"季节性图开始日期"`
  246. SeasonEndDate string `description:"季节性图开始日期"`
  247. Children []*ChartClassifyItems `gorm:"-"`
  248. Button ChartClassifyItemsButton `gorm:"-" description:"按钮权限"`
  249. IsJoinPermission int `description:"是否加入权限管控,0:不加入;1:加入;默认:0"`
  250. HaveOperaAuth bool `description:"是否有数据权限,默认:false"`
  251. Disable bool `description:"勾选是否禁用"`
  252. IsSelected int `description:"是否精选资源,0:否;1:是"`
  253. }
  254. // ChartClassifyItemsButton 操作按钮
  255. type ChartClassifyItemsButton struct {
  256. AddButton bool `description:"是否可添加"`
  257. OpButton bool `description:"是否可编辑"`
  258. DeleteButton bool `description:"是否可删除"`
  259. MoveButton bool `description:"是否可移动"`
  260. }
  261. type ChartClassifyListResp struct {
  262. AllNodes []*ChartClassifyItems
  263. Language string `description:"指标的展示语言,CN:中文,EN:英文"`
  264. CanOpClassify bool `description:"是否允许操作分类"`
  265. }
  266. type ChartClassifyDeleteCheckResp struct {
  267. DeleteStatus int `description:"检测状态:0:默认值,如果为0,继续走其他校验,1:该分类下关联图表不可删除,2:确认删除当前目录及包含的子目录吗"`
  268. TipsMsg string `description:"提示信息"`
  269. }
  270. type ChartClassifyDeleteCheckReq struct {
  271. ChartClassifyId int `description:"分类id"`
  272. ChartInfoId int `description:"指标id"`
  273. }
  274. func GetChartClassifyCountByClassifyId(chartClassifyId int) (count int, err error) {
  275. var strSql string
  276. var pars []interface{}
  277. if utils.DbDriverName == utils.DbDriverByDm {
  278. strSql = `WITH RECURSIVE chart_classify_cte (chart_classify_id, parent_id) AS (
  279. SELECT chart_classify_id, parent_id
  280. FROM chart_classify
  281. WHERE parent_id = ?
  282. UNION ALL
  283. SELECT c.chart_classify_id, c.parent_id
  284. FROM chart_classify c
  285. INNER JOIN chart_classify_cte ct ON c.parent_id = ct.chart_classify_id
  286. )
  287. SELECT COUNT(1) AS count
  288. FROM (
  289. SELECT chart_classify_id
  290. FROM chart_classify_cte
  291. UNION
  292. SELECT chart_classify_id
  293. FROM chart_classify
  294. WHERE chart_classify_id = ?
  295. ) AS t
  296. WHERE t.chart_classify_id <> ?`
  297. pars = append(pars, chartClassifyId, chartClassifyId, chartClassifyId)
  298. } else {
  299. strSql = ` SELECT COUNT(1) AS count FROM (
  300. SELECT rd.*
  301. FROM (SELECT * FROM chart_classify WHERE parent_id IS NOT NULL) rd,
  302. (SELECT @pid := ?) pd
  303. WHERE FIND_IN_SET(parent_id, @pid) > 0
  304. AND @pid := CONCAT(@pid, ',', chart_classify_id)
  305. UNION SELECT * FROM chart_classify WHERE chart_classify_id = @pid
  306. )AS t
  307. WHERE t.chart_classify_id<>? `
  308. pars = append(pars, chartClassifyId, chartClassifyId)
  309. }
  310. err = global.DbMap[utils.DbNameIndex].Raw(strSql, pars...).Scan(&count).Error
  311. return
  312. }
  313. func GetChartClassifyByCondition(condition string, pars []interface{}) (item *ChartClassify, err error) {
  314. o := global.DbMap[utils.DbNameIndex]
  315. sql := ` SELECT * FROM chart_classify WHERE 1=1 `
  316. if condition != "" {
  317. sql += condition
  318. }
  319. err = o.Raw(sql, pars...).First(&item).Error
  320. return
  321. }
  322. // MoveChartClassifyReq 移动图表分类请求参数
  323. type MoveChartClassifyReq struct {
  324. ClassifyId int `description:"分类id"`
  325. ParentClassifyId int `description:"父级分类id"`
  326. PrevClassifyId int `description:"上一个兄弟节点分类id"`
  327. NextClassifyId int `description:"下一个兄弟节点分类id"`
  328. ChartInfoId int `description:"图表ID, 如果图表ID有值,则移动对象为图表,否则认为移动对象分类"`
  329. PrevChartInfoId int `description:"上一个图表ID"`
  330. NextChartInfoId int `description:"下一个图表ID"`
  331. }
  332. // GetFirstChartClassifyByParentId 获取当前父级图表分类下的排序第一条的数据
  333. func GetFirstChartClassifyByParentId(parentId int) (item *ChartClassify, err error) {
  334. o := global.DbMap[utils.DbNameIndex]
  335. sql := ` SELECT * FROM chart_classify WHERE parent_id=? order by sort asc,chart_classify_id asc limit 1`
  336. err = o.Raw(sql, parentId).First(&item).Error
  337. return
  338. }
  339. // GetFirstChartClassifyByParentIdAndSource 获取当前父级图表分类下的排序第一条的数据
  340. func GetFirstChartClassifyByParentIdAndSource(parentId, source int) (item *ChartClassify, err error) {
  341. o := global.DbMap[utils.DbNameIndex]
  342. sql := ` SELECT * FROM chart_classify WHERE parent_id=? and source = ? order by sort asc,chart_classify_id asc limit 1`
  343. err = o.Raw(sql, parentId, source).First(&item).Error
  344. return
  345. }
  346. // UpdateChartClassifySortByParentId 根据图表父类id更新排序
  347. func UpdateChartClassifySortByParentId(parentId, classifyId, nowSort int, updateSort string) (err error) {
  348. o := global.DbMap[utils.DbNameIndex]
  349. sql := ` update chart_classify set sort = ` + updateSort + ` WHERE parent_id=? and sort > ? `
  350. if classifyId > 0 {
  351. sql += ` or ( chart_classify_id > ` + fmt.Sprint(classifyId) + ` and sort= ` + fmt.Sprint(nowSort) + `)`
  352. }
  353. err = o.Exec(sql, parentId, nowSort).Error
  354. return
  355. }
  356. // Update 更新图表分类基础信息
  357. func (chartClassify *ChartClassify) Update(cols []string) (err error) {
  358. err = global.DbMap[utils.DbNameIndex].Select(cols).Updates(chartClassify).Error
  359. return
  360. }
  361. // GetChartClassifyMaxSort 获取图表分类下最大的排序数
  362. func GetChartClassifyMaxSort(parentId, source int) (sort int, err error) {
  363. o := global.DbMap[utils.DbNameIndex]
  364. sql := `SELECT COALESCE(MAX(sort),0) AS sort FROM chart_classify WHERE parent_id=? AND source = ? `
  365. err = o.Raw(sql, parentId, source).Scan(&sort).Error
  366. return
  367. }
  368. type ChartClassifyView struct {
  369. ChartClassifyId int `orm:"column(chart_classify_id);pk" gorm:"primaryKey" `
  370. ChartClassifyName string `description:"分类名称"`
  371. ParentId int `description:"父级id"`
  372. }
  373. func GetChartClassifyViewById(classifyId int) (item *ChartClassifyView, err error) {
  374. o := global.DbMap[utils.DbNameIndex]
  375. sql := `SELECT * FROM chart_classify WHERE chart_classify_id=? `
  376. err = o.Raw(sql, classifyId).First(&item).Error
  377. return
  378. }
  379. func GetChartClassifyByParentIdFromETA(parentId int) (items []*ChartClassifyItems, err error) {
  380. sql := ` SELECT * FROM chart_classify WHERE parent_id=? AND source = 1 order by sort asc,chart_classify_id asc`
  381. err = global.DbMap[utils.DbNameIndex].Raw(sql, parentId).Find(&items).Error
  382. return
  383. }
  384. func GetChartClassifyAllFromETA() (items []*ChartClassifyItems, err error) {
  385. sql := ` SELECT * FROM chart_classify WHERE parent_id<>0 AND source = 1 order by sort asc,chart_classify_id asc`
  386. err = global.DbMap[utils.DbNameIndex].Raw(sql).Find(&items).Error
  387. return
  388. }
  389. func GetChartClassifyTwo(source int) (items []*ChartClassifyItems, err error) {
  390. sql := ` SELECT * FROM chart_classify WHERE level=3 AND source = ? order by sort asc,chart_classify_id asc`
  391. err = global.DbMap[utils.DbNameIndex].Raw(sql, source).Find(&items).Error
  392. return
  393. }
  394. func GetChartClassifyByLevel(level, source int) (items []*ChartClassifyItems, err error) {
  395. o := global.DbMap[utils.DbNameIndex]
  396. sql := ` SELECT * FROM chart_classify WHERE level=? AND source = ? order by sort asc,chart_classify_id asc`
  397. err = o.Raw(sql, level, source).Scan(&items).Error
  398. return
  399. }
  400. func EditChartClassifySysUser(classifyId, sysUserId int, chartClassifyName string) (err error) {
  401. o := global.DbMap[utils.DbNameIndex]
  402. sql := `UPDATE chart_classify SET sys_user_id=?,sys_user_real_name=?,modify_time=NOW() WHERE parent_id=? AND level=3 `
  403. err = o.Exec(sql, sysUserId, chartClassifyName, classifyId).Error
  404. return
  405. }
  406. func GetAllChartClassifyItemsBySource(source int) (items []*ChartClassifyItems, err error) {
  407. o := global.DbMap[utils.DbNameIndex]
  408. sql := ` SELECT * FROM chart_classify WHERE source = ? order by sort ASC,chart_classify_id ASC`
  409. err = o.Raw(sql, source).Scan(&items).Error
  410. return
  411. }
  412. // GetCrossVarietyChartClassifyBySysUserId
  413. // @Description: 根据创建人获取跨品种分析的分类
  414. // @author: Roc
  415. // @datetime 2023-11-24 14:05:43
  416. // @param sysUserId int
  417. // @return item *ChartClassify
  418. // @return err error
  419. func GetCrossVarietyChartClassifyBySysUserId(sysUserId int) (item *ChartClassify, err error) {
  420. o := global.DbMap[utils.DbNameIndex]
  421. sql := `SELECT * FROM chart_classify WHERE source = ? AND sys_user_id=?`
  422. err = o.Raw(sql, utils.CHART_SOURCE_CROSS_HEDGING, sysUserId).First(&item).Error
  423. return
  424. }
  425. // GetChartClassifyByParentIdAndName 根据父级ID和名称获取分类
  426. func GetChartClassifyByParentIdAndName(parentId int, classifyName string, classifyId int) (item *ChartClassify, err error) {
  427. sql := `SELECT * FROM chart_classify WHERE parent_id = ? AND chart_classify_name = ? AND chart_classify_id <> ? LIMIT 1`
  428. err = global.DbMap[utils.DbNameIndex].Raw(sql, parentId, classifyName, classifyId).First(&item).Error
  429. return
  430. }
  431. // GetChartClassifyByIdList
  432. // @Description: 根据分类ID列表获取分类
  433. // @author: Roc
  434. // @datetime 2024-04-01 13:21:48
  435. // @param classifyId []int
  436. // @return items []*ChartClassify
  437. // @return err error
  438. func GetChartClassifyByIdList(classifyIdList []int) (items []*ChartClassify, err error) {
  439. num := len(classifyIdList)
  440. if num <= 0 {
  441. return
  442. }
  443. o := global.DbMap[utils.DbNameIndex]
  444. sql := `SELECT * FROM chart_classify WHERE chart_classify_id in (` + utils.GetOrmInReplace(num) + `) `
  445. err = o.Raw(sql, classifyIdList).Find(&items).Error
  446. return
  447. }
  448. // GetChartClassifyBySourceAndIsJoinPermission
  449. // @Description: 根据分类类型和是否涉密获取分类列表
  450. // @author: Roc
  451. // @datetime 2024-03-29 10:30:21
  452. // @param source int
  453. // @param isJoinPermission int
  454. // @return items []*ChartClassify
  455. // @return err error
  456. func GetChartClassifyBySourceAndIsJoinPermission(source, isJoinPermission int) (items []*ChartClassify, err error) {
  457. o := global.DbMap[utils.DbNameIndex]
  458. sql := `SELECT * FROM chart_classify WHERE source = ? AND is_join_permission = ? `
  459. err = o.Raw(sql, source, isJoinPermission).Scan(&items).Error
  460. return
  461. }
  462. type ChartClassifyIdItems struct {
  463. ChartClassifyId int `description:"分类id"`
  464. ChartClassifyName string `description:"分类名称"`
  465. UniqueCode string `description:"唯一编码"`
  466. ParentId int `description:"父级分类id"`
  467. Level int `description:"层级"`
  468. RootId int `description:"顶级分类id"`
  469. IsJoinPermission int `description:"是否加入权限管控,0:不加入;1:加入;默认:0" json:"-"`
  470. HaveOperaAuth bool `description:"是否有该数据权限,默认:false"`
  471. }
  472. func GetChartClassifyByRootIdLevel(rootId, source int, orderStr string) (items []*ChartClassifyIdItems, err error) {
  473. o := global.DbMap[utils.DbNameIndex]
  474. sql := ` SELECT * FROM chart_classify WHERE root_id = ? AND source = ? `
  475. if orderStr != "" {
  476. sql += orderStr
  477. } else {
  478. sql += ` order by level desc, sort asc, chart_classify_id asc`
  479. }
  480. err = o.Raw(sql, rootId, source).Scan(&items).Error
  481. return
  482. }
  483. // UpdateChartClassifySortByParentIdAndSource 根据图表父类id更新排序
  484. func UpdateChartClassifySortByParentIdAndSource(parentId, classifyId, nowSort int, updateSort string, source int) (err error) {
  485. o := global.DbMap[utils.DbNameIndex]
  486. sql := ` update chart_classify set sort = ` + updateSort + ` WHERE parent_id=? and sort > ? AND source = ? `
  487. if classifyId > 0 {
  488. sql += ` or ( chart_classify_id > ` + fmt.Sprint(classifyId) + ` and sort= ` + fmt.Sprint(nowSort) + `)`
  489. }
  490. err = o.Exec(sql, parentId, nowSort, source).Error
  491. return
  492. }
  493. // GetChartClassifyAllBySource 根据来源获取所有分类
  494. func GetChartClassifyAllBySource(source int) (items []*ChartClassifyItems, err error) {
  495. sql := ` SELECT * FROM chart_classify WHERE source = ? ORDER BY parent_id ASC, sort ASC, chart_classify_id ASC`
  496. err = global.DbMap[utils.DbNameIndex].Raw(sql, source).Find(&items).Error
  497. return
  498. }
  499. // GetChartClassifyInfoSelectedBySource 获取所有的精选目录
  500. func GetChartClassifyInfoSelectedBySource(source int) (items []*ChartClassify, err error) {
  501. sql := ` SELECT * FROM chart_classify WHERE source = ? AND is_selected = ? ORDER BY parent_id ASC, sort ASC, chart_classify_id ASC`
  502. err = global.DbMap[utils.DbNameIndex].Raw(sql, source, utils.ChartClassifyIsSelected).Find(&items).Error
  503. return
  504. }
  505. // GetChartClassifyIdListByAdminId
  506. // @Description: 根据用户id和指标类型获取其关联的所有指标分类id列表
  507. // @author: Roc
  508. // @datetime 2024-09-11 15:52:48
  509. // @param adminId int
  510. // @param source int
  511. // @return chartClassifyIdList []int
  512. // @return err error
  513. func GetChartClassifyIdListByAdminId(adminId, source int) (chartClassifyIdList []int, err error) {
  514. sql := ` SELECT chart_classify_id FROM chart_info WHERE sys_user_id = ? AND source = ?`
  515. pars := []interface{}{adminId, source}
  516. sql += ` group by chart_classify_id `
  517. err = global.DbMap[utils.DbNameIndex].Raw(sql, pars...).Find(&chartClassifyIdList).Error
  518. return
  519. }
  520. // GetChartClassifyAndInfoByParentId
  521. func GetChartClassifyAndInfoByParentId(parentId int, isSelected int) (items []*ChartClassifyItems, err error) {
  522. where := ""
  523. if isSelected == utils.ChartClassifyIsSelected {
  524. where = " AND is_selected = 1"
  525. }
  526. sql := ` SELECT
  527. 0 AS chart_info_id,
  528. chart_classify_id,
  529. chart_classify_name,
  530. chart_classify_name_en,
  531. parent_id,
  532. create_time,
  533. modify_time,
  534. sys_user_id,
  535. sys_user_real_name,
  536. sort,
  537. level,
  538. unique_code,
  539. source,
  540. 0 as date_type,
  541. '' as start_date,
  542. '' as end_date,
  543. 0 as chart_type,
  544. '' as calendar,
  545. '' as season_start_date,
  546. '' as season_end_date,
  547. is_selected
  548. FROM
  549. chart_classify
  550. WHERE
  551. parent_id = ? and source = 1` + where + ` UNION ALL
  552. SELECT
  553. chart_info_id,
  554. chart_classify_id,
  555. chart_name AS chart_classify_name,
  556. chart_name_en AS chart_classify_name_en,
  557. 0 AS parent_id,
  558. create_time,
  559. modify_time,
  560. sys_user_id,
  561. sys_user_real_name,
  562. sort,
  563. 0 AS level,
  564. unique_code,
  565. source,
  566. date_type,
  567. start_date,
  568. end_date,
  569. chart_type,
  570. calendar,
  571. season_start_date,
  572. season_end_date
  573. FROM
  574. chart_info
  575. WHERE
  576. chart_classify_id = ? AND source = 1
  577. ORDER BY
  578. sort ASC,
  579. chart_classify_id ASC`
  580. err = global.DbMap[utils.DbNameIndex].Raw(sql, parentId, parentId).Find(&items).Error
  581. return
  582. }
  583. // GetChartClassifyAndInfoByParentId
  584. func GetChartClassifyAndInfoByParentIdForMe(parentId, adminId, isSelected int) (items []*ChartClassifyItems, err error) {
  585. where := ""
  586. if isSelected == utils.ChartClassifyIsSelected {
  587. where = " AND is_selected = 1"
  588. }
  589. sql := ` SELECT
  590. 0 AS chart_info_id,
  591. chart_classify_id,
  592. chart_classify_name,
  593. chart_classify_name_en,
  594. parent_id,
  595. create_time,
  596. modify_time,
  597. sys_user_id,
  598. sys_user_real_name,
  599. sort,
  600. level,
  601. unique_code,
  602. is_selected
  603. FROM
  604. chart_classify
  605. WHERE
  606. parent_id = ? and source = 1` + where + ` UNION ALL
  607. SELECT
  608. chart_info_id,
  609. chart_classify_id,
  610. chart_name AS chart_classify_name,
  611. chart_name_en AS chart_classify_name_en,
  612. 0 AS parent_id,
  613. create_time,
  614. modify_time,
  615. sys_user_id,
  616. sys_user_real_name,
  617. sort,
  618. 0 AS level,
  619. unique_code
  620. FROM
  621. chart_info
  622. WHERE
  623. chart_classify_id = ? AND sys_user_id = ? AND source = 1
  624. ORDER BY
  625. sort ASC,
  626. chart_classify_id ASC`
  627. err = global.DbMap[utils.DbNameIndex].Raw(sql, parentId, parentId, adminId, isSelected).Find(&items).Error
  628. return
  629. }
  630. func GetChartClassifiesById(chartClassifyId int) (items []*ChartClassifyItems, err error) {
  631. sql := `
  632. SELECT *
  633. FROM chart_classify
  634. WHERE chart_classify_id = ?
  635. OR parent_id = ?
  636. OR parent_id IN (
  637. SELECT chart_classify_id
  638. FROM chart_classify
  639. WHERE parent_id = ?
  640. );`
  641. err = global.DbMap[utils.DbNameIndex].Raw(sql, chartClassifyId, chartClassifyId, chartClassifyId).Find(&items).Error
  642. return
  643. }
  644. func GetChartClassifyByLevelPath(levelPath string) (items []*ChartClassify, err error) {
  645. sql := `SELECT * FROM chart_classify where level_path like '` + levelPath + `%'`
  646. err = global.DbMap[utils.DbNameIndex].Raw(sql).Find(&items).Error
  647. return
  648. }
  649. func UpdateChartClassifyIsSelected(source int, isSelected int, rootId int, levelPath string) (err error) {
  650. sql := `UPDATE chart_classify SET is_selected = ? WHERE source = ? AND root_id = ? AND level_path LIKE '` + levelPath + `%'`
  651. err = global.DbMap[utils.DbNameIndex].Exec(sql, isSelected, source, rootId).Error
  652. return
  653. }