hongze_task/models/data_source_longzhong.go

package models

import (
	"rdluck_tools/orm"
	"time"
)

type Longzhongdata struct {
	LongzhongdataId int `orm:"column(longzhongdata_id);pk"`
	LongzhonginfoId int
	TradeCode       string
	Dt              string
	Close           float64
	CreateTime      time.Time
	ModifyTime      time.Time
	UnitDesc        string
	UpdTime         string
}

type Longzhonginfo struct {
	LongzhonginfoId int `orm:"column(longzhonginfo_id);pk"`
	TradeCode       string
	SecName         string
	Frequency       string
	ClassifyId      int
	ClassifyName    string
	Unit            string
	CreateTime      time.Time
	Remark          string
	IsNormal        string
	LastModifyDate  string
	Unitid          int64
	TempId          int64
	TempName        string
	ModifyTime      time.Time
}

func GetLongzhonginfoBySecName(secName string) (item *Longzhongdata, err error) {
	o := orm.NewOrm()
	o.Using("edb")
	sql := `SELECT * FROM longzhonginfo WHERE sec_name=? `
	err = o.Raw(sql, secName).QueryRow(&item)
	return
}

//判断指标数据是否已经录入
func GetLongzhongdataCount(longzhonginfoId int, dt string) (count int, err error) {
	o := orm.NewOrm()
	o.Using("edb")
	sql := `SELECT COUNT(1) AS count FROM longzhongdata WHERE longzhonginfo_id=? AND dt=? `
	err = o.Raw(sql, longzhonginfoId, dt).QueryRow(&count)
	return
}

func AddLongzhongdata(item *Longzhongdata) (err error) {
	o := orm.NewOrm()
	o.Using("edb")
	_, err = o.Insert(item)
	return
}

func ModifyLongzhongdata(item *Longzhongdata) (err error) {
	o := orm.NewOrm()
	o.Using("edb")
	sql := ` UPDATE  longzhongdata
			SET
			  close = ?,
			  modify_time= NOW(),
			  unit_desc= ?,
			  upd_time = ?
			WHERE longzhonginfo_id = ? AND dt=?
			 `
	_, err = o.Raw(sql, item.Close, item.UnitDesc, item.UpdTime, item.LongzhonginfoId, item.Dt).Exec()
	return
}

func AddLongzhonginfo(item *Longzhonginfo) (newId int64, err error) {
	o := orm.NewOrm()
	o.Using("edb")
	newId, err = o.Insert(item)
	return
}

func ModifyLongzhonginfo(item *Longzhonginfo) (err error) {
	o := orm.NewOrm()
	o.Using("edb")
	sql := `UPDATE  longzhonginfo
		SET
		  sec_name = ?,
		  unit = ?,
		  frequency = ?,
		  classify_id = ?,
		  classify_name = ?,
		  remark = ?,
		  last_modify_date = ?,
		  temp_id = ?,
		  temp_name = ?,
		  is_normal = ?,
		  modify_time=NOW()
		WHERE unitid = ? `
	_, err = o.Raw(sql, item.SecName, item.Unit, item.Frequency, item.ClassifyId, item.ClassifyName, item.Remark, item.LastModifyDate, item.TempId, item.TempName, item.IsNormal, item.Unitid).Exec()
	return
}

//判断指标数据是否已经录入
func GetLongzhonginfoCount(classifyId int, unitid int64) (count int, err error) {
	o := orm.NewOrm()
	o.Using("edb")
	sql := `SELECT COUNT(1) AS count FROM longzhonginfo WHERE classify_id=? AND unitid=? `
	err = o.Raw(sql, classifyId, unitid).QueryRow(&count)
	return
}

type LzProductInfoResp struct {
	Msg  string           `json:"msg"`
	Code string           `json:"code"`
	Data []*LzProductInfo `json:"data"`
}

type LzProductInfo struct {
	Unitid      int64  `json:"unitid"`
	ProUnitName string `json:"proUnitName"`
	Tempid      int64  `json:"tempid"`
	TempName    string `json:"tempName"`
	Proid       int    `json:"proid"`
	ProName     string `json:"proName"`
	Unit        string `json:"unit"`
	LzType      string `json:"type"`
	Maxdate     string `json:"maxdate"`
	IsNormal    string `json:"isNormal"`
}

func GetLongzhonginfoMaxId() (count int, err error) {
	o := orm.NewOrm()
	o.Using("edb")
	sql := `SELECT MAX(longzhonginfo_id) AS count FROM longzhonginfo`
	err = o.Raw(sql).QueryRow(&count)
	return
}

type LzProductInfoDetail struct {
	UnitId    int64   `json:"unit_id"`
	UnitName  string  `json:"unit_name"`
	TempId    int     `json:"temp_id"`
	TempName  string  `json:"temp_name"`
	TempUnit  string  `json:"temp_unit"`
	ProId     int     `json:"pro_id"`
	ProCnName string  `json:"pro_cn_name"`
	UnitValue float64 `json:"unit_value"`
	UnitDesc  string  `json:"unit_desc"`
	DataTime  string  `json:"data_time"`
	UpdTime   string  `json:"upd_time"`
}

type LzProductInfoDetailResp struct {
	Msg      string                 `json:"msg"`
	Code     string                 `json:"code"`
	Data     []*LzProductInfoDetail `json:"data"`
	Pagesize int                    `json:"pagesize"`
	Page     int                    `json:"page"`
}

func GetLongzhonginfoByUnitId(classifyName string, unitId int64) (item *Longzhongdata, err error) {
	o := orm.NewOrm()
	o.Using("edb")
	sql := `SELECT * FROM longzhonginfo WHERE unitid=? AND classify_name=?`
	err = o.Raw(sql, unitId, classifyName).QueryRow(&item)
	return
}

func GetLongzhongDataById(lzInfoId int) (items []*Longzhongdata, err error) {
	o := orm.NewOrm()
	o.Using("edb")
	sql := `SELECT * FROM longzhongdata WHERE longzhonginfo_id=? ORDER BY dt DESC `
	_, err = o.Raw(sql, lzInfoId).QueryRows(&items)
	return
}

func GetLongzhongDataMaxCount(classifyId int) (count int, err error) {
	o := orm.NewOrm()
	o.Using("edb")
	sql := `SELECT MAX(t.num) AS count FROM (
				SELECT COUNT(1) AS num  FROM longzhonginfo AS a
				INNER JOIN longzhongdata AS b ON a.longzhonginfo_id=b.longzhonginfo_id
				WHERE a.classify_id=?
				GROUP BY a.longzhonginfo_id
			)AS t `
	err = o.Raw(sql, classifyId).QueryRow(&count)
	return
}

type LongzhonginfoList struct {
	LongzhonginfoId int `orm:"column(longzhonginfo_id);pk"`
	TradeCode       string
	SecName         string
	Frequency       string
	ClassifyId      int
	ClassifyName    string
	Unit            string
	IsNormal        string
	LastModifyDate  string
	Unitid          int
}

func GetLongzhonginfoList() (items []*LongzhonginfoList, err error) {
	o := orm.NewOrm()
	o.Using("edb")
	sql := ` SELECT * FROM longzhonginfo WHERE is_normal=1 `
	_, err = o.Raw(sql).QueryRows(&items)
	return
}

func ModifyLongzhonginfoIsNormal(longzhonginfoId int) (err error) {
	o := orm.NewOrm()
	o.Using("edb")
	sql := ` UPDATE longzhonginfo SET is_normal=0 WHERE longzhonginfo_id=? `
	_, err = o.Raw(sql,longzhonginfoId).Exec()
	return
}