|
@@ -1,11 +1,10 @@
|
|
|
package data_manage
|
|
|
|
|
|
import (
|
|
|
+ "eta_gn/eta_task/global"
|
|
|
"eta_gn/eta_task/utils"
|
|
|
|
|
|
"time"
|
|
|
-
|
|
|
- "github.com/beego/beego/v2/client/orm"
|
|
|
)
|
|
|
|
|
|
// BaseFromEiaSteoIndex EiaSteo指标
|
|
@@ -30,46 +29,62 @@ type BaseFromEiaSteoIndex struct {
|
|
|
}
|
|
|
|
|
|
func AddBaseFromEiaSteoIndex(item *BaseFromEiaSteoIndex) (lastId int64, err error) {
|
|
|
- o := orm.NewOrm()
|
|
|
- lastId, err = o.Insert(item)
|
|
|
+ //o := orm.NewOrm()
|
|
|
+ //lastId, err = o.Insert(item)
|
|
|
+ err = global.DEFAULT_DmSQL.Create(item).Error
|
|
|
+ if err != nil {
|
|
|
+ return
|
|
|
+ }
|
|
|
+ lastId = int64(item.BaseFromEiaSteoIndexId)
|
|
|
return
|
|
|
}
|
|
|
|
|
|
func InsertOrUpdateBaseFromEiaSteoIndex(item *BaseFromEiaSteoIndex) (lastId int64, err error) {
|
|
|
- o := orm.NewOrm()
|
|
|
- lastId, err = o.InsertOrUpdate(item)
|
|
|
+ //o := orm.NewOrm()
|
|
|
+ //lastId, err = o.InsertOrUpdate(item)
|
|
|
+ err = global.DEFAULT_DmSQL.Create(item).Error
|
|
|
+ if err != nil {
|
|
|
+ return
|
|
|
+ }
|
|
|
+ lastId = int64(item.BaseFromEiaSteoIndexId)
|
|
|
return
|
|
|
}
|
|
|
|
|
|
// Add 新增指标
|
|
|
func (item *BaseFromEiaSteoIndex) Add() (err error) {
|
|
|
- o := orm.NewOrm()
|
|
|
- lastId, err := o.Insert(item)
|
|
|
- if err != nil {
|
|
|
- return
|
|
|
- }
|
|
|
- item.BaseFromEiaSteoIndexId = int(lastId)
|
|
|
+ //o := orm.NewOrm()
|
|
|
+ //lastId, err := o.Insert(item)
|
|
|
+ //if err != nil {
|
|
|
+ // return
|
|
|
+ //}
|
|
|
+ //item.BaseFromEiaSteoIndexId = int(lastId)
|
|
|
+
|
|
|
+ err = global.DEFAULT_DmSQL.Create(item).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
|
func GetBaseFromEiaSteoIndexAll(dateStr string) (list []*BaseFromEiaSteoIndex, err error) {
|
|
|
- o := orm.NewOrm()
|
|
|
+ //o := orm.NewOrm()
|
|
|
sql := `SELECT * FROM base_from_eia_steo_index WHERE create_time>=?`
|
|
|
- _, err = o.Raw(sql, dateStr).QueryRows(&list)
|
|
|
+ //_, err = o.Raw(sql, dateStr).QueryRows(&list)
|
|
|
+ err = global.DEFAULT_DmSQL.Raw(sql, dateStr).Find(&list).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
|
func GetBaseFromEiaSteoIndexCodeListByDate(dateStr string) (list []string, err error) {
|
|
|
- o := orm.NewOrm()
|
|
|
+ //o := orm.NewOrm()
|
|
|
sql := `SELECT index_code FROM base_from_eia_steo_index WHERE modify_time>=?`
|
|
|
- _, err = o.Raw(sql, dateStr).QueryRows(&list)
|
|
|
+ //_, err = o.Raw(sql, dateStr).QueryRows(&list)
|
|
|
+ err = global.DEFAULT_DmSQL.Raw(sql, dateStr).Find(&list).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
|
// Update 更新作者基础信息
|
|
|
func (item *BaseFromEiaSteoIndex) Update(cols []string) (err error) {
|
|
|
- o := orm.NewOrm()
|
|
|
- _, err = o.Update(item, cols...)
|
|
|
+ //o := orm.NewOrm()
|
|
|
+ //_, err = o.Update(item, cols...)
|
|
|
+
|
|
|
+ err = global.DEFAULT_DmSQL.Select(cols).Updates(item).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
@@ -86,12 +101,14 @@ type BaseFromEiaSteoClassify struct {
|
|
|
|
|
|
// 新增分类
|
|
|
func (item *BaseFromEiaSteoClassify) AddBaseFromEiaSteoClassify() (err error) {
|
|
|
- o := orm.NewOrm()
|
|
|
- lastId, err := o.Insert(item)
|
|
|
- if err != nil {
|
|
|
- return
|
|
|
- }
|
|
|
- item.BaseFromEiaSteoClassifyId = int(lastId)
|
|
|
+ //o := orm.NewOrm()
|
|
|
+ //lastId, err := o.Insert(item)
|
|
|
+ //if err != nil {
|
|
|
+ // return
|
|
|
+ //}
|
|
|
+ //item.BaseFromEiaSteoClassifyId = int(lastId)
|
|
|
+
|
|
|
+ err = global.DEFAULT_DmSQL.Create(item).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
@@ -107,34 +124,48 @@ type BaseFromEiaSteoData struct {
|
|
|
}
|
|
|
|
|
|
func (b *BaseFromEiaSteoData) Update(cols []string) (err error) {
|
|
|
- o := orm.NewOrm()
|
|
|
- _, err = o.Update(b, cols...)
|
|
|
+ //o := orm.NewOrm()
|
|
|
+ //_, err = o.Update(b, cols...)
|
|
|
+ err = global.DEFAULT_DmSQL.Select(cols).Updates(b).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
|
func AddBaseFromEiaSteoData(item *BaseFromEiaSteoData) (lastId int64, err error) {
|
|
|
- o := orm.NewOrm()
|
|
|
- lastId, err = o.Insert(item)
|
|
|
+ //o := orm.NewOrm()
|
|
|
+ //lastId, err = o.Insert(item)
|
|
|
+
|
|
|
+ err = global.DEFAULT_DmSQL.Create(item).Error
|
|
|
+ if err != nil {
|
|
|
+ return
|
|
|
+ }
|
|
|
+ lastId = int64(item.BaseFromEiaSteoDataId)
|
|
|
return
|
|
|
}
|
|
|
|
|
|
// GetBaseFromEiaSteoClassifyAll 获取所有的分类
|
|
|
func GetBaseFromEiaSteoClassifyAll() (list []*BaseFromEiaSteoClassify, err error) {
|
|
|
- o := orm.NewOrm()
|
|
|
+ //o := orm.NewOrm()
|
|
|
sql := `SELECT * FROM base_from_eia_steo_classify`
|
|
|
- _, err = o.Raw(sql).QueryRows(&list)
|
|
|
+ //_, err = o.Raw(sql).QueryRows(&list)
|
|
|
+ err = global.DEFAULT_DmSQL.Raw(sql).Find(&list).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
|
func AddBaseEiaSteoClassify(item *BaseFromEiaSteoClassify) (lastId int64, err error) {
|
|
|
- o := orm.NewOrm()
|
|
|
- lastId, err = o.Insert(item)
|
|
|
+ //o := orm.NewOrm()
|
|
|
+ //lastId, err = o.Insert(item)
|
|
|
+ err = global.DEFAULT_DmSQL.Create(item).Error
|
|
|
+ if err != nil {
|
|
|
+ return
|
|
|
+ }
|
|
|
+ lastId = int64(item.BaseFromEiaSteoClassifyId)
|
|
|
return
|
|
|
}
|
|
|
|
|
|
func (item *BaseFromEiaSteoClassify) Update(cols []string) (err error) {
|
|
|
- o := orm.NewOrm()
|
|
|
- _, err = o.Update(item, cols...)
|
|
|
+ //o := orm.NewOrm()
|
|
|
+ //_, err = o.Update(item, cols...)
|
|
|
+ err = global.DEFAULT_DmSQL.Select(cols).Updates(item).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
@@ -142,9 +173,10 @@ func BatchDeleteBaseFromEiaSteoClassify(id []int) (err error) {
|
|
|
if len(id) == 0 {
|
|
|
return
|
|
|
}
|
|
|
- o := orm.NewOrm()
|
|
|
+ //o := orm.NewOrm()
|
|
|
sql := `DELETE FROM base_from_eia_steo_classify WHERE base_from_eia_steo_classify_id IN (` + utils.GetOrmInReplace(len(id)) + `)`
|
|
|
- _, err = o.Raw(sql, id).Exec()
|
|
|
+ //_, err = o.Raw(sql, id).Exec()
|
|
|
+ err = global.DEFAULT_DmSQL.Raw(sql, id).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
@@ -173,36 +205,41 @@ type EiaSteoClassifyDataResp struct {
|
|
|
}
|
|
|
|
|
|
func GetBaseFromEiaSteoIndexDataAll(dateStr string) (list []*BaseFromEiaSteoData, err error) {
|
|
|
- o := orm.NewOrm()
|
|
|
+ //o := orm.NewOrm()
|
|
|
sql := `SELECT * FROM base_from_eia_steo_data WHERE create_time>=?`
|
|
|
- _, err = o.Raw(sql, dateStr).QueryRows(&list)
|
|
|
+ //_, err = o.Raw(sql, dateStr).QueryRows(&list)
|
|
|
+ err = global.DEFAULT_DmSQL.Raw(sql, dateStr).Find(&list).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
|
func GetBaseFromEiaSteoIndexDataByIndexCode(indexCode, dateStr string) (list []*BaseFromEiaSteoData, err error) {
|
|
|
- o := orm.NewOrm()
|
|
|
+ //o := orm.NewOrm()
|
|
|
sql := `SELECT * FROM base_from_eia_steo_data WHERE index_code=? AND modify_time>=?`
|
|
|
- _, err = o.Raw(sql, indexCode, dateStr).QueryRows(&list)
|
|
|
+ //_, err = o.Raw(sql, indexCode, dateStr).QueryRows(&list)
|
|
|
+ err = global.DEFAULT_DmSQL.Raw(sql, indexCode, dateStr).Find(&list).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
|
func GetBaseFromEiaSteoIndexDataByDate(dateStr string) (list []*BaseFromEiaSteoData, err error) {
|
|
|
- o := orm.NewOrm()
|
|
|
+ //o := orm.NewOrm()
|
|
|
sql := `SELECT * FROM base_from_eia_steo_data WHERE modify_time>=?`
|
|
|
- _, err = o.Raw(sql, dateStr).QueryRows(&list)
|
|
|
+ //_, err = o.Raw(sql, dateStr).QueryRows(&list)
|
|
|
+ err = global.DEFAULT_DmSQL.Raw(sql, dateStr).Find(&list).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
|
func GetBaseFromEiaSteoIndexMaxDate() (max_date time.Time, err error) {
|
|
|
- o := orm.NewOrm()
|
|
|
+ //o := orm.NewOrm()
|
|
|
sql := ` SELECT max(a.data_time)as max_date FROM base_from_eia_steo_data as a `
|
|
|
- err = o.Raw(sql).QueryRow(&max_date)
|
|
|
+ //err = o.Raw(sql).QueryRow(&max_date)
|
|
|
+ err = global.DEFAULT_DmSQL.Raw(sql).Scan(&max_date).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
|
func GetBaseFromEiaSteoIndexMaxCreateDate() (max_date time.Time, err error) {
|
|
|
- o := orm.NewOrm()
|
|
|
+ //o := orm.NewOrm()
|
|
|
sql := ` SELECT max(a.create_time)as max_date FROM base_from_eia_steo_index as a `
|
|
|
- err = o.Raw(sql).QueryRow(&max_date)
|
|
|
+ //err = o.Raw(sql).QueryRow(&max_date)
|
|
|
+ err = global.DEFAULT_DmSQL.Raw(sql).Scan(&max_date).Error
|
|
|
return
|
|
|
}
|