edb_info.go 43 KB

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