eta_hub/models/data_manage/edb_info.go

package data_manage

import (
	"eta/eta_hub/utils"
	"fmt"
	"github.com/beego/beego/v2/client/orm"
	"strings"
	"time"
)

type EdbInfo struct {
	EdbInfoId        int       `orm:"column(edb_info_id);pk"`
	EdbInfoType      int       `description:"指标类型，0：普通指标，1：预测指标"`
	SourceName       string    `description:"来源名称"`
	Source           int       `description:"来源ID"`
	EdbCode          string    `description:"指标编码"`
	EdbName          string    `description:"指标名称"`
	EdbNameEn        string    `description:"英文指标名称"`
	EdbNameSource    string    `description:"指标名称来源"`
	Frequency        string    `description:"频率"`
	Unit             string    `description:"单位"`
	UnitEn           string    `description:"英文单位"`
	StartDate        string    `description:"起始日期"`
	EndDate          string    `description:"终止日期"`
	ClassifyId       int       `description:"分类ID"`
	SysUserId        int       `description:"创建人ID"`
	SysUserRealName  string    `description:"创建人姓名"`
	UniqueCode       string    `description:"指标唯一编码"`
	CreateTime       time.Time `description:"创建时间"`
	ModifyTime       time.Time `description:"更新时间"`
	MinValue         float64   `description:"指标最小值"`
	MaxValue         float64   `description:"指标最大值"`
	CalculateFormula string    `description:"计算公式"`
	EdbType          int       `description:"指标类型:1:基础指标,2:计算指标"`
	Sort             int       `description:"排序字段"`
	LatestDate       string    `description:"数据最新日期"`
	LatestValue      float64   `description:"数据最新值"`
	MoveType         int       `description:"移动方式:1:领先(默认),2:滞后"`
	MoveFrequency    string    `description:"移动频度"`
	NoUpdate         int8      `description:"是否停止更新，0：继续更新；1：停止更新"`
	ServerUrl        string    `description:"服务器地址"`
	ChartImage       string    `description:"图表图片"`
	Calendar         string    `description:"公历/农历" orm:"default(公历);"`
	DataDateType     string    `orm:"column(data_date_type);size(255);null;default(交易日)"`
	ManualSave       int       `description:"是否有手动保存过上下限: 0-否; 1-是"`
}

func (m *EdbInfo) GetItemsByCondition(cond string, pars []interface{}, fieldArr []string, orderRule string) (items []*EdbInfo, err error) {
	o := orm.NewOrmUsingDB("data")
	fields := strings.Join(fieldArr, ",")
	if len(fieldArr) == 0 {
		fields = `*`
	}
	order := `ORDER BY create_time DESC`
	if orderRule != "" {
		order = ` ORDER BY ` + orderRule
	}
	sql := fmt.Sprintf(`SELECT %s FROM edb_info WHERE 1=1 %s %s`, fields, cond, order)
	_, err = o.Raw(sql, pars).QueryRows(&items)
	return
}

func (m *EdbInfo) GetItemByUniCode(code string) (item *EdbInfo, err error) {
	o := orm.NewOrmUsingDB("data")
	sql := fmt.Sprintf(`SELECT * FROM edb_info WHERE unique_code = ? LIMIT 1`)
	err = o.Raw(sql, code).QueryRow(&item)
	return
}

func (m *EdbInfo) GetItemByEdbCode(code string) (item *EdbInfo, err error) {
	o := orm.NewOrmUsingDB("data")
	sql := fmt.Sprintf(`SELECT * FROM edb_info WHERE edb_code = ? LIMIT 1`)
	err = o.Raw(sql, code).QueryRow(&item)
	return
}

// EdbInfoItem 指标信息
type EdbInfoItem struct {
	EdbInfoId    int            `description:"指标信息ID"`
	UniqueCode   string         `description:"指标唯一编码"`
	EdbCode      string         `description:"指标编码"`
	EdbInfoType  int            `description:"指标类型: 0-普通指标; 1-预测指标"`
	EdbType      int            `description:"指标类型: 1-基础指标; 2-计算指标"`
	EdbName      string         `description:"指标名称"`
	ClassifyId   int            `description:"分类ID"`
	Source       int            `description:"来源ID"`
	SourceName   string         `description:"来源名称"`
	Frequency    string         `description:"频率"`
	Unit         string         `description:"单位"`
	StartDate    string         `description:"起始日期"`
	EndDate      string         `description:"终止日期"`
	MinValue     float64        `description:"指标最小值"`
	MaxValue     float64        `description:"指标最大值"`
	LatestDate   string         `description:"数据最新日期"`
	LatestValue  float64        `description:"数据最新值"`
	ChartImage   string         `description:"图表图片"`
	Calendar     string         `description:"公历/农历"`
	DataDateType string         `description:"数据日期类型: 交易日/自然日"`
	Sort         int            `description:"排序"`
	NoUpdate     int            `description:"是否停更: 0-继续更新; 1-停止更新"`
	CreateTime   string         `description:"创建时间"`
	ModifyTime   string         `description:"更新时间"`
	EdbData      []*EdbDataItem `description:"指标数据"`
}

func FormatEdbInfo2Item(origin *EdbInfo, edbData []*EdbDataItem) (item *EdbInfoItem) {
	if origin == nil {
		return
	}
	item = new(EdbInfoItem)
	item.EdbInfoId = origin.EdbInfoId
	item.EdbInfoType = origin.EdbInfoType
	item.SourceName = origin.SourceName
	item.Source = origin.Source
	item.EdbCode = origin.EdbCode
	item.EdbName = origin.EdbName
	item.Frequency = origin.Frequency
	item.Unit = origin.Unit
	item.StartDate = origin.StartDate
	item.EndDate = origin.EndDate
	item.ClassifyId = origin.ClassifyId
	item.UniqueCode = origin.UniqueCode
	item.MinValue = origin.MinValue
	item.MaxValue = origin.MaxValue
	item.EdbType = origin.EdbType
	item.Sort = origin.Sort
	item.LatestDate = origin.LatestDate
	item.LatestValue = origin.LatestValue
	item.NoUpdate = int(origin.NoUpdate)
	item.ChartImage = origin.ChartImage
	item.Calendar = origin.Calendar
	item.DataDateType = origin.DataDateType
	item.CreateTime = utils.TimeTransferString(utils.FormatDateTime, origin.CreateTime)
	item.ModifyTime = utils.TimeTransferString(utils.FormatDateTime, origin.ModifyTime)
	item.EdbData = edbData
	return
}