edb_info.go 43 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159
  1. package models
  2. import (
  3. "encoding/json"
  4. "errors"
  5. "eta/eta_index_lib/utils"
  6. "fmt"
  7. "github.com/beego/beego/v2/client/orm"
  8. "github.com/shopspring/decimal"
  9. "strconv"
  10. "time"
  11. )
  12. type EdbInfo struct {
  13. EdbInfoId int `orm:"column(edb_info_id);pk"`
  14. EdbInfoType int `description:"指标类型,0:普通指标,1:预测指标"`
  15. SourceName string `description:"来源名称"`
  16. Source int `description:"来源id"`
  17. EdbCode string `description:"指标编码"`
  18. EdbName string `description:"指标名称"`
  19. EdbNameEn string `description:"英文指标名称"`
  20. EdbNameSource string `description:"指标名称来源"`
  21. Frequency string `description:"频率"`
  22. Unit string `description:"单位"`
  23. UnitEn string `description:"英文单位"`
  24. StartDate string `description:"起始日期"`
  25. EndDate string `description:"终止日期"`
  26. ClassifyId int `description:"分类id"`
  27. SysUserId int
  28. SysUserRealName string
  29. UniqueCode string `description:"指标唯一编码"`
  30. CreateTime time.Time
  31. ModifyTime time.Time
  32. MinValue float64 `description:"指标最小值"`
  33. MaxValue float64 `description:"指标最大值"`
  34. CalculateFormula string `description:"计算公式"`
  35. EdbType int `description:"指标类型:1:基础指标,2:计算指标"`
  36. IsUpdate int `description:"当天是否已更新,1:未更新,2:已更新"`
  37. Sort int `description:"排序字段"`
  38. LatestDate string `description:"数据最新日期"`
  39. LatestValue float64 `description:"数据最新值"`
  40. MoveType int `description:"移动方式:1:领先(默认),2:滞后"`
  41. MoveFrequency string `description:"移动频度"`
  42. NoUpdate int8 `description:"是否停止更新,0:继续更新;1:停止更新"`
  43. ServerUrl string `description:"服务器地址"`
  44. ChartImage string `description:"图表图片"`
  45. Calendar string `description:"公历/农历" orm:"default(公历);"`
  46. DataDateType string `orm:"column(data_date_type);size(255);null;default(交易日)"`
  47. ManualSave int `description:"是否有手动保存过上下限: 0-否; 1-是"`
  48. SubSource int `description:"子数据来源:0:经济数据库,1:日期序列"`
  49. SubSourceName string `description:"子数据来源名称"`
  50. IndicatorCode string `description:"指标代码"`
  51. StockCode string `description:"证券代码"`
  52. }
  53. func (e *EdbInfo) Add() (err error) {
  54. o := orm.NewOrm()
  55. _, err = o.Insert(e)
  56. return
  57. }
  58. // AddEdbInfo 添加指标
  59. func AddEdbInfo(item *EdbInfo) (lastId int64, err error) {
  60. o := orm.NewOrm()
  61. lastId, err = o.Insert(item)
  62. return
  63. }
  64. // EdbInfoList 指标数据列表
  65. type EdbInfoList struct {
  66. EdbInfoId int `orm:"column(edb_info_id);pk"`
  67. EdbInfoType int `description:"指标类型,0:普通指标,1:预测指标"`
  68. SourceName string `description:"来源名称"`
  69. Source int `description:"来源id"`
  70. EdbCode string `description:"指标编码"`
  71. EdbNameEn string `description:"英文指标名称"`
  72. EdbName string `description:"指标名称"`
  73. Frequency string `description:"频率"`
  74. FrequencyEn string `description:"英文频率"`
  75. Unit string `description:"单位"`
  76. UnitEn string `description:"英文单位"`
  77. StartDate string `description:"起始日期"`
  78. EndDate string `description:"终止日期"`
  79. LatestDate string `description:"数据最新日期"`
  80. LatestValue float64 `description:"数据最新值"`
  81. ClassifyId int `description:"分类id"`
  82. UniqueCode string `description:"指标唯一编码"`
  83. SysUserId int `description:"创建人id"`
  84. SysUserRealName string `description:"创建人姓名"`
  85. ModifyTime string `description:"最新修改时间"`
  86. CreateTime string `description:"创建时间"`
  87. EdbNameAlias string `json:"-" description:"指标名称,别名"`
  88. EdbType int `description:"指标类型:1:基础指标,2:计算指标"`
  89. ChartImage string `description:"图表图片"`
  90. RuleType int `description:"预测规则,1:最新,2:固定值"`
  91. FixedValue float64 `description:"固定值"`
  92. DataList []*EdbData `description:"实际指标数据"`
  93. PredictDataList []*EdbData `description:"预测指标数据"`
  94. Button EdbClassifyItemsButton `description:"操作权限"`
  95. IsEnEdb bool `description:"是否展示英文标识"`
  96. }
  97. // EdbClassifyItemsButton 操作按钮
  98. type EdbClassifyItemsButton struct {
  99. AddButton bool `description:"是否可添加"`
  100. OpButton bool `description:"是否可编辑"`
  101. DeleteButton bool `description:"是否可删除"`
  102. MoveButton bool `description:"是否可移动"`
  103. ShowEdbRelation bool `description:"是否展示关联指标"`
  104. ShowChartRelation bool `description:"是否展示关联图表"`
  105. }
  106. // GetEdbInfoByName 根据指标名称获取所有的指标数据列表
  107. func GetEdbInfoByName(edbName string) (items []*EdbInfoList, err error) {
  108. o := orm.NewOrm()
  109. sql := ` SELECT * FROM edb_info WHERE edb_name=? `
  110. _, err = o.Raw(sql, edbName).QueryRows(&items)
  111. return
  112. }
  113. // ModifyEdbInfoNameSource 根据来源修改指标名称
  114. func ModifyEdbInfoNameSource(edbNameSource string, edbInfoId int) (err error) {
  115. o := orm.NewOrm()
  116. sql := ` UPDATE edb_info SET edb_name_source=? WHERE edb_info_id = ? `
  117. _, err = o.Raw(sql, edbNameSource, edbInfoId).Exec()
  118. return
  119. }
  120. // GetEdbInfoById 根据指标id获取指标信息
  121. func GetEdbInfoById(edbInfoId int) (item *EdbInfo, err error) {
  122. o := orm.NewOrm()
  123. sql := ` SELECT * FROM edb_info WHERE edb_info_id=? `
  124. err = o.Raw(sql, edbInfoId).QueryRow(&item)
  125. return
  126. }
  127. // GetEdbInfoByUniqueCode 根据指标唯一编码获取指标信息
  128. func GetEdbInfoByUniqueCode(uniqueCode string) (item *EdbInfo, err error) {
  129. o := orm.NewOrm()
  130. sql := ` SELECT * FROM edb_info WHERE unique_code=? `
  131. err = o.Raw(sql, uniqueCode).QueryRow(&item)
  132. return
  133. }
  134. // GetEdbInfoByIdList 根据指标id列表获取指标信息
  135. func GetEdbInfoByIdList(edbInfoIdList []int) (items []*EdbInfo, err error) {
  136. num := len(edbInfoIdList)
  137. if num <= 0 {
  138. return
  139. }
  140. o := orm.NewOrm()
  141. sql := ` SELECT * FROM edb_info WHERE edb_info_id in (` + utils.GetOrmInReplace(num) + `) `
  142. _, err = o.Raw(sql, edbInfoIdList).QueryRows(&items)
  143. return
  144. }
  145. // Update 更新EdbInfo信息
  146. func (edbInfo *EdbInfo) Update(cols []string) (err error) {
  147. o := orm.NewOrm()
  148. _, err = o.Update(edbInfo, cols...)
  149. return
  150. }
  151. // EdbInfoSearchData
  152. type EdbInfoSearchData struct {
  153. EdbDataId int `description:"数据ID"`
  154. DataTime string `description:"数据日期"`
  155. Value float64 `description:"数据"`
  156. EdbCode string `description:"指标编码"`
  157. }
  158. // GetEdbDataListAll 获取指标数据列表 order:1升序,其余值为降序
  159. func GetEdbDataListAll(condition string, pars []interface{}, source, subSource, order int) (item []*EdbInfoSearchData, err error) {
  160. o := orm.NewOrm()
  161. sql := ``
  162. tableName := GetEdbDataTableName(source, subSource)
  163. sql = ` SELECT * FROM %s WHERE 1=1 `
  164. sql = fmt.Sprintf(sql, tableName)
  165. if condition != "" {
  166. sql += condition
  167. }
  168. if order == 1 {
  169. sql += ` ORDER BY data_time ASC `
  170. } else {
  171. sql += ` ORDER BY data_time DESC `
  172. }
  173. _, err = o.Raw(sql, pars).QueryRows(&item)
  174. return
  175. }
  176. // GetEdbDataListAllByTo 根据事务链接获取指标数据列表 order:1升序,其余值为降序
  177. func GetEdbDataListAllByTo(to orm.TxOrmer, condition string, pars []interface{}, source, subSource, order int) (item []*EdbInfoSearchData, err error) {
  178. sql := ``
  179. tableName := GetEdbDataTableName(source, subSource)
  180. sql = ` SELECT * FROM %s WHERE 1=1 `
  181. sql = fmt.Sprintf(sql, tableName)
  182. if condition != "" {
  183. sql += condition
  184. }
  185. if order == 1 {
  186. sql += ` ORDER BY data_time ASC `
  187. } else {
  188. sql += ` ORDER BY data_time DESC `
  189. }
  190. _, err = to.Raw(sql, pars).QueryRows(&item)
  191. return
  192. }
  193. // EdbInfoMaxAndMinInfo 指标最新数据记录结构体
  194. type EdbInfoMaxAndMinInfo struct {
  195. MinDate string `description:"最小日期"`
  196. MaxDate string `description:"最大日期"`
  197. MinValue float64 `description:"最小值"`
  198. MaxValue float64 `description:"最大值"`
  199. LatestValue float64 `description:"最新值"`
  200. LatestDate string `description:"实际数据最新日期"`
  201. }
  202. // GetEdbInfoMaxAndMinInfo 获取指标的最新数据记录信息
  203. func GetEdbInfoMaxAndMinInfo(source, subSource int, edbCode string) (item *EdbInfoMaxAndMinInfo, err error) {
  204. o := orm.NewOrm()
  205. sql := ``
  206. tableName := GetEdbDataTableName(source, subSource)
  207. 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=? `
  208. sql = fmt.Sprintf(sql, tableName)
  209. err = o.Raw(sql, edbCode).QueryRow(&item)
  210. var latest_value float64
  211. sql = ` SELECT value AS latest_value FROM %s WHERE edb_code=? ORDER BY data_time DESC LIMIT 1 `
  212. sql = fmt.Sprintf(sql, tableName)
  213. err = o.Raw(sql, edbCode).QueryRow(&latest_value)
  214. item.LatestValue = latest_value
  215. return
  216. }
  217. // ModifyEdbInfoMaxAndMinInfo 修改指标的最新数据信息
  218. func ModifyEdbInfoMaxAndMinInfo(edbInfoId int, item *EdbInfoMaxAndMinInfo) (err error) {
  219. o := orm.NewOrm()
  220. sql := ` UPDATE edb_info SET start_date=?,end_date=?,min_value=?,max_value=?,is_update=2,latest_date=?,latest_value=?,modify_time=NOW() WHERE edb_info_id=? `
  221. _, err = o.Raw(sql, item.MinDate, item.MaxDate, item.MinValue, item.MaxValue, item.MaxDate, item.LatestValue, edbInfoId).Exec()
  222. return
  223. }
  224. // ModifyEdbTimeAndLastInfo 修改指标的最新数据信息(除上下限)
  225. func ModifyEdbTimeAndLastInfo(edbInfoId int, item *EdbInfoMaxAndMinInfo) (err error) {
  226. o := orm.NewOrm()
  227. sql := `UPDATE edb_info SET start_date = ?, end_date = ?, is_update = 2,latest_date = ?, latest_value = ?, modify_time = NOW() WHERE edb_info_id = ?`
  228. _, err = o.Raw(sql, item.MinDate, item.MaxDate, item.MaxDate, item.LatestValue, edbInfoId).Exec()
  229. return
  230. }
  231. // GetEdbDataCount 获取edb指标数据的数量; order:1升序,其余值为降序
  232. func GetEdbDataCount(condition string, pars []interface{}, source, subSource int) (count int, err error) {
  233. o := orm.NewOrm()
  234. sql := ``
  235. tableName := GetEdbDataTableName(source, subSource)
  236. sql = ` SELECT COUNT(1) AS count FROM %s WHERE 1=1 `
  237. sql = fmt.Sprintf(sql, tableName)
  238. if condition != "" {
  239. sql += condition
  240. }
  241. err = o.Raw(sql, pars).QueryRow(&count)
  242. return
  243. }
  244. // GetLastEdbData 获取最近的一条指标数据
  245. func GetLastEdbData(condition string, pars []interface{}, source, subSource int) (item *EdbInfoSearchData, err error) {
  246. o := orm.NewOrm()
  247. sql := ``
  248. tableName := GetEdbDataTableName(source, subSource)
  249. sql = ` SELECT * FROM %s WHERE 1=1 `
  250. sql = fmt.Sprintf(sql, tableName)
  251. if condition != "" {
  252. sql += condition
  253. }
  254. sql += ` ORDER BY data_time DESC `
  255. err = o.Raw(sql, pars).QueryRow(&item)
  256. return
  257. }
  258. // GetEdbInfoByEdbCode 根据指标code获取指标信息
  259. func GetEdbInfoByEdbCode(source int, edbCode string) (item *EdbInfo, err error) {
  260. o := orm.NewOrm()
  261. sql := ` SELECT * FROM edb_info WHERE source=? AND edb_code=? `
  262. err = o.Raw(sql, source, edbCode).QueryRow(&item)
  263. return
  264. }
  265. // GetEdbInfoOnlyByEdbCode 仅根据指标code获取指标信息
  266. func GetEdbInfoOnlyByEdbCode(edbCode string) (item *EdbInfo, err error) {
  267. o := orm.NewOrm()
  268. sql := ` SELECT * FROM edb_info WHERE edb_code=? `
  269. err = o.Raw(sql, edbCode).QueryRow(&item)
  270. return
  271. }
  272. // GetEdbInfoCalculateListByCondition 获取指标关系列表
  273. func GetEdbInfoCalculateListByCondition(condition string, pars []interface{}) (items []*EdbInfoCalculateMapping, err error) {
  274. o := orm.NewOrm()
  275. sql := ` SELECT a.*,b.sub_source AS from_sub_source FROM edb_info_calculate_mapping AS a LEFT JOIN edb_info AS b ON a.from_edb_info_id = b.edb_info_id WHERE 1=1 `
  276. if condition != "" {
  277. sql += condition
  278. }
  279. _, err = o.Raw(sql, pars).QueryRows(&items)
  280. return
  281. }
  282. // GetEdbInfoCalculateCountByCondition 获取关联指标数量
  283. func GetEdbInfoCalculateCountByCondition(condition string, pars []interface{}) (count int, err error) {
  284. o := orm.NewOrm()
  285. sql := ` SELECT COUNT(1) AS count FROM edb_info_calculate_mapping WHERE 1=1 `
  286. if condition != "" {
  287. sql += condition
  288. }
  289. err = o.Raw(sql, pars).QueryRow(&count)
  290. return
  291. }
  292. // 优化版本-处理数据精度问题
  293. type EdbInfoSearchDataV1 struct {
  294. EdbDataId int `description:"数据ID"`
  295. DataTime string `description:"数据日期"`
  296. Value string `description:"数据"`
  297. }
  298. // 优化版本-处理数据精度问题
  299. func GetEdbDataListAllV1(condition string, pars []interface{}, source, subSource, order int) (item []*EdbInfoSearchDataV1, err error) {
  300. o := orm.NewOrm()
  301. sql := ``
  302. tableName := GetEdbDataTableName(source, subSource)
  303. sql = ` SELECT * FROM %s WHERE 1=1 `
  304. sql = fmt.Sprintf(sql, tableName)
  305. if condition != "" {
  306. sql += condition
  307. }
  308. if order == 1 {
  309. sql += ` ORDER BY data_time ASC `
  310. } else {
  311. sql += ` ORDER BY data_time DESC `
  312. }
  313. _, err = o.Raw(sql, pars).QueryRows(&item)
  314. return
  315. }
  316. // GetEdbDataListAllV1ByTo 通过事务链接获取数据列表
  317. func GetEdbDataListAllV1ByTo(to orm.TxOrmer, condition string, pars []interface{}, source, subSource, order int) (item []*EdbInfoSearchDataV1, err error) {
  318. sql := ``
  319. tableName := GetEdbDataTableName(source, subSource)
  320. sql = ` SELECT * FROM %s WHERE 1=1 `
  321. sql = fmt.Sprintf(sql, tableName)
  322. if condition != "" {
  323. sql += condition
  324. }
  325. if order == 1 {
  326. sql += ` ORDER BY data_time ASC `
  327. } else {
  328. sql += ` ORDER BY data_time DESC `
  329. }
  330. _, err = to.Raw(sql, pars).QueryRows(&item)
  331. return
  332. }
  333. // GetEdbInfoByCondition 获取指标列表
  334. func GetEdbInfoByCondition(condition string, pars []interface{}, order int) (item []*EdbInfo, err error) {
  335. o := orm.NewOrm()
  336. sql := ` SELECT * FROM edb_info WHERE 1=1 `
  337. if condition != "" {
  338. sql += condition
  339. }
  340. if order == 1 {
  341. sql += ` ORDER BY end_date ASC `
  342. } else {
  343. sql += ` ORDER BY edb_info_id ASC `
  344. }
  345. _, err = o.Raw(sql, pars).QueryRows(&item)
  346. return
  347. }
  348. // UnifiedModifyEdbInfoMaxAndMinInfo 统一修改指标的最大最小值
  349. func UnifiedModifyEdbInfoMaxAndMinInfo(edbInfo *EdbInfo) (err error, errMsg string) {
  350. // 修改最大最小值
  351. maxAndMinItem, err := GetEdbInfoMaxAndMinInfo(edbInfo.Source, edbInfo.SubSource, edbInfo.EdbCode)
  352. if err != nil {
  353. if err.Error() == utils.ErrNoRow() {
  354. err = nil
  355. return
  356. }
  357. errMsg = "刷新指标失败!"
  358. err = errors.New("获取指标最大最小值失败,err:" + err.Error())
  359. return
  360. }
  361. if maxAndMinItem != nil {
  362. // ETA1.0.3改-如果指标有手动保存过, 那么就不更新指标的最大最小值
  363. if edbInfo.ManualSave == 1 {
  364. err = ModifyEdbTimeAndLastInfo(edbInfo.EdbInfoId, maxAndMinItem)
  365. if err != nil {
  366. errMsg = "刷新指标失败!"
  367. err = errors.New("修改指标最新日期和数据失败,err:" + err.Error())
  368. return
  369. }
  370. } else {
  371. err = ModifyEdbInfoMaxAndMinInfo(edbInfo.EdbInfoId, maxAndMinItem)
  372. if err != nil {
  373. errMsg = "刷新指标失败!"
  374. err = errors.New("修改指标最大最小值失败,err:" + err.Error())
  375. return
  376. }
  377. }
  378. }
  379. // 修改关联的预测指标
  380. go ModifyPredictEdbInfoMaxAndMinInfoBySourceEdbInfoId(edbInfo.EdbInfoId, maxAndMinItem)
  381. return
  382. }
  383. // UnifiedModifyPredictEdbInfoMaxAndMinInfo 统一修改预测运算指标的最大最小值
  384. func UnifiedModifyPredictEdbInfoMaxAndMinInfo(edbInfo *EdbInfo, latestDateStr string, latestValue float64) (err error, errMsg string) {
  385. // 修改最大最小值
  386. maxAndMinItem, err := GetEdbInfoMaxAndMinInfo(edbInfo.Source, edbInfo.SubSource, edbInfo.EdbCode)
  387. if err != nil {
  388. if err.Error() == utils.ErrNoRow() {
  389. err = nil
  390. return
  391. }
  392. errMsg = "刷新指标失败!"
  393. err = errors.New("获取指标最大最小值失败,err:" + err.Error())
  394. return
  395. }
  396. if maxAndMinItem != nil {
  397. maxAndMinItem.LatestDate = latestDateStr
  398. maxAndMinItem.LatestValue = latestValue
  399. err = ModifyPredictEdbInfoMaxAndMinInfo(edbInfo.EdbInfoId, maxAndMinItem)
  400. if err != nil {
  401. errMsg = "刷新指标失败!"
  402. err = errors.New("修改指标最大最小值失败,err:" + err.Error())
  403. return
  404. }
  405. }
  406. return
  407. }
  408. // GetChartPredictEdbInfoDataListByConfList 获取图表的预测指标的未来数据
  409. func GetChartPredictEdbInfoDataListByConfList(predictEdbConfList []*PredictEdbConfAndData, filtrateStartDateStr, latestDateStr, endDateStr, frequency, dataDateType string, realPredictEdbInfoData []*EdbInfoSearchData) (predictEdbInfoData []*EdbInfoSearchData, minValue, maxValue float64, err error) {
  410. endDate, err := time.ParseInLocation(utils.FormatDate, endDateStr, time.Local)
  411. if err != nil {
  412. return
  413. }
  414. latestDate, err := time.ParseInLocation(utils.FormatDate, latestDateStr, time.Local)
  415. if err != nil {
  416. return
  417. }
  418. // 开始预测数据的时间
  419. startDate := latestDate
  420. // 如果有筛选时间的话
  421. if filtrateStartDateStr != `` {
  422. filtrateStartDate, tmpErr := time.ParseInLocation(utils.FormatDate, filtrateStartDateStr, time.Local)
  423. if tmpErr != nil {
  424. err = tmpErr
  425. return
  426. }
  427. //如果筛选时间晚于实际数据时间,那么就以筛选时间作为获取预测数据的时间
  428. if filtrateStartDate.After(latestDate) {
  429. startDate = filtrateStartDate.AddDate(0, 0, -1)
  430. }
  431. }
  432. //var dateArr []string
  433. // 对应日期的值
  434. existMap := make(map[string]float64)
  435. for _, v := range realPredictEdbInfoData {
  436. //dateArr = append(dateArr, v.DataTime)
  437. existMap[v.DataTime] = v.Value
  438. }
  439. predictEdbInfoData = make([]*EdbInfoSearchData, 0)
  440. //dataValue := lastDataValue
  441. //预测规则,1:最新,2:固定值,3:同比,4:同差,5:环比,6:环差,7:N期移动均值,8:N期段线性外推值
  442. for _, predictEdbConf := range predictEdbConfList {
  443. dataEndTime := endDate
  444. if predictEdbConf.EndDate.Before(dataEndTime) {
  445. dataEndTime = predictEdbConf.EndDate
  446. }
  447. var tmpMinValue, tmpMaxValue float64 // 当前预测结果中的最大/最小值
  448. dayList := getPredictEdbDayList(startDate, dataEndTime, frequency, dataDateType)
  449. if len(dayList) <= 0 { // 如果未来没有日期的话,那么就退出当前循环,进入下一个循环
  450. continue
  451. }
  452. switch predictEdbConf.RuleType {
  453. case 1: //1:最新
  454. var lastDataValue float64 //最新值
  455. tmpAllData := make([]*EdbInfoSearchData, 0)
  456. tmpAllData = append(tmpAllData, realPredictEdbInfoData...)
  457. tmpAllData = append(tmpAllData, predictEdbInfoData...)
  458. lenTmpAllData := len(tmpAllData)
  459. if lenTmpAllData > 0 {
  460. lastDataValue = tmpAllData[lenTmpAllData-1].Value
  461. }
  462. predictEdbInfoData = GetChartPredictEdbInfoDataListByRule1(predictEdbConf.PredictEdbInfoId, lastDataValue, dayList, predictEdbInfoData, existMap)
  463. tmpMaxValue = lastDataValue
  464. tmpMinValue = lastDataValue
  465. case 2: //2:固定值
  466. tmpValDecimal, tmpErr := decimal.NewFromString(predictEdbConf.Value)
  467. if tmpErr != nil {
  468. err = tmpErr
  469. return
  470. }
  471. dataValue, _ := tmpValDecimal.Float64()
  472. predictEdbInfoData = GetChartPredictEdbInfoDataListByRule1(predictEdbConf.PredictEdbInfoId, dataValue, dayList, predictEdbInfoData, existMap)
  473. tmpMaxValue = dataValue
  474. tmpMinValue = dataValue
  475. case 3: //3:同比
  476. tmpValDecimal, tmpErr := decimal.NewFromString(predictEdbConf.Value)
  477. if tmpErr != nil {
  478. err = tmpErr
  479. return
  480. }
  481. tbValue, _ := tmpValDecimal.Float64()
  482. predictEdbInfoData, tmpMinValue, tmpMaxValue = GetChartPredictEdbInfoDataListByRuleTb(predictEdbConf.PredictEdbInfoId, tbValue, dayList, frequency, realPredictEdbInfoData, predictEdbInfoData, existMap)
  483. case 4: //4:同差
  484. tmpValDecimal, tmpErr := decimal.NewFromString(predictEdbConf.Value)
  485. if tmpErr != nil {
  486. err = tmpErr
  487. return
  488. }
  489. tcValue, _ := tmpValDecimal.Float64()
  490. predictEdbInfoData, tmpMinValue, tmpMaxValue = GetChartPredictEdbInfoDataListByRuleTc(predictEdbConf.PredictEdbInfoId, tcValue, dayList, frequency, realPredictEdbInfoData, predictEdbInfoData, existMap)
  491. case 5: //5:环比
  492. tmpValDecimal, tmpErr := decimal.NewFromString(predictEdbConf.Value)
  493. if tmpErr != nil {
  494. err = tmpErr
  495. return
  496. }
  497. hbValue, _ := tmpValDecimal.Float64()
  498. predictEdbInfoData, tmpMinValue, tmpMaxValue = GetChartPredictEdbInfoDataListByRuleHb(predictEdbConf.PredictEdbInfoId, hbValue, dayList, realPredictEdbInfoData, predictEdbInfoData, existMap)
  499. case 6: //6:环差
  500. tmpValDecimal, tmpErr := decimal.NewFromString(predictEdbConf.Value)
  501. if tmpErr != nil {
  502. err = tmpErr
  503. return
  504. }
  505. hcValue, _ := tmpValDecimal.Float64()
  506. predictEdbInfoData, tmpMinValue, tmpMaxValue = GetChartPredictEdbInfoDataListByRuleHc(predictEdbConf.PredictEdbInfoId, hcValue, dayList, realPredictEdbInfoData, predictEdbInfoData, existMap)
  507. case 7: //7:N期移动均值
  508. nValue, tmpErr := strconv.Atoi(predictEdbConf.Value)
  509. if tmpErr != nil {
  510. err = tmpErr
  511. return
  512. }
  513. predictEdbInfoData, tmpMinValue, tmpMaxValue = GetChartPredictEdbInfoDataListByRuleNMoveMeanValue(predictEdbConf.PredictEdbInfoId, nValue, dayList, realPredictEdbInfoData, predictEdbInfoData, existMap)
  514. case 8: //8:N期段线性外推值
  515. nValue, tmpErr := strconv.Atoi(predictEdbConf.Value)
  516. if tmpErr != nil {
  517. err = tmpErr
  518. return
  519. }
  520. predictEdbInfoData, tmpMinValue, tmpMaxValue, err = GetChartPredictEdbInfoDataListByRuleNLinearRegression(predictEdbConf.PredictEdbInfoId, nValue, dayList, realPredictEdbInfoData, predictEdbInfoData, existMap)
  521. if err != nil {
  522. return
  523. }
  524. case 9: //9:动态环差”预测规则;
  525. hcDataMap := make(map[string]float64) //规则计算的环差值map
  526. if predictEdbConf.PredictEdbInfoId > 0 {
  527. tmpPredictEdbRuleDataList, tmpErr := GetPredictEdbRuleDataItemList(predictEdbConf.PredictEdbInfoId, predictEdbConf.ConfigId, startDate.Format(utils.FormatDate), endDate.Format(utils.FormatDate))
  528. if tmpErr != nil {
  529. err = tmpErr
  530. return
  531. }
  532. for _, v := range tmpPredictEdbRuleDataList {
  533. hcDataMap[v.DataTime] = v.Value
  534. }
  535. } else {
  536. if len(predictEdbConf.DataList) <= 0 {
  537. return
  538. }
  539. for _, v := range predictEdbConf.DataList {
  540. currentDate, tmpErr := time.ParseInLocation(utils.FormatDate, v.DataTime, time.Local)
  541. if tmpErr != nil {
  542. continue
  543. }
  544. // 只处理时间段内的数据
  545. if currentDate.Before(startDate) || currentDate.After(endDate) {
  546. continue
  547. }
  548. hcDataMap[v.DataTime] = v.Value
  549. }
  550. }
  551. predictEdbInfoData, tmpMinValue, tmpMaxValue = GetChartPredictEdbInfoDataListByRuleTrendsHC(predictEdbConf.PredictEdbInfoId, dayList, realPredictEdbInfoData, predictEdbInfoData, hcDataMap, existMap)
  552. case 10: //10:根据 给定终值后插值 规则获取预测数据
  553. tmpValDecimal, tmpErr := decimal.NewFromString(predictEdbConf.Value)
  554. if tmpErr != nil {
  555. err = tmpErr
  556. return
  557. }
  558. finalValue, _ := tmpValDecimal.Float64()
  559. predictEdbInfoData, tmpMinValue, tmpMaxValue = GetChartPredictEdbInfoDataListByRuleFinalValueHc(predictEdbConf.PredictEdbInfoId, finalValue, dayList, realPredictEdbInfoData, predictEdbInfoData, existMap)
  560. case 11: //11:根据 季节性 规则获取预测数据
  561. var seasonConf SeasonConf
  562. tmpErr := json.Unmarshal([]byte(predictEdbConf.Value), &seasonConf)
  563. if tmpErr != nil {
  564. err = errors.New("季节性配置信息异常:" + tmpErr.Error())
  565. return
  566. }
  567. calendar := "公历"
  568. if seasonConf.Calendar == "农历" {
  569. calendar = "农历"
  570. }
  571. yearList := make([]int, 0)
  572. //选择方式,1:连续N年;2:指定年份
  573. if seasonConf.YearType == 1 {
  574. if seasonConf.NValue < 1 {
  575. err = errors.New("连续N年不允许小于1")
  576. return
  577. }
  578. currYear := time.Now().Year()
  579. for i := 0; i < seasonConf.NValue; i++ {
  580. yearList = append(yearList, currYear-i-1)
  581. }
  582. } else {
  583. yearList = seasonConf.YearList
  584. }
  585. predictEdbInfoData, tmpMinValue, tmpMaxValue, err = GetChartPredictEdbInfoDataListByRuleSeason(predictEdbConf.PredictEdbInfoId, yearList, calendar, dayList, realPredictEdbInfoData, predictEdbInfoData, existMap)
  586. if err != nil {
  587. return
  588. }
  589. case 12: //12:根据 移动平均同比 规则获取预测数据
  590. var moveAverageConf MoveAverageConf
  591. tmpErr := json.Unmarshal([]byte(predictEdbConf.Value), &moveAverageConf)
  592. if tmpErr != nil {
  593. err = errors.New("季节性配置信息异常:" + tmpErr.Error())
  594. return
  595. }
  596. predictEdbInfoData, tmpMinValue, tmpMaxValue, err = GetChartPredictEdbInfoDataListByRuleMoveAverageTb(predictEdbConf.PredictEdbInfoId, moveAverageConf.NValue, moveAverageConf.Year, dayList, realPredictEdbInfoData, predictEdbInfoData, existMap)
  597. if err != nil {
  598. return
  599. }
  600. case 13: //13:根据 同比增速差值 规则获取预测数据
  601. tmpValDecimal, tmpErr := decimal.NewFromString(predictEdbConf.Value)
  602. if tmpErr != nil {
  603. err = tmpErr
  604. return
  605. }
  606. tbEndValue, _ := tmpValDecimal.Float64()
  607. predictEdbInfoData, tmpMinValue, tmpMaxValue = GetChartPredictEdbInfoDataListByRuleTbzscz(predictEdbConf.PredictEdbInfoId, tbEndValue, dayList, frequency, realPredictEdbInfoData, predictEdbInfoData, existMap)
  608. case 14: //14:根据 一元线性拟合 规则获取预测数据
  609. var ruleConf RuleLineNhConf
  610. err = json.Unmarshal([]byte(predictEdbConf.Value), &ruleConf)
  611. if err != nil {
  612. err = errors.New("一元线性拟合规则配置信息异常:" + err.Error())
  613. return
  614. }
  615. // 规则计算的拟合残差值map
  616. newNhccDataMap := make(map[string]float64)
  617. if predictEdbConf.PredictEdbInfoId > 0 { //已经生成的动态数据
  618. tmpPredictEdbRuleDataList, tmpErr := GetPredictEdbRuleDataItemList(predictEdbConf.PredictEdbInfoId, predictEdbConf.ConfigId, startDate.Format(utils.FormatDate), endDate.Format(utils.FormatDate))
  619. if tmpErr != nil {
  620. err = tmpErr
  621. return
  622. }
  623. for _, v := range tmpPredictEdbRuleDataList {
  624. newNhccDataMap[v.DataTime] = v.Value
  625. }
  626. } else { //未生成的动态数据,需要使用外部传入的数据进行计算
  627. newNhccDataMap, err, _ = getCalculateNhccData(append(realPredictEdbInfoData, predictEdbInfoData...), ruleConf)
  628. }
  629. predictEdbInfoData, tmpMinValue, tmpMaxValue, err = GetChartPredictEdbInfoDataListByRuleLineNh(predictEdbConf.PredictEdbInfoId, dayList, realPredictEdbInfoData, predictEdbInfoData, newNhccDataMap, existMap)
  630. if err != nil {
  631. return
  632. }
  633. case 15: //15:N年均值:过去N年同期均值。过去N年可以连续或者不连续,指标数据均用线性插值补全为日度数据后计算;
  634. predictEdbInfoData, tmpMinValue, tmpMaxValue, err = GetChartPredictEdbInfoDataListByRuleNAnnualAverage(predictEdbConf.PredictEdbInfoId, predictEdbConf.Value, dayList, realPredictEdbInfoData, predictEdbInfoData, existMap)
  635. if err != nil {
  636. return
  637. }
  638. case 16: //16:年度值倒推
  639. predictEdbInfoData, tmpMinValue, tmpMaxValue, err = GetChartPredictEdbInfoDataListByRuleAnnualValueInversion(predictEdbConf.PredictEdbInfoId, predictEdbConf.Value, dayList, frequency, realPredictEdbInfoData, predictEdbInfoData, existMap)
  640. if err != nil {
  641. return
  642. }
  643. }
  644. // 下一个规则的开始日期
  645. {
  646. lenPredictEdbInfoData := len(predictEdbInfoData)
  647. if lenPredictEdbInfoData > 0 {
  648. tmpDataEndTime, _ := time.ParseInLocation(utils.FormatDate, predictEdbInfoData[lenPredictEdbInfoData-1].DataTime, time.Local)
  649. if startDate.Before(tmpDataEndTime) {
  650. startDate = tmpDataEndTime
  651. }
  652. }
  653. }
  654. if tmpMinValue < minValue {
  655. minValue = tmpMinValue
  656. }
  657. if tmpMaxValue < maxValue {
  658. maxValue = tmpMaxValue
  659. }
  660. }
  661. return
  662. }
  663. // GetPredictEdbDayList 获取预测指标日期列表
  664. // GetPredictEdbDayList 获取预测指标日期列表
  665. func getPredictEdbDayList(startDate, endDate time.Time, frequency, dataDateType string) (dayList []time.Time) {
  666. if dataDateType == `` {
  667. dataDateType = `交易日`
  668. }
  669. switch frequency {
  670. case "日度":
  671. for currDate := startDate.AddDate(0, 0, 1); currDate.Before(endDate) || currDate.Equal(endDate); currDate = currDate.AddDate(0, 0, 1) {
  672. // 如果日期类型是交易日的时候,那么需要将周六、日排除
  673. if dataDateType == `交易日` && (currDate.Weekday() == time.Sunday || currDate.Weekday() == time.Saturday) {
  674. continue
  675. }
  676. dayList = append(dayList, currDate)
  677. }
  678. case "周度":
  679. //nextDate := startDate.AddDate(0, 0, 7)
  680. for currDate := startDate.AddDate(0, 0, 7); currDate.Before(endDate) || currDate.Equal(endDate); currDate = currDate.AddDate(0, 0, 7) {
  681. dayList = append(dayList, currDate)
  682. }
  683. case "旬度":
  684. for currDate := startDate.AddDate(0, 0, 1); currDate.Before(endDate) || currDate.Equal(endDate); {
  685. nextDate := currDate.AddDate(0, 0, 1)
  686. //每个月的10号、20号、最后一天,那么就写入
  687. if nextDate.Day() == 11 || nextDate.Day() == 21 || nextDate.Day() == 1 {
  688. dayList = append(dayList, currDate)
  689. }
  690. currDate = nextDate
  691. }
  692. case "月度":
  693. for currDate := startDate; currDate.Before(endDate) || currDate.Equal(endDate); {
  694. currDate = time.Date(currDate.Year(), currDate.Month(), 1, 0, 0, 0, 0, time.Now().Location()).AddDate(0, 1, -1)
  695. if !currDate.After(endDate) && !currDate.Equal(startDate) {
  696. dayList = append(dayList, currDate)
  697. }
  698. currDate = currDate.AddDate(0, 0, 1)
  699. }
  700. case "季度":
  701. for currDate := startDate; currDate.Before(endDate) || currDate.Equal(endDate); {
  702. // 每月的最后一天
  703. currDate = time.Date(currDate.Year(), currDate.Month(), 1, 0, 0, 0, 0, time.Now().Location()).AddDate(0, 1, -1)
  704. if !currDate.After(endDate) && !currDate.Equal(startDate) {
  705. // 季度日期就写入,否则不写入
  706. if currDate.Month() == 3 || currDate.Month() == 6 || currDate.Month() == 9 || currDate.Month() == 12 {
  707. dayList = append(dayList, currDate)
  708. }
  709. }
  710. currDate = currDate.AddDate(0, 0, 1)
  711. }
  712. case "半年度":
  713. for currDate := startDate; currDate.Before(endDate) || currDate.Equal(endDate); {
  714. // 每月的最后一天
  715. currDate = time.Date(currDate.Year(), currDate.Month(), 1, 0, 0, 0, 0, time.Now().Location()).AddDate(0, 1, -1)
  716. if !currDate.After(endDate) && !currDate.Equal(startDate) {
  717. // 半年度日期就写入,否则不写入
  718. if currDate.Month() == 6 || currDate.Month() == 12 {
  719. dayList = append(dayList, currDate)
  720. }
  721. }
  722. currDate = currDate.AddDate(0, 0, 1)
  723. }
  724. case "年度":
  725. for currDate := startDate; currDate.Before(endDate) || currDate.Equal(endDate); {
  726. currDate = time.Date(currDate.Year()+1, 12, 31, 0, 0, 0, 0, time.Now().Location())
  727. if !currDate.After(endDate) && !currDate.Equal(startDate) {
  728. dayList = append(dayList, currDate)
  729. }
  730. }
  731. }
  732. return
  733. }
  734. // GetPredictDataListByPredictEdbInfo 根据预测指标信息获取预测指标的数据,order:1升序,其余值为降序
  735. func GetPredictDataListByPredictEdbInfo(edbInfo *EdbInfo, order int, startDate string) (dataList []*EdbInfoSearchData, sourceEdbInfoItem *EdbInfo, err error, errMsg string) {
  736. // 查找该预测指标配置
  737. predictEdbConfList, err := GetPredictEdbConfAndDataListById(edbInfo.EdbInfoId)
  738. if err != nil && err.Error() != utils.ErrNoRow() {
  739. errMsg = "获取预测指标配置信息失败"
  740. return
  741. }
  742. if len(predictEdbConfList) == 0 {
  743. errMsg = "获取预测指标配置信息失败"
  744. err = errors.New(errMsg)
  745. return
  746. }
  747. predictEdbConf := predictEdbConfList[0]
  748. // 来源指标
  749. sourceEdbInfoItem, err = GetEdbInfoById(predictEdbConf.SourceEdbInfoId)
  750. if err != nil {
  751. if err.Error() == utils.ErrNoRow() {
  752. errMsg = "找不到来源指标信息"
  753. err = errors.New(errMsg)
  754. }
  755. return
  756. }
  757. dataList, err, errMsg = GetPredictDataListByPredictEdbConfList(edbInfo, sourceEdbInfoItem, predictEdbConfList, order, startDate)
  758. return
  759. }
  760. // GetPredictDataListByPredictEdbConfList 根据预测指标信息获取预测指标的数据,order:1升序,其余值为降序
  761. func GetPredictDataListByPredictEdbConfList(edbInfo, sourceEdbInfoItem *EdbInfo, predictEdbConfList []*PredictEdbConfAndData, order int, startDate string) (dataList []*EdbInfoSearchData, err error, errMsg string) {
  762. allDataList := make([]*EdbInfoSearchData, 0)
  763. //获取指标数据(实际已生成)
  764. var condition string
  765. var pars []interface{}
  766. condition += " AND edb_info_id=? "
  767. pars = append(pars, sourceEdbInfoItem.EdbInfoId)
  768. if startDate != "" {
  769. condition += " AND data_time>=? "
  770. pars = append(pars, startDate)
  771. }
  772. tmpDataList, err := GetEdbDataListAll(condition, pars, sourceEdbInfoItem.Source, sourceEdbInfoItem.SubSource, 1)
  773. if err != nil {
  774. return
  775. }
  776. // 如果选择了日期,那么需要筛选所有的数据,用于未来指标的生成
  777. if startDate != `` {
  778. allDataList, err = GetEdbDataListAll(" AND edb_info_id=? ", []interface{}{sourceEdbInfoItem.EdbInfoId}, sourceEdbInfoItem.Source, sourceEdbInfoItem.SubSource, 1)
  779. if err != nil {
  780. return
  781. }
  782. } else {
  783. allDataList = tmpDataList
  784. }
  785. // 获取预测指标未来的数据
  786. predictDataList := make([]*EdbInfoSearchData, 0)
  787. endDateStr := edbInfo.EndDate //预测指标的结束日期
  788. var predictMinValue, predictMaxValue float64
  789. // 如果有配置的预测规则,那么就进行预测
  790. if len(predictEdbConfList) > 0 {
  791. predictDataList, predictMinValue, predictMaxValue, err = GetChartPredictEdbInfoDataListByConfList(predictEdbConfList, startDate, sourceEdbInfoItem.LatestDate, endDateStr, edbInfo.Frequency, edbInfo.DataDateType, allDataList)
  792. if err != nil {
  793. return
  794. }
  795. }
  796. //order:1升序,其余值为降序
  797. if order == 1 {
  798. dataList = append(tmpDataList, predictDataList...)
  799. } else {
  800. // 先倒序预测数据
  801. lenPredictDataList := len(predictDataList)
  802. if lenPredictDataList > 0 {
  803. for k := range predictDataList {
  804. dataList = append(dataList, predictDataList[lenPredictDataList-k-1])
  805. }
  806. }
  807. // 接着倒序实际指标
  808. lenDataList := len(tmpDataList)
  809. for k := range tmpDataList {
  810. dataList = append(dataList, tmpDataList[lenDataList-k-1])
  811. }
  812. }
  813. if len(predictDataList) > 0 {
  814. // 如果最小值 大于 预测值,那么将预测值作为最小值数据返回
  815. if edbInfo.MinValue > predictMinValue {
  816. edbInfo.MinValue = predictMinValue
  817. }
  818. // 如果最大值 小于 预测值,那么将预测值作为最大值数据返回
  819. if edbInfo.MaxValue < predictMaxValue {
  820. edbInfo.MaxValue = predictMaxValue
  821. }
  822. }
  823. return
  824. }
  825. // GetPredictEdbDataListAll 获取该预测指标所有的数据 ,order:1升序,其余值为降序
  826. func GetPredictEdbDataListAll(edbInfo *EdbInfo, order int) (items []*EdbInfoSearchData, err error) {
  827. if edbInfo.Source == utils.DATA_SOURCE_PREDICT { //普通的预测指标是没有入库数据的,直接往配置里面获取
  828. items, _, err, _ = GetPredictDataListByPredictEdbInfo(edbInfo, 1, "")
  829. } else {
  830. var condition string
  831. var pars []interface{}
  832. condition += " AND edb_info_id=? "
  833. pars = append(pars, edbInfo.EdbInfoId)
  834. items, err = GetEdbDataListAll(condition, pars, edbInfo.Source, edbInfo.SubSource, order)
  835. }
  836. return
  837. }
  838. // GetPredictEdbDataListAllByStartDate 根据开始日期获取该预测指标所有的数据 ,order:1升序,其余值为降序
  839. func GetPredictEdbDataListAllByStartDate(edbInfo *EdbInfo, order int, startDate string) (items []*EdbInfoSearchData, err error) {
  840. if edbInfo.Source == utils.DATA_SOURCE_PREDICT { //普通的预测指标是没有入库数据的,直接往配置里面获取
  841. items, _, err, _ = GetPredictDataListByPredictEdbInfo(edbInfo, order, startDate)
  842. } else {
  843. var condition string
  844. var pars []interface{}
  845. condition += " AND edb_info_id=? "
  846. pars = append(pars, edbInfo.EdbInfoId)
  847. if startDate != "" {
  848. condition += " AND data_time>=? "
  849. pars = append(pars, startDate)
  850. }
  851. items, err = GetEdbDataListAll(condition, pars, edbInfo.Source, edbInfo.SubSource, order)
  852. }
  853. return
  854. }
  855. // ModifyPredictEdbInfoMaxAndMinInfo 修改预测指标的最新数据信息
  856. func ModifyPredictEdbInfoMaxAndMinInfo(edbInfoId int, item *EdbInfoMaxAndMinInfo) (err error) {
  857. o := orm.NewOrm()
  858. sql := ` UPDATE edb_info SET start_date=?,end_date=?,min_value=?,max_value=?,is_update=2,latest_date=?,latest_value=?,modify_time=NOW() WHERE edb_info_id=? `
  859. _, err = o.Raw(sql, item.MinDate, item.MaxDate, item.MinValue, item.MaxValue, item.LatestDate, item.LatestValue, edbInfoId).Exec()
  860. return
  861. }
  862. // ModifyCalculateEdbInfo 修改计算指标信息
  863. func ModifyCalculateEdbInfo(edbName, frequency, unit, calculateFormula string, classifyId, edbInfoId int) (err error) {
  864. o := orm.NewOrm()
  865. sql := ` UPDATE edb_info
  866. SET
  867. edb_name =?,
  868. edb_name_source =?,
  869. frequency = ?,
  870. unit = ?,
  871. classify_id = ?,
  872. calculate_formula=?,
  873. modify_time = NOW()
  874. WHERE edb_info_id = ?`
  875. _, err = o.Raw(sql, edbName, edbName, frequency, unit, classifyId, calculateFormula, edbInfoId).Exec()
  876. return
  877. }
  878. func GetEdbInfoItemByCondition(condition string, pars []interface{}) (item *EdbInfoList, err error) {
  879. o := orm.NewOrm()
  880. sql := ` SELECT * FROM edb_info WHERE 1=1 `
  881. if condition != "" {
  882. sql += condition
  883. }
  884. err = o.Raw(sql, pars).QueryRow(&item)
  885. return
  886. }
  887. func GetEdbInfoItemListByCondition() (items []*EdbInfoList, err error) {
  888. o := orm.NewOrm()
  889. sql := ` SELECT * FROM edb_info WHERE 1=1 `
  890. _, err = o.Raw(sql).QueryRows(&items)
  891. return
  892. }
  893. // EdbInfoDetailReq 指标详情请求
  894. type EdbInfoDetailReq struct {
  895. UniqueCode string `description:"指标唯一编码"`
  896. EdbCode string `description:"指标编码"`
  897. }
  898. // EdbInfoDetailResp 指标详情返回
  899. type EdbInfoDetailResp struct {
  900. SourceName string `description:"来源名称"`
  901. EdbCode string `description:"指标编码"`
  902. EdbName string `description:"指标名称"`
  903. EdbNameSource string `description:"指标名称来源"`
  904. Frequency string `description:"频率"`
  905. Unit string `description:"单位"`
  906. StartDate string `description:"起始日期"`
  907. EndDate string `description:"终止日期"`
  908. UniqueCode string `description:"指标唯一编码"`
  909. CreateTime time.Time
  910. ModifyTime time.Time
  911. MinValue float64 `description:"指标最小值"`
  912. MaxValue float64 `description:"指标最大值"`
  913. EdbNameEn string `description:"英文指标名称"`
  914. UnitEn string `description:"英文单位"`
  915. LatestDate string `description:"数据最新日期"`
  916. LatestValue float64 `description:"数据最新值"`
  917. ChartImage string `description:"图表图片"`
  918. DataList []*EdbInfoSearchData
  919. }
  920. type AddEdbInfoParams struct {
  921. Source int `description:"来源id"`
  922. EdbCode string `description:"指标编码"`
  923. EdbName string `description:"指标名称"`
  924. Frequency string `description:"频率"`
  925. Unit string `description:"单位"`
  926. ClassifyId int `description:"分类id"`
  927. AdminId int `description:"创建人ID"`
  928. AdminName string `description:"创建人姓名"`
  929. }
  930. // EdbInfoAdd 添加指标到指标库
  931. func EdbInfoAdd(req *AddEdbInfoParams, serverUrl string, sysUserId int, sysUserRealName string) (edbInfo *EdbInfo, err error) {
  932. source := req.Source
  933. //判断指标名称是否存在
  934. var condition string
  935. var pars []interface{}
  936. //condition += " AND source=? "
  937. //pars = append(pars, source)
  938. condition += " AND edb_code=? "
  939. pars = append(pars, req.EdbCode)
  940. items, err := GetEdbInfoByCondition(condition, pars, 0)
  941. if err != nil {
  942. err = errors.New("判断指标是否存在失败,Err:" + err.Error())
  943. return
  944. }
  945. if len(items) > 0 {
  946. sql := ` UPDATE edb_info SET classify_id=?,modify_time=NOW() WHERE edb_code=? `
  947. o := orm.NewOrm()
  948. _, err = o.Raw(sql, req.ClassifyId, req.EdbCode).Exec()
  949. if err != nil {
  950. err = errors.New("更新指标分类信息失败,Err:" + err.Error())
  951. return
  952. }
  953. edbInfo = items[0]
  954. return
  955. }
  956. edbInfo = new(EdbInfo)
  957. edbInfo.Source = source
  958. sourceNameMap := map[int]string{
  959. utils.DATA_SOURCE_THS: "同花顺",
  960. utils.DATA_SOURCE_WIND: "wind",
  961. utils.DATA_SOURCE_PB: "彭博",
  962. utils.DATA_SOURCE_PB_FINANCE: "彭博财务",
  963. utils.DATA_SOURCE_MANUAL: "手工数据",
  964. utils.DATA_SOURCE_LZ: "隆众",
  965. utils.DATA_SOURCE_YS: "SMM",
  966. utils.DATA_SOURCE_GL: "钢联",
  967. utils.DATA_SOURCE_ZZ: "郑商所",
  968. utils.DATA_SOURCE_DL: "大商所",
  969. utils.DATA_SOURCE_SH: "上期所",
  970. utils.DATA_SOURCE_CFFEX: "中金所",
  971. utils.DATA_SOURCE_SHFE: "上期能源",
  972. utils.DATA_SOURCE_GIE: "欧洲天然气",
  973. utils.DATA_SOURCE_LT: "路透",
  974. utils.DATA_SOURCE_COAL: "中国煤炭网",
  975. utils.DATA_SOURCE_GOOGLE_TRAVEL: "our world in data",
  976. utils.DATA_SOURCE_MYSTEEL_CHEMICAL: "钢联化工",
  977. utils.DATA_SOURCE_EIA_STEO: "EIA STEO报告",
  978. utils.DATA_SOURCE_COM_TRADE: "UN",
  979. utils.DATA_SOURCE_SCI: "SCI",
  980. utils.DATA_SOURCE_BAIINFO: "BAIINFO",
  981. utils.DATA_SOURCE_STOCK_PLANT: "存量装置",
  982. utils.DATA_SOURCE_NATIONAL_STATISTICS: "国家统计局",
  983. utils.DATA_SOURCE_FUBAO: "富宝数据",
  984. }
  985. sourceName, ok := sourceNameMap[source]
  986. if !ok {
  987. edbSource := EdbSourceIdMap[source]
  988. if edbSource != nil {
  989. sourceName = edbSource.SourceName
  990. }
  991. if sourceName == "" {
  992. err = errors.New("指标来源异常")
  993. return
  994. }
  995. }
  996. edbInfo.SourceName = sourceName
  997. edbType := 1 //基础指标
  998. if source == utils.DATA_SOURCE_STOCK_PLANT {
  999. edbType = 2 //计算指标
  1000. }
  1001. // 钢联化工需要校验下信息是否完整
  1002. if source == utils.DATA_SOURCE_MYSTEEL_CHEMICAL {
  1003. indexObj := new(BaseFromMysteelChemicalIndex)
  1004. tmpItem, tmpErr := indexObj.GetIndexItem(req.EdbCode)
  1005. if tmpErr != nil {
  1006. err = tmpErr
  1007. return
  1008. }
  1009. if tmpItem == nil || tmpItem.IndexName == `` {
  1010. err = errors.New("指标信息不全")
  1011. return
  1012. }
  1013. }
  1014. edbInfo.EdbCode = req.EdbCode
  1015. edbInfo.EdbName = req.EdbName
  1016. edbInfo.EdbNameSource = req.EdbName
  1017. edbInfo.Frequency = req.Frequency
  1018. edbInfo.Unit = req.Unit
  1019. edbInfo.ClassifyId = req.ClassifyId
  1020. edbInfo.SysUserId = sysUserId
  1021. edbInfo.SysUserRealName = sysUserRealName
  1022. edbInfo.StartDate = "1990-01-01"
  1023. edbInfo.CreateTime = time.Now()
  1024. edbInfo.ModifyTime = time.Now()
  1025. edbInfo.ServerUrl = serverUrl
  1026. timestamp := strconv.FormatInt(time.Now().UnixNano(), 10)
  1027. edbInfo.UniqueCode = utils.MD5(utils.DATA_PREFIX + "_" + req.EdbCode + timestamp)
  1028. itemVal, err := GetEdbInfoMaxAndMinInfo(source, utils.DATA_SUB_SOURCE_EDB, req.EdbCode)
  1029. if itemVal != nil && err == nil {
  1030. edbInfo.MaxValue = itemVal.MaxValue
  1031. edbInfo.MinValue = itemVal.MinValue
  1032. }
  1033. edbInfo.EdbType = edbType
  1034. edbInfoId, err := AddEdbInfo(edbInfo)
  1035. if err != nil {
  1036. err = errors.New("保存失败,Err:" + err.Error())
  1037. return
  1038. }
  1039. edbInfo.EdbInfoId = int(edbInfoId)
  1040. return
  1041. }
  1042. // EdbInfoRefreshCheckReq 指标数据更新情况查询
  1043. type EdbInfoRefreshCheckReq struct {
  1044. Source int `description:"来源id"`
  1045. LatestDate string `description:"数据最新日期"`
  1046. FrequencyBatch string `description:"更新频度"`
  1047. }
  1048. // EdbInfoRefreshCheckResp 指标数据更新情况查询
  1049. type EdbInfoRefreshCheckResp struct {
  1050. UpdateNum int `description:"已更新指标数"`
  1051. UnUpdateNum int `description:"未更新指标数"`
  1052. }
  1053. // ResetEdbClassifyReq 指标重新分类请求体
  1054. type ResetEdbClassifyReq struct {
  1055. IndexCode string `description:"指标编码"`
  1056. ClassifyFirst string `description:"一级分类名称"`
  1057. ClassifySecond string `description:"二级分类名称"`
  1058. ClassifyThird string `description:"三级分类名称"`
  1059. }