123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131 |
- package models
- import (
- "eta_gn/eta_index_lib/global"
- "eta_gn/eta_index_lib/models/supply_analysis"
- "eta_gn/eta_index_lib/utils"
- "fmt"
- "strconv"
- "strings"
- "time"
- )
- type StockPlantEdbdata struct {
- VarietyEdbId string `gorm:"column:variety_edb_id;primaryKey" description:"指标编码"`
- DataTime string `gorm:"column:data_time" description:"日期"`
- Value string `gorm:"column:value" description:"值"`
- ModifyTime time.Time `gorm:"column:modify_time" description:"修改时间"`
- }
- func (m *StockPlantEdbdata) TableName() string {
- return "base_from_stock_plant_data"
- }
- func GetStockPlantEdbDataByCondition(condition string, pars []interface{}) (item []*StockPlantEdbdata, err error) {
- sql := ` SELECT * FROM base_from_stock_plant_data WHERE 1=1 `
- if condition != "" {
- sql += condition
- }
- sql += ` ORDER BY data_time DESC `
- err = global.DEFAULT_DmSQL.Raw(sql, pars...).Find(&item).Error
- return
- }
- // RefreshEdbDataFromStockPlant 刷新存量装置数据
- func RefreshEdbDataFromStockPlant(edbInfoId int, edbCode, startDate string) (err error) {
- source := utils.DATA_SOURCE_STOCK_PLANT
- subSource := utils.DATA_SUB_SOURCE_EDB
- if err != nil {
- return
- }
- edbInfoIdStr := strconv.Itoa(edbInfoId)
- //计算数据
- var condition string
- var pars []interface{}
- // 获取供应分析的指标信息
- varietyEdbInfo, err := supply_analysis.GetVarietyEdbInfoByEdbInfoId(edbInfoId)
- if err != nil {
- return
- }
- // 获取基础库的指标数据
- condition += " AND variety_edb_id=? "
- pars = append(pars, varietyEdbInfo.VarietyEdbId)
- if startDate != "" {
- condition += " AND data_time >=? "
- pars = append(pars, startDate)
- } else {
- condition += " AND data_time != ? "
- pars = append(pars, `0000-00-00`)
- }
- baseDataList, err := GetStockPlantEdbDataByCondition(condition, pars)
- var existCondition string
- var existPars []interface{}
- existCondition += " AND edb_info_id=? "
- existPars = append(existPars, edbInfoId)
- if startDate != "" {
- existCondition += " AND data_time>=? "
- existPars = append(existPars, startDate)
- }
- existList, err := GetEdbDataByCondition(source, subSource, existCondition, existPars)
- if err != nil {
- return err
- }
- existMap := make(map[string]*EdbInfoSearchData)
- for _, v := range existList {
- existMap[v.DataTime] = v
- }
- addSql := ` INSERT INTO edb_data_stock_plant (edb_info_id,edb_code,data_time,value,create_time,modify_time,data_timestamp) values `
- var isAdd bool
- manualMap := make(map[string]*StockPlantEdbdata)
- for _, v := range baseDataList {
- item := v
- if findItem, ok := existMap[v.DataTime]; !ok {
- eDate := item.DataTime
- sValue := item.Value
- 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 {
- if findItem != nil && utils.SubFloatToString(findItem.Value, 4) != item.Value {
- err = ModifyEdbDataById(source, subSource, findItem.EdbDataId, item.Value)
- if err != nil {
- return err
- }
- }
- }
- manualMap[v.DataTime] = v
- }
- // 删除不存在的日期数据
- for _, v := range existList {
- if _, ok := manualMap[v.DataTime]; !ok {
- go DeleteEdbDataById(source, subSource, v.EdbDataId)
- }
- }
- // 将数据入库
- if isAdd {
- addSql = strings.TrimRight(addSql, ",")
- err = global.DEFAULT_DmSQL.Exec(addSql).Error
- if err != nil {
- fmt.Println("RefreshAllEdbDataByStockPlant add Err", err.Error())
- return
- }
- }
- return
- }
|