edb_info.go 64 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681
  1. package models
  2. import (
  3. "encoding/json"
  4. "errors"
  5. "eta_gn/eta_index_lib/global"
  6. "eta_gn/eta_index_lib/models/mgo"
  7. "eta_gn/eta_index_lib/utils"
  8. "fmt"
  9. "gorm.io/gorm"
  10. "strconv"
  11. "time"
  12. "go.mongodb.org/mongo-driver/bson"
  13. "github.com/beego/beego/v2/client/orm"
  14. "github.com/shopspring/decimal"
  15. )
  16. type EdbInfo struct {
  17. EdbInfoId int `gorm:"primaryKey;autoIncrement;column:edb_info_id" description:"唯一标识"`
  18. EdbInfoType int `gorm:"column:edb_info_type" description:"指标类型,0:普通指标,1:预测指标"`
  19. SourceName string `gorm:"column:source_name" description:"来源名称"`
  20. Source int `gorm:"column:source" description:"来源id"`
  21. EdbCode string `gorm:"column:edb_code" description:"指标编码"`
  22. EdbName string `gorm:"column:edb_name" description:"指标名称"`
  23. EdbNameEn string `gorm:"column:edb_name_en" description:"英文指标名称"`
  24. EdbNameSource string `gorm:"column:edb_name_source" description:"指标名称来源"`
  25. Frequency string `gorm:"column:frequency" description:"频率"`
  26. Unit string `gorm:"column:unit" description:"单位"`
  27. UnitEn string `gorm:"column:unit_en" description:"英文单位"`
  28. StartDate string `gorm:"column:start_date" description:"起始日期"`
  29. EndDate string `gorm:"column:end_date" description:"终止日期"`
  30. ClassifyId int `gorm:"column:classify_id" description:"分类id"`
  31. SysUserId int `gorm:"column:sys_user_id" description:"系统用户ID"`
  32. SysUserRealName string `gorm:"column:sys_user_real_name" description:"系统用户真实姓名"`
  33. UniqueCode string `gorm:"column:unique_code" description:"指标唯一编码"`
  34. CreateTime time.Time `gorm:"column:create_time" description:"创建时间"`
  35. ModifyTime time.Time `gorm:"column:modify_time" description:"修改时间"`
  36. BaseModifyTime time.Time `gorm:"column:base_modify_time" description:"基础修改时间"`
  37. MinValue float64 `gorm:"column:min_value" description:"指标最小值"`
  38. MaxValue float64 `gorm:"column:max_value" description:"指标最大值"`
  39. CalculateFormula string `gorm:"column:calculate_formula" description:"计算公式"`
  40. EdbType int `gorm:"column:edb_type" description:"指标类型:1:基础指标,2:计算指标"`
  41. IsUpdate int `gorm:"column:is_update" description:"当天是否已更新,1:未更新,2:已更新"`
  42. Sort int `gorm:"column:sort" description:"排序字段"`
  43. LatestDate string `gorm:"column:latest_date" description:"数据最新日期(实际日期)"`
  44. LatestValue float64 `gorm:"column:latest_value" description:"数据最新值(实际值)"`
  45. EndValue float64 `gorm:"column:end_value" description:"数据的最新值(预测日期的最新值)"`
  46. MoveType int `gorm:"column:move_type" description:"移动方式:1:领先(默认),2:滞后"`
  47. MoveFrequency string `gorm:"column:move_frequency" description:"移动频度"`
  48. NoUpdate int8 `gorm:"column:no_update" description:"是否停止更新,0:继续更新;1:停止更新"`
  49. ServerUrl string `gorm:"column:server_url" description:"服务器地址"`
  50. ChartImage string `gorm:"column:chart_image" description:"图表图片"`
  51. Calendar string `gorm:"column:calendar;default:公历" description:"公历/农历"`
  52. EmptyType int `gorm:"column:empty_type" description:"空值处理类型(0查找前后35天,1不计算,2前值填充,3后值填充,4等于0)"`
  53. MaxEmptyType int `gorm:"column:max_empty_type" description:"MAX、MIN公式空值处理类型(1、等于0;2、跳过空值)"`
  54. DataDateType string `gorm:"column:data_date_type;size:255;default:交易日" description:"数据日期类型"`
  55. ManualSave int `gorm:"column:manual_save" description:"是否有手动保存过上下限: 0-否; 1-是"`
  56. TerminalCode string `gorm:"column:terminal_code" description:"终端编码,用于配置在机器上"`
  57. DataUpdateTime string `gorm:"column:data_update_time" description:"最近一次数据发生变化的时间"`
  58. ErDataUpdateDate string `gorm:"column:er_data_update_date" description:"本次更新,数据发生变化的最早日期"`
  59. SourceIndexName string `gorm:"column:source_index_name" description:"数据源中的指标名称"`
  60. SubSource int `gorm:"column:sub_source" description:"子数据来源:0:经济数据库,1:日期序列"`
  61. SubSourceName string `gorm:"column:sub_source_name" description:"子数据来源名称"`
  62. IndicatorCode string `gorm:"column:indicator_code" description:"指标代码"`
  63. StockCode string `gorm:"column:stock_code" description:"证券代码"`
  64. Extra string `gorm:"column:extra" description:"指标的额外配置"`
  65. }
  66. func (e *EdbInfo) Add() (err error) {
  67. err = global.DEFAULT_DmSQL.Create(e).Error
  68. return
  69. }
  70. // AddEdbInfo 添加指标
  71. func AddEdbInfo(item *EdbInfo) (lastId int64, err error) {
  72. err = global.DEFAULT_DmSQL.Create(item).Error
  73. return
  74. }
  75. // EdbInfoList 指标数据列表
  76. type EdbInfoList struct {
  77. FrequencyEn string `gorm:"-" description:"英文频率"`
  78. EdbNameAlias string `gorm:"-" json:"-" description:"指标名称,别名"`
  79. RuleType int `gorm:"-" description:"预测规则,1:最新,2:固定值"`
  80. FixedValue float64 `gorm:"-" description:"固定值"`
  81. DataList []*EdbData `gorm:"-" description:"实际指标数据"`
  82. PredictDataList []*EdbData `gorm:"-" description:"预测指标数据"`
  83. Button EdbClassifyItemsButton `gorm:"-" description:"操作权限"`
  84. IsEnEdb bool `gorm:"-" description:"是否展示英文标识"`
  85. EdbInfoId int `gorm:"primaryKey;autoIncrement;column:edb_info_id" description:"唯一标识"`
  86. EdbInfoType int `gorm:"column:edb_info_type" description:"指标类型,0:普通指标,1:预测指标"`
  87. SourceName string `gorm:"column:source_name" description:"来源名称"`
  88. Source int `gorm:"column:source" description:"来源id"`
  89. EdbCode string `gorm:"column:edb_code" description:"指标编码"`
  90. EdbName string `gorm:"column:edb_name" description:"指标名称"`
  91. EdbNameEn string `gorm:"column:edb_name_en" description:"英文指标名称"`
  92. //EdbNameSource string `gorm:"column:edb_name_source" description:"指标名称来源"`
  93. Frequency string `gorm:"column:frequency" description:"频率"`
  94. Unit string `gorm:"column:unit" description:"单位"`
  95. UnitEn string `gorm:"column:unit_en" description:"英文单位"`
  96. StartDate string `gorm:"column:start_date" description:"起始日期"`
  97. EndDate string `gorm:"column:end_date" description:"终止日期"`
  98. ClassifyId int `gorm:"column:classify_id" description:"分类id"`
  99. SysUserId int `gorm:"column:sys_user_id" description:"系统用户ID"`
  100. SysUserRealName string `gorm:"column:sys_user_real_name" description:"系统用户真实姓名"`
  101. UniqueCode string `gorm:"column:unique_code" description:"指标唯一编码"`
  102. CreateTime time.Time `gorm:"column:create_time" description:"创建时间"`
  103. ModifyTime time.Time `gorm:"column:modify_time" description:"修改时间"`
  104. //BaseModifyTime time.Time `gorm:"column:base_modify_time" description:"基础修改时间"`
  105. //MinValue float64 `gorm:"column:min_value" description:"指标最小值"`
  106. /*MaxValue float64 `gorm:"column:max_value" description:"指标最大值"`
  107. CalculateFormula string `gorm:"column:calculate_formula" description:"计算公式"`*/
  108. EdbType int `gorm:"column:edb_type" description:"指标类型:1:基础指标,2:计算指标"`
  109. //IsUpdate int `gorm:"column:is_update" description:"当天是否已更新,1:未更新,2:已更新"`
  110. //Sort int `gorm:"column:sort" description:"排序字段"`
  111. LatestDate string `gorm:"column:latest_date" description:"数据最新日期(实际日期)"`
  112. LatestValue float64 `gorm:"column:latest_value" description:"数据最新值(实际值)"`
  113. EndValue float64 `gorm:"column:end_value" description:"数据的最新值(预测日期的最新值)"`
  114. //MoveType int `gorm:"column:move_type" description:"移动方式:1:领先(默认),2:滞后"`
  115. //MoveFrequency string `gorm:"column:move_frequency" description:"移动频度"`
  116. NoUpdate int8 `gorm:"column:no_update" description:"是否停止更新,0:继续更新;1:停止更新"`
  117. //ServerUrl string `gorm:"column:server_url" description:"服务器地址"`
  118. ChartImage string `gorm:"column:chart_image" description:"图表图片"`
  119. //Calendar string `gorm:"column:calendar;default:公历" description:"公历/农历"`
  120. EmptyType int `gorm:"column:empty_type" description:"空值处理类型(0查找前后35天,1不计算,2前值填充,3后值填充,4等于0)"`
  121. MaxEmptyType int `gorm:"column:max_empty_type" description:"MAX、MIN公式空值处理类型(1、等于0;2、跳过空值)"`
  122. DataDateType string `gorm:"column:data_date_type;size:255;default:交易日" description:"数据日期类型"`
  123. /*ManualSave int `gorm:"column:manual_save" description:"是否有手动保存过上下限: 0-否; 1-是"`
  124. TerminalCode string `gorm:"column:terminal_code" description:"终端编码,用于配置在机器上"`
  125. DataUpdateTime string `gorm:"column:data_update_time" description:"最近一次数据发生变化的时间"`
  126. ErDataUpdateDate string `gorm:"column:er_data_update_date" description:"本次更新,数据发生变化的最早日期"`
  127. SourceIndexName string `gorm:"column:source_index_name" description:"数据源中的指标名称"`*/
  128. SubSource int `gorm:"column:sub_source" description:"子数据来源:0:经济数据库,1:日期序列"`
  129. SubSourceName string `gorm:"column:sub_source_name" description:"子数据来源名称"`
  130. IndicatorCode string `gorm:"column:indicator_code" description:"指标代码"`
  131. StockCode string `gorm:"column:stock_code" description:"证券代码"`
  132. //Extra string `gorm:"column:extra" description:"指标的额外配置"`
  133. }
  134. // EdbClassifyItemsButton 操作按钮
  135. type EdbClassifyItemsButton struct {
  136. AddButton bool `description:"是否可添加"`
  137. OpButton bool `description:"是否可编辑"`
  138. DeleteButton bool `description:"是否可删除"`
  139. MoveButton bool `description:"是否可移动"`
  140. ShowEdbRelation bool `description:"是否展示关联指标"`
  141. ShowChartRelation bool `description:"是否展示关联图表"`
  142. }
  143. // GetEdbInfoByName 根据指标名称获取所有的指标数据列表
  144. func GetEdbInfoByName(edbName string) (items []*EdbInfoList, err error) {
  145. sql := ` SELECT * FROM edb_info WHERE edb_name=? `
  146. err = global.DEFAULT_DmSQL.Raw(sql, edbName).Scan(&items).Error
  147. return
  148. }
  149. // ModifyEdbInfoNameSource 根据来源修改指标名称
  150. func ModifyEdbInfoNameSource(edbNameSource string, edbInfoId int) (err error) {
  151. sql := ` UPDATE edb_info SET edb_name_source=? WHERE edb_info_id = ? `
  152. err = global.DEFAULT_DmSQL.Exec(sql, edbNameSource, edbInfoId).Error
  153. return
  154. }
  155. // GetEdbInfoById 根据指标id获取指标信息
  156. func GetEdbInfoById(edbInfoId int) (item *EdbInfo, err error) {
  157. sql := ` SELECT * FROM edb_info WHERE edb_info_id=? `
  158. err = global.DEFAULT_DmSQL.Raw(sql, edbInfoId).First(&item).Error
  159. return
  160. }
  161. // GetEdbInfoByUniqueCode 根据指标唯一编码获取指标信息
  162. func GetEdbInfoByUniqueCode(uniqueCode string) (item *EdbInfo, err error) {
  163. sql := ` SELECT * FROM edb_info WHERE unique_code=? `
  164. err = global.DEFAULT_DmSQL.Raw(sql, uniqueCode).First(&item).Error
  165. return
  166. }
  167. // GetEdbInfoByIdList 根据指标id列表获取指标信息
  168. func GetEdbInfoByIdList(edbInfoIdList []int) (items []*EdbInfo, err error) {
  169. num := len(edbInfoIdList)
  170. if num <= 0 {
  171. return
  172. }
  173. sql := ` SELECT * FROM edb_info WHERE edb_info_id in (` + utils.GetOrmInReplace(num) + `) `
  174. err = global.DEFAULT_DmSQL.Raw(sql, edbInfoIdList).Scan(&items).Error
  175. return
  176. }
  177. // Update 更新EdbInfo信息
  178. func (edbInfo *EdbInfo) Update(cols []string) (err error) {
  179. err = global.DEFAULT_DmSQL.Model(edbInfo).Select(cols).Updates(edbInfo).Error
  180. return
  181. }
  182. // EdbInfoSearchData
  183. type EdbInfoSearchData struct {
  184. EdbDataId int `description:"数据ID"`
  185. EdbInfoId int `description:"指标ID"`
  186. DataTime string `description:"数据日期"`
  187. Value float64 `description:"数据"`
  188. EdbCode string `description:"指标编码"`
  189. DataTimestamp int64 `description:"时间戳"`
  190. }
  191. type FindEdbDataListAllCond struct {
  192. EdbInfoId int
  193. StartDataTime string
  194. StartDataTimeCond string
  195. EndDataTime string
  196. EndDataTimeCond string
  197. }
  198. // GetEdbDataListAll
  199. // @Description: 获取指标数据列表 order:1升序,其余值为降序
  200. // @author: Roc
  201. // @datetime 2024-05-08 15:34:01
  202. // @param source int
  203. // @param subSource int
  204. // @param findEdbDataListAllCond FindEdbDataListAllCond
  205. // @param order int order:1升序,其余值为降序
  206. // @return item []*EdbInfoSearchData
  207. // @return err error
  208. func GetEdbDataListAll(source, subSource int, findEdbDataListAllCond FindEdbDataListAllCond, order int) (item []*EdbInfoSearchData, err error) {
  209. if source == utils.DATA_SOURCE_BUSINESS && utils.UseMongo {
  210. return GetEdbDataListAllByMongo(source, subSource, findEdbDataListAllCond, order)
  211. }
  212. return GetEdbDataListAllByMysql(source, subSource, findEdbDataListAllCond, order)
  213. }
  214. // GetEdbDataListAllByMysql
  215. // @Description: 从mysql数据库中获取指标数据列表 order:1升序,其余值为降序
  216. // @author: Roc
  217. // @datetime 2024-05-08 15:32:55
  218. // @param source int
  219. // @param subSource int
  220. // @param findEdbDataListAllCond FindEdbDataListAllCond
  221. // @param order int
  222. // @return item []*EdbInfoSearchData
  223. // @return err error
  224. func GetEdbDataListAllByMysql(source, subSource int, findEdbDataListAllCond FindEdbDataListAllCond, order int) (item []*EdbInfoSearchData, err error) {
  225. if findEdbDataListAllCond.EdbInfoId <= 0 {
  226. return
  227. }
  228. sql := ``
  229. tableName := GetEdbDataTableName(source, subSource)
  230. sql = ` SELECT * FROM %s WHERE 1=1 `
  231. sql = fmt.Sprintf(sql, tableName)
  232. var condition string
  233. var pars []interface{}
  234. condition += " AND edb_info_id=? "
  235. pars = append(pars, findEdbDataListAllCond.EdbInfoId)
  236. // 开始日期
  237. if findEdbDataListAllCond.StartDataTime != "" && findEdbDataListAllCond.StartDataTimeCond != `` {
  238. condition += fmt.Sprintf(" AND data_time %s ? ", findEdbDataListAllCond.StartDataTimeCond)
  239. pars = append(pars, findEdbDataListAllCond.StartDataTime)
  240. }
  241. // 结束日期
  242. if findEdbDataListAllCond.EndDataTime != "" && findEdbDataListAllCond.EndDataTimeCond != `` {
  243. condition += fmt.Sprintf(" AND data_time %s ? ", findEdbDataListAllCond.EndDataTimeCond)
  244. pars = append(pars, findEdbDataListAllCond.EndDataTime)
  245. }
  246. if condition != "" {
  247. sql += condition
  248. }
  249. if order == 1 {
  250. sql += ` ORDER BY data_time ASC `
  251. } else {
  252. sql += ` ORDER BY data_time DESC `
  253. }
  254. // todo
  255. err = global.DEFAULT_DmSQL.Raw(sql, pars...).Scan(&item).Error
  256. return
  257. }
  258. // GetEdbDataListAllByTo
  259. // @Description: 根据事务链接获取指标数据列表 order:1升序,其余值为降序
  260. // @author: Roc
  261. // @datetime 2024-05-08 15:34:06
  262. // @param to *gorm.DB
  263. // @param source int
  264. // @param subSource int
  265. // @param findEdbDataListAllCond FindEdbDataListAllCond
  266. // @param order int
  267. // @return item []*EdbInfoSearchData
  268. // @return err error
  269. func GetEdbDataListAllByTo(to *gorm.DB, source, subSource int, findEdbDataListAllCond FindEdbDataListAllCond, order int) (item []*EdbInfoSearchData, err error) {
  270. if source == utils.DATA_SOURCE_BUSINESS && utils.UseMongo {
  271. return GetEdbDataListAllByMongo(source, subSource, findEdbDataListAllCond, order)
  272. }
  273. return GetEdbDataListAllByMysqlTo(to, source, subSource, findEdbDataListAllCond, order)
  274. }
  275. // GetEdbDataListAllByMysqlTo
  276. // @Description: 根据事务链接获取指标数据列表 order:1升序,其余值为降序(Mysql)
  277. // @author: Roc
  278. // @datetime 2024-05-08 15:34:13
  279. // @param to *gorm.DB
  280. // @param source int
  281. // @param subSource int
  282. // @param findEdbDataListAllCond FindEdbDataListAllCond
  283. // @param order int
  284. // @return item []*EdbInfoSearchData
  285. // @return err error
  286. func GetEdbDataListAllByMysqlTo(to *gorm.DB, source, subSource int, findEdbDataListAllCond FindEdbDataListAllCond, order int) (item []*EdbInfoSearchData, err error) {
  287. if findEdbDataListAllCond.EdbInfoId <= 0 {
  288. return
  289. }
  290. sql := ``
  291. tableName := GetEdbDataTableName(source, subSource)
  292. sql = ` SELECT * FROM %s WHERE 1=1 `
  293. sql = fmt.Sprintf(sql, tableName)
  294. var condition string
  295. var pars []interface{}
  296. condition += " AND edb_info_id=? "
  297. pars = append(pars, findEdbDataListAllCond.EdbInfoId)
  298. // 开始日期
  299. if findEdbDataListAllCond.StartDataTime != "" && findEdbDataListAllCond.StartDataTimeCond != `` {
  300. condition += fmt.Sprintf(" AND data_time %s ? ", findEdbDataListAllCond.StartDataTimeCond)
  301. pars = append(pars, findEdbDataListAllCond.StartDataTime)
  302. }
  303. // 结束日期
  304. if findEdbDataListAllCond.EndDataTime != "" && findEdbDataListAllCond.EndDataTimeCond != `` {
  305. condition += fmt.Sprintf(" AND data_time %s ? ", findEdbDataListAllCond.EndDataTimeCond)
  306. pars = append(pars, findEdbDataListAllCond.EndDataTime)
  307. }
  308. if condition != "" {
  309. sql += condition
  310. }
  311. if order == 1 {
  312. sql += ` ORDER BY data_time ASC `
  313. } else {
  314. sql += ` ORDER BY data_time DESC `
  315. }
  316. err = to.Raw(sql, pars...).Scan(&item).Error
  317. return
  318. }
  319. // GetEdbDataListAllByMongo
  320. // @Description: 根据事务链接获取指标数据列表 order:1升序,其余值为降序(Mongo)
  321. // @author: Roc
  322. // @datetime 2024-05-08 15:34:24
  323. // @param source int
  324. // @param subSource int
  325. // @param findEdbDataListAllCond FindEdbDataListAllCond
  326. // @param order int
  327. // @return dataList []*EdbInfoSearchData
  328. // @return err error
  329. func GetEdbDataListAllByMongo(source, subSource int, findEdbDataListAllCond FindEdbDataListAllCond, order int) (dataList []*EdbInfoSearchData, err error) {
  330. dataList = make([]*EdbInfoSearchData, 0)
  331. if findEdbDataListAllCond.EdbInfoId <= 0 {
  332. return
  333. }
  334. mogDataObj := mgo.EdbDataBusiness{}
  335. // 构建查询条件
  336. queryConditions := bson.M{
  337. "edb_info_id": findEdbDataListAllCond.EdbInfoId,
  338. }
  339. dateCondition, err := mgo.BuildDateCondition(findEdbDataListAllCond.StartDataTime, findEdbDataListAllCond.EndDataTime)
  340. if err != nil {
  341. return
  342. }
  343. if len(dateCondition) > 0 {
  344. queryConditions["data_time"] = dateCondition
  345. }
  346. sortList := make([]string, 0)
  347. if order == 1 {
  348. sortList = append(sortList, "data_time")
  349. } else {
  350. sortList = append(sortList, "-data_time")
  351. }
  352. // 获取列表数据
  353. tmpDataList, tmpErr := mogDataObj.GetAllDataList(queryConditions, sortList)
  354. if tmpErr != nil {
  355. err = tmpErr
  356. return
  357. }
  358. for k, v := range tmpDataList {
  359. dataList = append(dataList, &EdbInfoSearchData{
  360. EdbDataId: k + 1,
  361. EdbInfoId: v.EdbInfoId,
  362. EdbCode: v.EdbCode,
  363. DataTime: v.DataTime.Format(utils.FormatDate),
  364. Value: v.Value,
  365. DataTimestamp: v.DataTimestamp,
  366. })
  367. }
  368. return
  369. }
  370. // EdbInfoMaxAndMinInfo 指标最新数据记录结构体
  371. type EdbInfoMaxAndMinInfo struct {
  372. MinDate string `description:"最小日期" bson:"min_date"`
  373. MaxDate string `description:"最大日期" bson:"max_date"`
  374. MinValue float64 `description:"最小值" bson:"min_value"`
  375. MaxValue float64 `description:"最大值" bson:"max_value"`
  376. LatestValue float64 `gorm:"-" description:"最新值" bson:"latest_value"`
  377. LatestDate string `gorm:"-" description:"实际数据最新日期" bson:"latest_date"`
  378. EndValue float64 `description:"最新值" bson:"end_value"`
  379. }
  380. // AfterFind 在该模型上设置钩子函数,把日期转成正确的string,所以查询函数只能用Find函数,First或者Scan是不会触发该函数的来获取数据
  381. func (m *EdbInfoMaxAndMinInfo) AfterFind(db *gorm.DB) (err error) {
  382. minDateTmp, err := time.ParseInLocation(utils.FormatDateWallWithLoc, m.MinDate, time.Local)
  383. if err != nil {
  384. return
  385. }
  386. m.MinDate = minDateTmp.Format(utils.FormatDate)
  387. maxDateTmp, err := time.ParseInLocation(utils.FormatDateWallWithLoc, m.MaxDate, time.Local)
  388. if err != nil {
  389. return
  390. }
  391. m.MaxDate = maxDateTmp.Format(utils.FormatDate)
  392. return
  393. }
  394. // EdbInfoMaxAndMinInfoTmp 指标最新数据记录结构体
  395. type EdbInfoMaxAndMinInfoTmp struct {
  396. MinDate time.Time `description:"最小日期" bson:"min_date"`
  397. MaxDate time.Time `description:"最大日期" bson:"max_date"`
  398. MinValue float64 `description:"最小值" bson:"min_value"`
  399. MaxValue float64 `description:"最大值" bson:"max_value"`
  400. LatestValue float64 `gorm:"-" description:"最新值" bson:"latest_value"`
  401. LatestDate time.Time `gorm:"-" description:"实际数据最新日期" bson:"latest_date"`
  402. EndValue float64 `description:"最新值" bson:"end_value"`
  403. }
  404. // GetEdbInfoMaxAndMinInfo 获取指标的最新数据记录信息
  405. func GetEdbInfoMaxAndMinInfo(source, subSource int, edbCode string) (item *EdbInfoMaxAndMinInfo, err error) {
  406. //itemTmp := new(EdbInfoMaxAndMinInfoTmp)
  407. sql := ``
  408. tableName := GetEdbDataTableName(source, subSource)
  409. sql = ` SELECT MIN(data_time) AS min_date,MAX(data_time) AS max_date,MIN(value) AS min_value,MAX(value) AS max_value FROM %s WHERE edb_code=? `
  410. sql = fmt.Sprintf(sql, tableName)
  411. err = global.DEFAULT_DmSQL.Raw(sql, edbCode).Find(&item).Error
  412. //fmt.Println("item.MinDate", item.MinDate)
  413. /*itemTmp.MinDate = itemTmp.MinDate.In(time.Local)
  414. itemTmp.MaxDate = itemTmp.MaxDate.In(time.Local)
  415. item = &EdbInfoMaxAndMinInfo{
  416. MinDate: itemTmp.MinDate.Format(utils.FormatDate),
  417. MaxDate: itemTmp.MaxDate.Format(utils.FormatDate),
  418. MinValue: itemTmp.MinValue,
  419. MaxValue: itemTmp.MaxValue,
  420. LatestValue: itemTmp.LatestValue,
  421. LatestDate: itemTmp.LatestDate.Format(utils.FormatDate),
  422. EndValue: itemTmp.EndValue,
  423. }*/
  424. var latest_value float64
  425. sql = ` SELECT value AS latest_value FROM %s WHERE edb_code=? ORDER BY data_time DESC LIMIT 1 `
  426. sql = fmt.Sprintf(sql, tableName)
  427. // todo 测试查询单个字段
  428. err = global.DEFAULT_DmSQL.Raw(sql, edbCode).Scan(&latest_value).Error
  429. item.LatestValue = latest_value
  430. return
  431. }
  432. type EdbIndoDataUpdateTime struct {
  433. MinDate string `description:"本次更新,数据发生变化的最早日期"`
  434. }
  435. // AfterFind 在该模型上设置钩子函数,把日期转成正确的string,所以查询函数只能用Find函数,First或者Scan是不会触发该函数的来获取数据
  436. func (m *EdbIndoDataUpdateTime) AfterFind(db *gorm.DB) (err error) {
  437. minDateTmp, err := time.ParseInLocation(utils.FormatDateWallWithLoc, m.MinDate, time.Local)
  438. if err != nil {
  439. return
  440. }
  441. m.MinDate = minDateTmp.Format(utils.FormatDate)
  442. return
  443. }
  444. // GetEdbDataUpdateTimeByModify 获取指标的刷新时间获取最早的一条记录
  445. func GetEdbDataUpdateTimeByModify(source, subSource int, edbCode string, dataUpdateTime string) (item *EdbIndoDataUpdateTime, err error) {
  446. sql := ``
  447. tableName := GetEdbDataTableName(source, subSource)
  448. sql = ` SELECT MIN(data_time) AS min_date FROM %s WHERE edb_code=? and modify_time>=?`
  449. sql = fmt.Sprintf(sql, tableName)
  450. err = global.DEFAULT_DmSQL.Raw(sql, edbCode, dataUpdateTime).Find(&item).Error
  451. return
  452. }
  453. // ModifyEdbInfoMaxAndMinInfo 修改指标的最新数据信息
  454. func ModifyEdbInfoMaxAndMinInfo(edbInfoId int, item *EdbInfoMaxAndMinInfo) (err error) {
  455. sql := ` UPDATE edb_info SET start_date=?,end_date=?,min_value=?,max_value=?,is_update=2,latest_date=?,latest_value=?, end_value = ?,modify_time=NOW() WHERE edb_info_id=? `
  456. err = global.DEFAULT_DmSQL.Exec(sql, item.MinDate, item.MaxDate, item.MinValue, item.MaxValue, item.MaxDate, item.LatestValue, item.LatestValue, edbInfoId).Error
  457. return
  458. }
  459. // ModifyEdbTimeAndLastInfo 修改指标的最新数据信息(除上下限)
  460. func ModifyEdbTimeAndLastInfo(edbInfoId int, item *EdbInfoMaxAndMinInfo) (err error) {
  461. sql := `UPDATE edb_info SET start_date = ?, end_date = ?, is_update = 2,latest_date = ?, latest_value = ?, end_value = ?, modify_time = NOW() WHERE edb_info_id = ?`
  462. err = global.DEFAULT_DmSQL.Exec(sql, item.MinDate, item.MaxDate, item.MaxDate, item.LatestValue, item.LatestValue, edbInfoId).Error
  463. return
  464. }
  465. // ModifyEdbDataUpdateTime 修改指标刷新,本次数据刷新的最早日期
  466. func ModifyEdbDataUpdateTime(edbInfoId int, dataUpdateTime, erDataUpdateDate string) (err error) {
  467. sql := `UPDATE edb_info SET data_update_time = ?, er_data_update_date = ?, modify_time = NOW() WHERE edb_info_id = ?`
  468. err = global.DEFAULT_DmSQL.Exec(sql, dataUpdateTime, erDataUpdateDate, edbInfoId).Error
  469. return
  470. }
  471. // GetLteZeroEdbDataCount
  472. // @Description: 获取小于等于0的数据数量
  473. // @author: Roc
  474. // @datetime 2024-05-31 10:44:39
  475. // @param source int
  476. // @param subSource int
  477. // @param edbInfoId int
  478. // @return count int
  479. // @return err error
  480. func GetLteZeroEdbDataCount(source, subSource, edbInfoId int) (count int, err error) {
  481. if source == utils.DATA_SOURCE_BUSINESS && utils.UseMongo {
  482. return GetLteZeroEdbDataCountByMongo(source, subSource, edbInfoId)
  483. }
  484. return GetLteZeroEdbDataCountByMysql(source, subSource, edbInfoId)
  485. }
  486. // GetLteZeroEdbDataCountByMongo
  487. // @Description: 获取小于等于0的数据数量(从mongo)
  488. // @author: Roc
  489. // @datetime 2024-05-31 10:41:04
  490. // @param source int
  491. // @param subSource int
  492. // @param edbInfoId int
  493. // @return count int
  494. // @return err error
  495. func GetLteZeroEdbDataCountByMongo(source, subSource, edbInfoId int) (count int, err error) {
  496. mogDataObj := mgo.EdbDataBusiness{}
  497. // 构建查询条件
  498. whereQuery := bson.M{
  499. "edb_info_id": edbInfoId,
  500. "value": bson.M{"$lte": 0},
  501. }
  502. // 获数量数据
  503. tmpCount, err := mogDataObj.GetCountDataList(whereQuery)
  504. if err != nil {
  505. return
  506. }
  507. count = int(tmpCount)
  508. return
  509. }
  510. // GetLteZeroEdbDataCountByMysql
  511. // @Description: 获取小于等于0的数据数量(从mysql)
  512. // @author: Roc
  513. // @datetime 2024-05-31 10:29:19
  514. // @param source int
  515. // @param subSource int
  516. // @param edbInfoId int
  517. // @return count int
  518. // @return err error
  519. func GetLteZeroEdbDataCountByMysql(source, subSource, edbInfoId int) (count int, err error) {
  520. sql := ``
  521. tableName := GetEdbDataTableName(source, subSource)
  522. sql = ` SELECT COUNT(1) AS count FROM %s WHERE 1=1 AND edb_info_id =? AND value <=0 `
  523. sql = fmt.Sprintf(sql, tableName)
  524. err = global.DEFAULT_DmSQL.Raw(sql, edbInfoId).Scan(&count).Error
  525. return
  526. }
  527. // GetLastEdbData 获取最近的一条指标数据
  528. func GetLastEdbData(condition string, pars []interface{}, source, subSource int) (item *EdbInfoSearchData, err error) {
  529. sql := ``
  530. tableName := GetEdbDataTableName(source, subSource)
  531. sql = ` SELECT * FROM %s WHERE 1=1 `
  532. sql = fmt.Sprintf(sql, tableName)
  533. if condition != "" {
  534. sql += condition
  535. }
  536. sql += ` ORDER BY data_time DESC `
  537. err = global.DEFAULT_DmSQL.Raw(sql, pars...).First(&item).Error
  538. return
  539. }
  540. // GetEdbInfoByEdbCode 根据指标code获取指标信息
  541. func GetEdbInfoByEdbCode(source int, edbCode string) (item *EdbInfo, err error) {
  542. sql := ` SELECT * FROM edb_info WHERE source=? AND edb_code=? `
  543. err = global.DEFAULT_DmSQL.Raw(sql, source, edbCode).First(&item).Error
  544. if errors.Is(err, orm.ErrNoRows) {
  545. err = nil
  546. }
  547. return
  548. }
  549. // GetEdbInfoOnlyByEdbCode 仅根据指标code获取指标信息
  550. func GetEdbInfoOnlyByEdbCode(edbCode string) (item *EdbInfo, err error) {
  551. sql := ` SELECT * FROM edb_info WHERE edb_code=? `
  552. err = global.DEFAULT_DmSQL.Raw(sql, edbCode).First(&item).Error
  553. return
  554. }
  555. // GetEdbInfoCalculateListByCondition 获取指标关系列表
  556. func GetEdbInfoCalculateListByCondition(condition string, pars []interface{}) (items []*EdbInfoCalculateMapping, err error) {
  557. sql := ` SELECT * FROM edb_info_calculate_mapping WHERE 1=1 `
  558. if condition != "" {
  559. sql += condition
  560. }
  561. err = global.DEFAULT_DmSQL.Raw(sql, pars...).Scan(&items).Error
  562. return
  563. }
  564. // GetEdbInfoCalculateCountByCondition 获取关联指标数量
  565. func GetEdbInfoCalculateCountByCondition(condition string, pars []interface{}) (count int, err error) {
  566. sql := ` SELECT COUNT(1) AS count FROM edb_info_calculate_mapping WHERE 1=1 `
  567. if condition != "" {
  568. sql += condition
  569. }
  570. err = global.DEFAULT_DmSQL.Raw(sql, pars...).Scan(&count).Error
  571. return
  572. }
  573. // 优化版本-处理数据精度问题
  574. type EdbInfoSearchDataV1 struct {
  575. EdbDataId int `description:"数据ID"`
  576. DataTime string `description:"数据日期"`
  577. Value string `description:"数据"`
  578. }
  579. // 优化版本-处理数据精度问题
  580. func GetEdbDataListAllV1(condition string, pars []interface{}, source, subSource, order int) (item []*EdbInfoSearchDataV1, err error) {
  581. sql := ``
  582. tableName := GetEdbDataTableName(source, subSource)
  583. sql = ` SELECT * FROM %s WHERE 1=1 `
  584. sql = fmt.Sprintf(sql, tableName)
  585. if condition != "" {
  586. sql += condition
  587. }
  588. if order == 1 {
  589. sql += ` ORDER BY data_time ASC `
  590. } else {
  591. sql += ` ORDER BY data_time DESC `
  592. }
  593. err = global.DEFAULT_DmSQL.Raw(sql, pars...).Scan(&item).Error
  594. return
  595. }
  596. // GetEdbDataListAllV1ByTo 通过事务链接获取数据列表
  597. func GetEdbDataListAllV1ByTo(to *gorm.DB, condition string, pars []interface{}, source, subSource, order int) (item []*EdbInfoSearchDataV1, err error) {
  598. sql := ``
  599. tableName := GetEdbDataTableName(source, subSource)
  600. sql = ` SELECT * FROM %s WHERE 1=1 `
  601. sql = fmt.Sprintf(sql, tableName)
  602. if condition != "" {
  603. sql += condition
  604. }
  605. if order == 1 {
  606. sql += ` ORDER BY data_time ASC `
  607. } else {
  608. sql += ` ORDER BY data_time DESC `
  609. }
  610. err = to.Raw(sql, pars...).Scan(&item).Error
  611. return
  612. }
  613. // GetEdbInfoByCondition 获取指标列表
  614. func GetEdbInfoByCondition(condition string, pars []interface{}, order int) (item []*EdbInfo, err error) {
  615. sql := ` SELECT * FROM edb_info WHERE 1=1 `
  616. if condition != "" {
  617. sql += condition
  618. }
  619. if order == 1 {
  620. sql += ` ORDER BY end_date ASC `
  621. } else {
  622. sql += ` ORDER BY edb_info_id ASC `
  623. }
  624. err = global.DEFAULT_DmSQL.Raw(sql, pars...).Scan(&item).Error
  625. return
  626. }
  627. // UnifiedModifyEdbInfoMaxAndMinInfo 统一修改指标的最大最小值
  628. func UnifiedModifyEdbInfoMaxAndMinInfo(edbInfo *EdbInfo) (err error, errMsg string) {
  629. // 修改最大最小值
  630. maxAndMinItem, err := GetEdbInfoMaxAndMinInfo(edbInfo.Source, edbInfo.SubSource, edbInfo.EdbCode)
  631. if err != nil {
  632. if err.Error() == utils.ErrNoRow() {
  633. err = nil
  634. return
  635. }
  636. errMsg = "刷新指标失败!"
  637. err = errors.New("获取指标最大最小值失败,err:" + err.Error())
  638. return
  639. }
  640. if maxAndMinItem != nil {
  641. // ETA1.0.3改-如果指标有手动保存过, 那么就不更新指标的最大最小值
  642. if edbInfo.ManualSave == 1 {
  643. err = ModifyEdbTimeAndLastInfo(edbInfo.EdbInfoId, maxAndMinItem)
  644. if err != nil {
  645. errMsg = "刷新指标失败!"
  646. err = errors.New("修改指标最新日期和数据失败,err:" + err.Error())
  647. return
  648. }
  649. } else {
  650. err = ModifyEdbInfoMaxAndMinInfo(edbInfo.EdbInfoId, maxAndMinItem)
  651. if err != nil {
  652. errMsg = "刷新指标失败!"
  653. err = errors.New("修改指标最大最小值失败,err:" + err.Error())
  654. return
  655. }
  656. }
  657. }
  658. // 刷新关联的预测指标
  659. go RefreshPredictStandardBaseByGeneralEdbInfoId(edbInfo.EdbInfoId)
  660. //go ModifyPredictEdbInfoMaxAndMinInfoBySourceEdbInfoId(edbInfo.EdbInfoId, maxAndMinItem)
  661. return
  662. }
  663. // UnifiedModifyEdbInfoMaxAndMinInfoDataUpdate 统一修改指标的最大最小值以及数据更新时间点
  664. func UnifiedModifyEdbInfoMaxAndMinInfoDataUpdate(edbInfo *EdbInfo, dataUpdateTime string) (erDataUpdateDate string, err error, errMsg string) {
  665. err, errMsg = UnifiedModifyEdbInfoMaxAndMinInfo(edbInfo)
  666. if err != nil {
  667. return
  668. }
  669. // 获取本次刷新,指标数据更新的最早日期
  670. erDataUpdateTime, tErr := GetEdbDataUpdateTimeByModify(edbInfo.Source, edbInfo.SubSource, edbInfo.EdbCode, dataUpdateTime)
  671. if tErr != nil {
  672. if tErr.Error() != utils.ErrNoRow() { //本次刷新指标数据无变化
  673. errMsg = "刷新指标失败!"
  674. err = errors.New("查询指标刷新,数据更新的最早日期失败,err:" + tErr.Error())
  675. return
  676. }
  677. } else if erDataUpdateTime.MinDate != "" {
  678. err = ModifyEdbDataUpdateTime(edbInfo.EdbInfoId, dataUpdateTime, erDataUpdateTime.MinDate)
  679. if err != nil {
  680. errMsg = "刷新指标失败!"
  681. err = errors.New("修改指标数据更新的最早日期失败,err:" + tErr.Error())
  682. return
  683. }
  684. erDataUpdateDate = erDataUpdateTime.MinDate
  685. }
  686. return
  687. }
  688. // UnifiedModifyPredictEdbInfoMaxAndMinInfo 统一修改预测运算指标的最大最小值
  689. func UnifiedModifyPredictEdbInfoMaxAndMinInfo(edbInfo *EdbInfo, latestDateStr string, latestValue float64) (err error, errMsg string) {
  690. // 修改最大最小值
  691. maxAndMinItem, err := GetEdbInfoMaxAndMinInfo(edbInfo.Source, edbInfo.SubSource, edbInfo.EdbCode)
  692. if err != nil {
  693. if err.Error() == utils.ErrNoRow() {
  694. err = nil
  695. return
  696. }
  697. errMsg = "刷新指标失败!"
  698. err = errors.New("获取指标最大最小值失败,err:" + err.Error())
  699. return
  700. }
  701. // 最晚的日期的值
  702. maxAndMinItem.EndValue = maxAndMinItem.LatestValue
  703. if maxAndMinItem != nil {
  704. maxAndMinItem.LatestDate = latestDateStr
  705. maxAndMinItem.LatestValue = latestValue
  706. err = ModifyPredictEdbInfoMaxAndMinInfo(edbInfo.EdbInfoId, maxAndMinItem)
  707. if err != nil {
  708. errMsg = "刷新指标失败!"
  709. err = errors.New("修改指标最大最小值失败,err:" + err.Error())
  710. return
  711. }
  712. }
  713. return
  714. }
  715. // GetChartPredictEdbInfoDataListByConfList 获取图表的预测指标的未来数据
  716. func GetChartPredictEdbInfoDataListByConfList(predictEdbConfList []*PredictEdbConfAndData, filtrateStartDateStr, latestDateStr, endDateStr, frequency, dataDateType string, realPredictEdbInfoData []*EdbInfoSearchData) (predictEdbInfoData []*EdbInfoSearchData, minValue, maxValue float64, err error) {
  717. endDate, err := time.ParseInLocation(utils.FormatDate, endDateStr, time.Local)
  718. if err != nil {
  719. return
  720. }
  721. latestDate, err := time.ParseInLocation(utils.FormatDate, latestDateStr, time.Local)
  722. if err != nil {
  723. return
  724. }
  725. // 开始预测数据的时间
  726. startDate := latestDate
  727. // 如果有筛选时间的话
  728. if filtrateStartDateStr != `` {
  729. filtrateStartDate, tmpErr := time.ParseInLocation(utils.FormatDate, filtrateStartDateStr, time.Local)
  730. if tmpErr != nil {
  731. err = tmpErr
  732. return
  733. }
  734. //如果筛选时间晚于实际数据时间,那么就以筛选时间作为获取预测数据的时间
  735. if filtrateStartDate.After(latestDate) {
  736. startDate = filtrateStartDate.AddDate(0, 0, -1)
  737. }
  738. }
  739. //var dateArr []string
  740. // 对应日期的值
  741. existMap := make(map[string]float64)
  742. for _, v := range realPredictEdbInfoData {
  743. //dateArr = append(dateArr, v.DataTime)
  744. existMap[v.DataTime] = v.Value
  745. }
  746. predictEdbInfoData = make([]*EdbInfoSearchData, 0)
  747. //dataValue := lastDataValue
  748. //预测规则,1:最新,2:固定值,3:同比,4:同差,5:环比,6:环差,7:N期移动均值,8:N期段线性外推值
  749. for _, predictEdbConf := range predictEdbConfList {
  750. //dataEndTime := endDate
  751. //if predictEdbConf.EndDate.Before(dataEndTime) {
  752. // dataEndTime = predictEdbConf.EndDate
  753. //}
  754. endDate = predictEdbConf.EndDate
  755. var tmpMinValue, tmpMaxValue float64 // 当前预测结果中的最大/最小值
  756. dayList := getPredictEdbDayList(startDate, endDate, frequency, dataDateType)
  757. if len(dayList) <= 0 { // 如果未来没有日期的话,那么就退出当前循环,进入下一个循环
  758. continue
  759. }
  760. switch predictEdbConf.RuleType {
  761. case 1: //1:最新
  762. var lastDataValue float64 //最新值
  763. tmpAllData := make([]*EdbInfoSearchData, 0)
  764. tmpAllData = append(tmpAllData, realPredictEdbInfoData...)
  765. tmpAllData = append(tmpAllData, predictEdbInfoData...)
  766. lenTmpAllData := len(tmpAllData)
  767. if lenTmpAllData > 0 {
  768. lastDataValue = tmpAllData[lenTmpAllData-1].Value
  769. }
  770. predictEdbInfoData = GetChartPredictEdbInfoDataListByRule1(predictEdbConf.PredictEdbInfoId, lastDataValue, dayList, predictEdbInfoData, existMap)
  771. tmpMaxValue = lastDataValue
  772. tmpMinValue = lastDataValue
  773. case 2: //2:固定值
  774. tmpValDecimal, tmpErr := decimal.NewFromString(predictEdbConf.Value)
  775. if tmpErr != nil {
  776. err = tmpErr
  777. return
  778. }
  779. dataValue, _ := tmpValDecimal.Float64()
  780. predictEdbInfoData = GetChartPredictEdbInfoDataListByRule1(predictEdbConf.PredictEdbInfoId, dataValue, dayList, predictEdbInfoData, existMap)
  781. tmpMaxValue = dataValue
  782. tmpMinValue = dataValue
  783. case 3: //3:同比
  784. tmpValDecimal, tmpErr := decimal.NewFromString(predictEdbConf.Value)
  785. if tmpErr != nil {
  786. err = tmpErr
  787. return
  788. }
  789. tbValue, _ := tmpValDecimal.Float64()
  790. predictEdbInfoData, tmpMinValue, tmpMaxValue = GetChartPredictEdbInfoDataListByRuleTb(predictEdbConf.PredictEdbInfoId, tbValue, dayList, frequency, realPredictEdbInfoData, predictEdbInfoData, existMap)
  791. case 4: //4:同差
  792. tmpValDecimal, tmpErr := decimal.NewFromString(predictEdbConf.Value)
  793. if tmpErr != nil {
  794. err = tmpErr
  795. return
  796. }
  797. tcValue, _ := tmpValDecimal.Float64()
  798. predictEdbInfoData, tmpMinValue, tmpMaxValue = GetChartPredictEdbInfoDataListByRuleTc(predictEdbConf.PredictEdbInfoId, tcValue, dayList, frequency, realPredictEdbInfoData, predictEdbInfoData, existMap)
  799. case 5: //5:环比
  800. tmpValDecimal, tmpErr := decimal.NewFromString(predictEdbConf.Value)
  801. if tmpErr != nil {
  802. err = tmpErr
  803. return
  804. }
  805. hbValue, _ := tmpValDecimal.Float64()
  806. predictEdbInfoData, tmpMinValue, tmpMaxValue = GetChartPredictEdbInfoDataListByRuleHb(predictEdbConf.PredictEdbInfoId, hbValue, dayList, realPredictEdbInfoData, predictEdbInfoData, existMap)
  807. case 6: //6:环差
  808. tmpValDecimal, tmpErr := decimal.NewFromString(predictEdbConf.Value)
  809. if tmpErr != nil {
  810. err = tmpErr
  811. return
  812. }
  813. hcValue, _ := tmpValDecimal.Float64()
  814. predictEdbInfoData, tmpMinValue, tmpMaxValue = GetChartPredictEdbInfoDataListByRuleHc(predictEdbConf.PredictEdbInfoId, hcValue, dayList, realPredictEdbInfoData, predictEdbInfoData, existMap)
  815. case 7: //7:N期移动均值
  816. nValue, tmpErr := strconv.Atoi(predictEdbConf.Value)
  817. if tmpErr != nil {
  818. err = tmpErr
  819. return
  820. }
  821. predictEdbInfoData, tmpMinValue, tmpMaxValue = GetChartPredictEdbInfoDataListByRuleNMoveMeanValue(predictEdbConf.PredictEdbInfoId, nValue, dayList, realPredictEdbInfoData, predictEdbInfoData, existMap)
  822. case 8: //8:N期段线性外推值
  823. nValue, tmpErr := strconv.Atoi(predictEdbConf.Value)
  824. if tmpErr != nil {
  825. err = tmpErr
  826. return
  827. }
  828. predictEdbInfoData, tmpMinValue, tmpMaxValue, err = GetChartPredictEdbInfoDataListByRuleNLinearRegression(predictEdbConf.PredictEdbInfoId, nValue, dayList, realPredictEdbInfoData, predictEdbInfoData, existMap)
  829. if err != nil {
  830. return
  831. }
  832. case 9: //9:动态环差”预测规则;
  833. hcDataMap := make(map[string]float64) //规则计算的环差值map
  834. if predictEdbConf.PredictEdbInfoId > 0 {
  835. tmpPredictEdbRuleDataList, tmpErr := GetPredictEdbRuleDataItemList(predictEdbConf.PredictEdbInfoId, predictEdbConf.ConfigId, startDate.Format(utils.FormatDate), endDate.Format(utils.FormatDate))
  836. if tmpErr != nil {
  837. err = tmpErr
  838. return
  839. }
  840. for _, v := range tmpPredictEdbRuleDataList {
  841. hcDataMap[v.DataTime] = v.Value
  842. }
  843. } else {
  844. if len(predictEdbConf.DataList) <= 0 {
  845. return
  846. }
  847. for _, v := range predictEdbConf.DataList {
  848. currentDate, tmpErr := time.ParseInLocation(utils.FormatDate, v.DataTime, time.Local)
  849. if tmpErr != nil {
  850. continue
  851. }
  852. // 只处理时间段内的数据
  853. if currentDate.Before(startDate) || currentDate.After(endDate) {
  854. continue
  855. }
  856. hcDataMap[v.DataTime] = v.Value
  857. }
  858. }
  859. predictEdbInfoData, tmpMinValue, tmpMaxValue = GetChartPredictEdbInfoDataListByRuleTrendsHC(predictEdbConf.PredictEdbInfoId, dayList, realPredictEdbInfoData, predictEdbInfoData, hcDataMap, existMap)
  860. case 10: //10:根据 给定终值后插值 规则获取预测数据
  861. tmpValDecimal, tmpErr := decimal.NewFromString(predictEdbConf.Value)
  862. if tmpErr != nil {
  863. err = tmpErr
  864. return
  865. }
  866. finalValue, _ := tmpValDecimal.Float64()
  867. predictEdbInfoData, tmpMinValue, tmpMaxValue = GetChartPredictEdbInfoDataListByRuleFinalValueHc(predictEdbConf.PredictEdbInfoId, finalValue, dayList, realPredictEdbInfoData, predictEdbInfoData, existMap)
  868. case 11: //11:根据 季节性 规则获取预测数据
  869. var seasonConf SeasonConf
  870. tmpErr := json.Unmarshal([]byte(predictEdbConf.Value), &seasonConf)
  871. if tmpErr != nil {
  872. err = errors.New("季节性配置信息异常:" + tmpErr.Error())
  873. return
  874. }
  875. calendar := "公历"
  876. if seasonConf.Calendar == "农历" {
  877. calendar = "农历"
  878. }
  879. yearList := make([]int, 0)
  880. //选择方式,1:连续N年;2:指定年份
  881. if seasonConf.YearType == 1 {
  882. if seasonConf.NValue < 1 {
  883. err = errors.New("连续N年不允许小于1")
  884. return
  885. }
  886. currYear := time.Now().Year()
  887. for i := 0; i < seasonConf.NValue; i++ {
  888. yearList = append(yearList, currYear-i-1)
  889. }
  890. } else {
  891. yearList = seasonConf.YearList
  892. }
  893. predictEdbInfoData, tmpMinValue, tmpMaxValue, err = GetChartPredictEdbInfoDataListByRuleSeason(predictEdbConf.PredictEdbInfoId, yearList, calendar, dayList, realPredictEdbInfoData, predictEdbInfoData, existMap)
  894. if err != nil {
  895. return
  896. }
  897. case 12: //12:根据 移动平均同比 规则获取预测数据
  898. var moveAverageConf MoveAverageConf
  899. tmpErr := json.Unmarshal([]byte(predictEdbConf.Value), &moveAverageConf)
  900. if tmpErr != nil {
  901. err = errors.New("季节性配置信息异常:" + tmpErr.Error())
  902. return
  903. }
  904. predictEdbInfoData, tmpMinValue, tmpMaxValue, err = GetChartPredictEdbInfoDataListByRuleMoveAverageTb(predictEdbConf.PredictEdbInfoId, moveAverageConf.NValue, moveAverageConf.Year, dayList, realPredictEdbInfoData, predictEdbInfoData, existMap)
  905. if err != nil {
  906. return
  907. }
  908. case 13: //13:根据 同比增速差值 规则获取预测数据
  909. tmpValDecimal, tmpErr := decimal.NewFromString(predictEdbConf.Value)
  910. if tmpErr != nil {
  911. err = tmpErr
  912. return
  913. }
  914. tbEndValue, _ := tmpValDecimal.Float64()
  915. predictEdbInfoData, tmpMinValue, tmpMaxValue = GetChartPredictEdbInfoDataListByRuleTbzscz(predictEdbConf.PredictEdbInfoId, tbEndValue, dayList, frequency, realPredictEdbInfoData, predictEdbInfoData, existMap)
  916. case 14: //14:根据 一元线性拟合 规则获取预测数据
  917. var ruleConf RuleLineNhConf
  918. err = json.Unmarshal([]byte(predictEdbConf.Value), &ruleConf)
  919. if err != nil {
  920. err = errors.New("一元线性拟合规则配置信息异常:" + err.Error())
  921. return
  922. }
  923. // 规则计算的拟合残差值map
  924. newNhccDataMap := make(map[string]float64)
  925. if predictEdbConf.PredictEdbInfoId > 0 { //已经生成的动态数据
  926. tmpPredictEdbRuleDataList, tmpErr := GetPredictEdbRuleDataItemList(predictEdbConf.PredictEdbInfoId, predictEdbConf.ConfigId, startDate.Format(utils.FormatDate), endDate.Format(utils.FormatDate))
  927. if tmpErr != nil {
  928. err = tmpErr
  929. return
  930. }
  931. for _, v := range tmpPredictEdbRuleDataList {
  932. newNhccDataMap[v.DataTime] = v.Value
  933. }
  934. } else { //未生成的动态数据,需要使用外部传入的数据进行计算
  935. newNhccDataMap, err, _ = getCalculateNhccData(append(realPredictEdbInfoData, predictEdbInfoData...), ruleConf)
  936. }
  937. predictEdbInfoData, tmpMinValue, tmpMaxValue, err = GetChartPredictEdbInfoDataListByRuleLineNh(predictEdbConf.PredictEdbInfoId, dayList, realPredictEdbInfoData, predictEdbInfoData, newNhccDataMap, existMap)
  938. if err != nil {
  939. return
  940. }
  941. case 15: //15:N年均值:过去N年同期均值。过去N年可以连续或者不连续,指标数据均用线性插值补全为日度数据后计算;
  942. predictEdbInfoData, tmpMinValue, tmpMaxValue, err = GetChartPredictEdbInfoDataListByRuleNAnnualAverage(predictEdbConf.PredictEdbInfoId, predictEdbConf.Value, dayList, realPredictEdbInfoData, predictEdbInfoData, existMap)
  943. if err != nil {
  944. return
  945. }
  946. case 16: //16:年度值倒推
  947. predictEdbInfoData, tmpMinValue, tmpMaxValue, err = GetChartPredictEdbInfoDataListByRuleAnnualValueInversion(predictEdbConf.PredictEdbInfoId, predictEdbConf.Value, dayList, frequency, realPredictEdbInfoData, predictEdbInfoData, existMap)
  948. if err != nil {
  949. return
  950. }
  951. }
  952. // 下一个规则的开始日期
  953. {
  954. lenPredictEdbInfoData := len(predictEdbInfoData)
  955. if lenPredictEdbInfoData > 0 {
  956. tmpDataEndTime, _ := time.ParseInLocation(utils.FormatDate, predictEdbInfoData[lenPredictEdbInfoData-1].DataTime, time.Local)
  957. if startDate.Before(tmpDataEndTime) {
  958. startDate = tmpDataEndTime
  959. }
  960. }
  961. }
  962. if tmpMinValue < minValue {
  963. minValue = tmpMinValue
  964. }
  965. if tmpMaxValue < maxValue {
  966. maxValue = tmpMaxValue
  967. }
  968. }
  969. return
  970. }
  971. // GetPredictEdbDayList 获取预测指标日期列表
  972. // GetPredictEdbDayList 获取预测指标日期列表
  973. func getPredictEdbDayList(startDate, endDate time.Time, frequency, dataDateType string) (dayList []time.Time) {
  974. if dataDateType == `` {
  975. dataDateType = `交易日`
  976. }
  977. switch frequency {
  978. case "日度":
  979. for currDate := startDate.AddDate(0, 0, 1); currDate.Before(endDate) || currDate.Equal(endDate); currDate = currDate.AddDate(0, 0, 1) {
  980. // 如果日期类型是交易日的时候,那么需要将周六、日排除
  981. if dataDateType == `交易日` && (currDate.Weekday() == time.Sunday || currDate.Weekday() == time.Saturday) {
  982. continue
  983. }
  984. dayList = append(dayList, currDate)
  985. }
  986. case "周度":
  987. //nextDate := startDate.AddDate(0, 0, 7)
  988. for currDate := startDate.AddDate(0, 0, 7); currDate.Before(endDate) || currDate.Equal(endDate); currDate = currDate.AddDate(0, 0, 7) {
  989. dayList = append(dayList, currDate)
  990. }
  991. case "旬度":
  992. for currDate := startDate.AddDate(0, 0, 1); currDate.Before(endDate) || currDate.Equal(endDate); {
  993. nextDate := currDate.AddDate(0, 0, 1)
  994. //每个月的10号、20号、最后一天,那么就写入
  995. if nextDate.Day() == 11 || nextDate.Day() == 21 || nextDate.Day() == 1 {
  996. dayList = append(dayList, currDate)
  997. }
  998. currDate = nextDate
  999. }
  1000. case "月度":
  1001. for currDate := startDate; currDate.Before(endDate) || currDate.Equal(endDate); {
  1002. currDate = time.Date(currDate.Year(), currDate.Month(), 1, 0, 0, 0, 0, time.Now().Location()).AddDate(0, 1, -1)
  1003. if !currDate.After(endDate) && !currDate.Equal(startDate) {
  1004. dayList = append(dayList, currDate)
  1005. }
  1006. currDate = currDate.AddDate(0, 0, 1)
  1007. }
  1008. case "季度":
  1009. for currDate := startDate; currDate.Before(endDate) || currDate.Equal(endDate); {
  1010. // 每月的最后一天
  1011. currDate = time.Date(currDate.Year(), currDate.Month(), 1, 0, 0, 0, 0, time.Now().Location()).AddDate(0, 1, -1)
  1012. if !currDate.After(endDate) && !currDate.Equal(startDate) {
  1013. // 季度日期就写入,否则不写入
  1014. if currDate.Month() == 3 || currDate.Month() == 6 || currDate.Month() == 9 || currDate.Month() == 12 {
  1015. dayList = append(dayList, currDate)
  1016. }
  1017. }
  1018. currDate = currDate.AddDate(0, 0, 1)
  1019. }
  1020. case "半年度":
  1021. for currDate := startDate; currDate.Before(endDate) || currDate.Equal(endDate); {
  1022. // 每月的最后一天
  1023. currDate = time.Date(currDate.Year(), currDate.Month(), 1, 0, 0, 0, 0, time.Now().Location()).AddDate(0, 1, -1)
  1024. if !currDate.After(endDate) && !currDate.Equal(startDate) {
  1025. // 半年度日期就写入,否则不写入
  1026. if currDate.Month() == 6 || currDate.Month() == 12 {
  1027. dayList = append(dayList, currDate)
  1028. }
  1029. }
  1030. currDate = currDate.AddDate(0, 0, 1)
  1031. }
  1032. case "年度":
  1033. for currDate := startDate; currDate.Before(endDate) || currDate.Equal(endDate); {
  1034. currDate = time.Date(currDate.Year()+1, 12, 31, 0, 0, 0, 0, time.Now().Location())
  1035. if !currDate.After(endDate) && !currDate.Equal(startDate) {
  1036. dayList = append(dayList, currDate)
  1037. }
  1038. }
  1039. }
  1040. return
  1041. }
  1042. // GetPredictDataListByPredictEdbInfo 根据预测指标信息获取预测指标的数据,order:1升序,其余值为降序
  1043. func GetPredictDataListByPredictEdbInfo(edbInfo *EdbInfo, order int, startDate string) (dataList []*EdbInfoSearchData, sourceEdbInfoItem *EdbInfo, err error, errMsg string) {
  1044. // 查找该预测指标配置
  1045. predictEdbConfList, err := GetPredictEdbConfAndDataListById(edbInfo.EdbInfoId)
  1046. if err != nil && err.Error() != utils.ErrNoRow() {
  1047. errMsg = "获取预测指标配置信息失败"
  1048. return
  1049. }
  1050. if len(predictEdbConfList) == 0 {
  1051. errMsg = "获取预测指标配置信息失败"
  1052. err = errors.New(errMsg)
  1053. return
  1054. }
  1055. predictEdbConf := predictEdbConfList[0]
  1056. // 来源指标
  1057. sourceEdbInfoItem, err = GetEdbInfoById(predictEdbConf.SourceEdbInfoId)
  1058. if err != nil {
  1059. if err.Error() == utils.ErrNoRow() {
  1060. errMsg = "找不到来源指标信息"
  1061. err = errors.New(errMsg)
  1062. }
  1063. return
  1064. }
  1065. dataList, err, errMsg = GetPredictDataListByPredictEdbConfList(edbInfo, sourceEdbInfoItem, predictEdbConfList, order, startDate)
  1066. return
  1067. }
  1068. // GetPredictDataListByPredictEdbConfList 根据预测指标信息获取预测指标的数据,order:1升序,其余值为降序
  1069. func GetPredictDataListByPredictEdbConfList(edbInfo, sourceEdbInfoItem *EdbInfo, predictEdbConfList []*PredictEdbConfAndData, order int, startDate string) (dataList []*EdbInfoSearchData, err error, errMsg string) {
  1070. allDataList := make([]*EdbInfoSearchData, 0)
  1071. //获取指标数据(实际已生成)
  1072. tmpDataList, err := GetEdbDataListAll(sourceEdbInfoItem.Source, sourceEdbInfoItem.SubSource, FindEdbDataListAllCond{
  1073. EdbInfoId: sourceEdbInfoItem.EdbInfoId,
  1074. StartDataTime: startDate,
  1075. StartDataTimeCond: ">=",
  1076. }, 1)
  1077. if err != nil {
  1078. return
  1079. }
  1080. // 默认认为上一个查询是把所有的数据查出来了
  1081. allDataList = tmpDataList
  1082. // 如果条件中有选择了日期,那么上次查询实际并不是把所有的数据查出来了
  1083. // 那么需要把日期给过滤掉,然后筛选所有的数据,用于未来指标的生成
  1084. if startDate != `` {
  1085. allDataList, err = GetEdbDataListAll(sourceEdbInfoItem.Source, sourceEdbInfoItem.SubSource, FindEdbDataListAllCond{
  1086. EdbInfoId: sourceEdbInfoItem.EdbInfoId,
  1087. }, 1)
  1088. if err != nil {
  1089. return
  1090. }
  1091. }
  1092. // 获取预测指标未来的数据
  1093. predictDataList := make([]*EdbInfoSearchData, 0)
  1094. endDateStr := edbInfo.EndDate //预测指标的结束日期
  1095. var predictMinValue, predictMaxValue float64
  1096. // 如果有配置的预测规则,那么就进行预测
  1097. if len(predictEdbConfList) > 0 {
  1098. predictDataList, predictMinValue, predictMaxValue, err = GetChartPredictEdbInfoDataListByConfList(predictEdbConfList, startDate, sourceEdbInfoItem.LatestDate, endDateStr, edbInfo.Frequency, edbInfo.DataDateType, allDataList)
  1099. if err != nil {
  1100. return
  1101. }
  1102. }
  1103. //order:1升序,其余值为降序
  1104. if order == 1 {
  1105. dataList = append(tmpDataList, predictDataList...)
  1106. } else {
  1107. // 先倒序预测数据
  1108. lenPredictDataList := len(predictDataList)
  1109. if lenPredictDataList > 0 {
  1110. for k := range predictDataList {
  1111. dataList = append(dataList, predictDataList[lenPredictDataList-k-1])
  1112. }
  1113. }
  1114. // 接着倒序实际指标
  1115. lenDataList := len(tmpDataList)
  1116. for k := range tmpDataList {
  1117. dataList = append(dataList, tmpDataList[lenDataList-k-1])
  1118. }
  1119. }
  1120. if len(predictDataList) > 0 {
  1121. // 如果最小值 大于 预测值,那么将预测值作为最小值数据返回
  1122. if edbInfo.MinValue > predictMinValue {
  1123. edbInfo.MinValue = predictMinValue
  1124. }
  1125. // 如果最大值 小于 预测值,那么将预测值作为最大值数据返回
  1126. if edbInfo.MaxValue < predictMaxValue {
  1127. edbInfo.MaxValue = predictMaxValue
  1128. }
  1129. }
  1130. return
  1131. }
  1132. // GetPredictEdbDataListAll 获取该预测指标所有的数据 ,order:1升序,其余值为降序
  1133. func GetPredictEdbDataListAll(edbInfo *EdbInfo, order int) (items []*EdbInfoSearchData, err error) {
  1134. if edbInfo.Source == utils.DATA_SOURCE_PREDICT { //普通的预测指标是没有入库数据的,直接往配置里面获取
  1135. items, _, err, _ = GetPredictDataListByPredictEdbInfo(edbInfo, 1, "")
  1136. } else {
  1137. items, err = GetEdbDataListAll(edbInfo.Source, edbInfo.SubSource, FindEdbDataListAllCond{
  1138. EdbInfoId: edbInfo.EdbInfoId,
  1139. }, order)
  1140. }
  1141. return
  1142. }
  1143. // GetPredictEdbDataListAllByStartDate 根据开始日期获取该预测指标所有的数据 ,order:1升序,其余值为降序
  1144. func GetPredictEdbDataListAllByStartDate(edbInfo *EdbInfo, order int, startDate string) (items []*EdbInfoSearchData, err error) {
  1145. if edbInfo.Source == utils.DATA_SOURCE_PREDICT { //普通的预测指标是没有入库数据的,直接往配置里面获取
  1146. items, _, err, _ = GetPredictDataListByPredictEdbInfo(edbInfo, order, startDate)
  1147. } else {
  1148. items, err = GetEdbDataListAll(edbInfo.Source, edbInfo.SubSource, FindEdbDataListAllCond{
  1149. EdbInfoId: edbInfo.EdbInfoId,
  1150. StartDataTime: startDate,
  1151. StartDataTimeCond: ">=",
  1152. }, order)
  1153. }
  1154. return
  1155. }
  1156. // ModifyPredictEdbInfoMaxAndMinInfo 修改预测指标的最新数据信息
  1157. func ModifyPredictEdbInfoMaxAndMinInfo(edbInfoId int, item *EdbInfoMaxAndMinInfo) (err error) {
  1158. sql := ` UPDATE edb_info SET start_date=?,end_date=?,min_value=?,max_value=?,is_update=2,latest_date=?,latest_value=?,end_value=?,modify_time=NOW() WHERE edb_info_id=? `
  1159. err = global.DEFAULT_DmSQL.Exec(sql, item.MinDate, item.MaxDate, item.MinValue, item.MaxValue, item.LatestDate, item.LatestValue, item.EndValue, edbInfoId).Error
  1160. return
  1161. }
  1162. // ModifyCalculateEdbInfo 修改计算指标信息
  1163. func ModifyCalculateEdbInfo(edbName, frequency, unit, calculateFormula string, classifyId, edbInfoId int) (err error) {
  1164. sql := ` UPDATE edb_info
  1165. SET
  1166. edb_name =?,
  1167. edb_name_source =?,
  1168. frequency = ?,
  1169. unit = ?,
  1170. classify_id = ?,
  1171. calculate_formula=?,
  1172. modify_time = NOW()
  1173. WHERE edb_info_id = ?`
  1174. err = global.DEFAULT_DmSQL.Exec(sql, edbName, edbName, frequency, unit, classifyId, calculateFormula, edbInfoId).Error
  1175. return
  1176. }
  1177. func GetEdbInfoItemByCondition(condition string, pars []interface{}) (item *EdbInfoList, err error) {
  1178. sql := ` SELECT * FROM edb_info WHERE 1=1 `
  1179. if condition != "" {
  1180. sql += condition
  1181. }
  1182. err = global.DEFAULT_DmSQL.Raw(sql, pars...).First(&item).Error
  1183. return
  1184. }
  1185. func GetEdbInfoItemListByCondition() (items []*EdbInfoList, err error) {
  1186. sql := ` SELECT * FROM edb_info WHERE 1=1 `
  1187. err = global.DEFAULT_DmSQL.Raw(sql).Scan(&items).Error
  1188. return
  1189. }
  1190. // EdbInfoDetailReq 指标详情请求
  1191. type EdbInfoDetailReq struct {
  1192. UniqueCode string `description:"指标唯一编码"`
  1193. EdbCode string `description:"指标编码"`
  1194. }
  1195. // EdbInfoDetailResp 指标详情返回
  1196. type EdbInfoDetailResp struct {
  1197. SourceName string `description:"来源名称"`
  1198. EdbCode string `description:"指标编码"`
  1199. EdbName string `description:"指标名称"`
  1200. EdbNameSource string `description:"指标名称来源"`
  1201. Frequency string `description:"频率"`
  1202. Unit string `description:"单位"`
  1203. StartDate string `description:"起始日期"`
  1204. EndDate string `description:"终止日期"`
  1205. UniqueCode string `description:"指标唯一编码"`
  1206. CreateTime time.Time
  1207. ModifyTime time.Time
  1208. MinValue float64 `description:"指标最小值"`
  1209. MaxValue float64 `description:"指标最大值"`
  1210. EdbNameEn string `description:"英文指标名称"`
  1211. UnitEn string `description:"英文单位"`
  1212. LatestDate string `description:"数据最新日期"`
  1213. LatestValue float64 `description:"数据最新值"`
  1214. ChartImage string `description:"图表图片"`
  1215. DataList []*EdbInfoSearchData
  1216. }
  1217. type AddEdbInfoParams struct {
  1218. Source int `description:"来源id"`
  1219. EdbCode string `description:"指标编码"`
  1220. EdbName string `description:"指标名称"`
  1221. Frequency string `description:"频率"`
  1222. Unit string `description:"单位"`
  1223. ClassifyId int `description:"分类id"`
  1224. TerminalCode string `description:"终端编码"`
  1225. AdminId int `description:"创建人ID"`
  1226. AdminName string `description:"创建人姓名"`
  1227. }
  1228. // EdbInfoAdd 添加指标到指标库
  1229. func EdbInfoAdd(req *AddEdbInfoParams, serverUrl string, sysUserId int, sysUserRealName string) (edbInfo *EdbInfo, err error) {
  1230. source := req.Source
  1231. //判断指标名称是否存在
  1232. var condition string
  1233. var pars []interface{}
  1234. //condition += " AND source=? "
  1235. //pars = append(pars, source)
  1236. condition += " AND edb_code=? "
  1237. pars = append(pars, req.EdbCode)
  1238. items, err := GetEdbInfoByCondition(condition, pars, 0)
  1239. if err != nil {
  1240. err = errors.New("判断指标是否存在失败,Err:" + err.Error())
  1241. return
  1242. }
  1243. if len(items) > 0 {
  1244. sql := ` UPDATE edb_info SET classify_id=?,modify_time=NOW() WHERE edb_code=? `
  1245. err = global.DEFAULT_DmSQL.Exec(sql, req.ClassifyId, req.EdbCode).Error
  1246. if err != nil {
  1247. err = errors.New("更新指标分类信息失败,Err:" + err.Error())
  1248. return
  1249. }
  1250. edbInfo = items[0]
  1251. return
  1252. }
  1253. edbInfo = new(EdbInfo)
  1254. edbInfo.Source = source
  1255. sourceNameMap := map[int]string{
  1256. utils.DATA_SOURCE_THS: "同花顺",
  1257. utils.DATA_SOURCE_WIND: "wind",
  1258. utils.DATA_SOURCE_PB: "彭博",
  1259. utils.DATA_SOURCE_PB_FINANCE: "彭博财务",
  1260. utils.DATA_SOURCE_MANUAL: "手工数据",
  1261. utils.DATA_SOURCE_LZ: "隆众",
  1262. utils.DATA_SOURCE_YS: "SMM",
  1263. utils.DATA_SOURCE_GL: "钢联",
  1264. utils.DATA_SOURCE_ZZ: "郑商所",
  1265. utils.DATA_SOURCE_DL: "大商所",
  1266. utils.DATA_SOURCE_SH: "上期所",
  1267. utils.DATA_SOURCE_CFFEX: "中金所",
  1268. utils.DATA_SOURCE_SHFE: "上期能源",
  1269. utils.DATA_SOURCE_GIE: "欧洲天然气",
  1270. utils.DATA_SOURCE_LT: "路透",
  1271. utils.DATA_SOURCE_COAL: "中国煤炭市场网",
  1272. utils.DATA_SOURCE_GOOGLE_TRAVEL: "our world in data",
  1273. utils.DATA_SOURCE_MYSTEEL_CHEMICAL: "钢联化工",
  1274. utils.DATA_SOURCE_EIA_STEO: "EIA STEO报告",
  1275. utils.DATA_SOURCE_COM_TRADE: "UN",
  1276. utils.DATA_SOURCE_SCI: "SCI",
  1277. utils.DATA_SOURCE_BAIINFO: "BAIINFO",
  1278. utils.DATA_SOURCE_STOCK_PLANT: "存量装置",
  1279. utils.DATA_SOURCE_NATIONAL_STATISTICS: "国家统计局",
  1280. utils.DATA_SOURCE_FUBAO: "富宝数据",
  1281. }
  1282. sourceName, ok := sourceNameMap[source]
  1283. if !ok {
  1284. edbSource := EdbSourceIdMap[source]
  1285. if edbSource != nil {
  1286. sourceName = edbSource.SourceName
  1287. }
  1288. if sourceName == "" {
  1289. err = errors.New("指标来源异常")
  1290. return
  1291. }
  1292. }
  1293. edbInfo.SourceName = sourceName
  1294. edbType := 1 //基础指标
  1295. if source == utils.DATA_SOURCE_STOCK_PLANT {
  1296. edbType = 2 //计算指标
  1297. }
  1298. var sourceIndexName string
  1299. // 钢联化工需要校验下信息是否完整
  1300. if source == utils.DATA_SOURCE_MYSTEEL_CHEMICAL {
  1301. indexObj := new(BaseFromMysteelChemicalIndex)
  1302. tmpItem, tmpErr := indexObj.GetIndexItem(req.EdbCode)
  1303. if tmpErr != nil {
  1304. err = tmpErr
  1305. return
  1306. }
  1307. if tmpItem == nil || tmpItem.IndexName == `` {
  1308. err = errors.New("指标信息不全")
  1309. return
  1310. }
  1311. // 兼容数据
  1312. {
  1313. if req.Frequency == `` {
  1314. req.Frequency = tmpItem.Frequency
  1315. }
  1316. if req.Unit == `` {
  1317. req.Unit = tmpItem.Unit
  1318. }
  1319. }
  1320. }
  1321. edbInfo.EdbCode = req.EdbCode
  1322. edbInfo.EdbName = req.EdbName
  1323. edbInfo.EdbNameSource = req.EdbName
  1324. edbInfo.EdbNameEn = req.EdbName
  1325. edbInfo.Frequency = req.Frequency
  1326. edbInfo.Unit = req.Unit
  1327. edbInfo.UnitEn = req.Unit
  1328. edbInfo.ClassifyId = req.ClassifyId
  1329. edbInfo.SysUserId = sysUserId
  1330. edbInfo.SysUserRealName = sysUserRealName
  1331. edbInfo.StartDate = "1990-01-01"
  1332. edbInfo.CreateTime = time.Now()
  1333. edbInfo.ModifyTime = time.Now()
  1334. edbInfo.ServerUrl = serverUrl
  1335. edbInfo.TerminalCode = req.TerminalCode
  1336. edbInfo.Sort = GetAddEdbMaxSortByClassifyId(req.ClassifyId, 0)
  1337. timestamp := strconv.FormatInt(time.Now().UnixNano(), 10)
  1338. edbInfo.UniqueCode = utils.MD5(utils.DATA_PREFIX + "_" + req.EdbCode + timestamp)
  1339. itemVal, err := GetEdbInfoMaxAndMinInfo(source, utils.DATA_SUB_SOURCE_EDB, req.EdbCode)
  1340. if itemVal != nil && err == nil {
  1341. edbInfo.MaxValue = itemVal.MaxValue
  1342. edbInfo.MinValue = itemVal.MinValue
  1343. }
  1344. edbInfo.EdbType = edbType
  1345. edbInfo.SourceIndexName = sourceIndexName
  1346. edbInfoId, err := AddEdbInfo(edbInfo)
  1347. if err != nil {
  1348. err = errors.New("保存失败,Err:" + err.Error())
  1349. return
  1350. }
  1351. edbInfo.EdbInfoId = int(edbInfoId)
  1352. return
  1353. }
  1354. func ModifyEdbInfoBaseTimeById(edbInfoId int, cTime time.Time) (err error) {
  1355. // 更新修改时间
  1356. sql := ` UPDATE edb_info SET base_modify_time = ? WHERE edb_info_id = ? `
  1357. err = global.DEFAULT_DmSQL.Exec(sql, cTime, edbInfoId).Error
  1358. return
  1359. }
  1360. // EdbInfoRefreshCheckReq 指标数据更新情况查询
  1361. type EdbInfoRefreshCheckReq struct {
  1362. Source int `description:"来源id"`
  1363. LatestDate string `description:"数据最新日期"`
  1364. FrequencyBatch string `description:"更新频度"`
  1365. }
  1366. // EdbInfoRefreshCheckResp 指标数据更新情况查询
  1367. type EdbInfoRefreshCheckResp struct {
  1368. UpdateNum int `description:"已更新指标数"`
  1369. UnUpdateNum int `description:"未更新指标数"`
  1370. }
  1371. // ResetEdbClassifyReq 指标重新分类请求体
  1372. type ResetEdbClassifyReq struct {
  1373. IndexCode string `description:"指标编码"`
  1374. ClassifyFirst string `description:"一级分类名称"`
  1375. ClassifySecond string `description:"二级分类名称"`
  1376. ClassifyThird string `description:"三级分类名称"`
  1377. }
  1378. // GetEdbInfoMaxSortByClassifyId 获取分类下指标的最大的排序数
  1379. func GetEdbInfoMaxSortByClassifyId(classifyId int) (sort int, err error) {
  1380. sql := `SELECT Max(sort) AS sort FROM edb_info WHERE classify_id=? `
  1381. err = global.DEFAULT_DmSQL.Raw(sql, classifyId).Scan(&sort).Error
  1382. return
  1383. }
  1384. func GetEdbAndClassifyMaxSort(parentId int, classifyType uint8) (maxSort int, err error) {
  1385. //获取该层级下最大的排序数
  1386. classifyMaxSort, err := GetEdbClassifyMaxSort(parentId, classifyType)
  1387. if err != nil {
  1388. return
  1389. }
  1390. maxSort = classifyMaxSort
  1391. edbMaxSort, err := GetEdbInfoMaxSortByClassifyId(parentId)
  1392. if err != nil {
  1393. return
  1394. }
  1395. if maxSort < edbMaxSort {
  1396. maxSort = edbMaxSort
  1397. }
  1398. return
  1399. }
  1400. // CalculateEdbExtra 指标运算额外配置
  1401. type CalculateEdbExtra struct {
  1402. DateTag string `description:"时间序列的生成方式,all 表示所选指标的时间序列并集"`
  1403. }
  1404. // CalculateLjzEdbExtra 累计值额外配置
  1405. type CalculateLjzEdbExtra struct {
  1406. LastValType int `description:"最新值处理:0默认、均值填充"`
  1407. }
  1408. type EdbInfoEditRecord struct {
  1409. EdbInfoId int `description:"指标ID"`
  1410. EdbName string `description:"指标名称"`
  1411. Frequency string `description:"频率"`
  1412. Unit string `description:"单位"`
  1413. ClassifyId int `description:"分类id"`
  1414. CalculateFormula string `description:"计算公式"`
  1415. OperateUserId int `description:"操作人id"`
  1416. OperateUserRealName string `description:"操作人姓名"`
  1417. }
  1418. // GetEdbInfoByEdbCodeList
  1419. // @Description: 根据来源和指标编码列表获取指标信息列表
  1420. // @author: Roc
  1421. // @datetime 2024-05-31 16:31:52
  1422. // @param source int
  1423. // @param edbCodeList []string
  1424. // @return items []*EdbInfo
  1425. // @return err error
  1426. func GetEdbInfoByEdbCodeList(source int, edbCodeList []string) (items []*EdbInfo, err error) {
  1427. num := len(edbCodeList)
  1428. if num <= 0 {
  1429. return
  1430. }
  1431. sql := ` SELECT * FROM edb_info WHERE source=? AND edb_code IN (` + utils.GetOrmInReplace(num) + `) `
  1432. err = global.DEFAULT_DmSQL.Raw(sql, source, edbCodeList).Scan(&items).Error
  1433. return
  1434. }
  1435. // GetAddEdbMaxSortByClassifyId
  1436. // @Description: 获取添加指标时,该分类下最大的排序(忽略错误信息)
  1437. // @author: Roc
  1438. // @datetime 2024-07-05 09:39:46
  1439. // @param classifyId int
  1440. // @param classifyType uint8
  1441. // @return sort int
  1442. func GetAddEdbMaxSortByClassifyId(classifyId int, classifyType uint8) (sort int) {
  1443. sort, _ = GetEdbAndClassifyMaxSort(classifyId, classifyType)
  1444. sort = sort + 1
  1445. return
  1446. }
  1447. // EdbInfoExtra 指标额外数据-extra字段
  1448. type EdbInfoExtra struct {
  1449. ApiExtraPars string `description:"API-额外参数(如同花顺日期序列)"`
  1450. }
  1451. // GetEdbInfoNoUpdateTotalByIdList 根据指标id列表获取指标信息
  1452. func GetEdbInfoNoUpdateTotalByIdList(edbInfoIdList []int) (total int, err error) {
  1453. num := len(edbInfoIdList)
  1454. if num <= 0 {
  1455. return
  1456. }
  1457. sql := ` SELECT count(*) FROM edb_info WHERE edb_info_id in (` + utils.GetOrmInReplace(num) + `) and no_update=1`
  1458. err = global.DEFAULT_DmSQL.Raw(sql, edbInfoIdList).Scan(&total).Error
  1459. return
  1460. }
  1461. func TransEdbInfoSearchData2DataList(items []*EdbInfoSearchData) (list []*EdbDataList) {
  1462. if len(items) == 0 {
  1463. return
  1464. }
  1465. list = make([]*EdbDataList, 0)
  1466. for _, v := range items {
  1467. list = append(list, &EdbDataList{
  1468. EdbDataId: v.EdbDataId,
  1469. EdbInfoId: v.EdbInfoId,
  1470. DataTime: v.DataTime,
  1471. DataTimestamp: v.DataTimestamp,
  1472. Value: v.Value,
  1473. })
  1474. }
  1475. return
  1476. }
  1477. func TransEdbInfoDataList2SearchData(items []*EdbDataList) (list []*EdbInfoSearchData) {
  1478. if len(items) == 0 {
  1479. return
  1480. }
  1481. list = make([]*EdbInfoSearchData, 0)
  1482. for _, v := range items {
  1483. list = append(list, &EdbInfoSearchData{
  1484. EdbDataId: v.EdbDataId,
  1485. EdbInfoId: v.EdbInfoId,
  1486. DataTime: v.DataTime,
  1487. DataTimestamp: v.DataTimestamp,
  1488. Value: v.Value,
  1489. })
  1490. }
  1491. return
  1492. }
  1493. type SortEdbDataList []*EdbInfoSearchData
  1494. func (m SortEdbDataList) Len() int {
  1495. return len(m)
  1496. }
  1497. func (m SortEdbDataList) Less(i, j int) bool {
  1498. return m[i].DataTime > m[j].DataTime
  1499. }
  1500. func (m SortEdbDataList) Swap(i, j int) {
  1501. m[i], m[j] = m[j], m[i]
  1502. }