package models import ( "eta_gn/eta_api/global" "eta_gn/eta_api/models/data_manage" "eta_gn/eta_api/utils" "fmt" "gorm.io/gorm" "sort" "strconv" "strings" "time" "github.com/rdlucklib/rdluck_tools/paging" ) type DataList struct { TradeCode string `gorm:"column:TRADE_CODE;type:varchar(255);not null" description:"指标编码"` SecName string `gorm:"column:SEC_NAME;type:varchar(255);not null" description:"指标名称"` Unit string `gorm:"column:UNIT;type:varchar(255);not null" description:"单位"` Remark string `gorm:"column:REMARK;type:varchar(255);not null" description:"备注"` Frequency string `gorm:"type:varchar(255);not null" description:"频度"` ClassifyId int `gorm:"type:int;not null" description:"分类id"` ClassifyName string `gorm:"type:varchar(255);not null" description:"分类名称"` Dt string `gorm:"column:DT;type:varchar(255);not null" description:"录入日期"` Close float64 `gorm:"column:CLOSE;type:float;not null" description:"录入值"` ModifyTime string `gorm:"type:varchar(255);not null" description:"修改时间"` } type DataListResp struct { List []*DataList Paging *paging.PagingItem `description:"分页数据"` } func GetDataList(condition string, pars []interface{}, startSize, pageSize int) (items []*DataList, err error) { sql := `select a.TRADE_CODE,a.SEC_NAME,a.UNIT,a.frequency,a.classify_id,b.classify_name,c.DT,c.CLOSE,c.modify_time FROM edbdata AS c inner join edbinfo AS a ON a.TRADE_CODE=c.TRADE_CODE left join edbdata_classify AS b ON a.classify_id=b.classify_id where a.classify_id>0` if condition != "" { sql += condition } sql += ` order by c.DT desc limit ?,? ` //o := orm.NewOrmUsingDB("edb") //_, err = o.Raw(sql, pars, startSize, pageSize).QueryRows(&items) err = global.DmSQL["edb"].Raw(sql, pars, startSize, pageSize).Find(&items).Error return } func GetDataListCount(condition string, pars []interface{}) (count int, err error) { sql := ` select count(1) as count FROM edbdata AS c inner join edbinfo AS a ON a.TRADE_CODE=c.TRADE_CODE left join edbdata_classify AS b ON a.classify_id=b.classify_id where a.classify_id>0 ` if condition != "" { sql += condition } //o := orm.NewOrmUsingDB("edb") //err = o.Raw(sql, pars).QueryRow(&count) err = global.DmSQL["edb"].Raw(sql, pars).Scan(&count).Error return } type DataAddReq struct { TradeCode string `description:"指标唯一编码"` CreateDate string `description:"创建日期"` Close string `description:"录入值"` } type Edbdata struct { TradeCode string `gorm:"column:TRADE_CODE;primaryKey;type:varchar(255)" orm:"column(TRADE_CODE);pk" gorm:"primaryKey" description:"指标编码"` Dt string `gorm:"column:DT;type:varchar(255)" orm:"column(DT)" description:"日期"` Close string `gorm:"column:CLOSE;type:varchar(255)" orm:"column(CLOSE)" description:"值"` ModifyTime time.Time `gorm:"column:modify_time;type:timestamp" orm:"column(modify_time)" description:"修改时间"` } func GetDataInfo(tradeCode, creteDate string) (item *Edbdata, err error) { sql := " SELECT * FROM edbdata WHERE TRADE_CODE=? AND DT=? " //o := orm.NewOrmUsingDB("edb") //err = o.Raw(sql, tradeCode, creteDate).QueryRow(&item) err = global.DmSQL["edb"].Raw(sql, tradeCode, creteDate).First(&item).Error return } func AddEdbdata(item *Edbdata) (lastId int64, err error) { //o := orm.NewOrmUsingDB("edb") //lastId, err = o.Insert(item) err = global.DmSQL["edb"].Create(item).Error if err != nil { return } return } type DataEditReq struct { TradeCode string `description:"指标唯一编码"` CreateDate string `description:"创建日期"` Close interface{} `description:"录入值"` OldCreateDate string `description:"旧的录入日期"` } // BatchDataEditReq 批量修改指标 type BatchDataEditReq struct { OldCreateDate string `description:"旧的录入日期"` CreateDate string `description:"新的录入日期"` List []DataEditReq `description:"需要修改的数据"` } // 编辑数据 func EditEdbdata(item *Edbdata) (err error) { //o := orm.NewOrmUsingDB("edb") sql := ` UPDATE edbdata SET CLOSE = ?,modify_time=NOW() WHERE TRADE_CODE = ? AND DT = ? ` //_, err = o.Raw(sql, item.Close, item.TradeCode, item.Dt).Exec() err = global.DmSQL["edb"].Exec(sql, item.Close, item.TradeCode, item.Dt).Error return } type EdbdataDeleteRecord struct { Id int `orm:"column(id);pk" gorm:"primaryKey" ` TradeCode string `orm:"column(TRADE_CODE)" description:"指标编码"` Dt string `orm:"column(DT)" description:"日期"` Close string `orm:"column(CLOSE)" description:"值"` ModifyTime time.Time `orm:"column(modify_time)" description:"修改时间"` CreateTime time.Time SysUserId int } func AddEdbdataDeleteRecord(item *EdbdataDeleteRecord) (lastId int64, err error) { //o := orm.NewOrmUsingDB("edb") //lastId, err = o.Insert(item) err = global.DmSQL["edb"].Create(item).Error lastId = int64(item.Id) return } // DeleteEdbData 根据指标code和日期删除数据 func DeleteEdbData(tradeCode, dt string) (err error) { //o := orm.NewOrmUsingDB("edb") sql := ` DELETE FROM edbdata WHERE TRADE_CODE = ? AND DT = ? ` //_, err = o.Raw(sql, tradeCode, dt).Exec() err = global.DmSQL["edb"].Exec(sql, tradeCode, dt).Error return } // DeleteAllEdbData 根据指标code删除数据 func DeleteAllEdbData(tradeCode string) (err error) { //o := orm.NewOrmUsingDB("edb") sql := ` DELETE FROM edbdata WHERE TRADE_CODE = ? ` //_, err = o.Raw(sql, tradeCode).Exec() err = global.DmSQL["edb"].Exec(sql, tradeCode).Error return } type Edbinfo struct { TradeCode string `gorm:"column:TRADE_CODE;primaryKey;type:varchar(255)" orm:"column(TRADE_CODE);pk" gorm:"primaryKey" description:"指标code"` SecName string `gorm:"column:SEC_NAME;type:varchar(255)" orm:"column(SEC_NAME);" description:"指标名称"` Unit string `gorm:"column:UNIT;type:varchar(255)" orm:"column(UNIT);" description:"单位"` Remark string `gorm:"column:REMARK;type:varchar(255)" orm:"column(REMARK);" description:"备注"` Frequency string `gorm:"column:FREQUENCY;type:varchar(255)" orm:"column(frequency)" description:"频度"` ClassifyId int `gorm:"column:CLASSIFY_ID;type:int" orm:"column(classify_id)" description:"分类id"` ClassifyName string `gorm:"-" orm:"-" description:"分类名称"` CreateDate string `gorm:"column:CREATE_DATE;type:varchar(255)" orm:"column(create_date)" description:"创建时间"` UserId int `gorm:"column:USER_ID;type:int" orm:"column(user_id)" description:"录入用户id"` UserName string `gorm:"column:USER_NAME;type:varchar(255)" orm:"column(user_name)" description:"录入用户名称"` NoticeTime string `gorm:"column:NOTICE_TIME;type:varchar(255)" orm:"column(notice_time)" description:"通知时间"` Mobile string `gorm:"column:MOBILE;type:varchar(255)" orm:"column(mobile)" description:"录入者手机号"` Sort int `gorm:"column:SORT;type:int" orm:"column(sort)" description:"排序"` ModifyTime string `gorm:"column:MODIFY_TIME;type:varchar(255)" description:"最近一次更新时间"` IsJoinEdb int8 `gorm:"column:IS_JOIN_EDB;type:tinyint" description:"指标库是否已添加:0-否;1-是"` StartDate string `gorm:"column:START_DATE;type:varchar(255)" description:"数据开始日期"` EndDate string `gorm:"column:END_DATE;type:varchar(255)" description:"数据结束日期"` LatestValue float64 `gorm:"column:LATEST_VALUE;type:decimal(10,2)" description:"指标最新值"` } func DeleteEdbinfoByTraceCodeList(tradeCodeList []string) (err error) { if len(tradeCodeList) == 0 { return } //o := orm.NewOrmUsingDB("edb") o := global.DmSQL["edb"] //err = o.DoTx(func(ctx context.Context, txOrm orm.TxOrmer) error { // var holder []string // for range tradeCodeList { // holder = append(holder, "?") // } // sql := ` DELETE FROM edbdata WHERE TRADE_CODE in (` + strings.Join(holder, ",") + `) ` // _, err := txOrm.Raw(sql, tradeCodeList).Exec() // if err != nil { // return err // } // sql = ` DELETE FROM edbinfo WHERE TRADE_CODE in (` + strings.Join(holder, ",") + `)` // _, err = txOrm.Raw(sql, tradeCodeList).Exec() // if err != nil { // return err // } // return nil //}) err = o.Transaction(func(tx *gorm.DB) error { var holder []string for range tradeCodeList { holder = append(holder, "?") } sql := ` DELETE FROM edbdata WHERE TRADE_CODE in (` + strings.Join(holder, ",") + `) ` txErr := tx.Exec(sql, tradeCodeList).Error if txErr != nil { return txErr } sql = ` DELETE FROM edbinfo WHERE TRADE_CODE in (` + strings.Join(holder, ",") + `)` txErr = tx.Exec(sql, tradeCodeList).Error if txErr != nil { return txErr } return nil }) return } func GetEdbinfoListCount(condition string, pars []interface{}, mobile string, roleType int) (count int, err error) { //o := orm.NewOrmUsingDB("edb") sql := `` if mobile != "" && roleType == 1 { sql = `SELECT COUNT(1) AS count FROM edbinfo AS a INNER JOIN edbinfo_user AS c ON a.TRADE_CODE=c.TRADE_CODE AND c.mobile=? WHERE a.classify_id>0` if condition != "" { sql += condition } //err = o.Raw(sql, mobile, pars).QueryRow(&count) err = global.DmSQL["edb"].Raw(sql, mobile, pars).Scan(&count).Error } else { sql = `SELECT COUNT(1) AS count FROM edbinfo AS a WHERE a.classify_id>0` if condition != "" { sql += condition } //err = o.Raw(sql, pars).QueryRow(&count) err = global.DmSQL["edb"].Raw(sql, pars).Scan(&count).Error } return } type EdbinfoItem struct { TradeCode string `gorm:"column:TRADE_CODE;primaryKey;type:varchar(255)"` //`orm:"column(TRADE_CODE);pk" gorm:"primaryKey" description:"指标code"` SecName string `gorm:"column:SEC_NAME;type:varchar(255)"` //`orm:"column(SEC_NAME);" description:"指标名称"` Unit string `gorm:"column:UNIT;type:varchar(255)"` //`orm:"column(UNIT);" description:"单位"` Remark string `gorm:"column:REMARK;type:varchar(255)"` //`orm:"column(REMARK);" description:"备注"` Frequency string `gorm:"column:frequency;"` //`description:"频度"` ClassifyId int `gorm:"column:classify_id;type:int"` //`description:"分类id"` ClassifyName string `gorm:"-"` // `description:"分类名称"` CreateDate string `gorm:"column:create_date;"` // `description:"创建时间"` UserId int `gorm:"column:user_id;"` // `description:"录入用户id"` UserName string `gorm:"column:user_name;"` // `description:"录入用户名称"` NoticeTime string `gorm:"column:notice_time;"` //`description:"通知时间"` Mobile string `gorm:"column:mobile"` // `description:"录入者手机号"` ModifyTime string `gorm:"column:modify_time;"` //`// `description:"最近一次更新时间"` StartDate string `gorm:"column:start_date;"` // `description:"数据开始日期"` EndDate string `gorm:"column:end_date;"` // `description:"数据结束日期"` LatestValue float64 `gorm:"column:latest_value;"` // `description:"指标最新值"` } func GetEdbinfoItemList(condition string, pars []interface{}, startSize, pageSize int, mobile string, roleType int) (items []*EdbinfoItem, err error) { //o := orm.NewOrmUsingDB("edb") sql := `` if mobile != "" && roleType == 1 { sql = ` SELECT DISTINCT a.*,b.classify_name FROM edbinfo AS a LEFT JOIN edbdata_classify AS b ON a.classify_id=b.classify_id INNER JOIN edbinfo_user AS c ON a.TRADE_CODE=c.TRADE_CODE AND c.mobile=? WHERE a.classify_id>0` if condition != "" { sql += condition } sql += ` ORDER BY a.create_date DESC LIMIT ?,? ` //_, err = o.Raw(sql, mobile, pars, startSize, pageSize).QueryRows(&items) err = global.DmSQL["edb"].Raw(sql, mobile, pars, startSize, pageSize).Find(&items).Error } else { sql = `SELECT DISTINCT a.*,b.classify_name FROM edbinfo AS a LEFT JOIN edbdata_classify AS b on a.classify_id=b.classify_id WHERE a.classify_id>0` if condition != "" { sql += condition } sql += ` ORDER BY a.create_date DESC LIMIT ?,? ` //_, err = o.Raw(sql, pars, startSize, pageSize).QueryRows(&items) err = global.DmSQL["edb"].Raw(sql, pars, startSize, pageSize).Find(&items).Error } return } // EdbParamsInfo 指标数据结构体 type EdbParamsInfo struct { Unit string `gorm:"column:UNIT;"` //`orm:"column(UNIT);" description:"单位"` Frequency string `gorm:"column:frequency;"` //`orm:"column(frequency);" description:"单位"` } // GetEdbUnitList 获取指标单位 func GetEdbUnitList() (items []*EdbParamsInfo, err error) { //o := orm.NewOrmUsingDB("edb") sql := `SELECT UNIT from edbinfo group by UNIT` //_, err = o.Raw(sql).QueryRows(&items) err = global.DmSQL["edb"].Raw(sql).Find(&items).Error return } // GetEdbFrequencyList 获取指标频度 func GetEdbFrequencyList(classifyId, userId int) (items []*EdbParamsInfo, err error) { //o := orm.NewOrmUsingDB("edb") sql := `SELECT frequency from edbinfo a join edbdata b on a.TRADE_CODE=b.TRADE_CODE where classify_id = ? ` if userId > 0 { sql += ` and a.user_id = ` + fmt.Sprint(userId) + ` ` } sql += ` group by a.frequency` //_, err = o.Raw(sql, classifyId).QueryRows(&items) err = global.DmSQL["edb"].Raw(sql, classifyId).Find(&items).Error return } type TargetListResp struct { List []*EdbinfoItem Paging *paging.PagingItem `description:"分页数据"` } type EdbinfoAddReq struct { SecName string `description:"指标名称"` Unit string `description:"单位"` Frequency string `description:"频度"` ClassifyId int `description:"分类id"` NoticeTime string `description:"通知时间"` } // GetMaxTradeCode 获取指标最大trade_code func GetMaxTradeCode() (max_trade_code string, err error) { sql := " SELECT MAX(TRADE_CODE) AS max_trade_code FROM edbinfo WHERE LEFT(TRADE_CODE,1)='W' " //o := orm.NewOrmUsingDB("edb") //err = o.Raw(sql).QueryRow(&max_trade_code) err = global.DmSQL["edb"].Raw(sql).Scan(&max_trade_code).Error if (err != nil && utils.IsErrNoRow(err)) || max_trade_code == `` { max_trade_code = "W00" } return } func GetEdbinfoBySecName(secName string) (item *Edbinfo, err error) { sql := `SELECT * FROM edbinfo WHERE SEC_NAME=? ` //o := orm.NewOrmUsingDB("edb") //err = o.Raw(sql, secName).QueryRow(&item) err = global.DmSQL["edb"].Raw(sql, secName).First(&item).Error return } func GetEdbinfoByTradeCode(tradeCode string) (item *Edbinfo, err error) { sql := `SELECT * FROM edbinfo WHERE TRADE_CODE=? ` //o := orm.NewOrmUsingDB("edb") //err = o.Raw(sql, tradeCode).QueryRow(&item) err = global.DmSQL["edb"].Raw(sql, tradeCode).First(&item).Error return } func AddEdbinfo(tradeCode, secName, unit, remark, frequency, noticeTime string, classifyId int, userId int, userName string) (err error) { //o := orm.NewOrmUsingDB("edb") currTime := time.Now().Format(utils.FormatDateTime) edbInfo := &Edbinfo{ TradeCode: tradeCode, SecName: secName, Unit: unit, Remark: remark, Frequency: frequency, ClassifyId: classifyId, CreateDate: currTime, UserId: userId, UserName: userName, NoticeTime: noticeTime, Mobile: "", ModifyTime: currTime, IsJoinEdb: 0, } //_, err = o.Insert(edbInfo) err = global.DmSQL["edb"].Create(edbInfo).Error return } func AddEdbinfoUser(tradeCode, mobile string) (err error) { //o := orm.NewOrmUsingDB("edb") sql := `INSERT INTO edbinfo_user(TRADE_CODE, mobile) VALUES (?,?)` //_, err = o.Raw(sql, tradeCode, mobile).Exec() err = global.DmSQL["edb"].Exec(sql, tradeCode, mobile).Error return } type EdbinfoEditReq struct { TradeCode string `description:"指标code"` SecName string `description:"指标名称"` Unit string `description:"单位"` Frequency string `description:"频度"` ClassifyId int `description:"分类id"` NoticeTime string `description:"通知时间"` } func EditEdbinfo(tradeCode, secName, unit, frequency, noticeTime string, classifyId int) (err error) { sql := `UPDATE edbinfo SET SEC_NAME= ?, UNIT = ?,classify_id=?,frequency=?,notice_time=?,create_date=NOW() WHERE TRADE_CODE=? ` //o := orm.NewOrmUsingDB("edb") //_, err = o.Raw(sql, secName, unit, classifyId, frequency, noticeTime, tradeCode).Exec() err = global.DmSQL["edb"].Exec(sql, secName, unit, classifyId, frequency, noticeTime, tradeCode).Error return } func SearchTargetEntry(classifyId int, keyWord string) (items []*Edbinfo, err error) { where := "" pars := make([]interface{}, 0) sql := `SELECT * FROM edbinfo WHERE classify_id>0 AND classify_id=? ` pars = append(pars, classifyId) if keyWord != "" { sql += `AND SEC_NAME LIKE ? ` pars = utils.GetLikeKeywordPars(pars, keyWord, 1) } sql += where //o := orm.NewOrmUsingDB("edb") //_, err = o.Raw(sql, pars...).QueryRows(&items) err = global.DmSQL["edb"].Raw(sql, pars).Find(&items).Error return } type SearchTargetListResp struct { List []*Edbinfo } type EdbdataClassify struct { ClassifyId int `gorm:"column:classify_id;type:int" xorm:"int 'classify_id'"` ClassifyName string `gorm:"column:classify_name;type:varchar(255)" xorm:"varchar(255) 'classify_name'"` ParentId int `gorm:"column:parent_id;type:int" xorm:"int 'parent_id'"` EdbInfoTotal int `gorm:"column:edb_info_total;type:int" xorm:"int 'edb_info_total'"` TradeCode string `gorm:"column:trade_code;type:varchar(255)" xorm:"varchar(255) 'trade_code'"` UniqueCode string `gorm:"column:unique_code;type:varchar(255)" xorm:"varchar(255) 'unique_code'"` } func GetEdbdataClassifyByClassifyName(classifyName string) (item *EdbdataClassify, err error) { sql := `SELECT * FROM edbdata_classify WHERE classify_name=? ` //o := orm.NewOrmUsingDB("edb") //err = o.Raw(sql, classifyName).QueryRow(&item) err = global.DmSQL["edb"].Raw(sql, classifyName).First(&item).Error return } type EdbdataClassifyList struct { ClassifyId int ClassifyName string ParentId int Child []*EdbdataClassify TradeCode string UniqueCode string } func GetEdbdataClassify(userId int64) (items []*EdbdataClassifyList, err error) { var newItems []*EdbdataClassifyList //o := orm.NewOrmUsingDB("edb") sql := ` SELECT classify_id,classify_name,parent_id FROM edbdata_classify WHERE parent_id=0 ` //_, err = o.Raw(sql).QueryRows(&newItems) err = global.DmSQL["edb"].Raw(sql).Find(&newItems).Error if err != nil { return } classifyLen := len(newItems) for i := 0; i < classifyLen; i++ { var childItems []*EdbdataClassify parentId := newItems[i].ClassifyId childSql := `` if userId > 0 { userClassifyList, _ := GetManualUserClassify(int(userId)) var userIdArr []string for _, v := range userClassifyList { userIdArr = append(userIdArr, strconv.Itoa(v.ClassifyId)) } userIdStr := strings.Join(userIdArr, ",") if userIdStr != "" { childSql = "SELECT a.classify_id,a.classify_name,a.parent_id FROM edbdata_classify AS a WHERE a.is_show=1 and a.classify_id IN(" + userIdStr + ") AND parent_id=? ORDER BY a.create_time ASC " // _, err = o.Raw(childSql, parentId).QueryRows(&childItems) err = global.DmSQL["edb"].Raw(childSql, parentId).Find(&childItems).Error } } else { childSql = "SELECT classify_id,classify_name,parent_id FROM edbdata_classify WHERE is_show=1 and parent_id=? ORDER BY create_time ASC " //_, err = o.Raw(childSql, parentId).QueryRows(&childItems) err = global.DmSQL["edb"].Raw(childSql, parentId).Find(&childItems).Error } if err != nil { return } newItems[i].Child = childItems } for _, v := range newItems { childLen := len(v.Child) if childLen > 0 { items = append(items, v) } } return } type ManualUserClassify struct { ManualUserClassifyId int `gorm:"column:manual_user_classify_id"` //`orm:"column(manual_user_classify_id);pk" gorm:"primaryKey" ` AdminId int `gorm:"column:admin_id"` ClassifyId int `gorm:"column:classify_id"` CreateTime time.Time `gorm:"column:create_time"` } func GetManualUserClassify(sysUserId int) (list []*ManualUserClassify, err error) { //o := orm.NewOrmUsingDB("data") sql := `SELECT * FROM manual_user_classify WHERE admin_id=? ` //_, err = o.Raw(sql, sysUserId).QueryRows(&list) err = global.DmSQL["data"].Raw(sql, sysUserId).Find(&list).Error return } type EdbdataClassifyResp struct { List []*EdbdataClassifyList } func GetTargetBySecName(secName string) (item *Edbinfo, err error) { sql := `SELECT * FROM edbinfo WHERE SEC_NAME=? ` //o := orm.NewOrmUsingDB("edb") //err = o.Raw(sql, secName).QueryRow(&item) err = global.DmSQL["edb"].Raw(sql, secName).First(&item).Error return } // 更新指标数据信息 func (edbinfo *Edbinfo) Update(cols []string) (err error) { //o := orm.NewOrmUsingDB("edb") //_, err = o.Update(edbinfo, cols...) err = global.DmSQL["edb"].Select(cols).Updates(edbinfo).Error return } func ModifyTargetClassifyId(tradeCode string, classifyId int) (err error) { sql := `UPDATE edbinfo SET classify_id=? WHERE TRADE_CODE=? ` //o := orm.NewOrmUsingDB("edb") //_, err = o.Raw(sql, classifyId, tradeCode).Exec() err = global.DmSQL["edb"].Exec(sql, classifyId, tradeCode).Error return } func GetTargetsDataCount(tradeCode, dt string) (count int, err error) { sql := `SELECT COUNT(1) AS count FROM edbdata WHERE TRADE_CODE=? AND DT=? ` //o := orm.NewOrmUsingDB("edb") //err = o.Raw(sql, tradeCode, dt).QueryRow(&count) err = global.DmSQL["edb"].Raw(sql, tradeCode, dt).Scan(&count).Error return } // GetTargetsDataList 根据code获取指标数据列表 func GetTargetsDataList(tradeCode string) (items []*Edbdata, err error) { //o := orm.NewOrmUsingDB("edb") sql := `SELECT * FROM edbdata WHERE TRADE_CODE=? ORDER BY DT ASC ` //_, err = o.Raw(sql, tradeCode).QueryRows(&items) err = global.DmSQL["edb"].Raw(sql, tradeCode).Find(&items).Error return } func GetTargetsData(tradeCode, dt string) (item *Edbdata, err error) { sql := `SELECT * FROM edbdata WHERE TRADE_CODE=? AND DT=? ` //o := orm.NewOrmUsingDB("edb") //err = o.Raw(sql, tradeCode, dt).QueryRow(&item) err = global.DmSQL["edb"].Raw(sql, tradeCode, dt).First(&item).Error return } func ModifyTargetsDataByImport(tradeCode, dt, close string) (err error) { sql := `UPDATE edbdata SET CLOSE=?,modify_time=NOW() WHERE TRADE_CODE=? AND DT=? ` //o := orm.NewOrmUsingDB("edb") //_, err = o.Raw(sql, close, tradeCode, dt).Exec() err = global.DmSQL["edb"].Exec(sql, close, tradeCode, dt).Error return } func AddTargetsDataByImport(tradeCode, dt, close string) (err error) { sql := `INSERT INTO edbdata(TRADE_CODE, DT,CLOSE, modify_time)VALUES(?,?,?,NOW()) ` //o := orm.NewOrmUsingDB("edb") //_, err = o.Raw(sql, tradeCode, dt, close).Exec() err = global.DmSQL["edb"].Exec(sql, tradeCode, dt, close).Error return } type EdbdataImportResp struct { Status int Msg string SuccessCount int FailCount int } func GetFailList(sysUserId int) (items []*EdbdataImportFail, err error) { //o := orm.NewOrmUsingDB("edb") sql := ` SELECT * FROM edbdata_import_fail WHERE sys_user_id=? ` //_, err = o.Raw(sql, sysUserId).QueryRows(&items) err = global.DmSQL["edb"].Raw(sql, sysUserId).Find(&items).Error return } type DataListForExport struct { TradeCode string `gorm:"column:TRADE_CODE"` //`orm:"column(TRADE_CODE)" description:"指标code"` SecName string `gorm:"column:SEC_NAME"` //`orm:"column(SEC_NAME)" description:"指标名称"` Unit string `gorm:"column:UNIT"` //`orm:"column(UNIT)" description:"单位"` Frequency string `gorm:"column:frequency"` //`description:"频度"` ClassifyId int `gorm:"column:classify_id"` //`description:"分类id"` NoticeTime string `gorm:"column:notice_time"` //`description:"通知时间"` ClassifyName string `gorm:"column:classify_name"` Dt string `gorm:"column:DT"` //`orm:"column(DT)" description:"日期"` Close float64 `gorm:"column:CLOSE"` //`orm:"column(CLOSE)" description:"值"` } func GetDataListForExport(startDate, endDate, frequency, keyWord string, classifyId int) (items []*DataListForExport, err error) { where := `` var pars []interface{} if keyWord != "" { where = ` AND SEC_NAME LIKE ? ` pars = utils.GetLikeKeywordPars(pars, keyWord, 1) } if startDate != "" { where += ` AND create_date>=? ` pars = append(pars, startDate) } if endDate != "" { where += ` AND create_date<=? ` pars = append(pars, endDate) } if frequency != "" { where += ` AND frequency=? ` pars = append(pars, frequency) } if classifyId > 0 { where += ` AND classify_id=? ` pars = append(pars, classifyId) } sql := ` SELECT a.TRADE_CODE,a.SEC_NAME,a.UNIT,a.frequency,a.classify_id,b.classify_name,c.DT,c.CLOSE FROM edbdata AS c INNER JOIN edbinfo AS a ON a.TRADE_CODE=c.TRADE_CODE LEFT JOIN edbdata_classify AS b ON a.classify_id=b.classify_id WHERE a.classify_id>0 ` if where != "" { sql += where } sql = sql + " ORDER BY c.DT DESC " //o := orm.NewOrmUsingDB("edb") //_, err = o.Raw(sql, pars).QueryRows(&items) err = global.DmSQL["edb"].Raw(sql, pars).Find(&items).Error return } type DataDeleteReq struct { TradeCode string `description:"指标唯一编码"` CreateDate string `description:"数据录入日期"` } func DataDelete(tradeCode, createDate, close string, modifyTime time.Time, sysUserId int) (err error) { //o := orm.NewOrmUsingDB("edb") //to, err := o.Begin() to := global.DmSQL["edb"].Begin() if err != nil { return } defer func() { if err != nil { _ = to.Rollback() } else { _ = to.Commit() } }() recordSql := ` INSERT INTO edbdata_delete_record(TRADE_CODE,DT,CLOSE,modify_time,create_time,sys_user_id) VALUES(?,?,?,?,?,?)` //_, err = to.Raw(recordSql, tradeCode, createDate, close, modifyTime, time.Now(), sysUserId).Exec() err = to.Exec(recordSql, tradeCode, createDate, close, modifyTime, time.Now(), sysUserId).Error sql := ` DELETE FROM edbdata WHERE TRADE_CODE = ? AND DT = ? ` //_, err = to.Raw(sql, tradeCode, createDate).Exec() err = to.Exec(sql, tradeCode, createDate).Error return } func GetTargetInfoCount(tradeCode string) (count int, err error) { sql := ` SELECT COUNT(1) AS count FROM edbdata AS c INNER JOIN edbinfo AS a ON a.TRADE_CODE=c.TRADE_CODE WHERE a.TRADE_CODE=? ` //o := orm.NewOrmUsingDB("edb") //err = o.Raw(sql, tradeCode).QueryRow(&count) err = global.DmSQL["edb"].Raw(sql, tradeCode).Scan(&count).Error return } type TargetDeleteReq struct { TradeCode string `description:"指标唯一编码"` } func TargetDelete(tradeCode string) (err error) { //o := orm.NewOrmUsingDB("edb") //to, err := o.Begin() to := global.DmSQL["edb"].Begin() if err != nil { return } defer func() { if err != nil { _ = to.Rollback() } else { _ = to.Commit() } }() sql := " DELETE FROM edbinfo WHERE TRADE_CODE = ? " //_, err = to.Raw(sql, tradeCode).Exec() err = to.Exec(sql, tradeCode).Error sql = " DELETE FROM edbdata WHERE TRADE_CODE = ? " //_, err = to.Raw(sql, tradeCode).Exec() err = to.Exec(sql, tradeCode).Error return } type Researcher struct { AdminId int `gorm:"column:admin_id"` //`description:"系统用户id"` AdminName string `gorm:"column:admin_name"` //`description:"系统用户名称"` RealName string `gorm:"column:real_name"` //`description:"系统用户姓名"` Role string `gorm:"column:role"` //`description:"系统用户角色"` Mobile string `gorm:"column:mobile"` //`description:"手机号"` TargetCount int `gorm:"column:target_count"` //`description:"指标数量"` } type ResearcherListResp struct { List []*Researcher } func GetResearcherEntry() (items []*Researcher, err error) { sql := ` SELECT admin_id,admin_name,real_name,mobile,0 as target_count FROM admin WHERE role_type=1 ` //o := orm.NewOrm() //_, err = o.Raw(sql).QueryRows(&items) err = global.DmSQL["edb"].Raw(sql).Find(&items).Error researchLen := len(items) //edbO := orm.NewOrmUsingDB("edb") for i := 0; i < researchLen; i++ { var count int mobile := items[i].Mobile sqlCount := ` SELECT COUNT(DISTINCT a.TRADE_CODE) AS count FROM edbinfo_user AS a INNER JOIN edbinfo AS b ON a.TRADE_CODE=b.TRADE_CODE WHERE a.mobile=? AND b.classify_id>0 ` //err = edbO.Raw(sqlCount, mobile).QueryRow(&count) err = global.DmSQL["edb"].Raw(sqlCount, mobile).Scan(&count).Error items[i].TargetCount = count } return } func GetResearcherEntryByMobile(mobile string) (items []*Researcher, err error) { sql := ` SELECT admin_id,admin_name,real_name,mobile,0 as target_count FROM admin WHERE role_type=1 ` if mobile != "" { sql += ` AND mobile IN(` + mobile + `)` } //o := orm.NewOrm() //_, err = o.Raw(sql).QueryRows(&items) err = global.DmSQL["edb"].Raw(sql).Find(&items).Error researchLen := len(items) //edbO := orm.NewOrmUsingDB("edb") for i := 0; i < researchLen; i++ { var count int mobile := items[i].Mobile sqlCount := ` SELECT COUNT(DISTINCT a.TRADE_CODE) AS count FROM edbinfo_user AS a INNER JOIN edbinfo AS b ON a.TRADE_CODE=b.TRADE_CODE WHERE a.mobile=? AND b.classify_id>0 ` //err = edbO.Raw(sqlCount, mobile).QueryRow(&count) err = global.DmSQL["edb"].Raw(sqlCount, mobile).Scan(&count).Error items[i].TargetCount = count } return } type EdbinfoItems struct { TradeCode string `gorm:"column:TRADE_CODE"` //`orm:"column(TRADE_CODE);pk" gorm:"primaryKey" description:"指标code"` SecName string `gorm:"column:SEC_NAME"` //`orm:"column(SEC_NAME);" description:"指标名称"` Unit string `gorm:"column:UNIT"` //`orm:"column(UNIT);" description:"单位"` Remark string `gorm:"column:REMARK"` //`orm:"column(REMARK);" description:"备注"` Frequency string `gorm:"column:frequency"` //`description:"频度"` ClassifyId int `gorm:"column:classify_id"` //`description:"分类id"` ClassifyName string `gorm:"column:classify_name"` //`description:"分类名称"` CreateDate string `gorm:"column:create_date"` //`description:"创建时间"` UserId int `gorm:"column:user_id"` //`description:"录入用户id"` NoticeTime string `gorm:"column:notice_time"` //`description:"通知时间"` Mobile string `gorm:"column:mobile"` //`description:"录入者手机号"` ModifyDate string `gorm:"column:modify_date"` //`description:"待更新日期"` Status string `gorm:"column:status"` // `description:"状态:未完成/完成"` } type TargetItemsResp struct { List SortEdbInfo } type SortEdbInfo []EdbinfoItems func GetTargetItems(mobile string, classifyId int) (lastItems SortEdbInfo, err error) { var items []*EdbinfoItems //o := orm.NewOrmUsingDB("edb") //sql := ` SELECT *,'' modify_date,'' status FROM edbinfo AS a WHERE a.classify_id>0 ` sql := ` SELECT *,'' modify_date,'' STATUS FROM edbinfo AS a WHERE a.classify_id>0 ` if classifyId > 0 { sql += ` AND a.classify_id=` + strconv.Itoa(classifyId) + `` } sql += ` GROUP BY a.TRADE_CODE ` //if classifyId > 0 { // sql = ` SELECT *,'' modify_date,'' status FROM edbinfo AS a // WHERE a.classify_id=` + strconv.Itoa(classifyId) + ` AND a.classify_id>0 // GROUP BY a.TRADE_CODE ` //} sql = sql + ` ORDER BY CONVERT(a.SEC_NAME USING gbk ) COLLATE gbk_chinese_ci ASC ` //_, err = o.Raw(sql).QueryRows(&items) err = global.DmSQL["edb"].Raw(sql).Find(&items).Error if err != nil { return } itemsLen := len(items) nowWeek := time.Now().Weekday().String() fmt.Println(nowWeek) finishEdbInfo := SortEdbInfo{} unFinishEdbInfo := SortEdbInfo{} for i := 0; i < itemsLen; i++ { noticeTime := items[i].NoticeTime frequency := items[i].Frequency tradeCode := items[i].TradeCode if noticeTime != "" { if frequency == "周度" { noticeArr := strings.Split(noticeTime, " ") noticeWeek := noticeArr[0] fmt.Println(noticeWeek) addDay := 0 if nowWeek == "Sunday" { if noticeWeek == "周日" { addDay = 0 } else if noticeWeek == "周一" { addDay = 1 } else if noticeWeek == "周二" { addDay = 2 } else if noticeWeek == "周三" { addDay = 3 } else if noticeWeek == "周四" { addDay = 4 } else if noticeWeek == "周五" { addDay = 5 } else if noticeWeek == "周六" { addDay = 6 } else { addDay = 0 } } else if nowWeek == "Monday" { if noticeWeek == "周日" { addDay = 6 } else if noticeWeek == "周一" { addDay = 0 } else if noticeWeek == "周二" { addDay = 1 } else if noticeWeek == "周三" { addDay = 2 } else if noticeWeek == "周四" { addDay = 3 } else if noticeWeek == "周五" { addDay = 4 } else if noticeWeek == "周六" { addDay = 5 } else { addDay = 0 } } else if nowWeek == "Tuesday" { if noticeWeek == "周日" { addDay = 5 } else if noticeWeek == "周一" { addDay = 6 } else if noticeWeek == "周二" { addDay = 0 } else if noticeWeek == "周三" { addDay = 1 } else if noticeWeek == "周四" { addDay = 2 } else if noticeWeek == "周五" { addDay = 3 } else if noticeWeek == "周六" { addDay = 4 } else { addDay = 0 } } else if nowWeek == "Wednesday" { if noticeWeek == "周日" { addDay = 4 } else if noticeWeek == "周一" { addDay = 5 } else if noticeWeek == "周二" { addDay = 6 } else if noticeWeek == "周三" { addDay = 0 } else if noticeWeek == "周四" { addDay = 1 } else if noticeWeek == "周五" { addDay = 2 } else if noticeWeek == "周六" { addDay = 3 } else { addDay = 0 } } else if nowWeek == "Thursday" { if noticeWeek == "周日" { addDay = 3 } else if noticeWeek == "周一" { addDay = 4 } else if noticeWeek == "周二" { addDay = 5 } else if noticeWeek == "周三" { addDay = 6 } else if noticeWeek == "周四" { addDay = 0 } else if noticeWeek == "周五" { addDay = 1 } else if noticeWeek == "周六" { addDay = 2 } else { addDay = 0 } } else if nowWeek == "Friday" { if noticeWeek == "周日" { addDay = 2 } else if noticeWeek == "周一" { addDay = 3 } else if noticeWeek == "周二" { addDay = 4 } else if noticeWeek == "周三" { addDay = 5 } else if noticeWeek == "周四" { addDay = 6 } else if noticeWeek == "周五" { addDay = 0 } else if noticeWeek == "周六" { addDay = 1 } else { addDay = 0 } } else if nowWeek == "Saturday" { if noticeWeek == "周日" { addDay = 1 } else if noticeWeek == "周一" { addDay = 2 } else if noticeWeek == "周二" { addDay = 3 } else if noticeWeek == "周三" { addDay = 4 } else if noticeWeek == "周四" { addDay = 5 } else if noticeWeek == "周五" { addDay = 6 } else if noticeWeek == "周六" { addDay = 0 } else { addDay = 0 } } modifyDate := time.Now().AddDate(0, 0, addDay) modifyDateStr := modifyDate.Format(utils.FormatDate) items[i].ModifyDate = modifyDateStr modifyDateEndStr := modifyDate.AddDate(0, 0, -7).Format(utils.FormatDate) fmt.Println("addDay:", addDay) fmt.Println("modifyDateEndStr:", modifyDateEndStr) count := 0 sqlCount := ` SELECT COUNT(1) AS num FROM edbdata WHERE TRADE_CODE=? AND DT >= ? AND DT <= ? ` //err = o.Raw(sqlCount, tradeCode, modifyDateEndStr, modifyDateStr).QueryRow(&count) err = global.DmSQL["edb"].Raw(sqlCount, tradeCode, modifyDateEndStr, modifyDateStr).Scan(&count).Error if err != nil { return nil, err } if count > 0 { items[i].Status = "完成" finishEdbInfo = append(finishEdbInfo, *items[i]) } else { items[i].Status = "未完成" unFinishEdbInfo = append(unFinishEdbInfo, *items[i]) } } else if frequency == "日度" { items[i].Status = "完成" finishEdbInfo = append(finishEdbInfo, *items[i]) } else if frequency == "月度" { myYear := time.Now().Year() myMonth := time.Now().Format("01") startDate, endDate := utils.GetMonthStartAndEnd(strconv.Itoa(myYear), myMonth) count := 0 sqlCount := ` SELECT COUNT(1) AS num FROM edbdata WHERE TRADE_CODE=? AND DT >= ? AND DT <= ? ` //err = o.Raw(sqlCount, tradeCode, startDate, endDate).QueryRow(&count) err = global.DmSQL["edb"].Raw(sqlCount, tradeCode, startDate, endDate).Scan(&count).Error if err != nil { return nil, err } if noticeTime != "" { var modifyDateStr string strArr := strings.Split(noticeTime, "日") myYear := time.Now().Year() myMonth := time.Now().Format("01") modifyDateStr = strconv.Itoa(myYear) + "-" + myMonth + "-" + strArr[0] items[i].ModifyDate = modifyDateStr } if count > 0 { items[i].Status = "完成" finishEdbInfo = append(finishEdbInfo, *items[i]) } else { items[i].Status = "未完成" unFinishEdbInfo = append(unFinishEdbInfo, *items[i]) } } else { items[i].Status = "完成" finishEdbInfo = append(finishEdbInfo, *items[i]) } } else { if frequency == "月度" { myYear := time.Now().Year() myMonth := time.Now().Format("01") startDate, endDate := utils.GetMonthStartAndEnd(strconv.Itoa(myYear), myMonth) count := 0 sqlCount := ` SELECT COUNT(1) AS num FROM edbdata WHERE TRADE_CODE=? AND DT >= ? AND DT <= ? ` //err = o.Raw(sqlCount, tradeCode, startDate, endDate).QueryRow(&count) err = global.DmSQL["edb"].Raw(sqlCount, tradeCode, startDate, endDate).Scan(&count).Error if err != nil { return nil, err } if count > 0 { items[i].Status = "完成" finishEdbInfo = append(finishEdbInfo, *items[i]) } else { items[i].Status = "未完成" unFinishEdbInfo = append(unFinishEdbInfo, *items[i]) } } else { items[i].Status = "完成" finishEdbInfo = append(finishEdbInfo, *items[i]) } } } sort.Sort(SortByModifyDate{finishEdbInfo}) sort.Sort(SortByModifyDate{unFinishEdbInfo}) lastItems = append(lastItems, unFinishEdbInfo...) lastItems = append(lastItems, finishEdbInfo...) return } // 获取此 slice 的长度 func (p SortEdbInfo) Len() int { return len(p) } // 根据元素的状态降序排序 func (p SortEdbInfo) Less(i, j int) bool { return p[i].Status > p[j].Status } // 交换数据 func (p SortEdbInfo) Swap(i, j int) { p[i], p[j] = p[j], p[i] } // 嵌套结构体 将继承 SortEdbInfo 的所有属性和方法 // 所以相当于SortByName 也实现了 Len() 和 Swap() 方法 type SortByStatus struct{ SortEdbInfo } // 根据元素的姓名长度降序排序 (此处按照自己的业务逻辑写) func (p SortByStatus) Less(i, j int) bool { return len(p.SortEdbInfo[i].Status) > len(p.SortEdbInfo[j].Status) } type SortByModifyDate struct{ SortEdbInfo } // 根据元素的年龄降序排序 (此处按照自己的业务逻辑写) func (p SortByModifyDate) Less(i, j int) bool { return p.SortEdbInfo[i].ModifyDate > p.SortEdbInfo[j].ModifyDate } type DataCheckResp struct { Status int `description:"状态:1:该日期已存在数据,是否确认修改?,0:数据不存在"` Close string `description:"值"` } type TargetCheckResp struct { Status int `description:"状态:1:该指标有关联数据,请先删除数据,0:指标不存在关联数据,可直接删除"` } type EdbdataExportList struct { TradeCode string `gorm:"column:TRADE_CODE;primaryKey;not null" description:"指标code"` SecName string `gorm:"column:SEC_NAME" description:"指标名称"` Unit string `gorm:"column:UNIT" description:"单位"` Remark string `gorm:"column:REMARK" description:"备注"` Frequency string `gorm:"column:frequency" description:"频度"` //`description:"频度"` ClassifyId int `gorm:"column:classify_id" description:"分类id"` //`description:"分类id"` ClassifyName string `gorm:"column:classify_name" description:"分类名称"` //`description:"分类名称"` CreateDate string `gorm:"column:create_date" description:"创建时间"` //`description:"创建时间"` Dt string `gorm:"column:DT" description:"最新一次录入时间"` } func GetEdbdataSecName(condition string, pars []interface{}) (items []*EdbdataExportList, err error) { //sql := `SELECT a.TRADE_CODE,a.SEC_NAME,a.frequency,a.UNIT,MAX(c.DT) AS Dt // FROM edbdata AS c // INNER JOIN edbinfo AS a ON a.TRADE_CODE=c.TRADE_CODE // INNER JOIN edbinfo_user AS d ON a.TRADE_CODE=d.TRADE_CODE // LEFT JOIN edbdata_classify AS b ON a.classify_id=b.classify_id // WHERE a.classify_id>0` sql := `SELECT a.TRADE_CODE,a.SEC_NAME,a.frequency,a.UNIT,MAX(c.DT) AS Dt,b.classify_name FROM edbdata AS c INNER JOIN edbinfo AS a ON a.TRADE_CODE=c.TRADE_CODE LEFT JOIN edbdata_classify AS b ON a.classify_id=b.classify_id WHERE a.classify_id>0` if condition != "" { sql += condition } sql += " GROUP BY a.TRADE_CODE ORDER BY a.TRADE_CODE ASC " //o := orm.NewOrmUsingDB("edb") //_, err = o.Raw(sql, pars).QueryRows(&items) err = global.DmSQL["edb"].Raw(sql, pars).Find(&items).Error return } func GetEdbDataFrequency(classifyId int) (items []*string, err error) { sql := `SELECT DISTINCT frequency FROM edbinfo where classify_id=? AND frequency IS NOT NULL ORDER BY FIELD(frequency,'日度','周度','月度','季度','半年度','年度') ` //o := orm.NewOrmUsingDB("edb") //_, err = o.Raw(sql, classifyId).QueryRows(&items) err = global.DmSQL["edb"].Raw(sql, classifyId).Find(&items).Error return } // GetEdbDataFrequencyByClassifyIdList // @Description: 根据分类列表获取所有的频度 // @author: Roc // @datetime 2024-08-15 17:51:13 // @param classifyIdList []int // @return items []*string // @return err error func GetEdbDataFrequencyByClassifyIdList(classifyIdList []int) (items []string, err error) { num := len(classifyIdList) if num <= 0 { return } sql := `SELECT DISTINCT frequency FROM edbinfo where classify_id in (` + utils.GetOrmInReplace(num) + `) AND frequency IS NOT NULL ORDER BY FIELD(frequency,'日度','周度','月度','季度','半年度','年度') ` //o := orm.NewOrmUsingDB("edb") //_, err = o.Raw(sql, classifyIdList).QueryRows(&items) err = global.DmSQL["edb"].Raw(sql, classifyIdList).Find(&items).Error return } func GetEdbDataFrequencyByKeyord(keyword string) (items []string, err error) { sql := `SELECT DISTINCT frequency FROM edbinfo where SEC_NAME=? ORDER BY FIELD(frequency,'日度','周度','月度','季度','半年度','年度') ` //o := orm.NewOrmUsingDB("edb") //_, err = o.Raw(sql, keyword).QueryRows(&items) err = global.DmSQL["edb"].Raw(sql, keyword).Find(&items).Error return } type EdbdataList struct { Dt string `gorm:"column:DT" description:"录入时间"` //`orm:"column(DT);" description:"录入时间"` } func GetEdbdataList(tradeCode string) (items []*EdbdataList, err error) { sql := ` SELECT DT FROM edbdata WHERE TRADE_CODE IN(` + tradeCode + `) GROUP BY DT ORDER BY DT DESC ` //o := orm.NewOrmUsingDB("edb") //_, err = o.Raw(sql).QueryRows(&items) err = global.DmSQL["edb"].Raw(sql).Find(&items).Error return } type EdbdataItem struct { TradeCode string `gorm:"column:TRADE_CODE" description:"指标code"` //`orm:"column(TRADE_CODE);" description:"指标code"` Dt string `gorm:"column:DT" description:"最新一次录入时间"` // `orm:"column(DT);" description:"最新一次录入时间"` Close float64 `gorm:"column:CLOSE" description:"值"` //`orm:"column(CLOSE);" description:"值"` } func GetEdbdataValueByTradeCode(tradeCode, dt string) (item *EdbdataItem, err error) { sql := ` SELECT TRADE_CODE,DT,CLOSE FROM edbdata WHERE TRADE_CODE=? AND DT=? ` //o := orm.NewOrmUsingDB("edb") //err = o.Raw(sql, tradeCode, dt).QueryRow(&item) err = global.DmSQL["edb"].Raw(sql, tradeCode, dt).First(&item).Error return } func GetEdbdataAllByTradeCode(tradeCode string) (items []*EdbdataItem, err error) { sql := ` SELECT * FROM edbdata WHERE TRADE_CODE=? ` //o := orm.NewOrmUsingDB("edb") //_, err = o.Raw(sql, tradeCode).QueryRows(&items) err = global.DmSQL["edb"].Raw(sql, tradeCode).Find(&items).Error return } func GetEdbdataClassifyByParentId(parentId int) (items []*EdbdataClassify, err error) { sql := ` SELECT * FROM edbdata_classify WHERE parent_id=? ` //o := orm.NewOrmUsingDB("edb") //_, err = o.Raw(sql, parentId).QueryRows(&items) err = global.DmSQL["edb"].Raw(sql, parentId).Find(&items).Error return } type LzPriceClassify struct { ProductName string `gorm:"column:product_name" description:"产品名称"` } func GetLzPriceClassify() (items []*LzPriceClassify, err error) { sql := ` SELECT product_name FROM longzhongpriceinfo GROUP BY product_name ORDER BY product_name DESC ` //o := orm.NewOrmUsingDB("edb") //_, err = o.Raw(sql).QueryRows(&items) err = global.DmSQL["edb"].Raw(sql).Find(&items).Error return } type Longzhongpriceinfo struct { LongzhongpriceinfoId int `gorm:"column:longzhongpriceinfo_id"` //`orm:"column(longzhongpriceinfo_id);pk" gorm:"primaryKey" ` Standard string `gorm:"column:standard"` ModelName string `gorm:"column:model_name"` Unit string `gorm:"column:unit"` AreaName string `gorm:"column:area_name"` PriceType string `gorm:"column:price_type"` Memo string `gorm:"column:memo"` PriceId string `gorm:"column:price_id"` ProductName string `gorm:"column:product_name"` InfoType string `gorm:"column:info_type"` InfoTypeRemark string `gorm:"column:info_type_remark"` MarketName string `gorm:"column:market_name"` ManufactureName string `gorm:"column:manufacture_name"` } func GetLongzhongpriceinfoByClassifyName(productName string) (items []*Longzhongpriceinfo, err error) { sql := `SELECT * FROM longzhongpriceinfo WHERE product_name=? ORDER BY longzhongpriceinfo_id ASC ` //o := orm.NewOrmUsingDB("edb") //_, err = o.Raw(sql, productName).QueryRows(&items) err = global.DmSQL["edb"].Raw(sql, productName).Find(&items).Error return } func GetLongzhongPriceDataMaxCount(productName string) (count int, err error) { //o := orm.NewOrmUsingDB("edb") sql := `SELECT MAX(t.num) AS count FROM ( SELECT COUNT(1) AS num FROM longzhongpriceinfo AS a INNER JOIN longzhongpricedata AS b ON a.longzhongpriceinfo_id=b.longzhongpriceinfo_id WHERE a.product_name=? GROUP BY a.product_name )AS t ` //err = o.Raw(sql, productName).QueryRow(&count) err = global.DmSQL["edb"].Raw(sql, productName).Scan(&count).Error return } type LongzhongpricedataItems struct { LongzhongpricedataId int `gorm:"column:longzhongpricedata_id"` //`orm:"column(longzhongpricedata_id);pk" gorm:"primaryKey" ` LongzhongpriceinfoId int `gorm:"column:longzhongpriceinfo_id"` PriceDate string `gorm:"column:price_date"` Memo string `gorm:"column:memo"` Price float64 `gorm:"column:price"` CnyPrice float64 `gorm:"column:cny_price"` ZsyPrice float64 `gorm:"column:zsy_price"` ZshPrice float64 `gorm:"column:zsh_price"` LowPrice float64 `gorm:"column:low_price"` HighPrice float64 `gorm:"column:high_price"` RisePrice float64 `gorm:"column:rise_price"` TonPrice float64 `gorm:"column:ton_price"` PriceType string `gorm:"column:price_type"` UpdateDate string `gorm:"column:update_date"` } func GetLongzhongPriceDataById(lzPriceInfoId int) (items []*LongzhongpricedataItems, err error) { //o := orm.NewOrmUsingDB("edb") sql := ` SELECT DISTINCT a.longzhongpriceinfo_id,a.price_date,a.memo,a.price,a.cny_price,a.zsy_price,a.zsh_price,a.low_price,a.high_price,a.rise_price,a.ton_price,a.price_type,a.update_date FROM longzhongpricedata AS a WHERE longzhongpriceinfo_id=? ORDER BY price_date DESC ` //_, err = o.Raw(sql, lzPriceInfoId).QueryRows(&items) err = global.DmSQL["edb"].Raw(sql, lzPriceInfoId).Find(&items).Error return } func GetLzSurveyClassify() (items []*LzPriceClassify, err error) { sql := ` SELECT breed_name AS product_name FROM longzhong_survey_product GROUP BY breed_name ORDER BY breed_name DESC ` //o := orm.NewOrmUsingDB("edb") //_, err = o.Raw(sql).QueryRows(&items) err = global.DmSQL["edb"].Raw(sql).Find(&items).Error return } // LongzhongSurveyProduct 结构体 type LongzhongSurveyProduct struct { SurveyProductId int `gorm:"column:survey_product_id;primaryKey;autoIncrement:false;not null" description:"调查产品ID"` ProjectQuotaId int64 `gorm:"column:project_quota_id" description:"项目配额ID"` BreedId string `gorm:"column:breed_id" description:"品种ID"` BreedName string `gorm:"column:breed_name" description:"品种名称"` QuotaId string `gorm:"column:quota_id" description:"配额ID"` QuotaName string `gorm:"column:quota_name" description:"配额名称"` UnitId string `gorm:"column:unit_id" description:"单位ID"` UnitName string `gorm:"column:unit_name" description:"单位名称"` SampleType int64 `gorm:"column:sample_type" description:"样本类型"` SampleId string `gorm:"column:sample_id" description:"样本ID"` SampleName string `gorm:"column:sample_name" description:"样本名称"` DeviceId string `gorm:"column:device_id" description:"设备ID"` Device string `gorm:"column:device" description:"设备名称"` ProductCraftId string `gorm:"column:product_craft_id" description:"工艺ID"` ProductCraft string `gorm:"column:product_craft" description:"工艺名称"` ProductLine string `gorm:"column:product_line" description:"生产线"` InputMode int64 `gorm:"column:input_mode" description:"输入模式"` Frequency int64 `gorm:"column:frequency" description:"频率"` InputValue string `gorm:"column:input_value" description:"输入值"` TaskShouldFinishTime int `gorm:"column:task_should_finish_time" description:"任务应完成时间"` CustomId string `gorm:"column:custom_id" description:"自定义ID"` CustomType int64 `gorm:"column:custom_type" description:"自定义类型"` Custom string `gorm:"column:custom" description:"自定义名称"` QuotaSampleId int64 `gorm:"column:quota_sample_id" description:"配额样本ID"` StartDate string `gorm:"column:start_date" description:"开始日期"` EndDate string `gorm:"column:end_date" description:"结束日期"` LzCode string `gorm:"column:lz_code" description:"龙种代码"` } func GetLongzhongSurveyProductByClassifyName(productName string) (items []*LongzhongSurveyProduct, err error) { sql := `SELECT * FROM longzhong_survey_product WHERE breed_name=? ORDER BY survey_product_id ASC ` //o := orm.NewOrmUsingDB("edb") //_, err = o.Raw(sql, productName).QueryRows(&items) err = global.DmSQL["edb"].Raw(sql, productName).Find(&items).Error return } func GetLzSurveyProductByNameAndFrequency(productName string, frequency int) (items []*LongzhongSurveyProduct, err error) { sql := `SELECT * FROM longzhong_survey_product WHERE breed_name=? AND frequency=? ORDER BY survey_product_id ASC ` //o := orm.NewOrmUsingDB("edb") //_, err = o.Raw(sql, productName, frequency).QueryRows(&items) err = global.DmSQL["edb"].Raw(sql, productName, frequency).Find(&items).Error return } func GetExportLzSurveyProductByBreedIds(breedIds []string) (items []*LongzhongSurveyProduct, err error) { if len(breedIds) == 0 { return } field := ` survey_product_id, breed_id, breed_name, sample_name, custom, quota_name, lz_code, frequency, unit_name, end_date, input_value ` sql := `SELECT ` + field + ` FROM longzhong_survey_product WHERE breed_id IN (` + utils.GetOrmInReplace(len(breedIds)) + `) ORDER BY breed_id ASC, frequency ASC, survey_product_id ASC ` //o := orm.NewOrmUsingDB("edb") //_, err = o.Raw(sql, breedIds).QueryRows(&items) err = global.DmSQL["edb"].Raw(sql, breedIds).Find(&items).Error return } func GetLzFrequency(productName string) (items []*int, err error) { sql := `SELECT DISTINCT frequency FROM longzhong_survey_product WHERE breed_name=? ORDER BY frequency` //o := orm.NewOrmUsingDB("edb") //_, err = o.Raw(sql, productName).QueryRows(&items) err = global.DmSQL["edb"].Raw(sql, productName).Find(&items).Error return } // EdbInfoItem type EdbInfoItem struct { TradeCode string `gorm:"column:TRADE_CODE;primaryKey;type:varchar(255)" orm:"column(TRADE_CODE);pk" gorm:"primaryKey" description:"指标code"` SecName string `gorm:"column:SEC_NAME;type:varchar(255)" orm:"column(SEC_NAME);" description:"指标名称"` Unit string `gorm:"column:UNIT;type:varchar(255)" orm:"column(UNIT);" description:"单位"` Remark string `gorm:"column:REMARK;type:varchar(255)" orm:"column(REMARK);" description:"备注"` Frequency string `gorm:"column:FREQUENCY;type:varchar(255)" description:"频度"` ClassifyId int `gorm:"column:CLASSIFY_ID;type:int" description:"分类id"` ClassifyName string `gorm:"column:CLASSIFY_NAME;type:varchar(255)" description:"分类名称"` CreateDate string `gorm:"column:CREATE_DATE;type:varchar(255)" description:"创建时间"` UserId int `gorm:"column:USER_ID;type:int" description:"录入用户id"` UserName string `gorm:"column:USER_NAME;type:varchar(255)" description:"录入用户名称"` NoticeTime string `gorm:"column:NOTICE_TIME;type:varchar(255)" description:"通知时间"` Mobile string `gorm:"column:MOBILE;type:varchar(255)" description:"录入者手机号"` ModifyDate string `gorm:"column:MODIFY_DATE;type:varchar(255)" description:"待更新日期"` ModifyTime string `gorm:"column:MODIFY_TIME;type:varchar(255)" description:"最近一次更新时间"` Status string `gorm:"column:STATUS;type:varchar(255)" description:"状态:未完成/完成"` IsJoinEdb int8 `gorm:"column:IS_JOIN_EDB;type:tinyint" description:"指标库是否已添加:0-否;1-是"` StartDate string `gorm:"column:START_DATE;type:varchar(255)" description:"数据开始日期"` EndDate string `gorm:"column:END_DATE;type:varchar(255)" description:"数据结束日期"` LatestValue float64 `gorm:"column:LATEST_VALUE;type:decimal(10,2)" description:"指标最新值"` DataList []*Edbdata `gorm:"-" description:"指标数据列表"` } // GetTargetItemList 获取指标列表数据 func GetTargetItemList(classifyId, edbShowType int, frequency, keyword, tradeCode string, classifyIdStrList []string) (items []*EdbInfoItem, err error) { //o := orm.NewOrmUsingDB("edb") pars := make([]interface{}, 0) sql := ` SELECT a.*,'' modify_date,'' STATUS FROM edbinfo AS a ` if edbShowType != 0 { sql = ` SELECT a.*,b.DT,'' modify_date,'' STATUS FROM edbinfo AS a left join edbdata b on a.TRADE_CODE=b.TRADE_CODE ` } sql += ` WHERE a.classify_id>0 ` //如果没有分类id集合列表,那么就没有数据了,不用往下执行了,直接返回好了 if len(classifyIdStrList) <= 0 { return } if len(classifyIdStrList) > 0 { sql += ` AND a.classify_id in (` + strings.Join(classifyIdStrList, ",") + `) ` } if classifyId > 0 { sql += ` AND a.classify_id=` + strconv.Itoa(classifyId) + ` ` } //频度 if frequency != "" { sql += ` AND a.frequency="` + frequency + `" ` } //关键字 if keyword != "" { sql += ` AND (a.SEC_NAME like ? or a.TRADE_CODE like ? )` pars = utils.GetLikeKeywordPars(pars, keyword, 2) } //指定指标 if tradeCode != "" { sql += ` AND a.TRADE_CODE = "` + tradeCode + `" ` } //指标里面是否有数据 switch edbShowType { case 1: sql += ` AND b.CLOSE is not null ` case 2: sql += ` AND b.CLOSE is null ` } sql += ` GROUP BY a.TRADE_CODE ` sql = sql + ` ORDER BY CONVERT(a.SEC_NAME USING gbk ) COLLATE gbk_chinese_ci ASC ` //_, err = o.Raw(sql, pars).QueryRows(&items) err = global.DmSQL["edb"].Raw(sql, pars).Find(&items).Error return } // GetLzItemList 模糊查询隆众数据库指标列表 func GetLzItemList(keyword string) (items []*data_manage.LongzhongSurveyProduct, err error) { //o := orm.NewOrmUsingDB("edb") sql := "SELECT * FROM longzhong_survey_product WHERE CONCAT(sample_name,breed_name,custom,quota_name,lz_code) LIKE ?" //_, err = o.Raw(sql, utils.GetLikeKeyword(keyword)).QueryRows(&items) err = global.DmSQL["edb"].Raw(sql, utils.GetLikeKeyword(keyword)).Find(&items).Error return } type lzSurveyData struct { DataTime string `gorm:"column:data_time"` //`orm:"column(data_time)" description:"日期"` InputValue string `gorm:"column:input_value"` //`orm:"column(input_value)" description:"值"` } // GetLzItemListByCode 根据code查询隆众数据列表 func GetLzItemListByCode(lzCode string) (items []*lzSurveyData, err error) { //o := orm.NewOrmUsingDB("edb") sql := "SELECT * FROM longzhong_survey_data WHERE survey_product_id=? GROUP BY data_time DESC" //_, err = o.Raw(sql, lzCode).QueryRows(&items) err = global.DmSQL["edb"].Raw(sql, lzCode).Find(&items).Error return } // GetEdbDataListByCodes 通过指标ID获取所有数据 func GetEdbDataListByCodes(tradeCode string) (items []*Edbdata, err error) { sql := ` SELECT TRADE_CODE,DT,round(CLOSE,4) CLOSE,modify_time FROM edbdata WHERE TRADE_CODE IN(` + tradeCode + `) GROUP BY TRADE_CODE,DT ORDER BY DT DESC ` //o := orm.NewOrmUsingDB("edb") //_, err = o.Raw(sql).QueryRows(&items) err = global.DmSQL["edb"].Raw(sql).Find(&items).Error return } // TargetItemListResp 指标数据结构体 type TargetItemListResp struct { List []*EdbInfoItem FrequencyList []string } // BatchDataDeleteReq 批量删除某日的指标数据请求结构体 type BatchDataDeleteReq struct { CreateDate string `description:"创建日期"` TradeCodeList []string `description:"指标唯一编码列表"` } // BatchDeleteEdbDataByDate 批量删除某日的指标数据 func BatchDeleteEdbDataByDate(tradeCodes, dt string, opUserId int) (err error) { //o := orm.NewOrmUsingDB("edb") var list []*Edbdata sql := ` select * FROM edbdata WHERE TRADE_CODE in (` + tradeCodes + `) AND DT = ? ` //_, err = o.Raw(sql, dt).QueryRows(&list) err = global.DmSQL["edb"].Raw(sql, dt).Find(&list).Error if err != nil { return } deleteRecordList := make([]*EdbdataDeleteRecord, 0) for _, edbDataInfo := range list { deleteRecord := &EdbdataDeleteRecord{ TradeCode: edbDataInfo.TradeCode, Dt: edbDataInfo.Dt, Close: edbDataInfo.Close, ModifyTime: time.Now(), CreateTime: time.Now(), SysUserId: opUserId, } deleteRecordList = append(deleteRecordList, deleteRecord) } if len(deleteRecordList) > 0 { //_, tmpErr := o.InsertMulti(len(deleteRecordList), deleteRecordList) tmpErr := global.DmSQL["edb"].CreateInBatches(deleteRecordList, utils.MultiAddNum).Error if tmpErr != nil { err = tmpErr return } } sql = ` DELETE FROM edbdata WHERE TRADE_CODE in (` + tradeCodes + `) AND DT = ? ` //_, err = o.Raw(sql, dt).Exec() err = global.DmSQL["edb"].Exec(sql, dt).Error return } // BatchDeleteEdbData 批量删除指标数据 func BatchDeleteEdbData(tradeCode string, opUserId int) (err error) { //o := orm.NewOrmUsingDB("edb") var list []*Edbdata sql := ` select * FROM edbdata WHERE TRADE_CODE = ? ` //_, err = o.Raw(sql, tradeCode).QueryRows(&list) err = global.DmSQL["edb"].Raw(sql, tradeCode).Find(&list).Error if err != nil { return } deleteRecordList := make([]*EdbdataDeleteRecord, 0) for _, edbDataInfo := range list { deleteRecord := &EdbdataDeleteRecord{ TradeCode: edbDataInfo.TradeCode, Dt: edbDataInfo.Dt, Close: edbDataInfo.Close, ModifyTime: time.Now(), CreateTime: time.Now(), SysUserId: opUserId, } deleteRecordList = append(deleteRecordList, deleteRecord) } //_, err = o.InsertMulti(len(deleteRecordList), deleteRecordList) err = global.DmSQL["edb"].CreateInBatches(deleteRecordList, utils.MultiAddNum).Error if err != nil { return } sql = ` DELETE FROM edbdata WHERE TRADE_CODE = ? ` //_, err = o.Raw(sql, tradeCode).Exec() err = global.DmSQL["edb"].Exec(sql, tradeCode).Error return } // GetEdbInfoCountByClassifyId 根据指标分类id获取当前分类下的指标数量 func GetEdbInfoCountByClassifyId(classifyId int) (count int, err error) { //o := orm.NewOrmUsingDB("edb") sql := `SELECT COUNT(1) AS count FROM ( SELECT a.*,b.CLOSE FROM edbinfo AS a INNER JOIN edbdata AS b ON a.TRADE_CODE=b.TRADE_CODE WHERE a.classify_id=? group by a.TRADE_CODE) d ` //err = o.Raw(sql, classifyId).QueryRow(&count) err = global.DmSQL["edb"].Raw(sql, classifyId).Scan(&count).Error return } // EdbInfoGroupCount 指标分类id获取当前分类下的指标数量 type EdbInfoGroupCount struct { Count int `gorm:"column:count"` ClassifyId int `gorm:"column:classify_id"` } // GetEdbInfoGroupCountByClassifyIds 根据指标分类id获取当前分类下的指标数量 func GetEdbInfoGroupCountByClassifyIds(classifyIds string) (list []*EdbInfoGroupCount, err error) { //o := orm.NewOrmUsingDB("edb") sql := `SELECT COUNT(1) AS count,classify_id FROM ( SELECT a.*,b.CLOSE FROM edbinfo AS a INNER JOIN edbdata AS b ON a.TRADE_CODE=b.TRADE_CODE WHERE a.classify_id in (` + classifyIds + `) group by a.TRADE_CODE) d GROUP BY classify_id ` //_, err = o.Raw(sql).QueryRows(&list) err = global.DmSQL["edb"].Raw(sql).Find(&list).Error return } // GetExcelData 获取excel样式数据 func GetExcelData() (list []*data_manage.ExcelStyle, err error) { //o := orm.NewOrmUsingDB("edb") sql := `SELECT * FROM excel_style ` //_, err = o.Raw(sql).QueryRows(&list) err = global.DmSQL["edb"].Raw(sql).Find(&list).Error return } // AddExcelData 添加excel样式数据 func AddExcelData(item *data_manage.ExcelStyle) (id int64, err error) { //o := orm.NewOrmUsingDB("edb") //id, err = o.Insert(item) err = global.DmSQL["edb"].Create(item).Error return } type EdbdataFloat struct { TradeCode string `gorm:"column:TRADE_CODE"` //`orm:"column(TRADE_CODE);pk" gorm:"primaryKey" description:"指标编码"` Dt string `gorm:"column:DT"` //`orm:"column(DT)" description:"日期"` Close float64 `gorm:"column:CLOSE"` //`orm:"column(CLOSE)" description:"值"` ModifyTime time.Time `gorm:"column:modify_time"` //`orm:"column(modify_time)" description:"修改时间"` } func GetTargetsDataFloat(tradeCode, dt string) (item *EdbdataFloat, err error) { sql := `SELECT * FROM edbdata WHERE TRADE_CODE=? AND DT=? ` //o := orm.NewOrmUsingDB("edb") //err = o.Raw(sql, tradeCode, dt).QueryRow(&item) err = global.DmSQL["edb"].Raw(sql, tradeCode, dt).First(&item).Error return } func ModifyEdbinfo(tradeCode, unit, frequency string, classifyId int) (err error) { sql := `UPDATE edbinfo SET UNIT = ?,frequency=?, classify_id=?, create_date=NOW() WHERE TRADE_CODE=? ` //o := orm.NewOrmUsingDB("edb") //_, err = o.Raw(sql, unit, frequency, classifyId, tradeCode).Exec() err = global.DmSQL["edb"].Exec(sql, unit, frequency, classifyId, tradeCode).Error return } func DeleteTargetsDataByImport(tradeCode, dt string) (err error) { sql := `DELETE FROM edbdata WHERE TRADE_CODE=? AND DT=? ` //o := orm.NewOrmUsingDB("edb") //_, err = o.Raw(sql, tradeCode, dt).Exec() err = global.DmSQL["edb"].Exec(sql, tradeCode, dt).Error return } // GetEdbinfoListByCodeListGroupByUserId 根据指标code列表、用户分组获取指标信息 func GetEdbinfoListByCodeListGroupByUserId(edbCodeList []string) (items []*Edbinfo, err error) { num := len(edbCodeList) if num <= 0 { return } //o := orm.NewOrmUsingDB("edb") sql := `SELECT * FROM edbinfo WHERE TRADE_CODE in (` + utils.GetOrmInReplace(num) + `) GROUP BY user_id ` //_, err = o.Raw(sql, edbCodeList).QueryRows(&items) err = global.DmSQL["edb"].Raw(sql, edbCodeList).Find(&items).Error return } // GetEdbinfoListByCodeListByCodeIdList // @Description: 根据指标code列表获取列表信息 // @param edbCodeList // @return items // @return err func GetEdbinfoListByCodeListByCodeIdList(edbCodeList []string) (items []*Edbinfo, err error) { num := len(edbCodeList) if num <= 0 { return } //o := orm.NewOrmUsingDB("edb") sql := `SELECT * FROM edbinfo WHERE TRADE_CODE in (` + utils.GetOrmInReplace(num) + `) ` //_, err = o.Raw(sql, edbCodeList).QueryRows(&items) err = global.DmSQL["edb"].Raw(sql, edbCodeList).Find(&items).Error return } // GetEdbinfoListByCodeListByUserId // @Description: 根据用户id列表获取指标列表信息 // @param userIdList // @return items // @return err func GetEdbinfoListByCodeListByUserId(userIdList []int) (items []*Edbinfo, err error) { num := len(userIdList) if num <= 0 { return } //o := orm.NewOrmUsingDB("edb") sql := `SELECT * FROM edbinfo WHERE user_id in (` + utils.GetOrmInReplace(num) + `) ` //_, err = o.Raw(sql, userIdList).QueryRows(&items) err = global.DmSQL["edb"].Raw(sql, userIdList).Find(&items).Error return } // ModifyEdbinfoUserIdByCodeList 根据指标code列表修改创建人 func ModifyEdbinfoUserIdByCodeList(edbCodeList []string, userId int, userName string) (err error) { num := len(edbCodeList) if num <= 0 { return } //o := orm.NewOrmUsingDB("edb") sql := `UPDATE edbinfo SET user_id = ?,user_name = ? WHERE TRADE_CODE in (` + utils.GetOrmInReplace(num) + `) ` //_, err = o.Raw(sql, userId, userName, edbCodeList).Exec() err = global.DmSQL["edb"].Exec(sql, userId, userName, edbCodeList).Error return } // ModifyEdbinfoUserIdByOldUserId // @Description: 根据旧用户id修改新用户id // @author: Roc // @datetime 2024-03-25 17:59:32 // @param oldUserId int // @param userId int // @return err error func ModifyEdbinfoUserIdByOldUserId(oldUserIdList []int, userId int) (err error) { num := len(oldUserIdList) if num <= 0 { return } //o := orm.NewOrmUsingDB("edb") sql := `UPDATE edbinfo SET user_id=? WHERE user_id in (` + utils.GetOrmInReplace(num) + `) ` //_, err = o.Raw(sql, userId, oldUserIdList).Exec() err = global.DmSQL["edb"].Exec(sql, userId, oldUserIdList).Error return } func GetEdbInfoAdminList() (list []int, err error) { sql := `SELECT user_id FROM edbinfo GROUP BY user_id ` //o := orm.NewOrmUsingDB("edb") //_, err = o.Raw(sql).QueryRows(&list) err = global.DmSQL["edb"].Raw(sql).Find(&list).Error return } // GetAllChildManualEdbClassify // @Description: 获取手工数据中所有的子分类 // @author: Roc // @datetime 2024-07-16 13:27:28 // @return items []*EdbdataClassify // @return err error func GetAllChildManualEdbClassify() (items []*EdbdataClassify, err error) { //o := orm.NewOrmUsingDB("edb") sql := ` SELECT classify_id,classify_name,parent_id FROM edbdata_classify WHERE parent_id > 0 ` //_, err = o.Raw(sql).QueryRows(&items) err = global.DmSQL["edb"].Raw(sql).Find(&items).Error return } // GetChildManualEdbClassifyByIdList // @Description: 获取手工数据中所有的子分类 // @author: Roc // @datetime 2024-07-16 13:33:57 // @param idList []int // @return items []*EdbdataClassify // @return err error func GetChildManualEdbClassifyByIdList(idList []int) (items []*EdbdataClassify, err error) { num := len(idList) if num <= 0 { return } //o := orm.NewOrmUsingDB("edb") sql := ` SELECT classify_id,classify_name,parent_id FROM edbdata_classify WHERE classify_id in (` + utils.GetOrmInReplace(num) + `) ` //_, err = o.Raw(sql, idList).QueryRows(&items) err = global.DmSQL["edb"].Raw(sql, idList).Find(&items).Error return } // EdbinfoMaxMinDate // @Description: 手工指标的最小最大日期 type EdbinfoMaxMinDate struct { MinDate string `gorm:"column:min_date"` MaxDate string `gorm:"column:max_date"` } // GetEdbdataMaxMinDate // @Description: 获取手工指标的最小最大日期 // @author: Roc // @datetime 2024-08-01 14:27:20 // @param tradeCode string // @return item EdbinfoMaxMinDate // @return err error func GetEdbdataMaxMinDate(tradeCode string) (item EdbinfoMaxMinDate, err error) { //o := orm.NewOrmUsingDB("edb") sql := ` SELECT MIN(DT) min_date,MAX(DT) max_date FROM edbdata WHERE TRADE_CODE = ? ` //err = o.Raw(sql, tradeCode).QueryRow(&item) err = global.DmSQL["edb"].Raw(sql, tradeCode).Find(&item).Error return } // GetEdbdataLatestValue // @Description: 获取手工数据的最新值 // @author: Roc // @datetime 2024-08-02 10:33:22 // @param tradeCode string // @return latestValue float64 // @return err error func GetEdbdataLatestValue(tradeCode string) (latestValue float64, err error) { //o := orm.NewOrmUsingDB("edb") sql := ` SELECT CLOSE FROM edbdata WHERE TRADE_CODE = ? ORDER BY DT DESC LIMIT 1` //err = o.Raw(sql, tradeCode).QueryRow(&latestValue) err = global.DmSQL["edb"].Raw(sql, tradeCode).Scan(&latestValue).Error return } // ModifyEdbinfoMaxMinDate // @Description: 修改手工指标的最小最大日期 // @author: Roc // @datetime 2024-08-01 15:33:45 // @param startDate string // @param endDate string // @param tradeCode string // @return err error func ModifyEdbinfoMaxMinDate(tradeCode, startDate, endDate string, latestValue float64) (err error) { //o := orm.NewOrmUsingDB("edb") sql := ` UPDATE edbinfo SET start_date = ?, end_date = ?, latest_value = ? , modify_time = now() WHERE TRADE_CODE = ? ` //_, err = o.Raw(sql, startDate, endDate, latestValue, tradeCode).Exec() err = global.DmSQL["edb"].Exec(sql, startDate, endDate, latestValue, tradeCode).Error return }