base_from_pb.go 3.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123
  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. func AddEdbDataFromPb(edbCode string, item *services.EdbDataFromPb) (err error) {
  12. var errMsg string
  13. o := orm.NewOrm()
  14. defer func() {
  15. if err != nil {
  16. go utils.SendEmail(utils.APP_NAME_CN+"【"+utils.RunMode+"】"+"失败提醒", " 同花顺数据获取失败:err:"+errMsg, utils.EmailSendToUsers)
  17. }
  18. }()
  19. if len(item.Date) > 0 {
  20. var isAdd bool
  21. dateMap := item.Date
  22. addSql := ` INSERT INTO edb_data_pb(edb_info_id,edb_code,data_time,value,create_time,modify_time,ticker,field,data_timestamp) values `
  23. nowStr := time.Now().Format(utils.FormatDateTime)
  24. for k, v := range dateMap {
  25. timeStr := fmt.Sprintf("%d", v)
  26. v = v / 1000
  27. t := time.Unix(v, 0)
  28. dateTime := t.Format(utils.FormatDate)
  29. val := item.Value[k]
  30. field := item.Field[k]
  31. ticker := item.Ticker[k]
  32. if field == "PX_LAST" {
  33. addSql += "("
  34. addSql += "0," + "'" + edbCode + "'" + "," + "'" + dateTime + "'" + "," + utils.SubFloatToString(val, 20) + "," + "'" + nowStr + "'" +
  35. "," + "'" + nowStr + "'" + "," + "'" + ticker + "'" + "," + "'" + field + "'" + "," + "'" + timeStr + "'"
  36. addSql += "),"
  37. isAdd = true
  38. }
  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 RefreshEdbDataFromPb(edbInfoId int, edbCode, startDate string, item *services.EdbDataFromPb) (err error) {
  53. o := orm.NewOrm()
  54. source := utils.DATA_SOURCE_PB
  55. var condition string
  56. var pars []interface{}
  57. condition += " AND edb_info_id=? "
  58. pars = append(pars, edbInfoId)
  59. condition += " AND data_time>=? "
  60. pars = append(pars, startDate)
  61. existList, err := GetEdbDataByCondition(source, condition, pars)
  62. existMap := make(map[string]*EdbInfoSearchData)
  63. for _, v := range existList {
  64. existMap[v.DataTime] = v
  65. }
  66. edbInfoIdStr := strconv.Itoa(edbInfoId)
  67. addMap := make(map[string]string)
  68. if len(item.Date) > 0 {
  69. dateMap := item.Date
  70. var isAdd bool
  71. addSql := ` INSERT INTO edb_data_pb(edb_info_id,edb_code,data_time,value,create_time,modify_time,ticker,field,data_timestamp) values `
  72. nowStr := time.Now().Format(utils.FormatDateTime)
  73. for k, v := range dateMap {
  74. timeStr := fmt.Sprintf("%d", v)
  75. v = v / 1000
  76. t := time.Unix(v, 0)
  77. dateTime := t.Format(utils.FormatDate)
  78. val := item.Value[k]
  79. field := item.Field[k]
  80. ticker := item.Ticker[k]
  81. saveValue := utils.SubFloatToString(val, 30)
  82. if field == "PX_LAST" {
  83. if findItem, ok := existMap[dateTime]; !ok {
  84. if _, addOk := addMap[dateTime]; !addOk {
  85. addSql += "("
  86. addSql += edbInfoIdStr + "," + "'" + edbCode + "'" + "," + "'" + dateTime + "'" + "," + saveValue + "," + "'" + nowStr + "'" +
  87. "," + "'" + nowStr + "'" + "," + "'" + ticker + "'" + "," + "'" + field + "'" + "," + "'" + timeStr + "'"
  88. addSql += "),"
  89. isAdd = true
  90. addMap[dateTime] = saveValue
  91. }
  92. } else {
  93. if findItem != nil && utils.SubFloatToString(findItem.Value, 30) != saveValue {
  94. err = ModifyEdbDataById(source, findItem.EdbDataId, saveValue)
  95. if err != nil {
  96. return err
  97. }
  98. }
  99. }
  100. }
  101. }
  102. if isAdd {
  103. addSql = strings.TrimRight(addSql, ",")
  104. _, err = o.Raw(addSql).Exec()
  105. if err != nil {
  106. fmt.Println("RefreshAllEdbDataByPb add Err", err.Error())
  107. return
  108. }
  109. }
  110. }
  111. return
  112. }