package data_manage import ( "eta/eta_task/utils" "time" "github.com/beego/beego/v2/client/orm" ) // BaseFromEiaSteoIndex EiaSteo指标 type BaseFromEiaSteoIndex struct { BaseFromEiaSteoIndexId int `orm:"column(base_from_eia_steo_index_id);pk"` BaseFromEiaSteoClassifyId int `description:"指标分类id"` IndexCode string `description:"指标编码"` IndexName string `description:"指标名称（中文名称）"` IndexNameOriginal string `description:"指标名称（原始名称）"` Frequency string `description:"频度"` Level int `description:"指标层级（原始层级）"` Unit string `description:"单位"` Super string `description:"我也不知道是个啥，反正先存起来，万一用到了呢"` Precision int `description:"精度，我也不知道会不会用到，反正先存起来，万一用到了呢"` LastHistorical string `description:"最近的历史记录，我也不知道会不会用到，反正先存起来，万一用到了呢"` Description string `description:"备注信息，我也不知道会不会用到，反正先存起来，万一用到了呢"` IsMappable int `description:"是否可映射，我也不知道会不会用到，反正先存起来，万一用到了呢"` StartDate string `description:"开始日期"` EndDate string `description:"结束日期"` ModifyTime string `description:"最新更新时间"` CreateTime string `description:"创建时间"` } func AddBaseFromEiaSteoIndex(item *BaseFromEiaSteoIndex) (lastId int64, err error) { o := orm.NewOrm() lastId, err = o.Insert(item) return } func InsertOrUpdateBaseFromEiaSteoIndex(item *BaseFromEiaSteoIndex) (lastId int64, err error) { o := orm.NewOrm() lastId, err = o.InsertOrUpdate(item) return } // Add 新增指标 func (item *BaseFromEiaSteoIndex) Add() (err error) { o := orm.NewOrm() lastId, err := o.Insert(item) if err != nil { return } item.BaseFromEiaSteoIndexId = int(lastId) return } func GetBaseFromEiaSteoIndexAll(dateStr string) (list []*BaseFromEiaSteoIndex, err error) { o := orm.NewOrm() sql := `SELECT * FROM base_from_eia_steo_index WHERE create_time>=?` _, err = o.Raw(sql, dateStr).QueryRows(&list) return } func GetBaseFromEiaSteoIndexCodeListByDate(dateStr string) (list []string, err error) { o := orm.NewOrm() sql := `SELECT index_code FROM base_from_eia_steo_index WHERE modify_time>=?` _, err = o.Raw(sql, dateStr).QueryRows(&list) return } // Update 更新作者基础信息 func (item *BaseFromEiaSteoIndex) Update(cols []string) (err error) { o := orm.NewOrm() _, err = o.Update(item, cols...) return } // BaseFromEiaSteoClassify EiaSteo分类 type BaseFromEiaSteoClassify struct { BaseFromEiaSteoClassifyId int `orm:"column(base_from_eia_steo_classify_id);pk"` ClassifyName string `description:"分类名称（中文名称）"` ClassifyNameOriginal string `description:"分类名称（原始名称）"` ParentId int `description:"父级id"` Level int `description:"层级"` ModifyTime time.Time `description:"最新更新时间"` CreateTime time.Time `description:"创建时间"` } // 新增分类 func (item *BaseFromEiaSteoClassify) AddBaseFromEiaSteoClassify() (err error) { o := orm.NewOrm() lastId, err := o.Insert(item) if err != nil { return } item.BaseFromEiaSteoClassifyId = int(lastId) return } // BaseFromEiaSteoData EiaSteo数据表 type BaseFromEiaSteoData struct { BaseFromEiaSteoDataId int `orm:"column(base_from_eia_steo_data_id);pk"` BaseFromEiaSteoIndexId int `description:"指标id"` IndexCode string `description:"指标编码"` DataTime time.Time `description:"数据日期"` Value float64 `description:"数据值"` ModifyTime time.Time `description:"最新更新时间"` CreateTime time.Time `description:"创建时间"` } func (b *BaseFromEiaSteoData) Update(cols []string) (err error) { o := orm.NewOrm() _, err = o.Update(b, cols...) return } func AddBaseFromEiaSteoData(item *BaseFromEiaSteoData) (lastId int64, err error) { o := orm.NewOrm() lastId, err = o.Insert(item) return } // GetBaseFromEiaSteoClassifyAll 获取所有的分类 func GetBaseFromEiaSteoClassifyAll() (list []*BaseFromEiaSteoClassify, err error) { o := orm.NewOrm() sql := `SELECT * FROM base_from_eia_steo_classify` _, err = o.Raw(sql).QueryRows(&list) return } func AddBaseEiaSteoClassify(item *BaseFromEiaSteoClassify) (lastId int64, err error) { o := orm.NewOrm() lastId, err = o.Insert(item) return } func (item *BaseFromEiaSteoClassify) Update(cols []string) (err error) { o := orm.NewOrm() _, err = o.Update(item, cols...) return } func BatchDeleteBaseFromEiaSteoClassify(id []int) (err error) { if len(id) == 0 { return } 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() return } type EiaSteoIndexResp struct { Ret int Msg string ErrMsg string ErrCode string Data []*BaseFromEiaSteoIndex } type EiaSteoIndexDataResp struct { Ret int Msg string ErrMsg string ErrCode string Data []*BaseFromEiaSteoData } type EiaSteoClassifyDataResp struct { Ret int Msg string ErrMsg string ErrCode string Data []*BaseFromEiaSteoClassify } func GetBaseFromEiaSteoIndexDataAll(dateStr string) (list []*BaseFromEiaSteoData, err error) { o := orm.NewOrm() sql := `SELECT * FROM base_from_eia_steo_data WHERE create_time>=?` _, err = o.Raw(sql, dateStr).QueryRows(&list) return } func GetBaseFromEiaSteoIndexDataByIndexCode(indexCode, dateStr string) (list []*BaseFromEiaSteoData, err error) { 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) return } func GetBaseFromEiaSteoIndexDataByDate(dateStr string) (list []*BaseFromEiaSteoData, err error) { o := orm.NewOrm() sql := `SELECT * FROM base_from_eia_steo_data WHERE modify_time>=?` _, err = o.Raw(sql, dateStr).QueryRows(&list) return } func GetBaseFromEiaSteoIndexMaxDate() (max_date time.Time, err error) { 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) return } func GetBaseFromEiaSteoIndexMaxCreateDate() (max_date time.Time, err error) { 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) return }