chart_classify.go 25 KB

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