|
@@ -1,8 +1,8 @@
|
|
|
package models
|
|
|
|
|
|
import (
|
|
|
+ "eta_gn/eta_api/global"
|
|
|
"eta_gn/eta_api/utils"
|
|
|
- "github.com/beego/beego/v2/client/orm"
|
|
|
"github.com/rdlucklib/rdluck_tools/paging"
|
|
|
"time"
|
|
|
)
|
|
@@ -22,8 +22,9 @@ type TargetDetailResp struct {
|
|
|
// @return err error
|
|
|
func GetTargetByTradeCode(tradeCode string) (item *EdbInfoItem, err error) {
|
|
|
sql := `SELECT * FROM edbinfo WHERE TRADE_CODE = ? `
|
|
|
- o := orm.NewOrmUsingDB("edb")
|
|
|
- err = o.Raw(sql, tradeCode).QueryRow(&item)
|
|
|
+ //o := orm.NewOrmUsingDB("edb")
|
|
|
+ //err = o.Raw(sql, tradeCode).QueryRow(&item)
|
|
|
+ err = global.DmSQL["edb"].Raw(sql, tradeCode).First(&item).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
@@ -36,17 +37,17 @@ func GetTargetByTradeCode(tradeCode string) (item *EdbInfoItem, err error) {
|
|
|
// @return total int
|
|
|
// @return err error
|
|
|
func GetCountManualUserClassify(sysUserId, classifyId int) (total int, err error) {
|
|
|
- o := orm.NewOrmUsingDB("data")
|
|
|
+ //o := orm.NewOrmUsingDB("data")
|
|
|
sql := `SELECT count(1) ct FROM manual_user_classify WHERE admin_id=? AND classify_id = ? `
|
|
|
- err = o.Raw(sql, sysUserId, classifyId).QueryRow(&total)
|
|
|
-
|
|
|
+ //err = o.Raw(sql, sysUserId, classifyId).QueryRow(&total)
|
|
|
+ err = global.DmSQL["data"].Raw(sql, sysUserId, classifyId).Scan(&total).Error
|
|
|
return
|
|
|
}
|
|
|
func GetManualClassifyByClassifyId(classifyId int) (item *EdbdataClassify, err error) {
|
|
|
- o := orm.NewOrmUsingDB("edb")
|
|
|
+ //o := orm.NewOrmUsingDB("edb")
|
|
|
sql := ` SELECT classify_id,classify_name,parent_id FROM edbdata_classify WHERE classify_id = ? `
|
|
|
- err = o.Raw(sql, classifyId).QueryRow(&item)
|
|
|
-
|
|
|
+ //err = o.Raw(sql, classifyId).QueryRow(&item)
|
|
|
+ err = global.DmSQL["edb"].Raw(sql, classifyId).First(&item).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
@@ -59,8 +60,9 @@ func GetManualClassifyByClassifyId(classifyId int) (item *EdbdataClassify, err e
|
|
|
// @return err error
|
|
|
func GetEdbDataListByCode(tradeCode string) (items []*Edbdata, err error) {
|
|
|
sql := ` SELECT TRADE_CODE,DT,round(CLOSE,4) CLOSE,modify_time FROM edbdata WHERE TRADE_CODE = ? GROUP BY TRADE_CODE,DT ORDER BY DT DESC `
|
|
|
- o := orm.NewOrmUsingDB("edb")
|
|
|
- _, err = o.Raw(sql, tradeCode).QueryRows(&items)
|
|
|
+ //o := orm.NewOrmUsingDB("edb")
|
|
|
+ //_, err = o.Raw(sql, tradeCode).QueryRows(&items)
|
|
|
+ err = global.DmSQL["edb"].Raw(sql, tradeCode).Find(&items).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
@@ -103,7 +105,7 @@ type EdbListResp struct {
|
|
|
// @return items []*Edbinfo
|
|
|
// @return err error
|
|
|
func GetEdbInfoList(condition string, pars []interface{}, startSize, pageSize int) (items []*EdbInfoListItem, err error) {
|
|
|
- o := orm.NewOrmUsingDB("edb")
|
|
|
+ //o := orm.NewOrmUsingDB("edb")
|
|
|
sql := `SELECT DISTINCT a.* FROM edbinfo AS a WHERE a.classify_id > 0 `
|
|
|
if condition != "" {
|
|
|
sql += condition
|
|
@@ -115,8 +117,8 @@ func GetEdbInfoList(condition string, pars []interface{}, startSize, pageSize in
|
|
|
pars = append(pars, startSize, pageSize)
|
|
|
}
|
|
|
|
|
|
- _, err = o.Raw(sql, pars).QueryRows(&items)
|
|
|
-
|
|
|
+ //_, err = o.Raw(sql, pars).QueryRows(&items)
|
|
|
+ err = global.DmSQL["edb"].Raw(sql, pars).Find(&items).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
@@ -129,15 +131,15 @@ func GetEdbInfoList(condition string, pars []interface{}, startSize, pageSize in
|
|
|
// @return total int
|
|
|
// @return err error
|
|
|
func GetCountEdbInfoList(condition string, pars []interface{}) (total int, err error) {
|
|
|
- o := orm.NewOrmUsingDB("edb")
|
|
|
+ //o := orm.NewOrmUsingDB("edb")
|
|
|
sql := `SELECT COUNT(1) ct FROM edbinfo AS a WHERE a.classify_id > 0 `
|
|
|
if condition != "" {
|
|
|
sql += condition
|
|
|
}
|
|
|
sql += ` ORDER BY a.create_date DESC `
|
|
|
|
|
|
- err = o.Raw(sql, pars).QueryRow(&total)
|
|
|
-
|
|
|
+ //err = o.Raw(sql, pars).QueryRow(&total)
|
|
|
+ err = global.DmSQL["edb"].Raw(sql, pars).Scan(&total).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
@@ -148,8 +150,9 @@ func GetCountEdbInfoList(condition string, pars []interface{}) (total int, err e
|
|
|
// @param items []*Edbdata
|
|
|
// @return err error
|
|
|
func OnlyMultiAddEdbdata(items []*Edbdata) (err error) {
|
|
|
- o := orm.NewOrmUsingDB("edb")
|
|
|
- _, err = o.InsertMulti(utils.MultiAddNum, items)
|
|
|
+ //o := orm.NewOrmUsingDB("edb")
|
|
|
+ //_, err = o.InsertMulti(utils.MultiAddNum, items)
|
|
|
+ err = global.DmSQL["edb"].CreateInBatches(items, utils.MultiAddNum).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
@@ -166,10 +169,10 @@ func DelEdbdataByCodeAndDateList(tradeCode string, dateList []string) (err error
|
|
|
return
|
|
|
}
|
|
|
|
|
|
- o := orm.NewOrmUsingDB("edb")
|
|
|
+ //o := orm.NewOrmUsingDB("edb")
|
|
|
sql := `delete from edbdata WHERE TRADE_CODE=? AND DT in (` + utils.GetOrmInReplace(num) + `) `
|
|
|
- _, err = o.Raw(sql, tradeCode, dateList).Exec()
|
|
|
-
|
|
|
+ //_, err = o.Raw(sql, tradeCode, dateList).Exec()
|
|
|
+ err = global.DmSQL["edb"].Exec(sql, tradeCode, dateList).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
@@ -181,9 +184,10 @@ func DelEdbdataByCodeAndDateList(tradeCode string, dateList []string) (err error
|
|
|
// @param isJoinEdb int 是否加入指标库
|
|
|
// @return err error
|
|
|
func UpdateManualIsJoinEdbStatus(edbCode string, isJoinEdb int8) (err error) {
|
|
|
- o := orm.NewOrmUsingDB("edb")
|
|
|
+ //o := orm.NewOrmUsingDB("edb")
|
|
|
sql := ` UPDATE edbinfo SET is_join_edb = ? WHERE TRADE_CODE =? `
|
|
|
- _, err = o.Raw(sql, isJoinEdb, edbCode).Exec()
|
|
|
+ //_, err = o.Raw(sql, isJoinEdb, edbCode).Exec()
|
|
|
+ err = global.DmSQL["edb"].Exec(sql, isJoinEdb, edbCode).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
@@ -211,7 +215,8 @@ func DelManualIndexByCodeList(codeList []string) (err error) {
|
|
|
if num <= 0 {
|
|
|
return
|
|
|
}
|
|
|
- to, err := orm.NewOrmUsingDB("edb").Begin()
|
|
|
+ //to, err := orm.NewOrmUsingDB("edb").Begin()
|
|
|
+ to := global.DmSQL["edb"].Begin()
|
|
|
if err != nil {
|
|
|
return
|
|
|
}
|
|
@@ -224,14 +229,16 @@ func DelManualIndexByCodeList(codeList []string) (err error) {
|
|
|
}()
|
|
|
// 删除指标
|
|
|
sql := `DELETE FROM edbinfo WHERE TRADE_CODE in (` + utils.GetOrmInReplace(num) + `) `
|
|
|
- _, err = to.Raw(sql, codeList).Exec()
|
|
|
+ //_, err = to.Raw(sql, codeList).Exec()
|
|
|
+ err = to.Exec(sql, codeList).Error
|
|
|
if err != nil {
|
|
|
|
|
|
}
|
|
|
|
|
|
// 删除指标数据
|
|
|
sql = `DELETE FROM edbdata WHERE TRADE_CODE in (` + utils.GetOrmInReplace(num) + `) `
|
|
|
- _, err = to.Raw(sql, codeList).Exec()
|
|
|
+ // _, err = to.Raw(sql, codeList).Exec()
|
|
|
+ err = to.Exec(sql, codeList).Error
|
|
|
if err != nil {
|
|
|
|
|
|
}
|
|
@@ -252,9 +259,9 @@ func GetEdbinfoListBySecNameList(secNameList []string) (items []*Edbinfo, err er
|
|
|
return
|
|
|
}
|
|
|
sql := `SELECT * FROM edbinfo WHERE SEC_NAME in (` + utils.GetOrmInReplace(num) + ` ) AND REMARK='手动' `
|
|
|
- o := orm.NewOrmUsingDB("edb")
|
|
|
- _, err = o.Raw(sql, secNameList).QueryRows(&items)
|
|
|
-
|
|
|
+ //o := orm.NewOrmUsingDB("edb")
|
|
|
+ //_, err = o.Raw(sql, secNameList).QueryRows(&items)
|
|
|
+ err = global.DmSQL["edb"].Raw(sql, secNameList).Find(&items).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
@@ -270,22 +277,22 @@ func GetTargetsDataListByCodeList(tradeCodeList []string) (items []*Edbdata, err
|
|
|
if num <= 0 {
|
|
|
return
|
|
|
}
|
|
|
- o := orm.NewOrmUsingDB("edb")
|
|
|
+ //o := orm.NewOrmUsingDB("edb")
|
|
|
sql := `SELECT * FROM edbdata WHERE TRADE_CODE in (` + utils.GetOrmInReplace(num) + ` ) ORDER BY DT ASC `
|
|
|
- _, err = o.Raw(sql, tradeCodeList).QueryRows(&items)
|
|
|
-
|
|
|
+ //_, err = o.Raw(sql, tradeCodeList).QueryRows(&items)
|
|
|
+ err = global.DmSQL["edb"].Raw(sql, tradeCodeList).Find(&items).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
|
// EdbinfoOpRecord
|
|
|
// @Description: 手工数据的操作日志
|
|
|
type EdbinfoOpRecord struct {
|
|
|
- EdbinfoOpRecordId int `orm:"column(edbinfo_op_record_id);pk"`
|
|
|
- TradeCode string `orm:"column(TRADE_CODE)" description:"指标编码"`
|
|
|
- Remark string `orm:"column(remark)" description:"操作信息"`
|
|
|
- UserId int `orm:"column(user_id)" description:"用户id"`
|
|
|
- UserName string `orm:"column(user_name)" description:"用户姓名"`
|
|
|
- CreateTime time.Time `orm:"column(create_time)" description:"创建时间"`
|
|
|
+ EdbinfoOpRecordId int `gorm:"column:edbinfo_op_record_id;primaryKey;type:int" orm:"column(edbinfo_op_record_id);pk" description:"操作记录ID"`
|
|
|
+ TradeCode string `gorm:"column:TRADE_CODE;type:varchar(255)" orm:"column(TRADE_CODE)" description:"指标编码"`
|
|
|
+ Remark string `gorm:"column:REMARK;type:varchar(255)" orm:"column(remark)" 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:"用户姓名"`
|
|
|
+ CreateTime time.Time `gorm:"column:CREATE_TIME;type:datetime" orm:"column(create_time)" description:"创建时间"`
|
|
|
}
|
|
|
|
|
|
// Remark备注:
|
|
@@ -301,8 +308,9 @@ type EdbinfoOpRecord struct {
|
|
|
// @datetime 2024-07-30 16:25:55
|
|
|
// @return err error
|
|
|
func (m *EdbinfoOpRecord) Create() (err error) {
|
|
|
- o := orm.NewOrmUsingDB("edb")
|
|
|
- _, err = o.Insert(m)
|
|
|
+ //o := orm.NewOrmUsingDB("edb")
|
|
|
+ //_, err = o.Insert(m)
|
|
|
+ err = global.DmSQL["edb"].Create(m).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
@@ -314,8 +322,9 @@ func (m *EdbinfoOpRecord) Create() (err error) {
|
|
|
// @param list []*EdbinfoOpRecord
|
|
|
// @return err error
|
|
|
func (m *EdbinfoOpRecord) MulCreate(list []*EdbinfoOpRecord) (err error) {
|
|
|
- o := orm.NewOrmUsingDB("edb")
|
|
|
- _, err = o.InsertMulti(utils.MultiAddNum, list)
|
|
|
+ //o := orm.NewOrmUsingDB("edb")
|
|
|
+ //_, err = o.InsertMulti(utils.MultiAddNum, list)
|
|
|
+ err = global.DmSQL["edb"].CreateInBatches(list, utils.MultiAddNum).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
@@ -346,16 +355,17 @@ type EdbinfoOpRecordListResp struct {
|
|
|
// @return items []*EdbinfoOpRecordItem
|
|
|
// @return err error
|
|
|
func GetEdbinfoOpRecordPageList(tradeCode string, startSize, pageSize int) (total int, items []*EdbinfoOpRecordItem, err error) {
|
|
|
- o := orm.NewOrmUsingDB("edb")
|
|
|
+ //o := orm.NewOrmUsingDB("edb")
|
|
|
sql := `SELECT count(1) FROM edbinfo_op_record AS a WHERE TRADE_CODE = ? `
|
|
|
- err = o.Raw(sql, tradeCode).QueryRow(&total)
|
|
|
+ //err = o.Raw(sql, tradeCode).QueryRow(&total)
|
|
|
+ err = global.DmSQL["edb"].Raw(sql, tradeCode).Scan(&total).Error
|
|
|
if err != nil {
|
|
|
return
|
|
|
}
|
|
|
|
|
|
sql = `SELECT a.* FROM edbinfo_op_record AS a WHERE TRADE_CODE = ? ORDER BY a.create_time DESC LIMIT ?,?`
|
|
|
- _, err = o.Raw(sql, tradeCode, startSize, pageSize).QueryRows(&items)
|
|
|
-
|
|
|
+ //_, err = o.Raw(sql, tradeCode, startSize, pageSize).QueryRows(&items)
|
|
|
+ err = global.DmSQL["edb"].Raw(sql, tradeCode, startSize, pageSize).Find(&items).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
@@ -368,12 +378,13 @@ func GetEdbinfoOpRecordPageList(tradeCode string, startSize, pageSize int) (tota
|
|
|
// @return count int
|
|
|
// @return err error
|
|
|
func GetManualEdbCountByCondition(condition string, pars []interface{}) (count int, err error) {
|
|
|
- o := orm.NewOrmUsingDB("edb")
|
|
|
+ //o := orm.NewOrmUsingDB("edb")
|
|
|
sql := ` SELECT COUNT(1) AS count FROM edbinfo WHERE 1=1 `
|
|
|
if condition != "" {
|
|
|
sql += condition
|
|
|
}
|
|
|
- err = o.Raw(sql, pars).QueryRow(&count)
|
|
|
+ //err = o.Raw(sql, pars).QueryRow(&count)
|
|
|
+ err = global.DmSQL["edb"].Raw(sql, pars).Scan(&count).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
@@ -384,9 +395,9 @@ func GetManualEdbCountByCondition(condition string, pars []interface{}) (count i
|
|
|
// @param tradeCode string
|
|
|
// @return err error
|
|
|
func DelEdbinfoOpRecordByTradeCode(tradeCode string) (err error) {
|
|
|
- o := orm.NewOrmUsingDB("edb")
|
|
|
+ //o := orm.NewOrmUsingDB("edb")
|
|
|
sql := ` DELETE FROM edbinfo_op_record WHERE TRADE_CODE = ? `
|
|
|
- _, err = o.Raw(sql, tradeCode).Exec()
|
|
|
-
|
|
|
+ //_, err = o.Raw(sql, tradeCode).Exec()
|
|
|
+ err = global.DmSQL["edb"].Exec(sql, tradeCode).Error
|
|
|
return
|
|
|
}
|