edb_info.go 62 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632
  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 `description:"英文频率"`
  78. EdbNameAlias string `json:"-" description:"指标名称,别名"`
  79. RuleType int `description:"预测规则,1:最新,2:固定值"`
  80. FixedValue float64 `description:"固定值"`
  81. DataList []*EdbData `description:"实际指标数据"`
  82. PredictDataList []*EdbData `description:"预测指标数据"`
  83. Button EdbClassifyItemsButton `description:"操作权限"`
  84. IsEnEdb bool `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 `description:"最新值" bson:"latest_value"`
  377. LatestDate string `description:"实际数据最新日期" bson:"latest_date"`
  378. EndValue float64 `description:"最新值" bson:"end_value"`
  379. }
  380. // GetEdbInfoMaxAndMinInfo 获取指标的最新数据记录信息
  381. func GetEdbInfoMaxAndMinInfo(source, subSource int, edbCode string) (item *EdbInfoMaxAndMinInfo, err error) {
  382. sql := ``
  383. tableName := GetEdbDataTableName(source, subSource)
  384. 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=? `
  385. sql = fmt.Sprintf(sql, tableName)
  386. err = global.DEFAULT_DmSQL.Raw(sql, edbCode).First(&item).Error
  387. var latest_value float64
  388. sql = ` SELECT value AS latest_value FROM %s WHERE edb_code=? ORDER BY data_time DESC LIMIT 1 `
  389. sql = fmt.Sprintf(sql, tableName)
  390. // todo 测试查询单个字段
  391. err = global.DEFAULT_DmSQL.Raw(sql, edbCode).Scan(&latest_value).Error
  392. item.LatestValue = latest_value
  393. return
  394. }
  395. type EdbIndoDataUpdateTime struct {
  396. MinDate string `description:"本次更新,数据发生变化的最早日期"`
  397. }
  398. // GetEdbDataUpdateTimeByModify 获取指标的刷新时间获取最早的一条记录
  399. func GetEdbDataUpdateTimeByModify(source, subSource int, edbCode string, dataUpdateTime string) (item *EdbIndoDataUpdateTime, err error) {
  400. sql := ``
  401. tableName := GetEdbDataTableName(source, subSource)
  402. sql = ` SELECT MIN(data_time) AS min_date FROM %s WHERE edb_code=? and modify_time>=?`
  403. sql = fmt.Sprintf(sql, tableName)
  404. err = global.DEFAULT_DmSQL.Raw(sql, edbCode, dataUpdateTime).First(&item).Error
  405. return
  406. }
  407. // ModifyEdbInfoMaxAndMinInfo 修改指标的最新数据信息
  408. func ModifyEdbInfoMaxAndMinInfo(edbInfoId int, item *EdbInfoMaxAndMinInfo) (err error) {
  409. 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=? `
  410. err = global.DEFAULT_DmSQL.Exec(sql, item.MinDate, item.MaxDate, item.MinValue, item.MaxValue, item.MaxDate, item.LatestValue, item.LatestValue, edbInfoId).Error
  411. return
  412. }
  413. // ModifyEdbTimeAndLastInfo 修改指标的最新数据信息(除上下限)
  414. func ModifyEdbTimeAndLastInfo(edbInfoId int, item *EdbInfoMaxAndMinInfo) (err error) {
  415. 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 = ?`
  416. err = global.DEFAULT_DmSQL.Exec(sql, item.MinDate, item.MaxDate, item.MaxDate, item.LatestValue, item.LatestValue, edbInfoId).Error
  417. return
  418. }
  419. // ModifyEdbDataUpdateTime 修改指标刷新,本次数据刷新的最早日期
  420. func ModifyEdbDataUpdateTime(edbInfoId int, dataUpdateTime, erDataUpdateDate string) (err error) {
  421. sql := `UPDATE edb_info SET data_update_time = ?, er_data_update_date = ?, modify_time = NOW() WHERE edb_info_id = ?`
  422. err = global.DEFAULT_DmSQL.Exec(sql, dataUpdateTime, erDataUpdateDate, edbInfoId).Error
  423. return
  424. }
  425. // GetLteZeroEdbDataCount
  426. // @Description: 获取小于等于0的数据数量
  427. // @author: Roc
  428. // @datetime 2024-05-31 10:44:39
  429. // @param source int
  430. // @param subSource int
  431. // @param edbInfoId int
  432. // @return count int
  433. // @return err error
  434. func GetLteZeroEdbDataCount(source, subSource, edbInfoId int) (count int, err error) {
  435. if source == utils.DATA_SOURCE_BUSINESS && utils.UseMongo {
  436. return GetLteZeroEdbDataCountByMongo(source, subSource, edbInfoId)
  437. }
  438. return GetLteZeroEdbDataCountByMysql(source, subSource, edbInfoId)
  439. }
  440. // GetLteZeroEdbDataCountByMongo
  441. // @Description: 获取小于等于0的数据数量(从mongo)
  442. // @author: Roc
  443. // @datetime 2024-05-31 10:41:04
  444. // @param source int
  445. // @param subSource int
  446. // @param edbInfoId int
  447. // @return count int
  448. // @return err error
  449. func GetLteZeroEdbDataCountByMongo(source, subSource, edbInfoId int) (count int, err error) {
  450. mogDataObj := mgo.EdbDataBusiness{}
  451. // 构建查询条件
  452. whereQuery := bson.M{
  453. "edb_info_id": edbInfoId,
  454. "value": bson.M{"$lte": 0},
  455. }
  456. // 获数量数据
  457. tmpCount, err := mogDataObj.GetCountDataList(whereQuery)
  458. if err != nil {
  459. return
  460. }
  461. count = int(tmpCount)
  462. return
  463. }
  464. // GetLteZeroEdbDataCountByMysql
  465. // @Description: 获取小于等于0的数据数量(从mysql)
  466. // @author: Roc
  467. // @datetime 2024-05-31 10:29:19
  468. // @param source int
  469. // @param subSource int
  470. // @param edbInfoId int
  471. // @return count int
  472. // @return err error
  473. func GetLteZeroEdbDataCountByMysql(source, subSource, edbInfoId int) (count int, err error) {
  474. sql := ``
  475. tableName := GetEdbDataTableName(source, subSource)
  476. sql = ` SELECT COUNT(1) AS count FROM %s WHERE 1=1 AND edb_info_id =? AND value <=0 `
  477. sql = fmt.Sprintf(sql, tableName)
  478. err = global.DEFAULT_DmSQL.Raw(sql, edbInfoId).Scan(&count).Error
  479. return
  480. }
  481. // GetLastEdbData 获取最近的一条指标数据
  482. func GetLastEdbData(condition string, pars []interface{}, source, subSource int) (item *EdbInfoSearchData, err error) {
  483. sql := ``
  484. tableName := GetEdbDataTableName(source, subSource)
  485. sql = ` SELECT * FROM %s WHERE 1=1 `
  486. sql = fmt.Sprintf(sql, tableName)
  487. if condition != "" {
  488. sql += condition
  489. }
  490. sql += ` ORDER BY data_time DESC `
  491. err = global.DEFAULT_DmSQL.Raw(sql, pars).First(&item).Error
  492. return
  493. }
  494. // GetEdbInfoByEdbCode 根据指标code获取指标信息
  495. func GetEdbInfoByEdbCode(source int, edbCode string) (item *EdbInfo, err error) {
  496. sql := ` SELECT * FROM edb_info WHERE source=? AND edb_code=? `
  497. err = global.DEFAULT_DmSQL.Raw(sql, source, edbCode).First(&item).Error
  498. if errors.Is(err, orm.ErrNoRows) {
  499. err = nil
  500. }
  501. return
  502. }
  503. // GetEdbInfoOnlyByEdbCode 仅根据指标code获取指标信息
  504. func GetEdbInfoOnlyByEdbCode(edbCode string) (item *EdbInfo, err error) {
  505. sql := ` SELECT * FROM edb_info WHERE edb_code=? `
  506. err = global.DEFAULT_DmSQL.Raw(sql, edbCode).First(&item).Error
  507. return
  508. }
  509. // GetEdbInfoCalculateListByCondition 获取指标关系列表
  510. func GetEdbInfoCalculateListByCondition(condition string, pars []interface{}) (items []*EdbInfoCalculateMapping, err error) {
  511. sql := ` SELECT * FROM edb_info_calculate_mapping WHERE 1=1 `
  512. if condition != "" {
  513. sql += condition
  514. }
  515. err = global.DEFAULT_DmSQL.Raw(sql, pars).Scan(&items).Error
  516. return
  517. }
  518. // GetEdbInfoCalculateCountByCondition 获取关联指标数量
  519. func GetEdbInfoCalculateCountByCondition(condition string, pars []interface{}) (count int, err error) {
  520. sql := ` SELECT COUNT(1) AS count FROM edb_info_calculate_mapping WHERE 1=1 `
  521. if condition != "" {
  522. sql += condition
  523. }
  524. err = global.DEFAULT_DmSQL.Raw(sql, pars).Scan(&count).Error
  525. return
  526. }
  527. // 优化版本-处理数据精度问题
  528. type EdbInfoSearchDataV1 struct {
  529. EdbDataId int `description:"数据ID"`
  530. DataTime string `description:"数据日期"`
  531. Value string `description:"数据"`
  532. }
  533. // 优化版本-处理数据精度问题
  534. func GetEdbDataListAllV1(condition string, pars []interface{}, source, subSource, order int) (item []*EdbInfoSearchDataV1, err error) {
  535. sql := ``
  536. tableName := GetEdbDataTableName(source, subSource)
  537. sql = ` SELECT * FROM %s WHERE 1=1 `
  538. sql = fmt.Sprintf(sql, tableName)
  539. if condition != "" {
  540. sql += condition
  541. }
  542. if order == 1 {
  543. sql += ` ORDER BY data_time ASC `
  544. } else {
  545. sql += ` ORDER BY data_time DESC `
  546. }
  547. err = global.DEFAULT_DmSQL.Raw(sql, pars).Scan(&item).Error
  548. return
  549. }
  550. // GetEdbDataListAllV1ByTo 通过事务链接获取数据列表
  551. func GetEdbDataListAllV1ByTo(to *gorm.DB, condition string, pars []interface{}, source, subSource, order int) (item []*EdbInfoSearchDataV1, err error) {
  552. sql := ``
  553. tableName := GetEdbDataTableName(source, subSource)
  554. sql = ` SELECT * FROM %s WHERE 1=1 `
  555. sql = fmt.Sprintf(sql, tableName)
  556. if condition != "" {
  557. sql += condition
  558. }
  559. if order == 1 {
  560. sql += ` ORDER BY data_time ASC `
  561. } else {
  562. sql += ` ORDER BY data_time DESC `
  563. }
  564. err = to.Raw(sql, pars).Scan(&item).Error
  565. return
  566. }
  567. // GetEdbInfoByCondition 获取指标列表
  568. func GetEdbInfoByCondition(condition string, pars []interface{}, order int) (item []*EdbInfo, err error) {
  569. sql := ` SELECT * FROM edb_info WHERE 1=1 `
  570. if condition != "" {
  571. sql += condition
  572. }
  573. if order == 1 {
  574. sql += ` ORDER BY end_date ASC `
  575. } else {
  576. sql += ` ORDER BY edb_info_id ASC `
  577. }
  578. err = global.DEFAULT_DmSQL.Raw(sql, pars).Scan(&item).Error
  579. return
  580. }
  581. // UnifiedModifyEdbInfoMaxAndMinInfo 统一修改指标的最大最小值
  582. func UnifiedModifyEdbInfoMaxAndMinInfo(edbInfo *EdbInfo) (err error, errMsg string) {
  583. // 修改最大最小值
  584. maxAndMinItem, err := GetEdbInfoMaxAndMinInfo(edbInfo.Source, edbInfo.SubSource, edbInfo.EdbCode)
  585. if err != nil {
  586. if err.Error() == utils.ErrNoRow() {
  587. err = nil
  588. return
  589. }
  590. errMsg = "刷新指标失败!"
  591. err = errors.New("获取指标最大最小值失败,err:" + err.Error())
  592. return
  593. }
  594. if maxAndMinItem != nil {
  595. // ETA1.0.3改-如果指标有手动保存过, 那么就不更新指标的最大最小值
  596. if edbInfo.ManualSave == 1 {
  597. err = ModifyEdbTimeAndLastInfo(edbInfo.EdbInfoId, maxAndMinItem)
  598. if err != nil {
  599. errMsg = "刷新指标失败!"
  600. err = errors.New("修改指标最新日期和数据失败,err:" + err.Error())
  601. return
  602. }
  603. } else {
  604. err = ModifyEdbInfoMaxAndMinInfo(edbInfo.EdbInfoId, maxAndMinItem)
  605. if err != nil {
  606. errMsg = "刷新指标失败!"
  607. err = errors.New("修改指标最大最小值失败,err:" + err.Error())
  608. return
  609. }
  610. }
  611. }
  612. // 刷新关联的预测指标
  613. go RefreshPredictStandardBaseByGeneralEdbInfoId(edbInfo.EdbInfoId)
  614. //go ModifyPredictEdbInfoMaxAndMinInfoBySourceEdbInfoId(edbInfo.EdbInfoId, maxAndMinItem)
  615. return
  616. }
  617. // UnifiedModifyEdbInfoMaxAndMinInfoDataUpdate 统一修改指标的最大最小值以及数据更新时间点
  618. func UnifiedModifyEdbInfoMaxAndMinInfoDataUpdate(edbInfo *EdbInfo, dataUpdateTime string) (erDataUpdateDate string, err error, errMsg string) {
  619. err, errMsg = UnifiedModifyEdbInfoMaxAndMinInfo(edbInfo)
  620. if err != nil {
  621. return
  622. }
  623. // 获取本次刷新,指标数据更新的最早日期
  624. erDataUpdateTime, tErr := GetEdbDataUpdateTimeByModify(edbInfo.Source, edbInfo.SubSource, edbInfo.EdbCode, dataUpdateTime)
  625. if tErr != nil {
  626. if tErr.Error() != utils.ErrNoRow() { //本次刷新指标数据无变化
  627. errMsg = "刷新指标失败!"
  628. err = errors.New("查询指标刷新,数据更新的最早日期失败,err:" + tErr.Error())
  629. return
  630. }
  631. } else if erDataUpdateTime.MinDate != "" {
  632. err = ModifyEdbDataUpdateTime(edbInfo.EdbInfoId, dataUpdateTime, erDataUpdateTime.MinDate)
  633. if err != nil {
  634. errMsg = "刷新指标失败!"
  635. err = errors.New("修改指标数据更新的最早日期失败,err:" + tErr.Error())
  636. return
  637. }
  638. erDataUpdateDate = erDataUpdateTime.MinDate
  639. }
  640. return
  641. }
  642. // UnifiedModifyPredictEdbInfoMaxAndMinInfo 统一修改预测运算指标的最大最小值
  643. func UnifiedModifyPredictEdbInfoMaxAndMinInfo(edbInfo *EdbInfo, latestDateStr string, latestValue float64) (err error, errMsg string) {
  644. // 修改最大最小值
  645. maxAndMinItem, err := GetEdbInfoMaxAndMinInfo(edbInfo.Source, edbInfo.SubSource, edbInfo.EdbCode)
  646. if err != nil {
  647. if err.Error() == utils.ErrNoRow() {
  648. err = nil
  649. return
  650. }
  651. errMsg = "刷新指标失败!"
  652. err = errors.New("获取指标最大最小值失败,err:" + err.Error())
  653. return
  654. }
  655. // 最晚的日期的值
  656. maxAndMinItem.EndValue = maxAndMinItem.LatestValue
  657. if maxAndMinItem != nil {
  658. maxAndMinItem.LatestDate = latestDateStr
  659. maxAndMinItem.LatestValue = latestValue
  660. err = ModifyPredictEdbInfoMaxAndMinInfo(edbInfo.EdbInfoId, maxAndMinItem)
  661. if err != nil {
  662. errMsg = "刷新指标失败!"
  663. err = errors.New("修改指标最大最小值失败,err:" + err.Error())
  664. return
  665. }
  666. }
  667. return
  668. }
  669. // GetChartPredictEdbInfoDataListByConfList 获取图表的预测指标的未来数据
  670. func GetChartPredictEdbInfoDataListByConfList(predictEdbConfList []*PredictEdbConfAndData, filtrateStartDateStr, latestDateStr, endDateStr, frequency, dataDateType string, realPredictEdbInfoData []*EdbInfoSearchData) (predictEdbInfoData []*EdbInfoSearchData, minValue, maxValue float64, err error) {
  671. endDate, err := time.ParseInLocation(utils.FormatDate, endDateStr, time.Local)
  672. if err != nil {
  673. return
  674. }
  675. latestDate, err := time.ParseInLocation(utils.FormatDate, latestDateStr, time.Local)
  676. if err != nil {
  677. return
  678. }
  679. // 开始预测数据的时间
  680. startDate := latestDate
  681. // 如果有筛选时间的话
  682. if filtrateStartDateStr != `` {
  683. filtrateStartDate, tmpErr := time.ParseInLocation(utils.FormatDate, filtrateStartDateStr, time.Local)
  684. if tmpErr != nil {
  685. err = tmpErr
  686. return
  687. }
  688. //如果筛选时间晚于实际数据时间,那么就以筛选时间作为获取预测数据的时间
  689. if filtrateStartDate.After(latestDate) {
  690. startDate = filtrateStartDate.AddDate(0, 0, -1)
  691. }
  692. }
  693. //var dateArr []string
  694. // 对应日期的值
  695. existMap := make(map[string]float64)
  696. for _, v := range realPredictEdbInfoData {
  697. //dateArr = append(dateArr, v.DataTime)
  698. existMap[v.DataTime] = v.Value
  699. }
  700. predictEdbInfoData = make([]*EdbInfoSearchData, 0)
  701. //dataValue := lastDataValue
  702. //预测规则,1:最新,2:固定值,3:同比,4:同差,5:环比,6:环差,7:N期移动均值,8:N期段线性外推值
  703. for _, predictEdbConf := range predictEdbConfList {
  704. //dataEndTime := endDate
  705. //if predictEdbConf.EndDate.Before(dataEndTime) {
  706. // dataEndTime = predictEdbConf.EndDate
  707. //}
  708. endDate = predictEdbConf.EndDate
  709. var tmpMinValue, tmpMaxValue float64 // 当前预测结果中的最大/最小值
  710. dayList := getPredictEdbDayList(startDate, endDate, frequency, dataDateType)
  711. if len(dayList) <= 0 { // 如果未来没有日期的话,那么就退出当前循环,进入下一个循环
  712. continue
  713. }
  714. switch predictEdbConf.RuleType {
  715. case 1: //1:最新
  716. var lastDataValue float64 //最新值
  717. tmpAllData := make([]*EdbInfoSearchData, 0)
  718. tmpAllData = append(tmpAllData, realPredictEdbInfoData...)
  719. tmpAllData = append(tmpAllData, predictEdbInfoData...)
  720. lenTmpAllData := len(tmpAllData)
  721. if lenTmpAllData > 0 {
  722. lastDataValue = tmpAllData[lenTmpAllData-1].Value
  723. }
  724. predictEdbInfoData = GetChartPredictEdbInfoDataListByRule1(predictEdbConf.PredictEdbInfoId, lastDataValue, dayList, predictEdbInfoData, existMap)
  725. tmpMaxValue = lastDataValue
  726. tmpMinValue = lastDataValue
  727. case 2: //2:固定值
  728. tmpValDecimal, tmpErr := decimal.NewFromString(predictEdbConf.Value)
  729. if tmpErr != nil {
  730. err = tmpErr
  731. return
  732. }
  733. dataValue, _ := tmpValDecimal.Float64()
  734. predictEdbInfoData = GetChartPredictEdbInfoDataListByRule1(predictEdbConf.PredictEdbInfoId, dataValue, dayList, predictEdbInfoData, existMap)
  735. tmpMaxValue = dataValue
  736. tmpMinValue = dataValue
  737. case 3: //3:同比
  738. tmpValDecimal, tmpErr := decimal.NewFromString(predictEdbConf.Value)
  739. if tmpErr != nil {
  740. err = tmpErr
  741. return
  742. }
  743. tbValue, _ := tmpValDecimal.Float64()
  744. predictEdbInfoData, tmpMinValue, tmpMaxValue = GetChartPredictEdbInfoDataListByRuleTb(predictEdbConf.PredictEdbInfoId, tbValue, dayList, frequency, realPredictEdbInfoData, predictEdbInfoData, existMap)
  745. case 4: //4:同差
  746. tmpValDecimal, tmpErr := decimal.NewFromString(predictEdbConf.Value)
  747. if tmpErr != nil {
  748. err = tmpErr
  749. return
  750. }
  751. tcValue, _ := tmpValDecimal.Float64()
  752. predictEdbInfoData, tmpMinValue, tmpMaxValue = GetChartPredictEdbInfoDataListByRuleTc(predictEdbConf.PredictEdbInfoId, tcValue, dayList, frequency, realPredictEdbInfoData, predictEdbInfoData, existMap)
  753. case 5: //5:环比
  754. tmpValDecimal, tmpErr := decimal.NewFromString(predictEdbConf.Value)
  755. if tmpErr != nil {
  756. err = tmpErr
  757. return
  758. }
  759. hbValue, _ := tmpValDecimal.Float64()
  760. predictEdbInfoData, tmpMinValue, tmpMaxValue = GetChartPredictEdbInfoDataListByRuleHb(predictEdbConf.PredictEdbInfoId, hbValue, dayList, realPredictEdbInfoData, predictEdbInfoData, existMap)
  761. case 6: //6:环差
  762. tmpValDecimal, tmpErr := decimal.NewFromString(predictEdbConf.Value)
  763. if tmpErr != nil {
  764. err = tmpErr
  765. return
  766. }
  767. hcValue, _ := tmpValDecimal.Float64()
  768. predictEdbInfoData, tmpMinValue, tmpMaxValue = GetChartPredictEdbInfoDataListByRuleHc(predictEdbConf.PredictEdbInfoId, hcValue, dayList, realPredictEdbInfoData, predictEdbInfoData, existMap)
  769. case 7: //7:N期移动均值
  770. nValue, tmpErr := strconv.Atoi(predictEdbConf.Value)
  771. if tmpErr != nil {
  772. err = tmpErr
  773. return
  774. }
  775. predictEdbInfoData, tmpMinValue, tmpMaxValue = GetChartPredictEdbInfoDataListByRuleNMoveMeanValue(predictEdbConf.PredictEdbInfoId, nValue, dayList, realPredictEdbInfoData, predictEdbInfoData, existMap)
  776. case 8: //8:N期段线性外推值
  777. nValue, tmpErr := strconv.Atoi(predictEdbConf.Value)
  778. if tmpErr != nil {
  779. err = tmpErr
  780. return
  781. }
  782. predictEdbInfoData, tmpMinValue, tmpMaxValue, err = GetChartPredictEdbInfoDataListByRuleNLinearRegression(predictEdbConf.PredictEdbInfoId, nValue, dayList, realPredictEdbInfoData, predictEdbInfoData, existMap)
  783. if err != nil {
  784. return
  785. }
  786. case 9: //9:动态环差”预测规则;
  787. hcDataMap := make(map[string]float64) //规则计算的环差值map
  788. if predictEdbConf.PredictEdbInfoId > 0 {
  789. tmpPredictEdbRuleDataList, tmpErr := GetPredictEdbRuleDataItemList(predictEdbConf.PredictEdbInfoId, predictEdbConf.ConfigId, startDate.Format(utils.FormatDate), endDate.Format(utils.FormatDate))
  790. if tmpErr != nil {
  791. err = tmpErr
  792. return
  793. }
  794. for _, v := range tmpPredictEdbRuleDataList {
  795. hcDataMap[v.DataTime] = v.Value
  796. }
  797. } else {
  798. if len(predictEdbConf.DataList) <= 0 {
  799. return
  800. }
  801. for _, v := range predictEdbConf.DataList {
  802. currentDate, tmpErr := time.ParseInLocation(utils.FormatDate, v.DataTime, time.Local)
  803. if tmpErr != nil {
  804. continue
  805. }
  806. // 只处理时间段内的数据
  807. if currentDate.Before(startDate) || currentDate.After(endDate) {
  808. continue
  809. }
  810. hcDataMap[v.DataTime] = v.Value
  811. }
  812. }
  813. predictEdbInfoData, tmpMinValue, tmpMaxValue = GetChartPredictEdbInfoDataListByRuleTrendsHC(predictEdbConf.PredictEdbInfoId, dayList, realPredictEdbInfoData, predictEdbInfoData, hcDataMap, existMap)
  814. case 10: //10:根据 给定终值后插值 规则获取预测数据
  815. tmpValDecimal, tmpErr := decimal.NewFromString(predictEdbConf.Value)
  816. if tmpErr != nil {
  817. err = tmpErr
  818. return
  819. }
  820. finalValue, _ := tmpValDecimal.Float64()
  821. predictEdbInfoData, tmpMinValue, tmpMaxValue = GetChartPredictEdbInfoDataListByRuleFinalValueHc(predictEdbConf.PredictEdbInfoId, finalValue, dayList, realPredictEdbInfoData, predictEdbInfoData, existMap)
  822. case 11: //11:根据 季节性 规则获取预测数据
  823. var seasonConf SeasonConf
  824. tmpErr := json.Unmarshal([]byte(predictEdbConf.Value), &seasonConf)
  825. if tmpErr != nil {
  826. err = errors.New("季节性配置信息异常:" + tmpErr.Error())
  827. return
  828. }
  829. calendar := "公历"
  830. if seasonConf.Calendar == "农历" {
  831. calendar = "农历"
  832. }
  833. yearList := make([]int, 0)
  834. //选择方式,1:连续N年;2:指定年份
  835. if seasonConf.YearType == 1 {
  836. if seasonConf.NValue < 1 {
  837. err = errors.New("连续N年不允许小于1")
  838. return
  839. }
  840. currYear := time.Now().Year()
  841. for i := 0; i < seasonConf.NValue; i++ {
  842. yearList = append(yearList, currYear-i-1)
  843. }
  844. } else {
  845. yearList = seasonConf.YearList
  846. }
  847. predictEdbInfoData, tmpMinValue, tmpMaxValue, err = GetChartPredictEdbInfoDataListByRuleSeason(predictEdbConf.PredictEdbInfoId, yearList, calendar, dayList, realPredictEdbInfoData, predictEdbInfoData, existMap)
  848. if err != nil {
  849. return
  850. }
  851. case 12: //12:根据 移动平均同比 规则获取预测数据
  852. var moveAverageConf MoveAverageConf
  853. tmpErr := json.Unmarshal([]byte(predictEdbConf.Value), &moveAverageConf)
  854. if tmpErr != nil {
  855. err = errors.New("季节性配置信息异常:" + tmpErr.Error())
  856. return
  857. }
  858. predictEdbInfoData, tmpMinValue, tmpMaxValue, err = GetChartPredictEdbInfoDataListByRuleMoveAverageTb(predictEdbConf.PredictEdbInfoId, moveAverageConf.NValue, moveAverageConf.Year, dayList, realPredictEdbInfoData, predictEdbInfoData, existMap)
  859. if err != nil {
  860. return
  861. }
  862. case 13: //13:根据 同比增速差值 规则获取预测数据
  863. tmpValDecimal, tmpErr := decimal.NewFromString(predictEdbConf.Value)
  864. if tmpErr != nil {
  865. err = tmpErr
  866. return
  867. }
  868. tbEndValue, _ := tmpValDecimal.Float64()
  869. predictEdbInfoData, tmpMinValue, tmpMaxValue = GetChartPredictEdbInfoDataListByRuleTbzscz(predictEdbConf.PredictEdbInfoId, tbEndValue, dayList, frequency, realPredictEdbInfoData, predictEdbInfoData, existMap)
  870. case 14: //14:根据 一元线性拟合 规则获取预测数据
  871. var ruleConf RuleLineNhConf
  872. err = json.Unmarshal([]byte(predictEdbConf.Value), &ruleConf)
  873. if err != nil {
  874. err = errors.New("一元线性拟合规则配置信息异常:" + err.Error())
  875. return
  876. }
  877. // 规则计算的拟合残差值map
  878. newNhccDataMap := make(map[string]float64)
  879. if predictEdbConf.PredictEdbInfoId > 0 { //已经生成的动态数据
  880. tmpPredictEdbRuleDataList, tmpErr := GetPredictEdbRuleDataItemList(predictEdbConf.PredictEdbInfoId, predictEdbConf.ConfigId, startDate.Format(utils.FormatDate), endDate.Format(utils.FormatDate))
  881. if tmpErr != nil {
  882. err = tmpErr
  883. return
  884. }
  885. for _, v := range tmpPredictEdbRuleDataList {
  886. newNhccDataMap[v.DataTime] = v.Value
  887. }
  888. } else { //未生成的动态数据,需要使用外部传入的数据进行计算
  889. newNhccDataMap, err, _ = getCalculateNhccData(append(realPredictEdbInfoData, predictEdbInfoData...), ruleConf)
  890. }
  891. predictEdbInfoData, tmpMinValue, tmpMaxValue, err = GetChartPredictEdbInfoDataListByRuleLineNh(predictEdbConf.PredictEdbInfoId, dayList, realPredictEdbInfoData, predictEdbInfoData, newNhccDataMap, existMap)
  892. if err != nil {
  893. return
  894. }
  895. case 15: //15:N年均值:过去N年同期均值。过去N年可以连续或者不连续,指标数据均用线性插值补全为日度数据后计算;
  896. predictEdbInfoData, tmpMinValue, tmpMaxValue, err = GetChartPredictEdbInfoDataListByRuleNAnnualAverage(predictEdbConf.PredictEdbInfoId, predictEdbConf.Value, dayList, realPredictEdbInfoData, predictEdbInfoData, existMap)
  897. if err != nil {
  898. return
  899. }
  900. case 16: //16:年度值倒推
  901. predictEdbInfoData, tmpMinValue, tmpMaxValue, err = GetChartPredictEdbInfoDataListByRuleAnnualValueInversion(predictEdbConf.PredictEdbInfoId, predictEdbConf.Value, dayList, frequency, realPredictEdbInfoData, predictEdbInfoData, existMap)
  902. if err != nil {
  903. return
  904. }
  905. }
  906. // 下一个规则的开始日期
  907. {
  908. lenPredictEdbInfoData := len(predictEdbInfoData)
  909. if lenPredictEdbInfoData > 0 {
  910. tmpDataEndTime, _ := time.ParseInLocation(utils.FormatDate, predictEdbInfoData[lenPredictEdbInfoData-1].DataTime, time.Local)
  911. if startDate.Before(tmpDataEndTime) {
  912. startDate = tmpDataEndTime
  913. }
  914. }
  915. }
  916. if tmpMinValue < minValue {
  917. minValue = tmpMinValue
  918. }
  919. if tmpMaxValue < maxValue {
  920. maxValue = tmpMaxValue
  921. }
  922. }
  923. return
  924. }
  925. // GetPredictEdbDayList 获取预测指标日期列表
  926. // GetPredictEdbDayList 获取预测指标日期列表
  927. func getPredictEdbDayList(startDate, endDate time.Time, frequency, dataDateType string) (dayList []time.Time) {
  928. if dataDateType == `` {
  929. dataDateType = `交易日`
  930. }
  931. switch frequency {
  932. case "日度":
  933. for currDate := startDate.AddDate(0, 0, 1); currDate.Before(endDate) || currDate.Equal(endDate); currDate = currDate.AddDate(0, 0, 1) {
  934. // 如果日期类型是交易日的时候,那么需要将周六、日排除
  935. if dataDateType == `交易日` && (currDate.Weekday() == time.Sunday || currDate.Weekday() == time.Saturday) {
  936. continue
  937. }
  938. dayList = append(dayList, currDate)
  939. }
  940. case "周度":
  941. //nextDate := startDate.AddDate(0, 0, 7)
  942. for currDate := startDate.AddDate(0, 0, 7); currDate.Before(endDate) || currDate.Equal(endDate); currDate = currDate.AddDate(0, 0, 7) {
  943. dayList = append(dayList, currDate)
  944. }
  945. case "旬度":
  946. for currDate := startDate.AddDate(0, 0, 1); currDate.Before(endDate) || currDate.Equal(endDate); {
  947. nextDate := currDate.AddDate(0, 0, 1)
  948. //每个月的10号、20号、最后一天,那么就写入
  949. if nextDate.Day() == 11 || nextDate.Day() == 21 || nextDate.Day() == 1 {
  950. dayList = append(dayList, currDate)
  951. }
  952. currDate = nextDate
  953. }
  954. case "月度":
  955. for currDate := startDate; currDate.Before(endDate) || currDate.Equal(endDate); {
  956. currDate = time.Date(currDate.Year(), currDate.Month(), 1, 0, 0, 0, 0, time.Now().Location()).AddDate(0, 1, -1)
  957. if !currDate.After(endDate) && !currDate.Equal(startDate) {
  958. dayList = append(dayList, currDate)
  959. }
  960. currDate = currDate.AddDate(0, 0, 1)
  961. }
  962. case "季度":
  963. for currDate := startDate; currDate.Before(endDate) || currDate.Equal(endDate); {
  964. // 每月的最后一天
  965. currDate = time.Date(currDate.Year(), currDate.Month(), 1, 0, 0, 0, 0, time.Now().Location()).AddDate(0, 1, -1)
  966. if !currDate.After(endDate) && !currDate.Equal(startDate) {
  967. // 季度日期就写入,否则不写入
  968. if currDate.Month() == 3 || currDate.Month() == 6 || currDate.Month() == 9 || currDate.Month() == 12 {
  969. dayList = append(dayList, currDate)
  970. }
  971. }
  972. currDate = currDate.AddDate(0, 0, 1)
  973. }
  974. case "半年度":
  975. for currDate := startDate; currDate.Before(endDate) || currDate.Equal(endDate); {
  976. // 每月的最后一天
  977. currDate = time.Date(currDate.Year(), currDate.Month(), 1, 0, 0, 0, 0, time.Now().Location()).AddDate(0, 1, -1)
  978. if !currDate.After(endDate) && !currDate.Equal(startDate) {
  979. // 半年度日期就写入,否则不写入
  980. if currDate.Month() == 6 || currDate.Month() == 12 {
  981. dayList = append(dayList, currDate)
  982. }
  983. }
  984. currDate = currDate.AddDate(0, 0, 1)
  985. }
  986. case "年度":
  987. for currDate := startDate; currDate.Before(endDate) || currDate.Equal(endDate); {
  988. currDate = time.Date(currDate.Year()+1, 12, 31, 0, 0, 0, 0, time.Now().Location())
  989. if !currDate.After(endDate) && !currDate.Equal(startDate) {
  990. dayList = append(dayList, currDate)
  991. }
  992. }
  993. }
  994. return
  995. }
  996. // GetPredictDataListByPredictEdbInfo 根据预测指标信息获取预测指标的数据,order:1升序,其余值为降序
  997. func GetPredictDataListByPredictEdbInfo(edbInfo *EdbInfo, order int, startDate string) (dataList []*EdbInfoSearchData, sourceEdbInfoItem *EdbInfo, err error, errMsg string) {
  998. // 查找该预测指标配置
  999. predictEdbConfList, err := GetPredictEdbConfAndDataListById(edbInfo.EdbInfoId)
  1000. if err != nil && err.Error() != utils.ErrNoRow() {
  1001. errMsg = "获取预测指标配置信息失败"
  1002. return
  1003. }
  1004. if len(predictEdbConfList) == 0 {
  1005. errMsg = "获取预测指标配置信息失败"
  1006. err = errors.New(errMsg)
  1007. return
  1008. }
  1009. predictEdbConf := predictEdbConfList[0]
  1010. // 来源指标
  1011. sourceEdbInfoItem, err = GetEdbInfoById(predictEdbConf.SourceEdbInfoId)
  1012. if err != nil {
  1013. if err.Error() == utils.ErrNoRow() {
  1014. errMsg = "找不到来源指标信息"
  1015. err = errors.New(errMsg)
  1016. }
  1017. return
  1018. }
  1019. dataList, err, errMsg = GetPredictDataListByPredictEdbConfList(edbInfo, sourceEdbInfoItem, predictEdbConfList, order, startDate)
  1020. return
  1021. }
  1022. // GetPredictDataListByPredictEdbConfList 根据预测指标信息获取预测指标的数据,order:1升序,其余值为降序
  1023. func GetPredictDataListByPredictEdbConfList(edbInfo, sourceEdbInfoItem *EdbInfo, predictEdbConfList []*PredictEdbConfAndData, order int, startDate string) (dataList []*EdbInfoSearchData, err error, errMsg string) {
  1024. allDataList := make([]*EdbInfoSearchData, 0)
  1025. //获取指标数据(实际已生成)
  1026. tmpDataList, err := GetEdbDataListAll(sourceEdbInfoItem.Source, sourceEdbInfoItem.SubSource, FindEdbDataListAllCond{
  1027. EdbInfoId: sourceEdbInfoItem.EdbInfoId,
  1028. StartDataTime: startDate,
  1029. StartDataTimeCond: ">=",
  1030. }, 1)
  1031. if err != nil {
  1032. return
  1033. }
  1034. // 默认认为上一个查询是把所有的数据查出来了
  1035. allDataList = tmpDataList
  1036. // 如果条件中有选择了日期,那么上次查询实际并不是把所有的数据查出来了
  1037. // 那么需要把日期给过滤掉,然后筛选所有的数据,用于未来指标的生成
  1038. if startDate != `` {
  1039. allDataList, err = GetEdbDataListAll(sourceEdbInfoItem.Source, sourceEdbInfoItem.SubSource, FindEdbDataListAllCond{
  1040. EdbInfoId: sourceEdbInfoItem.EdbInfoId,
  1041. }, 1)
  1042. if err != nil {
  1043. return
  1044. }
  1045. }
  1046. // 获取预测指标未来的数据
  1047. predictDataList := make([]*EdbInfoSearchData, 0)
  1048. endDateStr := edbInfo.EndDate //预测指标的结束日期
  1049. var predictMinValue, predictMaxValue float64
  1050. // 如果有配置的预测规则,那么就进行预测
  1051. if len(predictEdbConfList) > 0 {
  1052. predictDataList, predictMinValue, predictMaxValue, err = GetChartPredictEdbInfoDataListByConfList(predictEdbConfList, startDate, sourceEdbInfoItem.LatestDate, endDateStr, edbInfo.Frequency, edbInfo.DataDateType, allDataList)
  1053. if err != nil {
  1054. return
  1055. }
  1056. }
  1057. //order:1升序,其余值为降序
  1058. if order == 1 {
  1059. dataList = append(tmpDataList, predictDataList...)
  1060. } else {
  1061. // 先倒序预测数据
  1062. lenPredictDataList := len(predictDataList)
  1063. if lenPredictDataList > 0 {
  1064. for k := range predictDataList {
  1065. dataList = append(dataList, predictDataList[lenPredictDataList-k-1])
  1066. }
  1067. }
  1068. // 接着倒序实际指标
  1069. lenDataList := len(tmpDataList)
  1070. for k := range tmpDataList {
  1071. dataList = append(dataList, tmpDataList[lenDataList-k-1])
  1072. }
  1073. }
  1074. if len(predictDataList) > 0 {
  1075. // 如果最小值 大于 预测值,那么将预测值作为最小值数据返回
  1076. if edbInfo.MinValue > predictMinValue {
  1077. edbInfo.MinValue = predictMinValue
  1078. }
  1079. // 如果最大值 小于 预测值,那么将预测值作为最大值数据返回
  1080. if edbInfo.MaxValue < predictMaxValue {
  1081. edbInfo.MaxValue = predictMaxValue
  1082. }
  1083. }
  1084. return
  1085. }
  1086. // GetPredictEdbDataListAll 获取该预测指标所有的数据 ,order:1升序,其余值为降序
  1087. func GetPredictEdbDataListAll(edbInfo *EdbInfo, order int) (items []*EdbInfoSearchData, err error) {
  1088. if edbInfo.Source == utils.DATA_SOURCE_PREDICT { //普通的预测指标是没有入库数据的,直接往配置里面获取
  1089. items, _, err, _ = GetPredictDataListByPredictEdbInfo(edbInfo, 1, "")
  1090. } else {
  1091. items, err = GetEdbDataListAll(edbInfo.Source, edbInfo.SubSource, FindEdbDataListAllCond{
  1092. EdbInfoId: edbInfo.EdbInfoId,
  1093. }, order)
  1094. }
  1095. return
  1096. }
  1097. // GetPredictEdbDataListAllByStartDate 根据开始日期获取该预测指标所有的数据 ,order:1升序,其余值为降序
  1098. func GetPredictEdbDataListAllByStartDate(edbInfo *EdbInfo, order int, startDate string) (items []*EdbInfoSearchData, err error) {
  1099. if edbInfo.Source == utils.DATA_SOURCE_PREDICT { //普通的预测指标是没有入库数据的,直接往配置里面获取
  1100. items, _, err, _ = GetPredictDataListByPredictEdbInfo(edbInfo, order, startDate)
  1101. } else {
  1102. items, err = GetEdbDataListAll(edbInfo.Source, edbInfo.SubSource, FindEdbDataListAllCond{
  1103. EdbInfoId: edbInfo.EdbInfoId,
  1104. StartDataTime: startDate,
  1105. StartDataTimeCond: ">=",
  1106. }, order)
  1107. }
  1108. return
  1109. }
  1110. // ModifyPredictEdbInfoMaxAndMinInfo 修改预测指标的最新数据信息
  1111. func ModifyPredictEdbInfoMaxAndMinInfo(edbInfoId int, item *EdbInfoMaxAndMinInfo) (err error) {
  1112. 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=? `
  1113. err = global.DEFAULT_DmSQL.Exec(sql, item.MinDate, item.MaxDate, item.MinValue, item.MaxValue, item.LatestDate, item.LatestValue, item.EndValue, edbInfoId).Error
  1114. return
  1115. }
  1116. // ModifyCalculateEdbInfo 修改计算指标信息
  1117. func ModifyCalculateEdbInfo(edbName, frequency, unit, calculateFormula string, classifyId, edbInfoId int) (err error) {
  1118. sql := ` UPDATE edb_info
  1119. SET
  1120. edb_name =?,
  1121. edb_name_source =?,
  1122. frequency = ?,
  1123. unit = ?,
  1124. classify_id = ?,
  1125. calculate_formula=?,
  1126. modify_time = NOW()
  1127. WHERE edb_info_id = ?`
  1128. err = global.DEFAULT_DmSQL.Exec(sql, edbName, edbName, frequency, unit, classifyId, calculateFormula, edbInfoId).Error
  1129. return
  1130. }
  1131. func GetEdbInfoItemByCondition(condition string, pars []interface{}) (item *EdbInfoList, err error) {
  1132. sql := ` SELECT * FROM edb_info WHERE 1=1 `
  1133. if condition != "" {
  1134. sql += condition
  1135. }
  1136. err = global.DEFAULT_DmSQL.Raw(sql, pars).First(&item).Error
  1137. return
  1138. }
  1139. func GetEdbInfoItemListByCondition() (items []*EdbInfoList, err error) {
  1140. sql := ` SELECT * FROM edb_info WHERE 1=1 `
  1141. err = global.DEFAULT_DmSQL.Raw(sql).Scan(&items).Error
  1142. return
  1143. }
  1144. // EdbInfoDetailReq 指标详情请求
  1145. type EdbInfoDetailReq struct {
  1146. UniqueCode string `description:"指标唯一编码"`
  1147. EdbCode string `description:"指标编码"`
  1148. }
  1149. // EdbInfoDetailResp 指标详情返回
  1150. type EdbInfoDetailResp struct {
  1151. SourceName string `description:"来源名称"`
  1152. EdbCode string `description:"指标编码"`
  1153. EdbName string `description:"指标名称"`
  1154. EdbNameSource string `description:"指标名称来源"`
  1155. Frequency string `description:"频率"`
  1156. Unit string `description:"单位"`
  1157. StartDate string `description:"起始日期"`
  1158. EndDate string `description:"终止日期"`
  1159. UniqueCode string `description:"指标唯一编码"`
  1160. CreateTime time.Time
  1161. ModifyTime time.Time
  1162. MinValue float64 `description:"指标最小值"`
  1163. MaxValue float64 `description:"指标最大值"`
  1164. EdbNameEn string `description:"英文指标名称"`
  1165. UnitEn string `description:"英文单位"`
  1166. LatestDate string `description:"数据最新日期"`
  1167. LatestValue float64 `description:"数据最新值"`
  1168. ChartImage string `description:"图表图片"`
  1169. DataList []*EdbInfoSearchData
  1170. }
  1171. type AddEdbInfoParams struct {
  1172. Source int `description:"来源id"`
  1173. EdbCode string `description:"指标编码"`
  1174. EdbName string `description:"指标名称"`
  1175. Frequency string `description:"频率"`
  1176. Unit string `description:"单位"`
  1177. ClassifyId int `description:"分类id"`
  1178. TerminalCode string `description:"终端编码"`
  1179. AdminId int `description:"创建人ID"`
  1180. AdminName string `description:"创建人姓名"`
  1181. }
  1182. // EdbInfoAdd 添加指标到指标库
  1183. func EdbInfoAdd(req *AddEdbInfoParams, serverUrl string, sysUserId int, sysUserRealName string) (edbInfo *EdbInfo, err error) {
  1184. source := req.Source
  1185. //判断指标名称是否存在
  1186. var condition string
  1187. var pars []interface{}
  1188. //condition += " AND source=? "
  1189. //pars = append(pars, source)
  1190. condition += " AND edb_code=? "
  1191. pars = append(pars, req.EdbCode)
  1192. items, err := GetEdbInfoByCondition(condition, pars, 0)
  1193. if err != nil {
  1194. err = errors.New("判断指标是否存在失败,Err:" + err.Error())
  1195. return
  1196. }
  1197. if len(items) > 0 {
  1198. sql := ` UPDATE edb_info SET classify_id=?,modify_time=NOW() WHERE edb_code=? `
  1199. err = global.DEFAULT_DmSQL.Exec(sql, req.ClassifyId, req.EdbCode).Error
  1200. if err != nil {
  1201. err = errors.New("更新指标分类信息失败,Err:" + err.Error())
  1202. return
  1203. }
  1204. edbInfo = items[0]
  1205. return
  1206. }
  1207. edbInfo = new(EdbInfo)
  1208. edbInfo.Source = source
  1209. sourceNameMap := map[int]string{
  1210. utils.DATA_SOURCE_THS: "同花顺",
  1211. utils.DATA_SOURCE_WIND: "wind",
  1212. utils.DATA_SOURCE_PB: "彭博",
  1213. utils.DATA_SOURCE_PB_FINANCE: "彭博财务",
  1214. utils.DATA_SOURCE_MANUAL: "手工数据",
  1215. utils.DATA_SOURCE_LZ: "隆众",
  1216. utils.DATA_SOURCE_YS: "SMM",
  1217. utils.DATA_SOURCE_GL: "钢联",
  1218. utils.DATA_SOURCE_ZZ: "郑商所",
  1219. utils.DATA_SOURCE_DL: "大商所",
  1220. utils.DATA_SOURCE_SH: "上期所",
  1221. utils.DATA_SOURCE_CFFEX: "中金所",
  1222. utils.DATA_SOURCE_SHFE: "上期能源",
  1223. utils.DATA_SOURCE_GIE: "欧洲天然气",
  1224. utils.DATA_SOURCE_LT: "路透",
  1225. utils.DATA_SOURCE_COAL: "中国煤炭市场网",
  1226. utils.DATA_SOURCE_GOOGLE_TRAVEL: "our world in data",
  1227. utils.DATA_SOURCE_MYSTEEL_CHEMICAL: "钢联化工",
  1228. utils.DATA_SOURCE_EIA_STEO: "EIA STEO报告",
  1229. utils.DATA_SOURCE_COM_TRADE: "UN",
  1230. utils.DATA_SOURCE_SCI: "SCI",
  1231. utils.DATA_SOURCE_BAIINFO: "BAIINFO",
  1232. utils.DATA_SOURCE_STOCK_PLANT: "存量装置",
  1233. utils.DATA_SOURCE_NATIONAL_STATISTICS: "国家统计局",
  1234. utils.DATA_SOURCE_FUBAO: "富宝数据",
  1235. }
  1236. sourceName, ok := sourceNameMap[source]
  1237. if !ok {
  1238. edbSource := EdbSourceIdMap[source]
  1239. if edbSource != nil {
  1240. sourceName = edbSource.SourceName
  1241. }
  1242. if sourceName == "" {
  1243. err = errors.New("指标来源异常")
  1244. return
  1245. }
  1246. }
  1247. edbInfo.SourceName = sourceName
  1248. edbType := 1 //基础指标
  1249. if source == utils.DATA_SOURCE_STOCK_PLANT {
  1250. edbType = 2 //计算指标
  1251. }
  1252. var sourceIndexName string
  1253. // 钢联化工需要校验下信息是否完整
  1254. if source == utils.DATA_SOURCE_MYSTEEL_CHEMICAL {
  1255. indexObj := new(BaseFromMysteelChemicalIndex)
  1256. tmpItem, tmpErr := indexObj.GetIndexItem(req.EdbCode)
  1257. if tmpErr != nil {
  1258. err = tmpErr
  1259. return
  1260. }
  1261. if tmpItem == nil || tmpItem.IndexName == `` {
  1262. err = errors.New("指标信息不全")
  1263. return
  1264. }
  1265. // 兼容数据
  1266. {
  1267. if req.Frequency == `` {
  1268. req.Frequency = tmpItem.Frequency
  1269. }
  1270. if req.Unit == `` {
  1271. req.Unit = tmpItem.Unit
  1272. }
  1273. }
  1274. }
  1275. edbInfo.EdbCode = req.EdbCode
  1276. edbInfo.EdbName = req.EdbName
  1277. edbInfo.EdbNameSource = req.EdbName
  1278. edbInfo.EdbNameEn = req.EdbName
  1279. edbInfo.Frequency = req.Frequency
  1280. edbInfo.Unit = req.Unit
  1281. edbInfo.UnitEn = req.Unit
  1282. edbInfo.ClassifyId = req.ClassifyId
  1283. edbInfo.SysUserId = sysUserId
  1284. edbInfo.SysUserRealName = sysUserRealName
  1285. edbInfo.StartDate = "1990-01-01"
  1286. edbInfo.CreateTime = time.Now()
  1287. edbInfo.ModifyTime = time.Now()
  1288. edbInfo.ServerUrl = serverUrl
  1289. edbInfo.TerminalCode = req.TerminalCode
  1290. edbInfo.Sort = GetAddEdbMaxSortByClassifyId(req.ClassifyId, 0)
  1291. timestamp := strconv.FormatInt(time.Now().UnixNano(), 10)
  1292. edbInfo.UniqueCode = utils.MD5(utils.DATA_PREFIX + "_" + req.EdbCode + timestamp)
  1293. itemVal, err := GetEdbInfoMaxAndMinInfo(source, utils.DATA_SUB_SOURCE_EDB, req.EdbCode)
  1294. if itemVal != nil && err == nil {
  1295. edbInfo.MaxValue = itemVal.MaxValue
  1296. edbInfo.MinValue = itemVal.MinValue
  1297. }
  1298. edbInfo.EdbType = edbType
  1299. edbInfo.SourceIndexName = sourceIndexName
  1300. edbInfoId, err := AddEdbInfo(edbInfo)
  1301. if err != nil {
  1302. err = errors.New("保存失败,Err:" + err.Error())
  1303. return
  1304. }
  1305. edbInfo.EdbInfoId = int(edbInfoId)
  1306. return
  1307. }
  1308. func ModifyEdbInfoBaseTimeById(edbInfoId int, cTime time.Time) (err error) {
  1309. // 更新修改时间
  1310. sql := ` UPDATE edb_info SET base_modify_time = ? WHERE edb_info_id = ? `
  1311. err = global.DEFAULT_DmSQL.Exec(sql, cTime, edbInfoId).Error
  1312. return
  1313. }
  1314. // EdbInfoRefreshCheckReq 指标数据更新情况查询
  1315. type EdbInfoRefreshCheckReq struct {
  1316. Source int `description:"来源id"`
  1317. LatestDate string `description:"数据最新日期"`
  1318. FrequencyBatch string `description:"更新频度"`
  1319. }
  1320. // EdbInfoRefreshCheckResp 指标数据更新情况查询
  1321. type EdbInfoRefreshCheckResp struct {
  1322. UpdateNum int `description:"已更新指标数"`
  1323. UnUpdateNum int `description:"未更新指标数"`
  1324. }
  1325. // ResetEdbClassifyReq 指标重新分类请求体
  1326. type ResetEdbClassifyReq struct {
  1327. IndexCode string `description:"指标编码"`
  1328. ClassifyFirst string `description:"一级分类名称"`
  1329. ClassifySecond string `description:"二级分类名称"`
  1330. ClassifyThird string `description:"三级分类名称"`
  1331. }
  1332. // GetEdbInfoMaxSortByClassifyId 获取分类下指标的最大的排序数
  1333. func GetEdbInfoMaxSortByClassifyId(classifyId int) (sort int, err error) {
  1334. sql := `SELECT Max(sort) AS sort FROM edb_info WHERE classify_id=? `
  1335. err = global.DEFAULT_DmSQL.Raw(sql, classifyId).Scan(&sort).Error
  1336. return
  1337. }
  1338. func GetEdbAndClassifyMaxSort(parentId int, classifyType uint8) (maxSort int, err error) {
  1339. //获取该层级下最大的排序数
  1340. classifyMaxSort, err := GetEdbClassifyMaxSort(parentId, classifyType)
  1341. if err != nil {
  1342. return
  1343. }
  1344. maxSort = classifyMaxSort
  1345. edbMaxSort, err := GetEdbInfoMaxSortByClassifyId(parentId)
  1346. if err != nil {
  1347. return
  1348. }
  1349. if maxSort < edbMaxSort {
  1350. maxSort = edbMaxSort
  1351. }
  1352. return
  1353. }
  1354. // CalculateEdbExtra 指标运算额外配置
  1355. type CalculateEdbExtra struct {
  1356. DateTag string `description:"时间序列的生成方式,all 表示所选指标的时间序列并集"`
  1357. }
  1358. // CalculateLjzEdbExtra 累计值额外配置
  1359. type CalculateLjzEdbExtra struct {
  1360. LastValType int `description:"最新值处理:0默认、均值填充"`
  1361. }
  1362. type EdbInfoEditRecord struct {
  1363. EdbInfoId int `description:"指标ID"`
  1364. EdbName string `description:"指标名称"`
  1365. Frequency string `description:"频率"`
  1366. Unit string `description:"单位"`
  1367. ClassifyId int `description:"分类id"`
  1368. CalculateFormula string `description:"计算公式"`
  1369. OperateUserId int `description:"操作人id"`
  1370. OperateUserRealName string `description:"操作人姓名"`
  1371. }
  1372. // GetEdbInfoByEdbCodeList
  1373. // @Description: 根据来源和指标编码列表获取指标信息列表
  1374. // @author: Roc
  1375. // @datetime 2024-05-31 16:31:52
  1376. // @param source int
  1377. // @param edbCodeList []string
  1378. // @return items []*EdbInfo
  1379. // @return err error
  1380. func GetEdbInfoByEdbCodeList(source int, edbCodeList []string) (items []*EdbInfo, err error) {
  1381. num := len(edbCodeList)
  1382. if num <= 0 {
  1383. return
  1384. }
  1385. sql := ` SELECT * FROM edb_info WHERE source=? AND edb_code IN (` + utils.GetOrmInReplace(num) + `) `
  1386. err = global.DEFAULT_DmSQL.Raw(sql, source, edbCodeList).Scan(&items).Error
  1387. return
  1388. }
  1389. // GetAddEdbMaxSortByClassifyId
  1390. // @Description: 获取添加指标时,该分类下最大的排序(忽略错误信息)
  1391. // @author: Roc
  1392. // @datetime 2024-07-05 09:39:46
  1393. // @param classifyId int
  1394. // @param classifyType uint8
  1395. // @return sort int
  1396. func GetAddEdbMaxSortByClassifyId(classifyId int, classifyType uint8) (sort int) {
  1397. sort, _ = GetEdbAndClassifyMaxSort(classifyId, classifyType)
  1398. sort = sort + 1
  1399. return
  1400. }
  1401. // EdbInfoExtra 指标额外数据-extra字段
  1402. type EdbInfoExtra struct {
  1403. ApiExtraPars string `description:"API-额外参数(如同花顺日期序列)"`
  1404. }
  1405. // GetEdbInfoNoUpdateTotalByIdList 根据指标id列表获取指标信息
  1406. func GetEdbInfoNoUpdateTotalByIdList(edbInfoIdList []int) (total int, err error) {
  1407. num := len(edbInfoIdList)
  1408. if num <= 0 {
  1409. return
  1410. }
  1411. sql := ` SELECT count(*) FROM edb_info WHERE edb_info_id in (` + utils.GetOrmInReplace(num) + `) and no_update=1`
  1412. err = global.DEFAULT_DmSQL.Raw(sql, edbInfoIdList).Scan(&total).Error
  1413. return
  1414. }
  1415. func TransEdbInfoSearchData2DataList(items []*EdbInfoSearchData) (list []*EdbDataList) {
  1416. if len(items) == 0 {
  1417. return
  1418. }
  1419. list = make([]*EdbDataList, 0)
  1420. for _, v := range items {
  1421. list = append(list, &EdbDataList{
  1422. EdbDataId: v.EdbDataId,
  1423. EdbInfoId: v.EdbInfoId,
  1424. DataTime: v.DataTime,
  1425. DataTimestamp: v.DataTimestamp,
  1426. Value: v.Value,
  1427. })
  1428. }
  1429. return
  1430. }
  1431. func TransEdbInfoDataList2SearchData(items []*EdbDataList) (list []*EdbInfoSearchData) {
  1432. if len(items) == 0 {
  1433. return
  1434. }
  1435. list = make([]*EdbInfoSearchData, 0)
  1436. for _, v := range items {
  1437. list = append(list, &EdbInfoSearchData{
  1438. EdbDataId: v.EdbDataId,
  1439. EdbInfoId: v.EdbInfoId,
  1440. DataTime: v.DataTime,
  1441. DataTimestamp: v.DataTimestamp,
  1442. Value: v.Value,
  1443. })
  1444. }
  1445. return
  1446. }
  1447. type SortEdbDataList []*EdbInfoSearchData
  1448. func (m SortEdbDataList) Len() int {
  1449. return len(m)
  1450. }
  1451. func (m SortEdbDataList) Less(i, j int) bool {
  1452. return m[i].DataTime > m[j].DataTime
  1453. }
  1454. func (m SortEdbDataList) Swap(i, j int) {
  1455. m[i], m[j] = m[j], m[i]
  1456. }