package models import ( "fmt" "github.com/beego/beego/v2/client/orm" "hongze/hongze_edb_lib/services" "hongze/hongze_edb_lib/utils" "strconv" "strings" "time" ) //新增同花顺指标数据 func AddEdbDataFromThs(edbCode string, item *services.EdbDataFromThs) (err error) { var errMsg string o := orm.NewOrm() defer func() { if err != nil { go utils.SendEmail(utils.APP_NAME_CN+"【"+utils.RunMode+"】"+"失败提醒", " 同花顺数据获取失败:err:"+errMsg, utils.EmailSendToUsers) } }() if len(item.Tables) > 0 { table := item.Tables[0] dataLen := len(table.Time) var isAdd bool addSql := ` INSERT INTO %s(edb_info_id,edb_code,data_time,value,create_time,modify_time,status,data_timestamp) values ` tableName := GetEdbDataTableName(utils.DATA_SOURCE_THS) addSql = fmt.Sprintf(addSql, tableName) for i := 0; i < dataLen; i++ { eDate := table.Time[i] sValue := table.Value[i] dataTime, err := time.Parse(utils.FormatDate, eDate) if err != nil { errMsg = " time.Parse :" + err.Error() return err } timestamp := dataTime.UnixNano() / 1e6 timeStr := fmt.Sprintf("%d", timestamp) addSql += GetAddSql("0", edbCode, eDate, timeStr, utils.SubFloatToString(sValue, 20)) isAdd = true } if isAdd { addSql = strings.TrimRight(addSql, ",") _, err = o.Raw(addSql).Exec() if err != nil { errMsg = " tx.Exec Err :" + err.Error() return } } } return } //刷新同花顺指标数据 func RefreshEdbDataFromThs(edbInfoId int, edbCode, startDate string, item *services.EdbDataFromThs) (err error) { o := orm.NewOrm() source := utils.DATA_SOURCE_THS if len(item.Tables) > 0 { var condition string var pars []interface{} condition += " AND edb_info_id=? " pars = append(pars, edbInfoId) condition += " AND data_time>=? " pars = append(pars, startDate) existList, err := GetEdbDataByCondition(source, condition, pars) existMap := make(map[string]*EdbInfoSearchData) for _, v := range existList { existMap[v.DataTime] = v } table := item.Tables[0] dataLen := len(table.Time) addSql := ` INSERT INTO edb_data_ths(edb_info_id,edb_code,data_time,value,create_time,modify_time,status,data_timestamp) values ` var isAdd bool addMap := make(map[string]string) edbInfoIdStr := strconv.Itoa(edbInfoId) for i := 0; i < dataLen; i++ { eDate := table.Time[i] sValue := table.Value[i] sValueStr := utils.SubFloatToString(sValue, 30) if findItem, ok := existMap[eDate]; !ok { if _, addOk := addMap[eDate]; !addOk { dataTime, err := time.Parse(utils.FormatDate, eDate) if err != nil { return err } timestamp := dataTime.UnixNano() / 1e6 timeStr := fmt.Sprintf("%d", timestamp) addSql += GetAddSql(edbInfoIdStr, edbCode, eDate, timeStr, sValueStr) isAdd = true addMap[eDate] = sValueStr } } else { if findItem != nil && utils.SubFloatToString(findItem.Value, 30) != utils.SubFloatToString(sValue, 30) { err = ModifyEdbDataById(source, findItem.EdbDataId, sValueStr) if err != nil { return err } } } } if isAdd { addSql = strings.TrimRight(addSql, ",") _, err = o.Raw(addSql).Exec() if err != nil { return err } } } return }