|
|
@@ -0,0 +1,129 @@
|
|
|
+package data_manage
|
|
|
+
|
|
|
+import (
|
|
|
+ "fmt"
|
|
|
+ "hongze/hongze_task/utils"
|
|
|
+ "rdluck_tools/orm"
|
|
|
+ "strconv"
|
|
|
+ "strings"
|
|
|
+ "time"
|
|
|
+)
|
|
|
+
|
|
|
+func AddEdbDataGlBySql(sqlStr string) (err error) {
|
|
|
+ o := orm.NewOrm()
|
|
|
+ o.Using("data")
|
|
|
+ _, err = o.Raw(sqlStr).Exec()
|
|
|
+ return
|
|
|
+}
|
|
|
+
|
|
|
+
|
|
|
+type GlData struct {
|
|
|
+ InputValue string `orm:"column(DATA_VALUE)" description:"日期"`
|
|
|
+ DataTime string `orm:"column(DATA_DATE)" description:"值"`
|
|
|
+}
|
|
|
+
|
|
|
+func GetGlDataByTradeCode(condition string, pars []interface{}) (item []*GlData, err error) {
|
|
|
+ sql := ` SELECT * FROM mb_index_main_data WHERE 1=1 `
|
|
|
+ o := orm.NewOrm()
|
|
|
+ o.Using("gl")
|
|
|
+ if condition != "" {
|
|
|
+ sql += condition
|
|
|
+ }
|
|
|
+ sql += ` ORDER BY PUBLISH_TIME DESC `
|
|
|
+ _, err = o.Raw(sql, pars).QueryRows(&item)
|
|
|
+ return
|
|
|
+}
|
|
|
+
|
|
|
+func GetEdbDataGlByCodeAndDate(edbCode string, startDate string) (count int, err error) {
|
|
|
+ o := orm.NewOrm()
|
|
|
+ o.Using("data")
|
|
|
+ sql := ` SELECT COUNT(1) AS count FROM edb_data_gl WHERE edb_code=? AND data_time=? `
|
|
|
+ err = o.Raw(sql, edbCode, startDate).QueryRow(&count)
|
|
|
+ return
|
|
|
+}
|
|
|
+
|
|
|
+func ModifyEdbDataGl(edbInfoId int64, dataTime, value string) (err error) {
|
|
|
+ o := orm.NewOrm()
|
|
|
+ o.Using("data")
|
|
|
+ sql := ` UPDATE edb_data_gl SET value=?,modify_time=NOW() WHERE edb_info_id=? AND data_time=? `
|
|
|
+ _, err = o.Raw(sql, value, edbInfoId, dataTime).Exec()
|
|
|
+ return
|
|
|
+}
|
|
|
+
|
|
|
+//刷新钢联指标数据
|
|
|
+func RefreshEdbDataByGl(edbInfoId int, edbCode, startDate, endDate string) (err error) {
|
|
|
+ o := orm.NewOrm()
|
|
|
+ o.Using("data")
|
|
|
+ o.Begin()
|
|
|
+ defer func() {
|
|
|
+ if err != nil {
|
|
|
+ o.Rollback()
|
|
|
+ } else {
|
|
|
+ o.Commit()
|
|
|
+ }
|
|
|
+ }()
|
|
|
+
|
|
|
+ if err != nil {
|
|
|
+ return
|
|
|
+ }
|
|
|
+ edbInfoIdStr := strconv.Itoa(edbInfoId)
|
|
|
+ //计算数据
|
|
|
+ var condition string
|
|
|
+ var pars []interface{}
|
|
|
+
|
|
|
+ if edbCode != "" {
|
|
|
+ condition += " AND INDEX_CODE=? "
|
|
|
+ pars = append(pars, edbCode)
|
|
|
+ }
|
|
|
+
|
|
|
+ if startDate != "" {
|
|
|
+ condition += " AND DATA_DATE>=? "
|
|
|
+ pars = append(pars, startDate)
|
|
|
+ }
|
|
|
+
|
|
|
+ if endDate != "" {
|
|
|
+ condition += " AND DATA_DATE<=? "
|
|
|
+ pars = append(pars, endDate)
|
|
|
+ }
|
|
|
+
|
|
|
+ glDataList, err := GetGlDataByTradeCode(condition, pars)
|
|
|
+
|
|
|
+ addSql := ` INSERT INTO edb_data_gl(edb_info_id,edb_code,data_time,value,create_time,modify_time,status,data_timestamp) values `
|
|
|
+ var isAdd bool
|
|
|
+ existMap := make(map[string]string)
|
|
|
+ for _, v := range glDataList {
|
|
|
+ item := v
|
|
|
+ if _, ok := existMap[v.DataTime]; !ok {
|
|
|
+ count, err := GetEdbDataGlByCodeAndDate(edbCode, v.DataTime)
|
|
|
+ if err != nil && err.Error() != utils.ErrNoRow() {
|
|
|
+ return err
|
|
|
+ }
|
|
|
+ if count <= 0 {
|
|
|
+ eDate := item.DataTime
|
|
|
+ sValue := item.InputValue
|
|
|
+ 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, sValue)
|
|
|
+ isAdd = true
|
|
|
+ } else {
|
|
|
+ err = ModifyEdbDataGl(int64(edbInfoId), v.DataTime, v.InputValue)
|
|
|
+ if err != nil {
|
|
|
+ return err
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+ existMap[v.DataTime] = v.InputValue
|
|
|
+ }
|
|
|
+ if isAdd {
|
|
|
+ addSql = strings.TrimRight(addSql, ",")
|
|
|
+ _, err = o.Raw(addSql).Exec()
|
|
|
+ if err != nil {
|
|
|
+ return err
|
|
|
+ }
|
|
|
+ }
|
|
|
+ return
|
|
|
+}
|
