|
@@ -3,6 +3,7 @@ package models
|
|
|
import (
|
|
|
"encoding/json"
|
|
|
"errors"
|
|
|
+ "eta/eta_index_lib/global"
|
|
|
"eta/eta_index_lib/models/mgo"
|
|
|
"eta/eta_index_lib/utils"
|
|
|
"fmt"
|
|
@@ -17,7 +18,8 @@ import (
|
|
|
)
|
|
|
|
|
|
type EdbInfo struct {
|
|
|
- EdbInfoId int `orm:"column(edb_info_id);pk"`
|
|
|
+ //EdbInfoId int `orm:"column(edb_info_id);pk"`
|
|
|
+ EdbInfoId int `gorm:"column:edb_info_id;primaryKey"`
|
|
|
EdbInfoType int `description:"指标类型,0:普通指标,1:预测指标"`
|
|
|
SourceName string `description:"来源名称"`
|
|
|
Source int `description:"来源id"`
|
|
@@ -70,21 +72,28 @@ type EdbInfo struct {
|
|
|
}
|
|
|
|
|
|
func (e *EdbInfo) Add() (err error) {
|
|
|
- o := orm.NewOrm()
|
|
|
- _, err = o.Insert(e)
|
|
|
+ //o := orm.NewOrm()
|
|
|
+ //_, err = o.Insert(e)
|
|
|
+ err = global.DEFAULT_DB.Create(&e).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
|
// AddEdbInfo 添加指标
|
|
|
func AddEdbInfo(item *EdbInfo) (lastId int64, err error) {
|
|
|
- o := orm.NewOrm()
|
|
|
- lastId, err = o.Insert(item)
|
|
|
+ //o := orm.NewOrm()
|
|
|
+ //lastId, err = o.Insert(item)
|
|
|
+ err = global.DEFAULT_DB.Create(&item).Error
|
|
|
+ if err != nil {
|
|
|
+ return
|
|
|
+ }
|
|
|
+ lastId = int64(item.EdbInfoId)
|
|
|
return
|
|
|
}
|
|
|
|
|
|
// EdbInfoList 指标数据列表
|
|
|
type EdbInfoList struct {
|
|
|
- EdbInfoId int `orm:"column(edb_info_id);pk"`
|
|
|
+ EdbInfoId int `gorm:"column:edb_info_id;primaryKey"`
|
|
|
+ //EdbInfoId int `orm:"column(edb_info_id);pk"`
|
|
|
EdbInfoType int `description:"指标类型,0:普通指标,1:预测指标"`
|
|
|
SourceName string `description:"来源名称"`
|
|
|
Source int `description:"来源id"`
|
|
@@ -137,33 +146,37 @@ type EdbClassifyItemsButton struct {
|
|
|
|
|
|
// GetEdbInfoByName 根据指标名称获取所有的指标数据列表
|
|
|
func GetEdbInfoByName(edbName string) (items []*EdbInfoList, err error) {
|
|
|
- o := orm.NewOrm()
|
|
|
+ //o := orm.NewOrm()
|
|
|
sql := ` SELECT * FROM edb_info WHERE edb_name=? `
|
|
|
- _, err = o.Raw(sql, edbName).QueryRows(&items)
|
|
|
+ //_, err = o.Raw(sql, edbName).QueryRows(&items)
|
|
|
+ err = global.DEFAULT_DB.Raw(sql, edbName).Find(&items).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
|
// ModifyEdbInfoNameSource 根据来源修改指标名称
|
|
|
func ModifyEdbInfoNameSource(edbNameSource string, edbInfoId int) (err error) {
|
|
|
- o := orm.NewOrm()
|
|
|
+ //o := orm.NewOrm()
|
|
|
sql := ` UPDATE edb_info SET edb_name_source=? WHERE edb_info_id = ? `
|
|
|
- _, err = o.Raw(sql, edbNameSource, edbInfoId).Exec()
|
|
|
+ //_, err = o.Raw(sql, edbNameSource, edbInfoId).Exec()
|
|
|
+ err = global.DEFAULT_DB.Exec(sql, edbNameSource, edbInfoId).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
|
// GetEdbInfoById 根据指标id获取指标信息
|
|
|
func GetEdbInfoById(edbInfoId int) (item *EdbInfo, err error) {
|
|
|
- o := orm.NewOrm()
|
|
|
+ //o := orm.NewOrm()
|
|
|
sql := ` SELECT * FROM edb_info WHERE edb_info_id=? `
|
|
|
- err = o.Raw(sql, edbInfoId).QueryRow(&item)
|
|
|
+ //err = o.Raw(sql, edbInfoId).QueryRow(&item)
|
|
|
+ err = global.DEFAULT_DB.Raw(sql, edbInfoId).First(&item).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
|
// GetEdbInfoByUniqueCode 根据指标唯一编码获取指标信息
|
|
|
func GetEdbInfoByUniqueCode(uniqueCode string) (item *EdbInfo, err error) {
|
|
|
- o := orm.NewOrm()
|
|
|
+ //o := orm.NewOrm()
|
|
|
sql := ` SELECT * FROM edb_info WHERE unique_code=? `
|
|
|
- err = o.Raw(sql, uniqueCode).QueryRow(&item)
|
|
|
+ //err = o.Raw(sql, uniqueCode).QueryRow(&item)
|
|
|
+ err = global.DEFAULT_DB.Raw(sql, uniqueCode).First(&item).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
@@ -173,9 +186,11 @@ func GetEdbInfoByIdList(edbInfoIdList []int) (items []*EdbInfo, err error) {
|
|
|
if num <= 0 {
|
|
|
return
|
|
|
}
|
|
|
- o := orm.NewOrm()
|
|
|
- sql := ` SELECT * FROM edb_info WHERE edb_info_id in (` + utils.GetOrmInReplace(num) + `) `
|
|
|
- _, err = o.Raw(sql, edbInfoIdList).QueryRows(&items)
|
|
|
+ //o := orm.NewOrm()
|
|
|
+ //sql := ` SELECT * FROM edb_info WHERE edb_info_id in (` + utils.GetOrmInReplace(num) + `) `
|
|
|
+ //_, err = o.Raw(sql, edbInfoIdList).QueryRows(&items)
|
|
|
+ sql := ` SELECT * FROM edb_info WHERE edb_info_id in ? `
|
|
|
+ err = global.DEFAULT_DB.Raw(sql, edbInfoIdList).Find(&items).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
@@ -184,16 +199,19 @@ func GetEdbInfoListByIds(edbInfoIdList []int) (items []*EdbInfoList, err error)
|
|
|
if num <= 0 {
|
|
|
return
|
|
|
}
|
|
|
- o := orm.NewOrm()
|
|
|
- sql := ` SELECT * FROM edb_info WHERE edb_info_id in (` + utils.GetOrmInReplace(num) + `) `
|
|
|
- _, err = o.Raw(sql, edbInfoIdList).QueryRows(&items)
|
|
|
+ //o := orm.NewOrm()
|
|
|
+ //sql := ` SELECT * FROM edb_info WHERE edb_info_id in (` + utils.GetOrmInReplace(num) + `) `
|
|
|
+ //_, err = o.Raw(sql, edbInfoIdList).QueryRows(&items)
|
|
|
+ sql := ` SELECT * FROM edb_info WHERE edb_info_id in ? `
|
|
|
+ err = global.DEFAULT_DB.Raw(sql, edbInfoIdList).Find(&items).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
|
// Update 更新EdbInfo信息
|
|
|
func (edbInfo *EdbInfo) Update(cols []string) (err error) {
|
|
|
- o := orm.NewOrm()
|
|
|
- _, err = o.Update(edbInfo, cols...)
|
|
|
+ //o := orm.NewOrm()
|
|
|
+ //_, err = o.Update(edbInfo, cols...)
|
|
|
+ err = global.DEFAULT_DB.Model(&edbInfo).Select(cols).Updates(&edbInfo).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
@@ -247,8 +265,8 @@ func GetEdbDataListAllByMysql(source, subSource int, findEdbDataListAllCond Find
|
|
|
if findEdbDataListAllCond.EdbInfoId <= 0 {
|
|
|
return
|
|
|
}
|
|
|
- to := orm.NewOrm()
|
|
|
-
|
|
|
+ //to := orm.NewOrm()
|
|
|
+ to := global.DEFAULT_DB.Begin()
|
|
|
sql := ``
|
|
|
tableName := GetEdbDataTableName(source, subSource)
|
|
|
sql = ` SELECT * FROM %s WHERE 1=1 `
|
|
@@ -278,7 +296,8 @@ func GetEdbDataListAllByMysql(source, subSource int, findEdbDataListAllCond Find
|
|
|
} else {
|
|
|
sql += ` ORDER BY data_time DESC `
|
|
|
}
|
|
|
- _, err = to.Raw(sql, pars).QueryRows(&item)
|
|
|
+ //_, err = to.Raw(sql, pars).QueryRows(&item)
|
|
|
+ err = to.Raw(sql, pars...).Find(&item).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
@@ -346,6 +365,7 @@ func GetEdbDataListAllByMysqlTo(to *gorm.DB, source, subSource int, findEdbDataL
|
|
|
sql += ` ORDER BY data_time DESC `
|
|
|
}
|
|
|
//_, err = to.Raw(sql, pars).QueryRows(&item)
|
|
|
+ //err = to.Raw(sql, pars...).Find(&item).Error
|
|
|
err = to.Raw(sql, pars...).Find(&item).Error
|
|
|
return
|
|
|
}
|
|
@@ -417,17 +437,18 @@ type EdbInfoMaxAndMinInfo struct {
|
|
|
|
|
|
// GetEdbInfoMaxAndMinInfo 获取指标的最新数据记录信息
|
|
|
func GetEdbInfoMaxAndMinInfo(source, subSource int, edbCode string) (item *EdbInfoMaxAndMinInfo, err error) {
|
|
|
- o := orm.NewOrm()
|
|
|
+ //o := orm.NewOrm()
|
|
|
sql := ``
|
|
|
tableName := GetEdbDataTableName(source, subSource)
|
|
|
sql = ` SELECT MIN(data_time) AS min_date,MAX(data_time) AS max_date,MIN(value) AS min_value,MAX(value) AS max_value FROM %s WHERE edb_code=? `
|
|
|
sql = fmt.Sprintf(sql, tableName)
|
|
|
- err = o.Raw(sql, edbCode).QueryRow(&item)
|
|
|
-
|
|
|
+ //err = o.Raw(sql, edbCode).QueryRow(&item)
|
|
|
+ err = global.DEFAULT_DB.Raw(sql, edbCode).First(&item).Error
|
|
|
var latest_value float64
|
|
|
sql = ` SELECT value AS latest_value FROM %s WHERE edb_code=? ORDER BY data_time DESC LIMIT 1 `
|
|
|
sql = fmt.Sprintf(sql, tableName)
|
|
|
- err = o.Raw(sql, edbCode).QueryRow(&latest_value)
|
|
|
+ //err = o.Raw(sql, edbCode).QueryRow(&latest_value)
|
|
|
+ err = global.DEFAULT_DB.Raw(sql, edbCode).Scan(&latest_value).Error
|
|
|
item.LatestValue = latest_value
|
|
|
return
|
|
|
}
|
|
@@ -438,36 +459,40 @@ type EdbIndoDataUpdateTime struct {
|
|
|
|
|
|
// GetEdbDataUpdateTimeByModify 获取指标的刷新时间获取最早的一条记录
|
|
|
func GetEdbDataUpdateTimeByModify(source, subSource int, edbCode string, dataUpdateTime string) (item *EdbIndoDataUpdateTime, err error) {
|
|
|
- o := orm.NewOrm()
|
|
|
+ //o := orm.NewOrm()
|
|
|
sql := ``
|
|
|
tableName := GetEdbDataTableName(source, subSource)
|
|
|
sql = ` SELECT MIN(data_time) AS min_date FROM %s WHERE edb_code=? and modify_time>=?`
|
|
|
sql = fmt.Sprintf(sql, tableName)
|
|
|
- err = o.Raw(sql, edbCode, dataUpdateTime).QueryRow(&item)
|
|
|
+ //err = o.Raw(sql, edbCode, dataUpdateTime).QueryRow(&item)
|
|
|
+ err = global.DEFAULT_DB.Raw(sql, edbCode, dataUpdateTime).First(&item).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
|
// ModifyEdbInfoMaxAndMinInfo 修改指标的最新数据信息
|
|
|
func ModifyEdbInfoMaxAndMinInfo(edbInfoId int, item *EdbInfoMaxAndMinInfo) (err error) {
|
|
|
- o := orm.NewOrm()
|
|
|
+ //o := orm.NewOrm()
|
|
|
sql := ` UPDATE edb_info SET start_date=?,end_date=?,min_value=?,max_value=?,is_update=2,latest_date=?,latest_value=?, end_value = ?,modify_time=NOW() WHERE edb_info_id=? `
|
|
|
- _, err = o.Raw(sql, item.MinDate, item.MaxDate, item.MinValue, item.MaxValue, item.MaxDate, item.LatestValue, item.LatestValue, edbInfoId).Exec()
|
|
|
+ //_, err = o.Raw(sql, item.MinDate, item.MaxDate, item.MinValue, item.MaxValue, item.MaxDate, item.LatestValue, item.LatestValue, edbInfoId).Exec()
|
|
|
+ err = global.DEFAULT_DB.Exec(sql, item.MinDate, item.MaxDate, item.MinValue, item.MaxValue, item.MaxDate, item.LatestValue, item.LatestValue, edbInfoId).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
|
// ModifyEdbTimeAndLastInfo 修改指标的最新数据信息(除上下限)
|
|
|
func ModifyEdbTimeAndLastInfo(edbInfoId int, item *EdbInfoMaxAndMinInfo) (err error) {
|
|
|
- o := orm.NewOrm()
|
|
|
+ //o := orm.NewOrm()
|
|
|
sql := `UPDATE edb_info SET start_date = ?, end_date = ?, is_update = 2,latest_date = ?, latest_value = ?, end_value = ?, modify_time = NOW() WHERE edb_info_id = ?`
|
|
|
- _, err = o.Raw(sql, item.MinDate, item.MaxDate, item.MaxDate, item.LatestValue, item.LatestValue, edbInfoId).Exec()
|
|
|
+ //_, err = o.Raw(sql, item.MinDate, item.MaxDate, item.MaxDate, item.LatestValue, item.LatestValue, edbInfoId).Exec()
|
|
|
+ err = global.DEFAULT_DB.Exec(sql, item.MinDate, item.MaxDate, item.MaxDate, item.LatestValue, item.LatestValue, edbInfoId).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
|
// ModifyEdbDataUpdateTime 修改指标刷新,本次数据刷新的最早日期
|
|
|
func ModifyEdbDataUpdateTime(edbInfoId int, dataUpdateTime, erDataUpdateDate string) (err error) {
|
|
|
- o := orm.NewOrm()
|
|
|
+ //o := orm.NewOrm()
|
|
|
sql := `UPDATE edb_info SET data_update_time = ?, er_data_update_date = ?, modify_time = NOW() WHERE edb_info_id = ?`
|
|
|
- _, err = o.Raw(sql, dataUpdateTime, erDataUpdateDate, edbInfoId).Exec()
|
|
|
+ //_, err = o.Raw(sql, dataUpdateTime, erDataUpdateDate, edbInfoId).Exec()
|
|
|
+ err = global.DEFAULT_DB.Exec(sql, dataUpdateTime, erDataUpdateDate, edbInfoId).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
@@ -525,18 +550,19 @@ func GetLteZeroEdbDataCountByMongo(source, subSource, edbInfoId int) (count int,
|
|
|
// @return count int
|
|
|
// @return err error
|
|
|
func GetLteZeroEdbDataCountByMysql(source, subSource, edbInfoId int) (count int, err error) {
|
|
|
- o := orm.NewOrm()
|
|
|
+ //o := orm.NewOrm()
|
|
|
sql := ``
|
|
|
tableName := GetEdbDataTableName(source, subSource)
|
|
|
sql = ` SELECT COUNT(1) AS count FROM %s WHERE 1=1 AND edb_info_id =? AND value <=0 `
|
|
|
sql = fmt.Sprintf(sql, tableName)
|
|
|
- err = o.Raw(sql, edbInfoId).QueryRow(&count)
|
|
|
+ //err = o.Raw(sql, edbInfoId).QueryRow(&count)
|
|
|
+ err = global.DEFAULT_DB.Raw(sql, edbInfoId).Scan(&count).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
|
// GetLastEdbData 获取最近的一条指标数据
|
|
|
func GetLastEdbData(condition string, pars []interface{}, source, subSource int) (item *EdbInfoSearchData, err error) {
|
|
|
- o := orm.NewOrm()
|
|
|
+ //o := orm.NewOrm()
|
|
|
sql := ``
|
|
|
tableName := GetEdbDataTableName(source, subSource)
|
|
|
sql = ` SELECT * FROM %s WHERE 1=1 `
|
|
@@ -546,16 +572,17 @@ func GetLastEdbData(condition string, pars []interface{}, source, subSource int)
|
|
|
sql += condition
|
|
|
}
|
|
|
sql += ` ORDER BY data_time DESC `
|
|
|
- err = o.Raw(sql, pars).QueryRow(&item)
|
|
|
+ //err = o.Raw(sql, pars).QueryRow(&item)
|
|
|
+ err = global.DEFAULT_DB.Raw(sql, pars...).First(&item).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
|
// GetEdbInfoByEdbCode 根据指标code获取指标信息
|
|
|
func GetEdbInfoByEdbCode(source int, edbCode string) (item *EdbInfo, err error) {
|
|
|
- o := orm.NewOrm()
|
|
|
+ //o := orm.NewOrm()
|
|
|
sql := ` SELECT * FROM edb_info WHERE source=? AND edb_code=? `
|
|
|
- err = o.Raw(sql, source, edbCode).QueryRow(&item)
|
|
|
-
|
|
|
+ //err = o.Raw(sql, source, edbCode).QueryRow(&item)
|
|
|
+ err = global.DEFAULT_DB.Raw(sql, source, edbCode).First(&item).Error
|
|
|
if errors.Is(err, orm.ErrNoRows) {
|
|
|
err = nil
|
|
|
}
|
|
@@ -565,33 +592,36 @@ func GetEdbInfoByEdbCode(source int, edbCode string) (item *EdbInfo, err error)
|
|
|
|
|
|
// GetEdbInfoOnlyByEdbCode 仅根据指标code获取指标信息
|
|
|
func GetEdbInfoOnlyByEdbCode(edbCode string) (item *EdbInfo, err error) {
|
|
|
- o := orm.NewOrm()
|
|
|
+ //o := orm.NewOrm()
|
|
|
sql := ` SELECT * FROM edb_info WHERE edb_code=? `
|
|
|
- err = o.Raw(sql, edbCode).QueryRow(&item)
|
|
|
+ //err = o.Raw(sql, edbCode).QueryRow(&item)
|
|
|
+ err = global.DEFAULT_DB.Raw(sql, edbCode).First(&item).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
|
// GetEdbInfoCalculateListByCondition 获取指标关系列表
|
|
|
func GetEdbInfoCalculateListByCondition(condition string, pars []interface{}) (items []*EdbInfoCalculateMapping, err error) {
|
|
|
- o := orm.NewOrm()
|
|
|
+ //o := orm.NewOrm()
|
|
|
|
|
|
sql := ` SELECT * FROM edb_info_calculate_mapping WHERE 1=1 `
|
|
|
if condition != "" {
|
|
|
sql += condition
|
|
|
}
|
|
|
- _, err = o.Raw(sql, pars).QueryRows(&items)
|
|
|
+ err = global.DEFAULT_DB.Raw(sql, pars...).Find(&items).Error
|
|
|
+ //_, err = o.Raw(sql, pars).QueryRows(&items)
|
|
|
return
|
|
|
}
|
|
|
|
|
|
// GetEdbInfoCalculateCountByCondition 获取关联指标数量
|
|
|
func GetEdbInfoCalculateCountByCondition(condition string, pars []interface{}) (count int, err error) {
|
|
|
- o := orm.NewOrm()
|
|
|
+ //o := orm.NewOrm()
|
|
|
|
|
|
sql := ` SELECT COUNT(1) AS count FROM edb_info_calculate_mapping WHERE 1=1 `
|
|
|
if condition != "" {
|
|
|
sql += condition
|
|
|
}
|
|
|
- err = o.Raw(sql, pars).QueryRow(&count)
|
|
|
+ //err = o.Raw(sql, pars).QueryRow(&count)
|
|
|
+ err = global.DEFAULT_DB.Raw(sql, pars...).Scan(&count).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
@@ -604,7 +634,7 @@ type EdbInfoSearchDataV1 struct {
|
|
|
|
|
|
// 优化版本-处理数据精度问题
|
|
|
func GetEdbDataListAllV1(condition string, pars []interface{}, source, subSource, order int) (item []*EdbInfoSearchDataV1, err error) {
|
|
|
- o := orm.NewOrm()
|
|
|
+ //o := orm.NewOrm()
|
|
|
sql := ``
|
|
|
tableName := GetEdbDataTableName(source, subSource)
|
|
|
sql = ` SELECT * FROM %s WHERE 1=1 `
|
|
@@ -618,12 +648,13 @@ func GetEdbDataListAllV1(condition string, pars []interface{}, source, subSource
|
|
|
} else {
|
|
|
sql += ` ORDER BY data_time DESC `
|
|
|
}
|
|
|
- _, err = o.Raw(sql, pars).QueryRows(&item)
|
|
|
+ //_, err = o.Raw(sql, pars).QueryRows(&item)
|
|
|
+ err = global.DEFAULT_DB.Raw(sql, pars...).Find(&item).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
|
// GetEdbDataListAllV1ByTo 通过事务链接获取数据列表
|
|
|
-func GetEdbDataListAllV1ByTo(to orm.TxOrmer, condition string, pars []interface{}, source, subSource, order int) (item []*EdbInfoSearchDataV1, err error) {
|
|
|
+func GetEdbDataListAllV1ByTo(to *gorm.DB, condition string, pars []interface{}, source, subSource, order int) (item []*EdbInfoSearchDataV1, err error) {
|
|
|
sql := ``
|
|
|
tableName := GetEdbDataTableName(source, subSource)
|
|
|
sql = ` SELECT * FROM %s WHERE 1=1 `
|
|
@@ -637,13 +668,14 @@ func GetEdbDataListAllV1ByTo(to orm.TxOrmer, condition string, pars []interface{
|
|
|
} else {
|
|
|
sql += ` ORDER BY data_time DESC `
|
|
|
}
|
|
|
- _, err = to.Raw(sql, pars).QueryRows(&item)
|
|
|
+ //_, err = to.Raw(sql, pars).QueryRows(&item)
|
|
|
+ err = to.Raw(sql, pars...).Find(&item).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
|
// GetEdbInfoByCondition 获取指标列表
|
|
|
func GetEdbInfoByCondition(condition string, pars []interface{}, order int) (item []*EdbInfo, err error) {
|
|
|
- o := orm.NewOrm()
|
|
|
+ //o := orm.NewOrm()
|
|
|
sql := ` SELECT * FROM edb_info WHERE 1=1 `
|
|
|
if condition != "" {
|
|
|
sql += condition
|
|
@@ -653,7 +685,8 @@ func GetEdbInfoByCondition(condition string, pars []interface{}, order int) (ite
|
|
|
} else {
|
|
|
sql += ` ORDER BY edb_info_id ASC `
|
|
|
}
|
|
|
- _, err = o.Raw(sql, pars).QueryRows(&item)
|
|
|
+ //_, err = o.Raw(sql, pars).QueryRows(&item)
|
|
|
+ err = global.DEFAULT_DB.Raw(sql, pars...).Find(&item).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
@@ -707,7 +740,7 @@ func UnifiedModifyEdbInfoMaxAndMinInfoDataUpdate(edbInfo *EdbInfo, dataUpdateTim
|
|
|
// 获取本次刷新,指标数据更新的最早日期
|
|
|
erDataUpdateTime, tErr := GetEdbDataUpdateTimeByModify(edbInfo.Source, edbInfo.SubSource, edbInfo.EdbCode, dataUpdateTime)
|
|
|
if tErr != nil {
|
|
|
- if tErr.Error() != utils.ErrNoRow() { //本次刷新指标数据无变化
|
|
|
+ if !utils.IsErrNoRow(tErr) { //本次刷新指标数据无变化
|
|
|
errMsg = "刷新指标失败!"
|
|
|
err = errors.New("查询指标刷新,数据更新的最早日期失败,err:" + tErr.Error())
|
|
|
return
|
|
@@ -1186,7 +1219,7 @@ func getPredictEdbDayList(startDate, endDate time.Time, frequency, dataDateType
|
|
|
func GetPredictDataListByPredictEdbInfo(edbInfo *EdbInfo, order int, startDate string) (dataList []*EdbInfoSearchData, sourceEdbInfoItem *EdbInfo, err error, errMsg string) {
|
|
|
// 查找该预测指标配置
|
|
|
predictEdbConfList, err := GetPredictEdbConfAndDataListById(edbInfo.EdbInfoId)
|
|
|
- if err != nil && err.Error() != utils.ErrNoRow() {
|
|
|
+ if err != nil && !utils.IsErrNoRow(err) {
|
|
|
errMsg = "获取预测指标配置信息失败"
|
|
|
return
|
|
|
}
|
|
@@ -1313,15 +1346,16 @@ func GetPredictEdbDataListAllByStartDate(edbInfo *EdbInfo, order int, startDate
|
|
|
|
|
|
// ModifyPredictEdbInfoMaxAndMinInfo 修改预测指标的最新数据信息
|
|
|
func ModifyPredictEdbInfoMaxAndMinInfo(edbInfoId int, item *EdbInfoMaxAndMinInfo) (err error) {
|
|
|
- o := orm.NewOrm()
|
|
|
+ //o := orm.NewOrm()
|
|
|
sql := ` UPDATE edb_info SET start_date=?,end_date=?,min_value=?,max_value=?,is_update=2,latest_date=?,latest_value=?,end_value=?,modify_time=NOW() WHERE edb_info_id=? `
|
|
|
- _, err = o.Raw(sql, item.MinDate, item.MaxDate, item.MinValue, item.MaxValue, item.LatestDate, item.LatestValue, item.EndValue, edbInfoId).Exec()
|
|
|
+ //_, err = o.Raw(sql, item.MinDate, item.MaxDate, item.MinValue, item.MaxValue, item.LatestDate, item.LatestValue, item.EndValue, edbInfoId).Exec()
|
|
|
+ err = global.DEFAULT_DB.Exec(sql, item.MinDate, item.MaxDate, item.MinValue, item.MaxValue, item.LatestDate, item.LatestValue, item.EndValue, edbInfoId).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
|
// ModifyCalculateEdbInfo 修改计算指标信息
|
|
|
func ModifyCalculateEdbInfo(edbName, frequency, unit, calculateFormula string, classifyId, edbInfoId int) (err error) {
|
|
|
- o := orm.NewOrm()
|
|
|
+ //o := orm.NewOrm()
|
|
|
sql := ` UPDATE edb_info
|
|
|
SET
|
|
|
edb_name =?,
|
|
@@ -1332,24 +1366,27 @@ func ModifyCalculateEdbInfo(edbName, frequency, unit, calculateFormula string, c
|
|
|
calculate_formula=?,
|
|
|
modify_time = NOW()
|
|
|
WHERE edb_info_id = ?`
|
|
|
- _, err = o.Raw(sql, edbName, edbName, frequency, unit, classifyId, calculateFormula, edbInfoId).Exec()
|
|
|
+ //_, err = o.Raw(sql, edbName, edbName, frequency, unit, classifyId, calculateFormula, edbInfoId).Exec()
|
|
|
+ err = global.DEFAULT_DB.Exec(sql, edbName, edbName, frequency, unit, classifyId, calculateFormula, edbInfoId).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
|
func GetEdbInfoItemByCondition(condition string, pars []interface{}) (item *EdbInfoList, err error) {
|
|
|
- o := orm.NewOrm()
|
|
|
+ //o := orm.NewOrm()
|
|
|
sql := ` SELECT * FROM edb_info WHERE 1=1 `
|
|
|
if condition != "" {
|
|
|
sql += condition
|
|
|
}
|
|
|
- err = o.Raw(sql, pars).QueryRow(&item)
|
|
|
+ //err = o.Raw(sql, pars).QueryRow(&item)
|
|
|
+ err = global.DEFAULT_DB.Raw(sql, pars...).Find(&item).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
|
func GetEdbInfoItemListByCondition() (items []*EdbInfoList, err error) {
|
|
|
- o := orm.NewOrm()
|
|
|
+ //o := orm.NewOrm()
|
|
|
sql := ` SELECT * FROM edb_info WHERE 1=1 `
|
|
|
- _, err = o.Raw(sql).QueryRows(&items)
|
|
|
+ //_, err = o.Raw(sql).QueryRows(&items)
|
|
|
+ err = global.DEFAULT_DB.Raw(sql).Find(&items).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
@@ -1414,8 +1451,9 @@ func EdbInfoAdd(req *AddEdbInfoParams, serverUrl string, sysUserId int, sysUserR
|
|
|
|
|
|
if len(items) > 0 {
|
|
|
sql := ` UPDATE edb_info SET classify_id=?,modify_time=NOW() WHERE edb_code=? `
|
|
|
- o := orm.NewOrm()
|
|
|
- _, err = o.Raw(sql, req.ClassifyId, req.EdbCode).Exec()
|
|
|
+ //o := orm.NewOrm()
|
|
|
+ //_, err = o.Raw(sql, req.ClassifyId, req.EdbCode).Exec()
|
|
|
+ err = global.DEFAULT_DB.Exec(sql, req.ClassifyId, req.EdbCode).Error
|
|
|
if err != nil {
|
|
|
err = errors.New("更新指标分类信息失败,Err:" + err.Error())
|
|
|
return
|
|
@@ -1533,10 +1571,11 @@ func EdbInfoAdd(req *AddEdbInfoParams, serverUrl string, sysUserId int, sysUserR
|
|
|
}
|
|
|
|
|
|
func ModifyEdbInfoBaseTimeById(edbInfoId int, cTime time.Time) (err error) {
|
|
|
- o := orm.NewOrm()
|
|
|
+ //o := orm.NewOrm()
|
|
|
// 更新修改时间
|
|
|
sql := ` UPDATE edb_info SET base_modify_time = ? WHERE edb_info_id = ? `
|
|
|
- _, err = o.Raw(sql, cTime, edbInfoId).Exec()
|
|
|
+ //_, err = o.Raw(sql, cTime, edbInfoId).Exec()
|
|
|
+ err = global.DEFAULT_DB.Exec(sql, cTime, edbInfoId).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
@@ -1563,9 +1602,10 @@ type ResetEdbClassifyReq struct {
|
|
|
|
|
|
// GetEdbInfoMaxSortByClassifyId 获取分类下指标的最大的排序数
|
|
|
func GetEdbInfoMaxSortByClassifyId(classifyId int) (sort int, err error) {
|
|
|
- o := orm.NewOrm()
|
|
|
+ //o := orm.NewOrm()
|
|
|
sql := `SELECT Max(sort) AS sort FROM edb_info WHERE classify_id=? `
|
|
|
- err = o.Raw(sql, classifyId).QueryRow(&sort)
|
|
|
+ //err = o.Raw(sql, classifyId).QueryRow(&sort)
|
|
|
+ err = global.DEFAULT_DB.Raw(sql, classifyId).Scan(&sort).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
@@ -1620,10 +1660,10 @@ func GetEdbInfoByEdbCodeList(source int, edbCodeList []string) (items []*EdbInfo
|
|
|
if num <= 0 {
|
|
|
return
|
|
|
}
|
|
|
- o := orm.NewOrm()
|
|
|
+ //o := orm.NewOrm()
|
|
|
sql := ` SELECT * FROM edb_info WHERE source=? AND edb_code IN (` + utils.GetOrmInReplace(num) + `) `
|
|
|
- _, err = o.Raw(sql, source, edbCodeList).QueryRows(&items)
|
|
|
-
|
|
|
+ //_, err = o.Raw(sql, source, edbCodeList).QueryRows(&items)
|
|
|
+ err = global.DEFAULT_DB.Raw(sql, source, edbCodeList).Find(&items).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
@@ -1652,9 +1692,10 @@ func GetEdbInfoNoUpdateTotalByIdList(edbInfoIdList []int) (total int, err error)
|
|
|
if num <= 0 {
|
|
|
return
|
|
|
}
|
|
|
- o := orm.NewOrm()
|
|
|
+ //o := orm.NewOrm()
|
|
|
sql := ` SELECT count(*) FROM edb_info WHERE edb_info_id in (` + utils.GetOrmInReplace(num) + `) and no_update=1`
|
|
|
- err = o.Raw(sql, edbInfoIdList).QueryRow(&total)
|
|
|
+ //err = o.Raw(sql, edbInfoIdList).QueryRow(&total)
|
|
|
+ err = global.DEFAULT_DB.Raw(sql, edbInfoIdList).Scan(&total).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
@@ -1707,11 +1748,12 @@ func (m SortEdbDataList) Swap(i, j int) {
|
|
|
}
|
|
|
|
|
|
func ModifyEdbInfoDataStatus(edbInfoId int64, source, subSource int, edbCode string) (err error) {
|
|
|
- o := orm.NewOrm()
|
|
|
+ //o := orm.NewOrm()
|
|
|
sql := ``
|
|
|
tableName := GetEdbDataTableName(source, subSource)
|
|
|
sql = ` UPDATE %s SET edb_info_id=?,modify_time=NOW() WHERE edb_code=? `
|
|
|
sql = fmt.Sprintf(sql, tableName)
|
|
|
- _, err = o.Raw(sql, edbInfoId, edbCode).Exec()
|
|
|
+ //_, err = o.Raw(sql, edbInfoId, edbCode).Exec()
|
|
|
+ err = global.DEFAULT_DB.Exec(sql, edbInfoId, edbCode).Error
|
|
|
return
|
|
|
}
|