edb_info_relation.go 5.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136
  1. package models
  2. import (
  3. "eta/eta_index_lib/utils"
  4. "github.com/beego/beego/v2/client/orm"
  5. "time"
  6. )
  7. type EdbInfoRelation struct {
  8. EdbInfoRelationId int `orm:"column(edb_info_relation_id);pk"`
  9. EdbInfoId int `description:"指标id"`
  10. Source int `description:"来源:1:同花顺,2:wind,3:彭博,4:指标运算,5:累计值转月,6:同比值,7:同差值,8:N数值移动平均计算,9:手工指标,10:隆众"`
  11. EdbName string `description:"指标名称"`
  12. EdbCode string `description:"指标编码"`
  13. ReferObjectId int `description:"引用对象ID(图表ID,ETA逻辑ID等)"`
  14. ReferObjectType int `description:"引用对象ID类型(1.图表,2.ETA逻辑)"`
  15. ReferObjectSubType int `description:"引用对象子类"`
  16. CreateTime time.Time `description:"创建时间"`
  17. ModifyTime time.Time `description:"修改时间"`
  18. RelationTime time.Time `description:"引用时间"`
  19. RelationType int `description:"引用类型,0:直接饮用,1间接引用"`
  20. RootEdbInfoId int `description:"间接引用时,关联的直接引用的指标ID"`
  21. ChildEdbInfoId int `description:"间接引用时,计算指标直接关联的指标ID"`
  22. RelationCode string `description:"引用标识"`
  23. ParentRelationId int `description:"间接引用关联的直接引用的ID"`
  24. }
  25. func (e *EdbInfoRelation) TableName() string {
  26. return "edb_info_relation"
  27. }
  28. // GetEdbInfoRelationByChildEdbInfoId 查询引用的指标ID
  29. func GetEdbInfoRelationByChildEdbInfoId(edbInfoId int) (item *EdbInfoRelation, err error) {
  30. o := orm.NewOrm()
  31. msql := ` SELECT * FROM edb_info_relation WHERE child_edb_info_id = ? or edb_info_id=?`
  32. err = o.Raw(msql, edbInfoId, edbInfoId).QueryRow(&item)
  33. return
  34. }
  35. // GetEdbInfoRelationListByChildEdbInfoId 根据间接引用中的的计算指标ID查询引用列表
  36. func GetEdbInfoRelationListByChildEdbInfoId(edbInfoId int) (items []*EdbInfoRelation, err error) {
  37. o := orm.NewOrm()
  38. msql := ` SELECT * FROM edb_info_relation WHERE relation_type=1 AND child_edb_info_id=?`
  39. _, err = o.Raw(msql, edbInfoId).QueryRows(&items)
  40. return
  41. }
  42. // GetEdbInfoRelationListByParentRelationId 根据间接引用中的的父级ID查询
  43. func GetEdbInfoRelationEdbIdsByParentRelationId(relationId, edbInfoId int) (items []int, err error) {
  44. o := orm.NewOrm()
  45. msql := ` SELECT edb_info_id FROM edb_info_relation WHERE parent_relation_id=? AND child_edb_info_id=?`
  46. _, err = o.Raw(msql, relationId, edbInfoId).QueryRows(&items)
  47. return
  48. }
  49. // GetEdbInfoRelationByRelationIds 查询引用的指标ID
  50. func GetEdbInfoRelationByRelationIds(ids []int) (items []*EdbInfoRelation, err error) {
  51. o := orm.NewOrm()
  52. msql := ` SELECT * FROM edb_info_relation WHERE edb_info_relation_id in (` + utils.GetOrmInReplace(len(ids)) + `) `
  53. _, err = o.Raw(msql, ids).QueryRows(&items)
  54. return
  55. }
  56. // UpdateSecondRelationEdbInfoId 更新指标替换后的间接引用记录
  57. func UpdateSecondRelationEdbInfoId(edbRelationIds []int, relationList []*EdbInfoRelation, refreshEdbInfoIds []int, indexCodeList []string) (err error) {
  58. o, err := orm.NewOrm().Begin()
  59. if err != nil {
  60. return
  61. }
  62. defer func() {
  63. if err != nil {
  64. _ = o.Rollback()
  65. return
  66. }
  67. _ = o.Commit()
  68. }()
  69. // 删除相关的间接引用
  70. sql := ` DELETE FROM edb_info_relation WHERE relation_type=1 and parent_relation_id in (` + utils.GetOrmInReplace(len(edbRelationIds)) + `)`
  71. _, err = o.Raw(sql, edbRelationIds).Exec()
  72. if err != nil {
  73. return
  74. }
  75. // 新增间接引用
  76. relationCodesMap := make(map[string]struct{}, 0)
  77. if len(relationList) > 0 {
  78. for _, relation := range relationList {
  79. if relation.RelationType == 1 {
  80. relationCodesMap[relation.RelationCode] = struct{}{}
  81. }
  82. }
  83. _, err = o.InsertMulti(len(relationList), relationList)
  84. if err != nil {
  85. return
  86. }
  87. }
  88. if len(refreshEdbInfoIds) > 0 {
  89. // todo 更新指标的刷新状态
  90. sql = ` UPDATE edb_info SET no_update = 0, set_update_time=? WHERE edb_info_id IN (` + utils.GetOrmInReplace(len(refreshEdbInfoIds)) + `) AND no_update = 1`
  91. _, err = o.Raw(sql, time.Now(), refreshEdbInfoIds).Exec()
  92. if err != nil {
  93. return
  94. }
  95. }
  96. //更新数据源上海钢联指标
  97. if len(indexCodeList) > 0 {
  98. // 更改数据源的更新状态
  99. sql = ` UPDATE base_from_mysteel_chemical_index SET is_stop = 0 WHERE index_code IN (` + utils.GetOrmInReplace(len(indexCodeList)) + `) and is_stop=1`
  100. _, err = o.Raw(sql, indexCodeList).Exec()
  101. if err != nil {
  102. return
  103. }
  104. }
  105. if len(relationList) > 0 {
  106. // 更新间接引用指标的关联ID
  107. relationCodes := make([]string, 0)
  108. for relationCode := range relationCodesMap {
  109. relationCodes = append(relationCodes, relationCode)
  110. }
  111. if len(relationCodes) > 0 {
  112. sql := ` UPDATE edb_info_relation e1
  113. JOIN edb_info_relation e2 ON e1.relation_code = e2.relation_code
  114. SET e1.parent_relation_id = e2.edb_info_relation_id
  115. WHERE
  116. e1.relation_type = 1
  117. AND e2.relation_type = 0 AND e1.parent_relation_id !=e2.edb_info_relation_id AND e1.relation_code in (` + utils.GetOrmInReplace(len(relationCodes)) + `)`
  118. _, err = o.Raw(sql, relationCodes).Exec()
  119. if err != nil {
  120. return
  121. }
  122. }
  123. }
  124. return
  125. }