edb_classify.go 22 KB

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