edb_classify.go 23 KB

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