edb_classify.go 5.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169
  1. package models
  2. import (
  3. "errors"
  4. "eta/eta_index_lib/global"
  5. "eta/eta_index_lib/utils"
  6. "strconv"
  7. "time"
  8. )
  9. type EdbClassify struct {
  10. ClassifyId int `gorm:"column:classify_id;type:int(11);primaryKey;not null;"`
  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. }
  24. func AddEdbClassify(item *EdbClassify) (lastId int64, err error) {
  25. err = global.DEFAULT_DB.Create(item).Error
  26. if err != nil {
  27. return
  28. }
  29. lastId = int64(item.ClassifyId)
  30. return
  31. }
  32. // Update 更新分类基础信息
  33. func (edbClassify *EdbClassify) Update(cols []string) (err error) {
  34. err = global.DEFAULT_DB.Model(edbClassify).Select(cols).Updates(edbClassify).Error
  35. return
  36. }
  37. func GetEdbClassifyById(classifyId int) (item *EdbClassify, err error) {
  38. sql := `SELECT * FROM edb_classify WHERE classify_id=? `
  39. err = global.DEFAULT_DB.Raw(sql, classifyId).First(&item).Error
  40. return
  41. }
  42. type EdbClassifySimplify struct {
  43. ClassifyId int `description:"分类id"`
  44. ClassifyName string `description:"分类名称"`
  45. ParentId int
  46. }
  47. // 递归获取所有目录
  48. func GetClassifyAllByClassifyId(classifyId int) (resultList []*EdbClassifySimplify, err error) {
  49. for {
  50. sql := ` SELECT * FROM edb_classify AS a WHERE a.classify_id=? order by sort asc,classify_id asc`
  51. item := new(EdbClassifySimplify)
  52. err = global.DEFAULT_DB.Raw(sql, classifyId).First(&item).Error
  53. if err != nil {
  54. break
  55. }
  56. if item.ClassifyId > 0 {
  57. resultList = append(resultList, item)
  58. classifyId = item.ParentId
  59. } else {
  60. break
  61. }
  62. }
  63. return resultList, err
  64. }
  65. // 判断分类名称是否存在
  66. func GetEdbClassifyCount(classifyName string, parentId int, classifyType uint8) (item *EdbClassify, err error) {
  67. sql := `SELECT * FROM edb_classify WHERE parent_id=? AND classify_name=? AND classify_type = ? `
  68. err = global.DEFAULT_DB.Raw(sql, parentId, classifyName, classifyType).First(&item).Error
  69. return
  70. }
  71. // GetEdbClassifyMaxSort 获取分类下最大的排序数
  72. func GetEdbClassifyMaxSort(parentId int, classifyType uint8) (sort int, err error) {
  73. sql := `SELECT Max(sort) AS sort FROM edb_classify WHERE parent_id=? AND classify_type=? `
  74. err = global.DEFAULT_DB.Raw(sql, parentId, classifyType).Scan(&sort).Error
  75. return
  76. }
  77. type AddEdbClassifyReq struct {
  78. ClassifyName string `description:"分类名称"`
  79. ParentId int `description:"父级id,第一级传0"`
  80. Level int `description:"层级,第一级传0,其余传上一级的层级"`
  81. ClassifyType uint8 `description:"分类类型,0:普通指标分类,1:预测指标分类"`
  82. }
  83. // AddEdbClassify 添加指标分类
  84. func SaveEdbClassify(classifyName string, parentId, level int, classifyType uint8, sysUserId, sysUserName string) (classifyInfo *EdbClassify, err error, errMsg string) {
  85. edbClassify, err := GetEdbClassifyCount(classifyName, parentId, classifyType)
  86. var isAdd bool
  87. if err != nil {
  88. if err.Error() == utils.ErrNoRow() {
  89. isAdd = true
  90. } else {
  91. errMsg = `判断名称是否已存在失败`
  92. return
  93. }
  94. }
  95. if edbClassify != nil && edbClassify.ClassifyId > 0 {
  96. classifyInfo = edbClassify
  97. return edbClassify, nil, ""
  98. } else {
  99. isAdd = true
  100. }
  101. if isAdd {
  102. sysUserIdInt, _ := strconv.Atoi(sysUserId)
  103. rootId := 0
  104. if parentId > 0 {
  105. parentClassify, tErr := GetEdbClassifyById(parentId)
  106. if tErr != nil {
  107. if tErr.Error() == utils.ErrNoRow() {
  108. errMsg = "父级分类不存在"
  109. err = errors.New(errMsg)
  110. return
  111. }
  112. errMsg = "获取失败"
  113. err = errors.New("获取分类信息失败,Err:" + tErr.Error())
  114. return
  115. }
  116. rootId = parentClassify.RootId
  117. }
  118. timestamp := strconv.FormatInt(time.Now().UnixNano(), 10)
  119. classifyInfo = &EdbClassify{
  120. //ClassifyId: 0,
  121. ClassifyType: classifyType,
  122. ClassifyName: classifyName,
  123. ParentId: parentId,
  124. RootId: rootId,
  125. HasData: 0,
  126. CreateTime: time.Now(),
  127. ModifyTime: time.Now(),
  128. SysUserId: sysUserIdInt,
  129. SysUserRealName: sysUserName,
  130. Level: level + 1,
  131. UniqueCode: utils.MD5(utils.DATA_PREFIX + "_" + timestamp),
  132. Sort: GetAddEdbMaxSortByClassifyId(parentId, classifyType),
  133. }
  134. var classifyId int64
  135. classifyId, err = AddEdbClassify(classifyInfo)
  136. if err != nil {
  137. errMsg = "保存分类失败"
  138. return
  139. }
  140. classifyInfo.ClassifyId = int(classifyId)
  141. if parentId == 0 { //一级目录的rootId等于自己本身
  142. classifyInfo.RootId = int(classifyId)
  143. err = classifyInfo.Update([]string{"RootId"})
  144. if err != nil {
  145. errMsg = "更新分类失败"
  146. return
  147. }
  148. }
  149. }
  150. return
  151. }