edb_classify.go 22 KB

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