hongze_task/models/data_manage/edb_info.go

package data_manage

import (
	"errors"
	"fmt"
	"hongze/hongze_task/utils"
	"rdluck_tools/orm"
	"strconv"
	"time"
)

type EdbInfo struct {
	EdbInfoId        int    `orm:"column(edb_info_id);pk"`
	SourceName       string `description:"来源名称"`
	Source           int    `description:"来源id"`
	EdbCode          string `description:"指标编码"`
	EdbName          string `description:"指标名称"`
	EdbNameSource    string `description:"指标名称来源"`
	Frequency        string `description:"频率"`
	Unit             string `description:"单位"`
	StartDate        string `description:"起始日期"`
	EndDate          string `description:"终止日期"`
	ClassifyId       int    `description:"分类id"`
	SysUserId        int
	SysUserRealName  string
	UniqueCode       string `description:"指标唯一编码"`
	CreateTime       time.Time
	ModifyTime       time.Time
	MinValue         float64 `description:"指标最小值"`
	MaxValue         float64 `description:"指标最大值"`
	CalculateFormula string  `description:"计算公式"`
}

type EdbInfoList struct {
	EdbInfoId        int       `orm:"column(edb_info_id);pk"`
	SourceName       string    `description:"来源名称"`
	Source           int       `description:"来源id"`
	EdbCode          string    `description:"指标编码"`
	EdbName          string    `description:"指标名称"`
	Frequency        string    `description:"频率"`
	Unit             string    `description:"单位"`
	StartDate        time.Time `description:"起始日期"`
	EndDate          time.Time `description:"终止日期"`
	ClassifyId       int       `description:"分类id"`
	UniqueCode       string    `description:"指标唯一编码"`
	CalculateFormula string    `description:"计算公式"`
}

type EdbInfoSearchData struct {
	DataTime string  `description:"数据日期"`
	Value    float64 `description:"数据"`
}

func GetEdbInfoByCondition(condition string, pars []interface{}) (item []*EdbInfoList, err error) {
	o := orm.NewOrm()
	o.Using("data")
	sql := ` SELECT * FROM edb_info WHERE 1=1 `
	if condition != "" {
		sql += condition
	}
	_, err = o.Raw(sql, pars).QueryRows(&item)
	return
}

func ModifyEdbDataInfoDate(edbInfoId int, maxDate string) (err error) {
	o := orm.NewOrm()
	o.Using("data")
	sql := ` UPDATE edb_info SET end_date=?,modify_time=NOW() WHERE edb_info_id=? `
	_, err = o.Raw(sql, maxDate, edbInfoId).Exec()
	return
}

type EdbInfoMaxAndMinInfo struct {
	MinDate  string  `description:"最小日期"`
	MaxDate  string  `description:"最大日期"`
	MinValue float64 `description:"最小值"`
	MaxValue float64 `description:"最大值"`
}

func GetEdbInfoMaxAndMinInfo(source int, edbCode string) (item *EdbInfoMaxAndMinInfo, err error) {
	o := orm.NewOrm()
	o.Using("data")
	sql := ``
	tableName := ``
	if source == utils.DATA_SOURCE_THS {
		tableName = `edb_data_ths`
	} else if source == utils.DATA_SOURCE_WIND {
		tableName = `edb_data_wind`
	} else if source == utils.DATA_SOURCE_PB {
		tableName = `edb_data_pb`
	} else if source == utils.DATA_SOURCE_CALCULATE {
		tableName = "edb_data_calculate"
	} else {
		errors.New("无效的渠道:" + strconv.Itoa(source))
		return
	}
	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)
	fmt.Println("sql:", sql)
	err = o.Raw(sql, edbCode).QueryRow(&item)
	return
}

func ModifyEdbInfoMaxAndMinInfo(edbInfoId int, item *EdbInfoMaxAndMinInfo) (err error) {
	o := orm.NewOrm()
	o.Using("data")
	sql := ` UPDATE edb_info SET start_date=?,end_date=?,min_value=?,max_value=?,modify_time=NOW() WHERE edb_info_id=? `
	_, err = o.Raw(sql, item.MinDate, item.MaxDate, item.MinValue, item.MaxValue, edbInfoId).Exec()
	return
}

func GetEdbDataListAll(condition string, pars []interface{}, source int) (item []*EdbInfoSearchData, err error) {
	o := orm.NewOrm()
	o.Using("data")
	sql := ``
	tableName := GetEdbDataTableName(source)
	sql = ` SELECT * FROM %s WHERE 1=1 `
	sql = fmt.Sprintf(sql, tableName)

	if condition != "" {
		sql += condition
	}

	sql += ` ORDER BY data_time ASC `
	_, err = o.Raw(sql, pars).QueryRows(&item)
	return
}

func GetEdbDataTableName(source int) (tableName string) {
	if source == utils.DATA_SOURCE_THS {
		tableName = `edb_data_ths`
	} else if source == utils.DATA_SOURCE_WIND {
		tableName = `edb_data_wind`
	} else if source == utils.DATA_SOURCE_PB {
		tableName = `edb_data_pb`
	} else if source == utils.DATA_SOURCE_CALCULATE {
		tableName = `edb_data_calculate`
	} else {
		tableName = ""
		errors.New("无效的渠道:" + strconv.Itoa(source))
		return
	}
	return
}

func GetEdbInfoById(edbInfoId int) (item *EdbInfo, err error) {
	o := orm.NewOrm()
	o.Using("data")
	sql := ` SELECT * FROM edb_info WHERE edb_info_id=? `
	err = o.Raw(sql, edbInfoId).QueryRow(&item)
	return
}