edb_classify.go 29 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786
  1. package data_manage
  2. import (
  3. "eta/eta_api/global"
  4. "eta/eta_api/utils"
  5. "fmt"
  6. "strconv"
  7. "time"
  8. )
  9. type EdbClassify struct {
  10. ClassifyId int `gorm:"column:classify_id;primaryKey"`
  11. ClassifyType uint8 `description:"分类类型,0:普通指标分类,1:预测指标分类"`
  12. ClassifyName string `description:"分类名称"`
  13. ParentId int `description:"父级id"`
  14. RootId int `description:"顶级id"`
  15. HasData int `description:"是否含有指标数据"`
  16. CreateTime time.Time `description:"创建时间"`
  17. ModifyTime time.Time `description:"修改时间"`
  18. SysUserId int `description:"创建人id"`
  19. SysUserRealName string `description:"创建人姓名"`
  20. Level int `description:"层级"`
  21. UniqueCode string `description:"唯一编码"`
  22. Sort int `description:"排序字段,越小越靠前,默认值:10"`
  23. LastModifyUserId int `description:"最后修改人ID"`
  24. LastModifyUserRealName string `description:"最后修改人姓名"`
  25. IsJoinPermission int `description:"是否加入权限管控,0:不加入;1:加入;默认:0"`
  26. ClassifyNameEn string `description:"英文分类名称"`
  27. }
  28. func AddEdbClassify(item *EdbClassify) (lastId int64, err error) {
  29. //o := orm.NewOrmUsingDB("data")
  30. //lastId, err = o.Insert(item)
  31. err = global.DbMap[utils.DbNameIndex].Create(item).Error
  32. if err != nil {
  33. return
  34. }
  35. lastId = int64(item.ClassifyId)
  36. return
  37. }
  38. type AddEdbClassifyReq struct {
  39. ClassifyName string `description:"分类名称"`
  40. ParentId int `description:"父级id,第一级传0"`
  41. Level int `description:"层级,第一级传0,其余传上一级的层级"`
  42. ClassifyType uint8 `description:"分类类型,0:普通指标分类,1:预测指标分类"`
  43. }
  44. // GetEdbClassifyCount
  45. // @Description: 根据父级id获取同级中文名称的数量
  46. // @author: Roc
  47. // @datetime 2024-04-17 14:12:22
  48. // @param classifyName string
  49. // @param parentId int
  50. // @param classifyType uint8
  51. // @return count int
  52. // @return err error
  53. func GetEdbClassifyCount(classifyName string, parentId int, classifyType uint8) (count int, err error) {
  54. //o := orm.NewOrmUsingDB("data")
  55. sql := `SELECT COUNT(1) AS count FROM edb_classify WHERE parent_id=? AND classify_name=? AND classify_type = ? `
  56. err = global.DbMap[utils.DbNameIndex].Raw(sql, parentId, classifyName, classifyType).Scan(&count).Error
  57. //err = o.Raw(sql, parentId, classifyName, classifyType).QueryRow(&count)
  58. return
  59. }
  60. // GetEdbClassifyEnCount
  61. // @Description: 根据父级id获取同级英文名称的数量
  62. // @author: Roc
  63. // @datetime 2024-04-17 14:12:09
  64. // @param classifyNameEn string
  65. // @param parentId int
  66. // @param classifyType uint8
  67. // @return count int
  68. // @return err error
  69. func GetEdbClassifyEnCount(classifyNameEn string, parentId int, classifyType uint8) (count int, err error) {
  70. //o := orm.NewOrmUsingDB("data")
  71. sql := `SELECT COUNT(1) AS count FROM edb_classify WHERE parent_id=? AND classify_name_en = ? AND classify_type = ? `
  72. //err = o.Raw(sql, parentId, classifyNameEn, classifyType).QueryRow(&count)
  73. err = global.DbMap[utils.DbNameIndex].Raw(sql, parentId, classifyNameEn, classifyType).Scan(&count).Error
  74. return
  75. }
  76. type EditEdbClassifyReq struct {
  77. ClassifyName string `description:"分类名称"`
  78. ClassifyId int `description:"分类名称"`
  79. ParentId int `description:"父级分类id"`
  80. }
  81. func GetEdbClassifyById(classifyId int) (item *EdbClassify, err error) {
  82. //o := orm.NewOrmUsingDB("data")
  83. sql := `SELECT * FROM edb_classify WHERE classify_id=? `
  84. //err = o.Raw(sql, classifyId).QueryRow(&item)
  85. err = global.DbMap[utils.DbNameIndex].Raw(sql, classifyId).First(&item).Error
  86. return
  87. }
  88. func GetEdbClassifyCountById(classifyId int) (count int, err error) {
  89. //o := orm.NewOrmUsingDB("data")
  90. sql := `SELECT count(1) AS count FROM edb_classify WHERE classify_id=? `
  91. //err = o.Raw(sql, classifyId).QueryRow(&count)
  92. err = global.DbMap[utils.DbNameIndex].Raw(sql, classifyId).Scan(&count).Error
  93. return
  94. }
  95. // EditEdbClassify
  96. // @Description: 修改指标分类的中文名称
  97. // @author: Roc
  98. // @datetime 2024-04-17 14:14:48
  99. // @param classifyId int
  100. // @param classifyName string
  101. // @param userId int
  102. // @param userRealName string
  103. // @return err error
  104. func EditEdbClassify(classifyId int, classifyName string, userId int, userRealName string) (err error) {
  105. //o := orm.NewOrmUsingDB("data")
  106. sql := `UPDATE edb_classify SET classify_name=?,modify_time=NOW(), last_modify_user_id = ?, last_modify_user_real_name = ? WHERE classify_id=? `
  107. //_, err = o.Raw(sql, classifyName, userId, userRealName, classifyId).Exec()
  108. err = global.DbMap[utils.DbNameIndex].Exec(sql, classifyName, userId, userRealName, classifyId).Error
  109. return
  110. }
  111. // EditEdbClassifyEn
  112. // @Description: 修改指标分类的英文名称
  113. // @author: Roc
  114. // @datetime 2024-04-17 14:14:36
  115. // @param classifyId int
  116. // @param classifyNameEn string
  117. // @param userId int
  118. // @param userRealName string
  119. // @return err error
  120. func EditEdbClassifyEn(classifyId int, classifyNameEn string, userId int, userRealName string) (err error) {
  121. //o := orm.NewOrmUsingDB("data")
  122. sql := `UPDATE edb_classify SET classify_name_en = ?,modify_time=NOW(), last_modify_user_id = ?, last_modify_user_real_name = ? WHERE classify_id=? `
  123. //_, err = o.Raw(sql, classifyNameEn, userId, userRealName, classifyId).Exec()
  124. err = global.DbMap[utils.DbNameIndex].Exec(sql, classifyNameEn, userId, userRealName, classifyId).Error
  125. return
  126. }
  127. type DeleteEdbClassifyReq struct {
  128. ClassifyId int `description:"分类id"`
  129. EdbInfoId int `description:"指标id"`
  130. }
  131. func GetEdbInfoCountByClassifyId(classifyId int) (count int, err error) {
  132. //o := orm.NewOrmUsingDB("data")
  133. var sql string
  134. var pars []interface{}
  135. if utils.DbDriverName == utils.DbDriverByDm {
  136. sql = `WITH RECURSIVE edb_classify_cte (classify_id, parent_id) AS (
  137. SELECT classify_id, parent_id
  138. FROM edb_classify
  139. WHERE parent_id = ?
  140. UNION ALL
  141. SELECT c.classify_id, c.parent_id
  142. FROM edb_classify c
  143. INNER JOIN edb_classify_cte ct ON c.parent_id = ct.classify_id
  144. )
  145. SELECT COUNT(1) AS count
  146. FROM edb_info a
  147. WHERE a.classify_id IN (
  148. SELECT classify_id
  149. FROM edb_classify_cte
  150. UNION
  151. SELECT classify_id
  152. FROM edb_classify
  153. WHERE classify_id = ?
  154. )`
  155. pars = append(pars, classifyId, classifyId)
  156. }else{
  157. sql = ` SELECT COUNT(1) AS count FROM edb_info AS a
  158. WHERE a.classify_id IN(
  159. SELECT t.classify_id FROM
  160. (
  161. SELECT rd.*
  162. FROM (SELECT * FROM edb_classify WHERE parent_id IS NOT NULL) rd,
  163. (SELECT @pid := ?) pd
  164. WHERE FIND_IN_SET(parent_id, @pid) > 0
  165. AND @pid := CONCAT(@pid, ',', classify_id)
  166. UNION SELECT * FROM edb_classify WHERE classify_id = @pid
  167. )AS t
  168. ) `
  169. pars = append(pars, classifyId)
  170. }
  171. //err = o.Raw(sql, classifyId).QueryRow(&count)
  172. err = global.DbMap[utils.DbNameIndex].Raw(sql, pars...).Scan(&count).Error
  173. return
  174. }
  175. func DeleteEdbClassify(classifyId int) (err error) {
  176. var pars []interface{}
  177. var sql string
  178. if utils.DbDriverName == utils.DbDriverByDm {
  179. sql = `WITH RECURSIVE edb_classify_cte (classify_id, parent_id) AS (
  180. SELECT classify_id, parent_id
  181. FROM edb_classify
  182. WHERE parent_id = ?
  183. UNION ALL
  184. SELECT c.classify_id, c.parent_id
  185. FROM edb_classify c
  186. INNER JOIN edb_classify_cte ct ON c.parent_id = ct.classify_id
  187. )
  188. DELETE FROM edb_classify
  189. WHERE classify_id IN (
  190. SELECT classify_id
  191. FROM edb_classify_cte
  192. UNION
  193. SELECT classify_id
  194. FROM edb_classify
  195. WHERE classify_id = ?
  196. )`
  197. pars = append(pars, classifyId, classifyId)
  198. } else {
  199. sql = ` DELETE FROM edb_classify
  200. WHERE classify_id IN(
  201. SELECT t.classify_id FROM
  202. (
  203. SELECT rd.*
  204. FROM (SELECT * FROM edb_classify WHERE parent_id IS NOT NULL) rd,
  205. (SELECT @pid := ?) pd
  206. WHERE FIND_IN_SET(parent_id, @pid) > 0
  207. AND @pid := CONCAT(@pid, ',', classify_id)
  208. UNION SELECT * FROM edb_classify WHERE classify_id = @pid
  209. )AS t
  210. ) `
  211. pars = append(pars, classifyId)
  212. }
  213. err = global.DbMap[utils.DbNameIndex].Exec(sql, pars...).Error
  214. return
  215. }
  216. func GetEdbClassifyByParentId(parentId int, classifyType int8) (items []*EdbClassifyItems, err error) {
  217. //o := orm.NewOrmUsingDB("data")
  218. sql := ` SELECT * FROM edb_classify WHERE parent_id=? AND classify_type = ? order by sort asc,classify_id asc`
  219. //_, err = o.Raw(sql, parentId, classifyType).QueryRows(&items)
  220. err = global.DbMap[utils.DbNameIndex].Raw(sql, parentId, classifyType).Find(&items).Error
  221. return
  222. }
  223. func GetEdbClassifyByRootIdLevel(rootId int, classifyType uint8, orderStr string) (items []*EdbClassifyItems, err error) {
  224. //o := orm.NewOrmUsingDB("data")
  225. sql := ` SELECT * FROM edb_classify WHERE root_id=? AND classify_type = ? `
  226. if orderStr != "" {
  227. sql += orderStr
  228. } else {
  229. sql += ` order by level desc, sort asc, classify_id asc`
  230. }
  231. //_, err = o.Raw(sql, rootId, classifyType).QueryRows(&items)
  232. err = global.DbMap[utils.DbNameIndex].Raw(sql, rootId, classifyType).Find(&items).Error
  233. return
  234. }
  235. func GetEdbClassifyByRootIdLevelOrder(rootId int, classifyType uint8) (items []*EdbClassifyItems, err error) {
  236. //o := orm.NewOrmUsingDB("data")
  237. sql := ` SELECT * FROM edb_classify WHERE root_id=? AND classify_type = ? order by level desc, sort asc, classify_id asc`
  238. //_, err = o.Raw(sql, rootId, classifyType).QueryRows(&items)
  239. err = global.DbMap[utils.DbNameIndex].Raw(sql, rootId, classifyType).Find(&items).Error
  240. return
  241. }
  242. // GetEdbClassifyAll 获取普通指标的分类列表
  243. func GetEdbClassifyAll() (items []*EdbClassifyItems, err error) {
  244. //o := orm.NewOrmUsingDB("data")
  245. sql := ` SELECT * FROM edb_classify WHERE parent_id<>0 AND classify_type = 0 order by sort asc,classify_id asc`
  246. //_, err = o.Raw(sql).QueryRows(&items)
  247. err = global.DbMap[utils.DbNameIndex].Raw(sql).Find(&items).Error
  248. return
  249. }
  250. // GetAllEdbClassify 获取所有的普通指标的分类列表(包含第一级)
  251. func GetAllEdbClassify() (items []*EdbClassifyItems, err error) {
  252. //o := orm.NewOrmUsingDB("data")
  253. sql := ` SELECT * FROM edb_classify WHERE classify_type = 0 order by sort asc,classify_id asc`
  254. //_, err = o.Raw(sql).QueryRows(&items)
  255. err = global.DbMap[utils.DbNameIndex].Raw(sql).Find(&items).Error
  256. return
  257. }
  258. // GetNormalEdbClassifyAll 获取普通指标的分类列表
  259. func GetNormalEdbClassifyAll() (items []*EdbClassifyItems, err error) {
  260. //o := orm.NewOrmUsingDB("data")
  261. sql := ` SELECT * FROM edb_classify WHERE classify_type = 0 order by sort asc,classify_id asc`
  262. //_, err = o.Raw(sql).QueryRows(&items)
  263. err = global.DbMap[utils.DbNameIndex].Raw(sql).Find(&items).Error
  264. return
  265. }
  266. // GetPredictEdbClassifyAll 获取预测指标的分类列表
  267. func GetPredictEdbClassifyAll() (items []*EdbClassifyItems, err error) {
  268. //o := orm.NewOrmUsingDB("data")
  269. sql := ` SELECT * FROM edb_classify WHERE classify_type = 1 order by sort asc,classify_id asc`
  270. //_, err = o.Raw(sql).QueryRows(&items)
  271. err = global.DbMap[utils.DbNameIndex].Raw(sql).Find(&items).Error
  272. return
  273. }
  274. // GetAllEdbClassifyByType 根据类型获取分类列表
  275. func GetAllEdbClassifyByType(classifyType int) (items []*EdbClassifyItems, err error) {
  276. //o := orm.NewOrmUsingDB("data")
  277. sql := ` SELECT * FROM edb_classify WHERE classify_type = ? order by sort ASC,classify_id ASC`
  278. //_, err = o.Raw(sql, classifyType).QueryRows(&items)
  279. err = global.DbMap[utils.DbNameIndex].Raw(sql, classifyType).Find(&items).Error
  280. return
  281. }
  282. type EdbClassifyItems struct {
  283. ClassifyId int `description:"分类id"`
  284. EdbInfoId int `description:"指标id"`
  285. ClassifyName string
  286. ClassifyNameEn string
  287. ParentId int
  288. RootId int `description:"顶级id"`
  289. Level int `description:"层级"`
  290. Sort int `description:"排序字段,越小越靠前,默认值:10"`
  291. UniqueCode string `description:"唯一编码"`
  292. Source int `description:"来源id"`
  293. SourceName string `description:"来源名称"`
  294. SysUserId int `description:"创建人id"`
  295. SysUserRealName string `description:"创建人姓名"`
  296. StartDate string
  297. EdbCode string
  298. EdbType int `description:"指标类型:1:基础指标,2:计算指标"`
  299. Children []*EdbClassifyItems `gorm:"-"`
  300. Button EdbClassifyItemsButton `description:"操作权限" gorm:"-"`
  301. IsJoinPermission int `description:"是否加入权限管控,0:不加入;1:加入;默认:0"`
  302. HaveOperaAuth bool `description:"是否有数据权限"`
  303. SelectDisable bool `description:"是否可以被选中 前端用"`
  304. }
  305. type EdbClassifyIdItems struct {
  306. ClassifyId int `description:"分类id"`
  307. ClassifyName string
  308. ClassifyNameEn string
  309. UniqueCode string `description:"唯一编码"`
  310. ParentId int `description:"父级分类id"`
  311. Level int `description:"层级"`
  312. RootId int `description:"顶级分类id"`
  313. IsJoinPermission int `description:"是否加入权限管控,0:不加入;1:加入;默认:0" json:"-"`
  314. HaveOperaAuth bool `description:"是否有该数据权限,默认:false"`
  315. }
  316. type EdbClassifyItemList []*EdbClassifyItems
  317. func (m EdbClassifyItemList) Len() int {
  318. return len(m)
  319. }
  320. func (m EdbClassifyItemList) Less(i, j int) bool {
  321. return m[i].Sort < m[j].Sort
  322. }
  323. func (m EdbClassifyItemList) Swap(i, j int) {
  324. m[i], m[j] = m[j], m[i]
  325. }
  326. // EdbClassifyItemsButton 操作按钮
  327. type EdbClassifyItemsButton struct {
  328. AddButton bool `description:"是否可添加"`
  329. OpButton bool `description:"是否可编辑"`
  330. DeleteButton bool `description:"是否可删除"`
  331. MoveButton bool `description:"是否可移动"`
  332. ShowEdbRelation bool `description:"是否展示关联指标"`
  333. ShowChartRelation bool `description:"是否展示关联图表"`
  334. InsertNewDataButton bool `description:"是否可插入最新值"`
  335. }
  336. type EdbClassifyListResp struct {
  337. AllNodes EdbClassifyItemList
  338. CanOpClassify bool `description:"是否允许操作分类"`
  339. Language string `description:"指标的展示语言,CN:中文,EN:英文"`
  340. }
  341. type ClassifyDeleteCheckResp struct {
  342. DeleteStatus int `description:"检测状态:0:默认值,如果为0,继续走其他校验,1:若目录关联指标不可删除,2:确认删除当前目录及包含的子目录吗,3:当前指标已用作画图,不可删除"`
  343. TipsMsg string `description:"提示信息"`
  344. TableList []*ExcelBaseInfo `description:"关联的表格"`
  345. }
  346. type ExcelBaseInfo struct {
  347. ExcelInfoId int `orm:"column(excel_info_id);pk"`
  348. Source int `description:"表格来源,1:excel插件的表格,2:自定义表格,3:混合表格,4:自定义分析,默认:1"`
  349. ExcelType int `description:"表格类型,1:指标列,2:日期列,默认:1"`
  350. ExcelName string `description:"表格名称"`
  351. UniqueCode string `description:"表格唯一编码"`
  352. ExcelClassifyId int `description:"表格分类id"`
  353. //SysUserId int `description:"操作人id"`
  354. //SysUserRealName string `description:"操作人真实姓名"`
  355. }
  356. type ClassifyDeleteCheckReq struct {
  357. ClassifyId int `description:"分类id"`
  358. EdbInfoId int `description:"指标id"`
  359. }
  360. func GetClassifyCountByClassifyId(classifyId int) (count int, err error) {
  361. //o := orm.NewOrmUsingDB("data")
  362. var sql string
  363. var pars []interface{}
  364. if utils.DbDriverName == utils.DbDriverByDm {
  365. sql = `WITH RECURSIVE edb_classify_cte (classify_id, parent_id) AS (
  366. SELECT classify_id, parent_id
  367. FROM edb_classify
  368. WHERE parent_id = ?
  369. UNION ALL
  370. SELECT c.classify_id, c.parent_id
  371. FROM edb_classify c
  372. INNER JOIN edb_classify_cte ct ON c.parent_id = ct.classify_id
  373. )
  374. SELECT COUNT(1) AS count
  375. FROM (
  376. SELECT classify_id
  377. FROM edb_classify_cte
  378. UNION
  379. SELECT classify_id
  380. FROM edb_classify
  381. WHERE classify_id = ?
  382. ) AS t
  383. WHERE t.classify_id <> ?`
  384. pars = append(pars, classifyId, classifyId, classifyId)
  385. }else{
  386. sql = ` SELECT COUNT(1) AS count FROM (
  387. SELECT rd.*
  388. FROM (SELECT * FROM edb_classify WHERE parent_id IS NOT NULL) rd,
  389. (SELECT @pid := ?) pd
  390. WHERE FIND_IN_SET(parent_id, @pid) > 0
  391. AND @pid := CONCAT(@pid, ',', classify_id)
  392. UNION SELECT * FROM edb_classify WHERE classify_id = @pid
  393. )AS t
  394. WHERE t.classify_id<>? `
  395. pars = append(pars, classifyId, classifyId)
  396. }
  397. //err = o.Raw(sql, classifyId, classifyId).QueryRow(&count)
  398. err = global.DbMap[utils.DbNameIndex].Raw(sql, pars...).Scan(&count).Error
  399. return
  400. }
  401. type EdbClassifySimplify struct {
  402. ClassifyId int `description:"分类id"`
  403. ClassifyName string `description:"分类名称"`
  404. ParentId int
  405. }
  406. // 递归获取所有目录
  407. func GetClassifyAllByClassifyId(classifyId int) (resultList []*EdbClassifySimplify, err error) {
  408. fmt.Println("185")
  409. //o := orm.NewOrmUsingDB("data")
  410. for {
  411. sql := ` SELECT * FROM edb_classify AS a
  412. WHERE a.classify_id=? order by sort asc,classify_id asc`
  413. item := new(EdbClassifySimplify)
  414. //err = o.Raw(sql, classifyId).QueryRow(&item)
  415. err = global.DbMap[utils.DbNameIndex].Raw(sql, classifyId).Scan(item).Error
  416. if err != nil {
  417. break
  418. }
  419. if item.ClassifyId > 0 {
  420. resultList = append(resultList, item)
  421. classifyId = item.ParentId
  422. } else {
  423. break
  424. }
  425. }
  426. return resultList, err
  427. }
  428. // MoveEdbClassifyReq 移动分类请求参数
  429. type MoveEdbClassifyReq struct {
  430. ClassifyId int `description:"分类id"`
  431. ParentClassifyId int `description:"父级分类id"`
  432. PrevClassifyId int `description:"上一个兄弟节点分类id"`
  433. NextClassifyId int `description:"下一个兄弟节点分类id"`
  434. EdbInfoId int `description:"指标ID, 如果指标ID有值,则移动对象为指标,否则认为移动对象为分类"`
  435. PrevEdbInfoId int `description:"上一个指标ID"`
  436. NextEdbInfoId int `description:"下一个指标ID"`
  437. }
  438. // GetFirstEdbClassifyByParentId 获取当前父级分类下,且排序数相同 的排序第一条的数据
  439. func GetFirstEdbClassifyByParentId(parentId int) (item *EdbClassify, err error) {
  440. //o := orm.NewOrmUsingDB("data")
  441. sql := ` SELECT * FROM edb_classify WHERE parent_id=? order by sort asc,classify_id asc limit 1`
  442. //err = o.Raw(sql, parentId).QueryRow(&item)
  443. err = global.DbMap[utils.DbNameIndex].Raw(sql, parentId).First(&item).Error
  444. return
  445. }
  446. // UpdateEdbClassifySortByParentId 根据父类id更新排序
  447. func UpdateEdbClassifySortByParentId(parentId, classifyId, nowSort int, updateSort string, classifyType uint8) (err error) {
  448. //o := orm.NewOrmUsingDB("data")
  449. sql := ` update edb_classify set sort = ` + updateSort + ` WHERE parent_id=? AND sort > ? AND classify_type = ? `
  450. if classifyId > 0 {
  451. sql += ` or ( classify_id > ` + fmt.Sprint(classifyId) + ` and sort = ` + fmt.Sprint(nowSort) + `)`
  452. }
  453. //_, err = o.Raw(sql, parentId, nowSort, classifyType).Exec()
  454. err = global.DbMap[utils.DbNameIndex].Exec(sql, parentId, nowSort, classifyType).Error
  455. return
  456. }
  457. // Update 更新分类基础信息
  458. func (edbClassify *EdbClassify) Update(cols []string) (err error) {
  459. //o := orm.NewOrmUsingDB("data")
  460. //_, err = o.Update(edbClassify, cols...)
  461. err = global.DbMap[utils.DbNameIndex].Model(&edbClassify).Select(cols).Updates(edbClassify).Error
  462. return
  463. }
  464. // GetEdbClassifyMaxSort 获取分类下最大的排序数
  465. func GetEdbClassifyMaxSort(parentId int, classifyType uint8) (sort int, err error) {
  466. //o := orm.NewOrmUsingDB("data")
  467. sql := `SELECT COALESCE(Max(sort),0) AS sort FROM edb_classify WHERE parent_id=? AND classify_type=? `
  468. //err = o.Raw(sql, parentId, classifyType).QueryRow(&sort)
  469. err = global.DbMap[utils.DbNameIndex].Raw(sql, parentId, classifyType).Scan(&sort).Error
  470. return
  471. }
  472. // GetEdbInfoMaxSortByClassifyId 获取分类下指标的最大的排序数
  473. func GetEdbInfoMaxSortByClassifyId(classifyId int) (sort int, err error) {
  474. //o := orm.NewOrmUsingDB("data")
  475. sql := `SELECT COALESCE(Max(sort),0) AS sort FROM edb_info WHERE classify_id=? `
  476. //err = o.Raw(sql, classifyId).QueryRow(&sort)
  477. err = global.DbMap[utils.DbNameIndex].Raw(sql, classifyId).Scan(&sort).Error
  478. return
  479. }
  480. type ClassifyEdbInfoListResp struct {
  481. EdbInfoList []*EdbClassifyItems
  482. }
  483. // GetEdbClassifyAllV2 获取普通指标的分类列表
  484. func GetEdbClassifyAllV2(classifyType int8) (items []*EdbClassifyItems, err error) {
  485. //o := orm.NewOrmUsingDB("data")
  486. sql := ` SELECT * FROM edb_classify WHERE parent_id <>0 AND classify_type = ? order by sort asc,classify_id asc`
  487. //_, err = o.Raw(sql, classifyType).QueryRows(&items)
  488. err = global.DbMap[utils.DbNameIndex].Raw(sql, classifyType).Find(&items).Error
  489. return
  490. }
  491. func GetEdbClassifyByParentIdTwo(classifyType int8) (items []*EdbClassifyItems, err error) {
  492. //o := orm.NewOrmUsingDB("data")
  493. sql := ` SELECT * FROM edb_classify WHERE level=2 AND classify_type = ? order by sort asc,classify_id asc`
  494. //_, err = o.Raw(sql, classifyType).QueryRows(&items)
  495. err = global.DbMap[utils.DbNameIndex].Raw(sql, classifyType).Find(&items).Error
  496. return
  497. }
  498. type AddEdbClassifyReqV2 struct {
  499. ClassifyName string `description:"分类名称"`
  500. ParentId int `description:"父级id,第一级传0"`
  501. Level int `description:"层级,第一级传0,其余传上一级的层级"`
  502. ClassifyType uint8 `description:"分类类型,0:普通指标分类,1:预测指标分类"`
  503. }
  504. // GetEdbClassifyMaxSortV2 获取分类下最大的排序数
  505. func GetEdbClassifyMaxSortV2(parentId int, classifyType uint8) (sort int, err error) {
  506. //o := orm.NewOrmUsingDB("data")
  507. sql := `SELECT Max(sort) AS sort FROM edb_classify WHERE parent_id=? AND parent_id_two=? AND classify_type=? `
  508. //err = o.Raw(sql, parentId, classifyType).QueryRow(&sort)
  509. err = global.DbMap[utils.DbNameIndex].Raw(sql, parentId, classifyType).Scan(&sort).Error
  510. return
  511. }
  512. func GetEdbClassifyByLevel(level, source int) (items []*EdbClassifyItems, err error) {
  513. //o := orm.NewOrmUsingDB("data")
  514. sql := ` SELECT * FROM edb_classify WHERE level=? AND classify_type = ? `
  515. //_, err = o.Raw(sql, level, source).QueryRows(&items)
  516. err = global.DbMap[utils.DbNameIndex].Raw(sql, level, source).Find(&items).Error
  517. return
  518. }
  519. func FIXEdbClassifyId(newId, oldId int) (err error) {
  520. //o := orm.NewOrmUsingDB("data")
  521. sql := ` UPDATE edb_info SET classify_id=? WHERE classify_id=? `
  522. //_, err = o.Raw(sql, newId, oldId).Exec()
  523. err = global.DbMap[utils.DbNameIndex].Exec(sql, newId, oldId).Error
  524. return
  525. }
  526. func FixEdbClassify() {
  527. //新增三级分类
  528. list, err := GetEdbClassifyByLevel(2, 0)
  529. if err != nil {
  530. return
  531. }
  532. for _, v := range list {
  533. timestamp := strconv.FormatInt(time.Now().UnixNano(), 10)
  534. classifyInfo := &EdbClassify{
  535. //ClassifyId: 0,
  536. ClassifyType: 0,
  537. ClassifyName: v.ClassifyName,
  538. ParentId: v.ClassifyId,
  539. HasData: 0,
  540. CreateTime: time.Now(),
  541. ModifyTime: time.Now(),
  542. SysUserRealName: "admin",
  543. Level: 3,
  544. UniqueCode: utils.MD5(utils.DATA_PREFIX + "_" + timestamp),
  545. Sort: v.Sort,
  546. }
  547. if utils.RunMode == "debug" {
  548. classifyInfo.SysUserId = 11
  549. } else {
  550. classifyInfo.SysUserId = 163
  551. }
  552. id, err := AddEdbClassify(classifyInfo)
  553. if err != nil {
  554. return
  555. }
  556. err = FIXEdbClassifyId(int(id), v.ClassifyId)
  557. if err != nil {
  558. return
  559. }
  560. }
  561. }
  562. func FixPredictEdbClassify() {
  563. //新增三级分类
  564. list, err := GetEdbClassifyByLevel(2, 1)
  565. if err != nil {
  566. return
  567. }
  568. for _, v := range list {
  569. timestamp := strconv.FormatInt(time.Now().UnixNano(), 10)
  570. classifyInfo := &EdbClassify{
  571. //ClassifyId: 0,
  572. ClassifyType: 1,
  573. ClassifyName: v.ClassifyName,
  574. ParentId: v.ClassifyId,
  575. HasData: 0,
  576. CreateTime: time.Now(),
  577. ModifyTime: time.Now(),
  578. SysUserRealName: "admin",
  579. Level: 3,
  580. UniqueCode: utils.MD5(utils.DATA_PREFIX + "_" + timestamp),
  581. Sort: v.Sort,
  582. }
  583. if utils.RunMode == "debug" {
  584. classifyInfo.SysUserId = 11
  585. } else {
  586. classifyInfo.SysUserId = 163
  587. }
  588. id, err := AddEdbClassify(classifyInfo)
  589. if err != nil {
  590. return
  591. }
  592. err = FIXEdbClassifyId(int(id), v.ClassifyId)
  593. if err != nil {
  594. return
  595. }
  596. }
  597. }
  598. func GetEdbClassify(classifyId int) (classifyIds string, err error) {
  599. //o := orm.NewOrmUsingDB("data")
  600. sql :=fmt.Sprintf( `SELECT %s AS classify_ids FROM (
  601. SELECT a.classify_id FROM edb_classify AS a
  602. WHERE a.classify_id=?
  603. UNION ALL
  604. SELECT a.classify_id FROM edb_classify AS a
  605. WHERE a.parent_id=? UNION ALL
  606. SELECT
  607. classify_id
  608. FROM
  609. edb_classify
  610. WHERE
  611. parent_id IN ( SELECT classify_id FROM edb_classify WHERE parent_id = ? )
  612. )AS t`, utils.GroupUnitFunc(utils.DbDriverName, "t.classify_id", ",", false, "", "", false))
  613. //err = o.Raw(sql, classifyId, classifyId, classifyId).QueryRow(&classifyIds)
  614. err = global.DbMap[utils.DbNameIndex].Raw(sql, classifyId, classifyId, classifyId).Scan(&classifyIds).Error
  615. return
  616. }
  617. func FixEdbClassifySysUser() {
  618. list, err := GetEdbClassifyByLevel(2, 0)
  619. if err != nil {
  620. return
  621. }
  622. for _, v := range list {
  623. err := EditEdbClassifySysUser(v.ClassifyId, v.SysUserId, v.SysUserRealName)
  624. if err != nil {
  625. return
  626. }
  627. fmt.Println("ClassifyId:", v.ClassifyId)
  628. }
  629. fmt.Println("FixEdbClassifySysUser end")
  630. }
  631. func EditEdbClassifySysUser(classifyId, sysUserId int, chartClassifyName string) (err error) {
  632. //o := orm.NewOrmUsingDB("data")
  633. sql := `UPDATE edb_classify SET sys_user_id=?,sys_user_real_name=?,modify_time=NOW() WHERE parent_id=? AND level=3 `
  634. //_, err = o.Raw(sql, sysUserId, chartClassifyName, classifyId).Exec()
  635. err = global.DbMap[utils.DbNameIndex].Exec(sql, sysUserId, chartClassifyName, classifyId).Error
  636. return
  637. }
  638. func FixPredictEdbClassifySysUser() {
  639. list, err := GetEdbClassifyByLevel(2, 1)
  640. if err != nil {
  641. return
  642. }
  643. for _, v := range list {
  644. err := EditEdbClassifySysUser(v.ClassifyId, v.SysUserId, v.SysUserRealName)
  645. if err != nil {
  646. return
  647. }
  648. fmt.Println("ClassifyId:", v.ClassifyId)
  649. }
  650. fmt.Println("EditChartClassifySysUser end")
  651. }
  652. func UpdateEdbClassifyChildByParentClassifyId(classifyIds []int, rootId int, levelStep int) (err error) {
  653. //o := orm.NewOrmUsingDB("data")
  654. var pars []interface{}
  655. pars = append(pars, rootId, levelStep)
  656. pars = append(pars, classifyIds)
  657. // 更新相关联的二级分类的parentId,和classify_name_second
  658. sql := `update edb_classify
  659. SET root_id = ?, level = level+?
  660. where classify_id IN (` + utils.GetOrmInReplace(len(classifyIds)) + `)`
  661. //_, err = o.Raw(sql, pars...).Exec()
  662. err = global.DbMap[utils.DbNameIndex].Exec(sql, pars...).Error
  663. if err != nil {
  664. return
  665. }
  666. return
  667. }
  668. // GetEdbClassifyByParentIdAndName 根据父级ID和名称获取分类
  669. func GetEdbClassifyByParentIdAndName(parentId int, classifyName string, classifyId int) (item *EdbClassify, err error) {
  670. //o := orm.NewOrmUsingDB("data")
  671. sql := `SELECT * FROM edb_classify WHERE parent_id = ? AND classify_name = ? AND classify_id <> ? LIMIT 1`
  672. //err = o.Raw(sql, parentId, classifyName, classifyId).QueryRow(&item)
  673. err = global.DbMap[utils.DbNameIndex].Raw(sql, parentId, classifyName, classifyId).First(&item).Error
  674. return
  675. }
  676. // GetEdbClassifyByIdList
  677. // @Description: 根据分类id列表获取分类列表
  678. // @author: Roc
  679. // @datetime 2024-03-29 10:30:21
  680. // @param classifyId int
  681. // @return item *EdbClassify
  682. // @return err error
  683. func GetEdbClassifyByIdList(classifyIdList []int) (item []*EdbClassify, err error) {
  684. num := len(classifyIdList)
  685. if num <= 0 {
  686. return
  687. }
  688. //o := orm.NewOrmUsingDB("data")
  689. //sql := `SELECT * FROM edb_classify WHERE classify_id in (` + utils.GetOrmInReplace(num) + `) `
  690. //_, err = o.Raw(sql, classifyIdList).QueryRows(&item)
  691. sql := `SELECT * FROM edb_classify WHERE classify_id in ? `
  692. err = global.DbMap[utils.DbNameIndex].Raw(sql, classifyIdList).Find(&item).Error
  693. return
  694. }
  695. // GetEdbClassifyByClassifyTypeAndIsJoinPermission
  696. // @Description: 根据分类类型和是否涉密获取分类列表
  697. // @author: Roc
  698. // @datetime 2024-03-29 10:30:21
  699. // @param classifyType int
  700. // @param isJoinPermission int
  701. // @return items []*EdbClassify
  702. // @return err error
  703. func GetEdbClassifyByClassifyTypeAndIsJoinPermission(classifyType, isJoinPermission int) (items []*EdbClassify, err error) {
  704. //o := orm.NewOrmUsingDB("data")
  705. sql := `SELECT * FROM edb_classify WHERE classify_type = ? AND is_join_permission = ? `
  706. //_, err = o.Raw(sql, classifyType, isJoinPermission).QueryRows(&items)
  707. err = global.DbMap[utils.DbNameIndex].Raw(sql, classifyType, isJoinPermission).Find(&items).Error
  708. return
  709. }
  710. // GetEdbClassifyRootIdsByClassifyIds 获取普通指标的顶级分类列表
  711. func GetEdbClassifyRootIdsByClassifyIds(classifyIds []int) (items []int, err error) {
  712. //o := orm.NewOrmUsingDB("data")
  713. //sql := `SELECT distinct root_id FROM edb_classify WHERE classify_type=0 and classify_id in (` + utils.GetOrmInReplace(len(classifyIds)) + `)`
  714. //_, err = o.Raw(sql, classifyIds).QueryRows(&items)
  715. sql := `SELECT distinct root_id FROM edb_classify WHERE classify_type=0 and classify_id in ?`
  716. err = global.DbMap[utils.DbNameIndex].Raw(sql, classifyIds).Find(&items).Error
  717. return
  718. }