|
@@ -0,0 +1,158 @@
|
|
|
+package models
|
|
|
+
|
|
|
+import (
|
|
|
+ "eta/eta_index_lib/utils"
|
|
|
+ "fmt"
|
|
|
+ "github.com/beego/beego/v2/client/orm"
|
|
|
+ "strconv"
|
|
|
+ "strings"
|
|
|
+ "time"
|
|
|
+)
|
|
|
+
|
|
|
+var windWsd = "wsd"
|
|
|
+
|
|
|
+func AddEdbDataFromWindWsd(securitiesCode string, item map[string]map[string]interface{}) (err error) {
|
|
|
+ o := orm.NewOrm()
|
|
|
+
|
|
|
+ var isAdd bool
|
|
|
+ addSql := ` INSERT INTO edb_data_wind_wsd(edb_info_id,edb_code,data_time,value,create_time,modify_time,data_timestamp) values `
|
|
|
+
|
|
|
+ for wk, wv := range item {
|
|
|
+ for vk, vv := range wv {
|
|
|
+ var indexCode string
|
|
|
+ wkInt, err := strconv.ParseInt(wk, 10, 64)
|
|
|
+ if err != nil {
|
|
|
+ return err
|
|
|
+ }
|
|
|
+
|
|
|
+ vk = strings.ToLower(vk)
|
|
|
+ indexCode = windWsd + securitiesCode + vk
|
|
|
+
|
|
|
+ timeStr := wk
|
|
|
+ wkInt = wkInt / 1000
|
|
|
+ t := time.Unix(wkInt, 0)
|
|
|
+ dateTime := t.Format(utils.FormatDate)
|
|
|
+
|
|
|
+ var saveVal string
|
|
|
+ switch vt := vv.(type) {
|
|
|
+ case int:
|
|
|
+ saveVal = strconv.Itoa(vt)
|
|
|
+ case float64:
|
|
|
+ saveVal = utils.SubFloatToFloatStr(vt, 20)
|
|
|
+ case string:
|
|
|
+ saveVal = vt
|
|
|
+ }
|
|
|
+ addSql += GetAddSql("0", indexCode, dateTime, timeStr, saveVal)
|
|
|
+ isAdd = true
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ if isAdd {
|
|
|
+ addSql = strings.TrimRight(addSql, ",")
|
|
|
+ _, err = o.Raw(addSql).Exec()
|
|
|
+ if err != nil {
|
|
|
+ return
|
|
|
+ }
|
|
|
+ }
|
|
|
+ return
|
|
|
+}
|
|
|
+
|
|
|
+
|
|
|
+func RefreshEdbDataFromWindWsd(edbInfoId int, edbCode, startDate string, item *EdbDataFromWind) (err error) {
|
|
|
+ o := orm.NewOrm()
|
|
|
+ source := utils.DATA_SOURCE_WIND
|
|
|
+
|
|
|
+
|
|
|
+ var realDataMaxDate, edbDataInsertConfigDate time.Time
|
|
|
+ var edbDataInsertConfig *EdbDataInsertConfig
|
|
|
+ var isFindConfigDateRealData bool
|
|
|
+ {
|
|
|
+ edbDataInsertConfig, err = GetEdbDataInsertConfigByEdbId(edbInfoId)
|
|
|
+ if err != nil && err.Error() != utils.ErrNoRow() {
|
|
|
+ return
|
|
|
+ }
|
|
|
+ if edbDataInsertConfig != nil {
|
|
|
+ edbDataInsertConfigDate = edbDataInsertConfig.Date
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ var condition string
|
|
|
+ var pars []interface{}
|
|
|
+
|
|
|
+ condition += " AND edb_info_id=? "
|
|
|
+ pars = append(pars, edbInfoId)
|
|
|
+
|
|
|
+ var startDateTime time.Time
|
|
|
+ if startDate != "" {
|
|
|
+ condition += " AND data_time>=? "
|
|
|
+ pars = append(pars, startDate)
|
|
|
+ startDateTime, _ = time.ParseInLocation(utils.FormatDate, startDate, time.Local)
|
|
|
+ }
|
|
|
+
|
|
|
+ existList, err := GetEdbDataByCondition(source, condition, pars)
|
|
|
+ existMap := make(map[string]*EdbInfoSearchData)
|
|
|
+ for _, v := range existList {
|
|
|
+ existMap[v.DataTime] = v
|
|
|
+ }
|
|
|
+ addSql := ` INSERT INTO edb_data_wind(edb_info_id,edb_code,data_time,value,create_time,modify_time,data_timestamp) values `
|
|
|
+ var isAdd bool
|
|
|
+ addMap := make(map[string]string)
|
|
|
+ edbInfoIdStr := strconv.Itoa(edbInfoId)
|
|
|
+ for k, v := range item.Dt {
|
|
|
+ timeStr := fmt.Sprintf("%d", v)
|
|
|
+ v = v / 1000
|
|
|
+ t := time.Unix(v, 0)
|
|
|
+ dateTime := t.Format(utils.FormatDate)
|
|
|
+ t, _ = time.ParseInLocation(utils.FormatDate, dateTime, time.Local)
|
|
|
+
|
|
|
+ val := item.Close[k]
|
|
|
+ saveVal := utils.SubFloatToString(val, 30)
|
|
|
+
|
|
|
+
|
|
|
+ if !startDateTime.IsZero() && t.Before(startDateTime) {
|
|
|
+ tmpItem, tmpErr := GetEdbDataByDate(source, edbCode, dateTime)
|
|
|
+ if tmpErr == nil && tmpItem != nil {
|
|
|
+ existMap[tmpItem.DataTime] = tmpItem
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ if findItem, ok := existMap[dateTime]; !ok {
|
|
|
+ if _, existOk := addMap[dateTime]; !existOk {
|
|
|
+ isAdd = true
|
|
|
+ addSql += GetAddSql(edbInfoIdStr, edbCode, dateTime, timeStr, saveVal)
|
|
|
+ addMap[dateTime] = saveVal
|
|
|
+ }
|
|
|
+ } else {
|
|
|
+ if findItem != nil && utils.SubFloatToString(findItem.Value, 30) != saveVal {
|
|
|
+ err = ModifyEdbDataById(source, findItem.EdbDataId, saveVal)
|
|
|
+ if err != nil {
|
|
|
+ return err
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ {
|
|
|
+ if realDataMaxDate.IsZero() || t.After(realDataMaxDate) {
|
|
|
+ realDataMaxDate = t
|
|
|
+ }
|
|
|
+ if edbDataInsertConfigDate.IsZero() || t.Equal(edbDataInsertConfigDate) {
|
|
|
+ isFindConfigDateRealData = true
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ HandleConfigInsertEdbData(realDataMaxDate, edbDataInsertConfig, edbInfoId, source, existMap, isFindConfigDateRealData)
|
|
|
+
|
|
|
+ if isAdd {
|
|
|
+ addSql = strings.TrimRight(addSql, ",")
|
|
|
+ _, err = o.Raw(addSql).Exec()
|
|
|
+ if err != nil {
|
|
|
+ fmt.Println("RefreshEdbDataFromWind add Err", err.Error())
|
|
|
+ return
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ return
|
|
|
+}
|