predict_edb_conf.go 8.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184
  1. package models
  2. import (
  3. "eta/eta_index_lib/global"
  4. "eta/eta_index_lib/utils"
  5. "gorm.io/gorm"
  6. "time"
  7. )
  8. // AddPredictEdbInfoReq 添加预测指标请求
  9. type AddPredictEdbInfoReq struct {
  10. EdbInfoId int `description:"指标ID"`
  11. ClassifyId int `description:"分类id"`
  12. AdminId int `description:"添加人id"`
  13. AdminName string `description:"添加人名称"`
  14. SourceEdbInfoId int `description:"来源指标id"`
  15. EdbName string `description:"指标名称"`
  16. RuleList []RuleConfig `description:"配置规则列表"`
  17. MaxValue float64 `description:"最大值"`
  18. MinValue float64 `description:"最小值"`
  19. DataDateType string `description:"日期类型,枚举值:交易日、自然日"`
  20. EndDateType int `description:"截止日期类型:0:未来日期,1未来期数"`
  21. }
  22. // RuleConfig 预测规则配置
  23. type RuleConfig struct {
  24. RuleType int `description:"预测规则,1:最新,2:固定值,3:同比,4:同差,5:环比,6:环差,7:N期移动均值,8:N期段线性外推值,9:动态环差"`
  25. Value string `description:"值/计算公式"`
  26. EmptyType int `description:"空值处理类型(0查找前后35天,1不计算,2前值填充,3后值填充,4等于0)"`
  27. MaxEmptyType int `description:"MAX、MIN公式空值处理类型(1、等于0;2、跳过空值)"`
  28. EndDate string `description:"截止日期"`
  29. EndNum int `description:"截止期数"`
  30. EdbInfoIdArr []EdbInfoFromTag `gorm:"-" description:"指标信息"`
  31. }
  32. // AfterFind 在该模型上设置钩子函数,把日期转成正确的string,所以查询函数只能用Find函数,First或者Scan是不会触发该函数的来获取数据
  33. func (m *RuleConfig) AfterFind(db *gorm.DB) (err error) {
  34. m.EndDate = utils.GormDateStrToDateStr(m.EndDate)
  35. return
  36. }
  37. // EditPredictEdbInfoReq 编辑预测指标请求
  38. type EditPredictEdbInfoReq struct {
  39. EdbInfoId int `description:"指标ID"`
  40. ClassifyId int `description:"分类id"`
  41. EdbName string `description:"指标名称"`
  42. RuleList []RuleConfig `gorm:"-" description:"配置规则列表"`
  43. }
  44. type PredictEdbConf struct {
  45. ConfigId int `gorm:"primaryKey;autoIncrement;column:config_id" description:"规则id"`
  46. PredictEdbInfoId int `gorm:"column:predict_edb_info_id" description:"预测指标id"`
  47. SourceEdbInfoId int `description:"来源指标id"`
  48. RuleType int `description:"预测规则,1:最新,2:固定值,3:同比,4:同差,5:环比,6:环差,7:N期移动均值,8:N期段线性外推值"`
  49. FixedValue float64 `description:"固定值"`
  50. Value string `description:"配置的值"`
  51. EmptyType int `description:"空值处理类型(0查找前后35天,1不计算,2前值填充,3后值填充,4等于0)"`
  52. MaxEmptyType int `description:"MAX、MIN公式空值处理类型(1、等于0;2、跳过空值)"`
  53. EndDate time.Time `description:"截止日期"`
  54. EndNum int `description:"截止期数"`
  55. ModifyTime time.Time `description:"修改时间"`
  56. CreateTime time.Time `description:"添加时间"`
  57. }
  58. // PredictEdbConfAndData 预测规则和其对应的动态数据
  59. type PredictEdbConfAndData struct {
  60. ConfigId int `gorm:"column:config_id;primaryKey" description:"规则id"`
  61. PredictEdbInfoId int `gorm:"column:predict_edb_info_id" description:"预测指标id"`
  62. SourceEdbInfoId int `description:"来源指标id"`
  63. RuleType int `description:"预测规则,1:最新,2:固定值,3:同比,4:同差,5:环比,6:环差,7:N期移动均值,8:N期段线性外推值,9:动态环差"`
  64. FixedValue float64 `description:"固定值"`
  65. Value string `description:"配置的值"`
  66. EmptyType int `description:"空值处理类型(0查找前后35天,1不计算,2前值填充,3后值填充,4等于0)"`
  67. MaxEmptyType int `description:"MAX、MIN公式空值处理类型(1、等于0;2、跳过空值)"`
  68. EndDate time.Time `description:"截止日期"`
  69. EndNum int `description:"截止期数"`
  70. ModifyTime time.Time `description:"修改时间"`
  71. CreateTime time.Time `description:"添加时间"`
  72. DataList []*EdbInfoSearchData `gorm:"-" description:"动态数据"`
  73. }
  74. // GetPredictEdbConfById 根据预测指标id获取预测指标配置信息
  75. func GetPredictEdbConfById(edbInfoId int) (item *PredictEdbConf, err error) {
  76. sql := ` SELECT * FROM predict_edb_conf WHERE predict_edb_info_id=? `
  77. err = global.DEFAULT_DB.Raw(sql, edbInfoId).First(&item).Error
  78. return
  79. }
  80. // GetPredictEdbConfCount 根据来源指标id获取被引用的次数
  81. func GetPredictEdbConfCount(sourceEdbInfoId int) (count int, err error) {
  82. sql := ` SELECT COUNT(1) AS count FROM predict_edb_conf WHERE source_edb_info_id=? `
  83. err = global.DEFAULT_DB.Raw(sql, sourceEdbInfoId).Scan(&count).Error
  84. return
  85. }
  86. // GetPredictEdbConfListById 根据预测指标id获取预测指标配置信息列表
  87. func GetPredictEdbConfListById(edbInfoId int) (items []*PredictEdbConf, err error) {
  88. sql := ` SELECT * FROM predict_edb_conf WHERE predict_edb_info_id=? ORDER BY config_id ASC`
  89. err = global.DEFAULT_DB.Raw(sql, edbInfoId).Find(&items).Error
  90. return
  91. }
  92. // GetPredictEdbConfAndDataListById 根据预测指标id获取预测指标配置信息列表
  93. func GetPredictEdbConfAndDataListById(edbInfoId int) (items []*PredictEdbConfAndData, err error) {
  94. sql := ` SELECT * FROM predict_edb_conf WHERE predict_edb_info_id=? ORDER BY config_id ASC`
  95. err = global.DEFAULT_DB.Raw(sql, edbInfoId).Find(&items).Error
  96. return
  97. }
  98. // GetPredictEdbConfListByConfigIdList 根据预测指标id列表获取预测指标配置信息列表
  99. func GetPredictEdbConfListByConfigIdList(configIdList []int) (items []*PredictEdbConf, err error) {
  100. num := len(configIdList)
  101. if num <= 0 {
  102. return
  103. }
  104. sql := ` SELECT * FROM predict_edb_conf WHERE config_id in (` + utils.GetOrmInReplace(num) + `) ORDER BY config_id ASC`
  105. err = global.DEFAULT_DB.Raw(sql, configIdList).Find(&items).Error
  106. return
  107. }
  108. // ModifyPredictEdbInfoMaxAndMinInfoBySourceEdbInfoId 根据来源指标修改预测指标的最新数据信息
  109. func ModifyPredictEdbInfoMaxAndMinInfoBySourceEdbInfoId(sourceEdbInfoId int, item *EdbInfoMaxAndMinInfo) (err error) {
  110. //return
  111. o := global.DEFAULT_DB
  112. var list []*PredictEdbConf
  113. sql := ` SELECT * FROM predict_edb_conf WHERE source_edb_info_id=? `
  114. result := o.Raw(sql, sourceEdbInfoId).Find(&list)
  115. err = result.Error
  116. if err != nil {
  117. return
  118. }
  119. total := result.RowsAffected
  120. if total > 0 {
  121. idList := make([]int, 0)
  122. for _, v := range list {
  123. idList = append(idList, v.PredictEdbInfoId)
  124. }
  125. //sql := ` UPDATE edb_info SET start_date=?,min_value=?,max_value=?,is_update=2,latest_date=?,latest_value=?,modify_time=NOW() WHERE edb_info_id in (` + utils.GetOrmInReplace(int(total)) + `) `
  126. //_, err = o.Raw(sql, item.MinDate, item.MinValue, item.MaxValue, item.MaxDate, item.LatestValue, idList).Exec()
  127. sql := ` UPDATE edb_info SET start_date=?,is_update=2,latest_date=?,latest_value=?,end_value=?,modify_time=NOW() WHERE edb_info_id in (?) `
  128. err = o.Exec(sql, item.MinDate, item.MaxDate, item.LatestValue, item.LatestValue, idList).Error
  129. }
  130. return
  131. }
  132. // GetPredictEdbConfBySourceEdbInfoId 根据来源指标id获取配置
  133. func GetPredictEdbConfBySourceEdbInfoId(sourceEdbInfoId int) (item *PredictEdbConf, err error) {
  134. sql := ` SELECT * FROM predict_edb_conf WHERE source_edb_info_id=? `
  135. err = global.DEFAULT_DB.Raw(sql, sourceEdbInfoId).First(&item).Error
  136. return
  137. }
  138. // AddPredictEdbConf 添加预测指标规则
  139. func AddPredictEdbConf(item *PredictEdbConf) (err error) {
  140. err = global.DEFAULT_DB.Create(item).Error
  141. return
  142. }
  143. // GetPredictEdbInfoAllCalculate 根据基础预测指标id集合 获取 所有的普通指标列表数据
  144. func GetPredictEdbInfoAllCalculate(edbInfoIdList []int) (list []*EdbInfo, err error) {
  145. num := len(edbInfoIdList)
  146. if num <= 0 {
  147. return
  148. }
  149. sql := ` SELECT b.* FROM predict_edb_conf AS a
  150. INNER JOIN edb_info AS b ON a.source_edb_info_id=b.edb_info_id
  151. WHERE a.predict_edb_info_id in (` + utils.GetOrmInReplace(num) + `)
  152. GROUP BY a.source_edb_info_id
  153. ORDER BY a.source_edb_info_id ASC `
  154. err = global.DEFAULT_DB.Raw(sql, edbInfoIdList).Find(&list).Error
  155. return
  156. }