edb_data_manual.go 5.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171
  1. package data_manage
  2. import (
  3. "eta/eta_task/global"
  4. "eta/eta_task/utils"
  5. )
  6. //func AddEdbDataManualBySql(sqlStr string) (err error) {
  7. // o := global.DbMap[utils.DbNameIndex]
  8. // err = o.Exec(sqlStr).Error
  9. // return
  10. //}
  11. //func GetEdbDataManualMaxOrMinDate(edbCode string) (min_date, max_date string, err error) {
  12. // o := global.DbMap[utils.DbNameIndex]
  13. // sql := ` SELECT MIN(data_time) AS min_date,MAX(data_time) AS max_date FROM edb_data_manual WHERE edb_code=? `
  14. // err = o.Raw(sql, edbCode).Row().Scan(&min_date, &max_date)
  15. // return
  16. //}
  17. //type ManualEdbdata struct {
  18. // TradeCode string `orm:"column(TRADE_CODE);pk" gorm:"primaryKey" description:"指标编码"`
  19. // Dt string `orm:"column(DT)" description:"日期"`
  20. // Close float64 `orm:"column(CLOSE)" description:"值"`
  21. // ModifyTime time.Time `orm:"column(modify_time)" description:"修改时间"`
  22. //}
  23. //func GetEdbdataManualByTradeCode(condition string, pars []interface{}) (item []*ManualEdbdata, err error) {
  24. // sql := ` SELECT * FROM edbdata WHERE 1=1 `
  25. // o := global.DbMap[utils.DbNameIndex]
  26. // if condition != "" {
  27. // sql += condition
  28. // }
  29. // sql += ` ORDER BY DT DESC `
  30. // err = o.Raw(sql, pars).Find(&item).Error
  31. // return
  32. //}
  33. //func GetEdbDataManualByCodeAndDate(edbCode string, startDate string) (count int, err error) {
  34. // o := global.DbMap[utils.DbNameIndex]
  35. // sql := ` SELECT COUNT(1) AS count FROM edb_data_manual WHERE edb_code=? AND data_time=? `
  36. // err = o.Raw(sql, edbCode, startDate).Scan(&count).Error
  37. // return
  38. //}
  39. //func ModifyEdbDataManual(edbInfoId int64, dataTime, value string) (err error) {
  40. // o := global.DbMap[utils.DbNameIndex]
  41. // sql := ` UPDATE edb_data_manual SET value=?,modify_time=NOW() WHERE edb_info_id=? AND data_time=? `
  42. // err = o.Exec(sql, value, edbInfoId, dataTime).Error
  43. // return
  44. //}
  45. // 刷新手工指标数据
  46. //func RefreshManual(item *EdbInfoList) (err error) {
  47. // //edbInfoId int, edbCode, startDate, endDate string
  48. // o := global.DbMap[utils.DbNameIndex]
  49. // tx := o.Begin()
  50. // if tx.Error != nil {
  51. // return tx.Error
  52. // }
  53. // defer func() {
  54. // if err != nil {
  55. // tx.Rollback()
  56. // } else {
  57. // tx.Commit()
  58. // }
  59. // }()
  60. //
  61. // edbInfoId := item.EdbInfoId
  62. // edbCode := item.EdbCode
  63. //
  64. // if err != nil {
  65. // return
  66. // }
  67. // edbInfoIdStr := strconv.Itoa(edbInfoId)
  68. // //计算数据
  69. // var condition string
  70. // var pars []interface{}
  71. // if edbCode != "" {
  72. // condition += " AND TRADE_CODE=? "
  73. // pars = append(pars, edbCode)
  74. // }
  75. //
  76. // //获取已存在的手工数据
  77. // manualDataList, err := GetEdbdataManualByTradeCode(condition, pars)
  78. //
  79. // var existCondition string
  80. // var existPars []interface{}
  81. // if edbCode != "" {
  82. // existCondition += " AND edb_code=? "
  83. // existPars = append(existPars, edbCode)
  84. // }
  85. // //获取ETA指标库已存在的所有数据
  86. // existDataList, err := GetEdbDataListAll(existCondition, existPars, utils.DATA_SOURCE_MANUAL, utils.DATA_SUB_SOURCE_EDB, 0)
  87. // if err != nil {
  88. // return err
  89. // }
  90. //
  91. // existMap := make(map[string]*EdbInfoSearchData)
  92. // for _, v := range existDataList {
  93. // existMap[v.DataTime] = v
  94. // }
  95. //
  96. // addSql := ` INSERT INTO edb_data_manual(edb_info_id,edb_code,data_time,value,create_time,modify_time,status,data_timestamp) values `
  97. // var isAdd bool
  98. // manualMap := make(map[string]*ManualEdbdata)
  99. // for _, v := range manualDataList {
  100. // item := v
  101. // eDate := item.Dt
  102. // sValue := utils.SubFloatToString(item.Close, 4)
  103. // if findItem, ok := existMap[v.Dt]; !ok {
  104. // nowStr := time.Now().Format(utils.FormatDateTime)
  105. // dataTime, err := time.Parse(utils.FormatDate, eDate)
  106. // if err != nil {
  107. // return err
  108. // }
  109. // timestamp := dataTime.UnixNano() / 1e6
  110. // timeStr := fmt.Sprintf("%d", timestamp)
  111. // addSql += "("
  112. // addSql += edbInfoIdStr + "," + "'" + edbCode + "'" + "," + "'" + eDate + "'" + "," + sValue + "," + "'" + nowStr + "'" +
  113. // "," + "'" + nowStr + "'" + "," + "1" + "," + "'" + timeStr + "'"
  114. // addSql += "),"
  115. // isAdd = true
  116. // } else {
  117. // if item.Close != findItem.Value {
  118. // err = ModifyEdbDataManual(int64(edbInfoId), eDate, sValue)
  119. // if err != nil {
  120. // return err
  121. // }
  122. // }
  123. // }
  124. // manualMap[v.Dt] = v
  125. // }
  126. //
  127. // for _, v := range existDataList {
  128. // if _, ok := manualMap[v.DataTime]; !ok {
  129. // go DeleteEdbDataByIdAndSource(v.EdbDataId, utils.DATA_SOURCE_MANUAL, utils.DATA_SUB_SOURCE_EDB)
  130. // }
  131. // }
  132. //
  133. // if isAdd {
  134. // addSql = strings.TrimRight(addSql, ",")
  135. // err = tx.Exec(addSql, pars...).Error
  136. // if err != nil {
  137. // return err
  138. // }
  139. // }
  140. //
  141. // maxAndMinItem, err := GetEdbInfoMaxAndMinInfo(utils.DATA_SOURCE_MANUAL, utils.DATA_SUB_SOURCE_EDB, edbCode)
  142. // if err != nil {
  143. // if utils.IsErrNoRow(err) {
  144. // return nil
  145. // } else {
  146. // return err
  147. // }
  148. // } else {
  149. // if maxAndMinItem != nil && maxAndMinItem.MinDate != "" && maxAndMinItem.MaxDate != "" {
  150. // err = ModifyEdbInfoMaxAndMinInfo(edbInfoId, maxAndMinItem)
  151. // if err != nil {
  152. // return err
  153. // }
  154. // }
  155. // }
  156. // return
  157. //}
  158. func AddEdbDataManualBySql(sqlStr string, params []interface{}) (err error) {
  159. o := global.DbMap[utils.DbNameManualIndex]
  160. err = o.Exec(sqlStr, params...).Error
  161. return
  162. }