base_from_ths.go 3.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122
  1. package models
  2. import (
  3. "fmt"
  4. "github.com/beego/beego/v2/client/orm"
  5. "hongze/hongze_edb_lib/services"
  6. "hongze/hongze_edb_lib/utils"
  7. "strconv"
  8. "strings"
  9. "time"
  10. )
  11. //新增同花顺指标数据
  12. func AddEdbDataFromThs(edbCode string, item *services.EdbDataFromThs) (err error) {
  13. var errMsg string
  14. o := orm.NewOrm()
  15. defer func() {
  16. if err != nil {
  17. go utils.SendEmail(utils.APP_NAME_CN+"【"+utils.RunMode+"】"+"失败提醒", " 同花顺数据获取失败:err:"+errMsg, utils.EmailSendToUsers)
  18. }
  19. }()
  20. if len(item.Tables) > 0 {
  21. table := item.Tables[0]
  22. dataLen := len(table.Time)
  23. var isAdd bool
  24. addSql := ` INSERT INTO %s(edb_info_id,edb_code,data_time,value,create_time,modify_time,data_timestamp) values `
  25. tableName := GetEdbDataTableName(utils.DATA_SOURCE_THS)
  26. addSql = fmt.Sprintf(addSql, tableName)
  27. for i := 0; i < dataLen; i++ {
  28. eDate := table.Time[i]
  29. sValue := table.Value[i]
  30. dataTime, err := time.Parse(utils.FormatDate, eDate)
  31. if err != nil {
  32. errMsg = " time.Parse :" + err.Error()
  33. return err
  34. }
  35. timestamp := dataTime.UnixNano() / 1e6
  36. timeStr := fmt.Sprintf("%d", timestamp)
  37. addSql += GetAddSql("0", edbCode, eDate, timeStr, utils.SubFloatToString(sValue, 20))
  38. isAdd = true
  39. }
  40. if isAdd {
  41. addSql = strings.TrimRight(addSql, ",")
  42. _, err = o.Raw(addSql).Exec()
  43. if err != nil {
  44. errMsg = " tx.Exec Err :" + err.Error()
  45. return
  46. }
  47. }
  48. }
  49. return
  50. }
  51. //刷新同花顺指标数据
  52. func RefreshEdbDataFromThs(edbInfoId int, edbCode, startDate string, item *services.EdbDataFromThs) (err error) {
  53. o := orm.NewOrm()
  54. source := utils.DATA_SOURCE_THS
  55. if len(item.Tables) > 0 {
  56. var condition string
  57. var pars []interface{}
  58. condition += " AND edb_info_id=? "
  59. pars = append(pars, edbInfoId)
  60. if startDate!="" {
  61. condition += " AND data_time>=? "
  62. pars = append(pars, startDate)
  63. }
  64. existList, err := GetEdbDataByCondition(source, condition, pars)
  65. if err != nil {
  66. return err
  67. }
  68. existMap := make(map[string]*EdbInfoSearchData)
  69. for _, v := range existList {
  70. existMap[v.DataTime] = v
  71. }
  72. table := item.Tables[0]
  73. dataLen := len(table.Time)
  74. addSql := ` INSERT INTO edb_data_ths(edb_info_id,edb_code,data_time,value,create_time,modify_time,data_timestamp) values `
  75. var isAdd bool
  76. addMap := make(map[string]string)
  77. edbInfoIdStr := strconv.Itoa(edbInfoId)
  78. for i := 0; i < dataLen; i++ {
  79. eDate := table.Time[i]
  80. sValue := table.Value[i]
  81. sValueStr := utils.SubFloatToString(sValue, 30)
  82. if findItem, ok := existMap[eDate]; !ok {
  83. if _, addOk := addMap[eDate]; !addOk {
  84. dataTime, err := time.Parse(utils.FormatDate, eDate)
  85. if err != nil {
  86. return err
  87. }
  88. timestamp := dataTime.UnixNano() / 1e6
  89. timeStr := fmt.Sprintf("%d", timestamp)
  90. addSql += GetAddSql(edbInfoIdStr, edbCode, eDate, timeStr, sValueStr)
  91. isAdd = true
  92. addMap[eDate] = sValueStr
  93. }
  94. } else {
  95. if findItem != nil && utils.SubFloatToString(findItem.Value, 30) != utils.SubFloatToString(sValue, 30) {
  96. err = ModifyEdbDataById(source, findItem.EdbDataId, sValueStr)
  97. if err != nil {
  98. return err
  99. }
  100. }
  101. }
  102. }
  103. if isAdd {
  104. addSql = strings.TrimRight(addSql, ",")
  105. _, err = o.Raw(addSql).Exec()
  106. if err != nil {
  107. return err
  108. }
  109. }
  110. }
  111. return
  112. }