1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798 |
- package models
- import (
- "eta_gn/eta_api/global"
- "eta_gn/eta_api/models/data_manage"
- "eta_gn/eta_api/utils"
- "fmt"
- "sort"
- "strconv"
- "strings"
- "time"
- "gorm.io/gorm"
- "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...).QueryRows(&items)
- pars = append(pars, startSize)
- pars = append(pars, pageSize)
- err = global.DmSQL["edb"].Raw(sql, pars...).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 `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, utils.ForwardPars(pars, mobile)...).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...).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...).QueryRows(&items)
- pars = append(pars, startSize, pageSize)
- err = global.DmSQL["edb"].Raw(sql, pars...).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 `gorm:"-"`
- 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()
- 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()
- 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 COALESCE(MAX(t.num), 0) 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:true;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
- }
|