edb_info.go 29 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788
  1. package models
  2. import (
  3. "errors"
  4. "fmt"
  5. "github.com/beego/beego/v2/client/orm"
  6. "github.com/shopspring/decimal"
  7. "hongze/hongze_edb_lib/utils"
  8. "strconv"
  9. "time"
  10. )
  11. type EdbInfo struct {
  12. EdbInfoId int `orm:"column(edb_info_id);pk"`
  13. SourceName string `description:"来源名称"`
  14. Source int `description:"来源id"`
  15. EdbCode string `description:"指标编码"`
  16. EdbName string `description:"指标名称"`
  17. EdbNameSource string `description:"指标名称来源"`
  18. Frequency string `description:"频率"`
  19. Unit string `description:"单位"`
  20. StartDate string `description:"起始日期"`
  21. EndDate string `description:"终止日期"`
  22. ClassifyId int `description:"分类id"`
  23. SysUserId int
  24. SysUserRealName string
  25. UniqueCode string `description:"指标唯一编码"`
  26. CreateTime time.Time
  27. ModifyTime time.Time
  28. MinValue float64 `description:"指标最小值"`
  29. MaxValue float64 `description:"指标最大值"`
  30. CalculateFormula string `description:"计算公式"`
  31. EdbType int `description:"指标类型:1:基础指标,2:计算指标"`
  32. Sort int `description:"排序字段"`
  33. MoveType int `description:"移动方式:1:领先(默认),2:滞后"`
  34. MoveFrequency string `description:"移动频度"`
  35. NoUpdate int8 `description:"是否停止更新,0:继续更新;1:停止更新"`
  36. ServerUrl string `description:"服务器地址"`
  37. EdbInfoType int `description:"指标类型,0:普通指标,1:预测指标"`
  38. EdbNameEn string `description:"英文指标名称"`
  39. UnitEn string `description:"英文单位"`
  40. LatestDate string `description:"数据最新日期"`
  41. LatestValue float64 `description:"数据最新值"`
  42. ChartImage string `description:"图表图片"`
  43. Calendar string `description:"公历/农历"`
  44. }
  45. // AddEdbInfo 添加指标
  46. func AddEdbInfo(item *EdbInfo) (lastId int64, err error) {
  47. o := orm.NewOrm()
  48. lastId, err = o.Insert(item)
  49. return
  50. }
  51. // EdbInfoList 指标数据列表
  52. type EdbInfoList struct {
  53. EdbInfoId int `orm:"column(edb_info_id);pk"`
  54. EdbInfoType int `description:"指标类型,0:普通指标,1:预测指标"`
  55. SourceName string `description:"来源名称"`
  56. Source int `description:"来源id"`
  57. EdbCode string `description:"指标编码"`
  58. EdbNameEn string `description:"英文指标名称"`
  59. EdbName string `description:"指标名称"`
  60. Frequency string `description:"频率"`
  61. FrequencyEn string `description:"英文频率"`
  62. Unit string `description:"单位"`
  63. UnitEn string `description:"英文单位"`
  64. StartDate string `description:"起始日期"`
  65. EndDate string `description:"终止日期"`
  66. LatestDate string `description:"数据最新日期"`
  67. LatestValue float64 `description:"数据最新值"`
  68. ClassifyId int `description:"分类id"`
  69. UniqueCode string `description:"指标唯一编码"`
  70. SysUserId int `description:"创建人id"`
  71. SysUserRealName string `description:"创建人姓名"`
  72. ModifyTime string `description:"最新修改时间"`
  73. CreateTime string `description:"创建时间"`
  74. EdbNameAlias string `json:"-" description:"指标名称,别名"`
  75. EdbType int `description:"指标类型:1:基础指标,2:计算指标"`
  76. ChartImage string `description:"图表图片"`
  77. RuleType int `description:"预测规则,1:最新,2:固定值"`
  78. FixedValue float64 `description:"固定值"`
  79. DataList []*EdbData `description:"实际指标数据"`
  80. PredictDataList []*EdbData `description:"预测指标数据"`
  81. Button EdbClassifyItemsButton `description:"操作权限"`
  82. IsEnEdb bool `description:"是否展示英文标识"`
  83. }
  84. // EdbClassifyItemsButton 操作按钮
  85. type EdbClassifyItemsButton struct {
  86. AddButton bool `description:"是否可添加"`
  87. OpButton bool `description:"是否可编辑"`
  88. DeleteButton bool `description:"是否可删除"`
  89. MoveButton bool `description:"是否可移动"`
  90. ShowEdbRelation bool `description:"是否展示关联指标"`
  91. ShowChartRelation bool `description:"是否展示关联图表"`
  92. }
  93. // GetEdbInfoByName 根据指标名称获取所有的指标数据列表
  94. func GetEdbInfoByName(edbName string) (items []*EdbInfoList, err error) {
  95. o := orm.NewOrm()
  96. sql := ` SELECT * FROM edb_info WHERE edb_name=? `
  97. _, err = o.Raw(sql, edbName).QueryRows(&items)
  98. return
  99. }
  100. // ModifyEdbInfoNameSource 根据来源修改指标名称
  101. func ModifyEdbInfoNameSource(edbNameSource string, edbInfoId int) (err error) {
  102. o := orm.NewOrm()
  103. sql := ` UPDATE edb_info SET edb_name_source=? WHERE edb_info_id = ? `
  104. _, err = o.Raw(sql, edbNameSource, edbInfoId).Exec()
  105. return
  106. }
  107. // GetEdbInfoById 根据指标id获取指标信息
  108. func GetEdbInfoById(edbInfoId int) (item *EdbInfo, err error) {
  109. o := orm.NewOrm()
  110. sql := ` SELECT * FROM edb_info WHERE edb_info_id=? `
  111. err = o.Raw(sql, edbInfoId).QueryRow(&item)
  112. return
  113. }
  114. // GetEdbInfoByIdList 根据指标id列表获取指标信息
  115. func GetEdbInfoByIdList(edbInfoIdList []int) (items []*EdbInfo, err error) {
  116. num := len(edbInfoIdList)
  117. if num <= 0 {
  118. return
  119. }
  120. o := orm.NewOrm()
  121. sql := ` SELECT * FROM edb_info WHERE edb_info_id in (` + utils.GetOrmInReplace(num) + `) `
  122. _, err = o.Raw(sql, edbInfoIdList).QueryRows(&items)
  123. return
  124. }
  125. // Update 更新EdbInfo信息
  126. func (edbInfo *EdbInfo) Update(cols []string) (err error) {
  127. o := orm.NewOrm()
  128. _, err = o.Update(edbInfo, cols...)
  129. return
  130. }
  131. // EdbInfoSearchData
  132. type EdbInfoSearchData struct {
  133. EdbDataId int `description:"数据ID"`
  134. DataTime string `description:"数据日期"`
  135. Value float64 `description:"数据"`
  136. }
  137. // GetEdbDataListAll 获取指标数据列表 order:1升序,其余值为降序
  138. func GetEdbDataListAll(condition string, pars []interface{}, source, order int) (item []*EdbInfoSearchData, err error) {
  139. o := orm.NewOrm()
  140. sql := ``
  141. tableName := GetEdbDataTableName(source)
  142. sql = ` SELECT * FROM %s WHERE 1=1 `
  143. sql = fmt.Sprintf(sql, tableName)
  144. if condition != "" {
  145. sql += condition
  146. }
  147. if order == 1 {
  148. sql += ` ORDER BY data_time ASC `
  149. } else {
  150. sql += ` ORDER BY data_time DESC `
  151. }
  152. _, err = o.Raw(sql, pars).QueryRows(&item)
  153. return
  154. }
  155. // GetEdbDataListAllByTo 根据事务链接获取指标数据列表 order:1升序,其余值为降序
  156. func GetEdbDataListAllByTo(to orm.TxOrmer, condition string, pars []interface{}, source, order int) (item []*EdbInfoSearchData, err error) {
  157. sql := ``
  158. tableName := GetEdbDataTableName(source)
  159. sql = ` SELECT * FROM %s WHERE 1=1 `
  160. sql = fmt.Sprintf(sql, tableName)
  161. if condition != "" {
  162. sql += condition
  163. }
  164. if order == 1 {
  165. sql += ` ORDER BY data_time ASC `
  166. } else {
  167. sql += ` ORDER BY data_time DESC `
  168. }
  169. _, err = to.Raw(sql, pars).QueryRows(&item)
  170. return
  171. }
  172. // EdbInfoMaxAndMinInfo 指标最新数据记录结构体
  173. type EdbInfoMaxAndMinInfo struct {
  174. MinDate string `description:"最小日期"`
  175. MaxDate string `description:"最大日期"`
  176. MinValue float64 `description:"最小值"`
  177. MaxValue float64 `description:"最大值"`
  178. LatestValue float64 `description:"最新值"`
  179. LatestDate string `description:"实际数据最新日期"`
  180. }
  181. // GetEdbInfoMaxAndMinInfo 获取指标的最新数据记录信息
  182. func GetEdbInfoMaxAndMinInfo(source int, edbCode string) (item *EdbInfoMaxAndMinInfo, err error) {
  183. o := orm.NewOrm()
  184. sql := ``
  185. tableName := GetEdbDataTableName(source)
  186. 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=? `
  187. sql = fmt.Sprintf(sql, tableName)
  188. err = o.Raw(sql, edbCode).QueryRow(&item)
  189. var latest_value float64
  190. sql = ` SELECT value AS latest_value FROM %s WHERE edb_code=? ORDER BY data_time DESC LIMIT 1 `
  191. sql = fmt.Sprintf(sql, tableName)
  192. err = o.Raw(sql, edbCode).QueryRow(&latest_value)
  193. item.LatestValue = latest_value
  194. return
  195. }
  196. // ModifyEdbInfoMaxAndMinInfo 修改指标的最新数据信息
  197. func ModifyEdbInfoMaxAndMinInfo(edbInfoId int, item *EdbInfoMaxAndMinInfo) (err error) {
  198. o := orm.NewOrm()
  199. 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=? `
  200. _, err = o.Raw(sql, item.MinDate, item.MaxDate, item.MinValue, item.MaxValue, item.MaxDate, item.LatestValue, edbInfoId).Exec()
  201. return
  202. }
  203. // GetEdbDataCount 获取edb指标数据的数量; order:1升序,其余值为降序
  204. func GetEdbDataCount(condition string, pars []interface{}, source int) (count int, err error) {
  205. o := orm.NewOrm()
  206. sql := ``
  207. tableName := GetEdbDataTableName(source)
  208. sql = ` SELECT COUNT(1) AS count FROM %s WHERE 1=1 `
  209. sql = fmt.Sprintf(sql, tableName)
  210. if condition != "" {
  211. sql += condition
  212. }
  213. err = o.Raw(sql, pars).QueryRow(&count)
  214. return
  215. }
  216. // GetLastEdbData 获取最近的一条指标数据
  217. func GetLastEdbData(condition string, pars []interface{}, source int) (item *EdbInfoSearchData, err error) {
  218. o := orm.NewOrm()
  219. sql := ``
  220. tableName := GetEdbDataTableName(source)
  221. sql = ` SELECT * FROM %s WHERE 1=1 `
  222. sql = fmt.Sprintf(sql, tableName)
  223. if condition != "" {
  224. sql += condition
  225. }
  226. sql += ` ORDER BY data_time DESC `
  227. err = o.Raw(sql, pars).QueryRow(&item)
  228. return
  229. }
  230. // GetEdbInfoByEdbCode 根据指标code获取指标信息
  231. func GetEdbInfoByEdbCode(source int, edbCode string) (item *EdbInfo, err error) {
  232. o := orm.NewOrm()
  233. sql := ` SELECT * FROM edb_info WHERE source=? AND edb_code=? `
  234. err = o.Raw(sql, source, edbCode).QueryRow(&item)
  235. return
  236. }
  237. // GetEdbInfoOnlyByEdbCode 仅根据指标code获取指标信息
  238. func GetEdbInfoOnlyByEdbCode(edbCode string) (item *EdbInfo, err error) {
  239. o := orm.NewOrm()
  240. sql := ` SELECT * FROM edb_info WHERE edb_code=? `
  241. err = o.Raw(sql, edbCode).QueryRow(&item)
  242. return
  243. }
  244. // GetEdbInfoCalculateListByCondition 获取指标关系列表
  245. func GetEdbInfoCalculateListByCondition(condition string, pars []interface{}) (items []*EdbInfoCalculateMapping, err error) {
  246. o := orm.NewOrm()
  247. sql := ` SELECT * FROM edb_info_calculate_mapping WHERE 1=1 `
  248. if condition != "" {
  249. sql += condition
  250. }
  251. _, err = o.Raw(sql, pars).QueryRows(&items)
  252. return
  253. }
  254. // GetEdbInfoCalculateCountByCondition 获取关联指标数量
  255. func GetEdbInfoCalculateCountByCondition(condition string, pars []interface{}) (count int, err error) {
  256. o := orm.NewOrm()
  257. sql := ` SELECT COUNT(1) AS count FROM edb_info_calculate_mapping WHERE 1=1 `
  258. if condition != "" {
  259. sql += condition
  260. }
  261. err = o.Raw(sql, pars).QueryRow(&count)
  262. return
  263. }
  264. //优化版本-处理数据精度问题
  265. type EdbInfoSearchDataV1 struct {
  266. EdbDataId int `description:"数据ID"`
  267. DataTime string `description:"数据日期"`
  268. Value string `description:"数据"`
  269. }
  270. //优化版本-处理数据精度问题
  271. func GetEdbDataListAllV1(condition string, pars []interface{}, source, order int) (item []*EdbInfoSearchDataV1, err error) {
  272. o := orm.NewOrm()
  273. sql := ``
  274. tableName := GetEdbDataTableName(source)
  275. sql = ` SELECT * FROM %s WHERE 1=1 `
  276. sql = fmt.Sprintf(sql, tableName)
  277. if condition != "" {
  278. sql += condition
  279. }
  280. if order == 1 {
  281. sql += ` ORDER BY data_time ASC `
  282. } else {
  283. sql += ` ORDER BY data_time DESC `
  284. }
  285. _, err = o.Raw(sql, pars).QueryRows(&item)
  286. return
  287. }
  288. // GetEdbDataListAllV1ByTo 通过事务链接获取数据列表
  289. func GetEdbDataListAllV1ByTo(to orm.TxOrmer, condition string, pars []interface{}, source, order int) (item []*EdbInfoSearchDataV1, err error) {
  290. sql := ``
  291. tableName := GetEdbDataTableName(source)
  292. sql = ` SELECT * FROM %s WHERE 1=1 `
  293. sql = fmt.Sprintf(sql, tableName)
  294. if condition != "" {
  295. sql += condition
  296. }
  297. if order == 1 {
  298. sql += ` ORDER BY data_time ASC `
  299. } else {
  300. sql += ` ORDER BY data_time DESC `
  301. }
  302. _, err = to.Raw(sql, pars).QueryRows(&item)
  303. return
  304. }
  305. // GetEdbInfoByCondition 获取指标列表
  306. func GetEdbInfoByCondition(condition string, pars []interface{}, order int) (item []*EdbInfo, err error) {
  307. o := orm.NewOrm()
  308. sql := ` SELECT * FROM edb_info WHERE 1=1 `
  309. if condition != "" {
  310. sql += condition
  311. }
  312. if order == 1 {
  313. sql += ` ORDER BY end_date ASC `
  314. } else {
  315. sql += ` ORDER BY edb_info_id ASC `
  316. }
  317. _, err = o.Raw(sql, pars).QueryRows(&item)
  318. return
  319. }
  320. // UnifiedModifyEdbInfoMaxAndMinInfo 统一修改指标的最大最小值
  321. func UnifiedModifyEdbInfoMaxAndMinInfo(edbInfo *EdbInfo) (err error, errMsg string) {
  322. // 修改最大最小值
  323. maxAndMinItem, err := GetEdbInfoMaxAndMinInfo(edbInfo.Source, edbInfo.EdbCode)
  324. if err != nil {
  325. if err.Error() == utils.ErrNoRow() {
  326. err = nil
  327. return
  328. }
  329. errMsg = "刷新指标失败!"
  330. err = errors.New("获取指标最大最小值失败,err:" + err.Error())
  331. return
  332. }
  333. if maxAndMinItem != nil {
  334. err = ModifyEdbInfoMaxAndMinInfo(edbInfo.EdbInfoId, maxAndMinItem)
  335. if err != nil {
  336. errMsg = "刷新指标失败!"
  337. err = errors.New("修改指标最大最小值失败,err:" + err.Error())
  338. return
  339. }
  340. }
  341. // 修改关联的预测指标
  342. go ModifyPredictEdbInfoMaxAndMinInfoBySourceEdbInfoId(edbInfo.EdbInfoId, maxAndMinItem)
  343. return
  344. }
  345. // UnifiedModifyPredictEdbInfoMaxAndMinInfo 统一修改预测运算指标的最大最小值
  346. func UnifiedModifyPredictEdbInfoMaxAndMinInfo(edbInfo *EdbInfo, latestDateStr string, latestValue float64) (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. maxAndMinItem.LatestDate = latestDateStr
  360. maxAndMinItem.LatestValue = latestValue
  361. err = ModifyPredictEdbInfoMaxAndMinInfo(edbInfo.EdbInfoId, maxAndMinItem)
  362. if err != nil {
  363. errMsg = "刷新指标失败!"
  364. err = errors.New("修改指标最大最小值失败,err:" + err.Error())
  365. return
  366. }
  367. }
  368. return
  369. }
  370. // GetChartPredictEdbInfoDataList 获取图表的预测指标的未来数据
  371. func GetChartPredictEdbInfoDataList(predictEdbConf PredictEdbConf, latestDateStr string, lastDataValue float64, endDateStr, frequency string, order int) (predictEdbInfoData []*EdbInfoSearchData, err error) {
  372. endDate, err := time.ParseInLocation(utils.FormatDate, endDateStr, time.Local)
  373. if err != nil {
  374. return
  375. }
  376. latestDate, err := time.ParseInLocation(utils.FormatDate, latestDateStr, time.Local)
  377. if err != nil {
  378. return
  379. }
  380. // 开始预测数据的时间
  381. startDate := latestDate
  382. dataValue := lastDataValue
  383. if predictEdbConf.RuleType == 2 {
  384. dataValue = predictEdbConf.FixedValue
  385. }
  386. //获取后面的预测数据
  387. dayList := getPredictEdbDayList(startDate, endDate, frequency)
  388. predictEdbInfoData = make([]*EdbInfoSearchData, 0)
  389. // order:1升序,其余值为降序
  390. if order == 1 {
  391. for k, v := range dayList {
  392. predictEdbInfoData = append(predictEdbInfoData, &EdbInfoSearchData{
  393. EdbDataId: predictEdbConf.PredictEdbInfoId + 10000000000 + k,
  394. DataTime: v.Format(utils.FormatDate),
  395. Value: dataValue,
  396. })
  397. }
  398. } else {
  399. lenDayList := len(dayList)
  400. if lenDayList > 0 {
  401. for i := lenDayList - 1; i >= 0; i-- {
  402. v := dayList[i]
  403. predictEdbInfoData = append(predictEdbInfoData, &EdbInfoSearchData{
  404. EdbDataId: predictEdbConf.PredictEdbInfoId + 10000000000 + i,
  405. DataTime: v.Format(utils.FormatDate),
  406. Value: dataValue,
  407. })
  408. }
  409. }
  410. }
  411. return
  412. }
  413. // GetChartPredictEdbInfoDataListByConfList 获取图表的预测指标的未来数据
  414. func GetChartPredictEdbInfoDataListByConfList(predictEdbConfList []*PredictEdbConf, filtrateStartDateStr, latestDateStr, endDateStr, frequency string, realPredictEdbInfoData []*EdbInfoSearchData) (predictEdbInfoData []*EdbInfoSearchData, minValue, maxValue float64, err error) {
  415. endDate, err := time.ParseInLocation(utils.FormatDate, endDateStr, time.Local)
  416. if err != nil {
  417. return
  418. }
  419. latestDate, err := time.ParseInLocation(utils.FormatDate, latestDateStr, time.Local)
  420. if err != nil {
  421. return
  422. }
  423. // 开始预测数据的时间
  424. startDate := latestDate
  425. // 如果有筛选时间的话
  426. if filtrateStartDateStr != `` {
  427. filtrateStartDate, tmpErr := time.ParseInLocation(utils.FormatDate, filtrateStartDateStr, time.Local)
  428. if tmpErr != nil {
  429. err = tmpErr
  430. return
  431. }
  432. //如果筛选时间晚于实际数据时间,那么就以筛选时间作为获取预测数据的时间
  433. if filtrateStartDate.After(latestDate) {
  434. startDate = filtrateStartDate.AddDate(0, 0, -1)
  435. }
  436. }
  437. //var dateArr []string
  438. // 对应日期的值
  439. existMap := make(map[string]float64)
  440. for _, v := range realPredictEdbInfoData {
  441. //dateArr = append(dateArr, v.DataTime)
  442. existMap[v.DataTime] = v.Value
  443. }
  444. predictEdbInfoData = make([]*EdbInfoSearchData, 0)
  445. //dataValue := lastDataValue
  446. //预测规则,1:最新,2:固定值,3:同比,4:同差,5:环比,6:环差,7:N期移动均值,8:N期段线性外推值
  447. for _, predictEdbConf := range predictEdbConfList {
  448. dataEndTime := endDate
  449. if predictEdbConf.EndDate.Before(dataEndTime) {
  450. dataEndTime = predictEdbConf.EndDate
  451. }
  452. var tmpMinValue, tmpMaxValue float64 // 当前预测结果中的最大/最小值
  453. switch predictEdbConf.RuleType {
  454. case 1: //1:最新
  455. var lastDataValue float64 //最新值
  456. tmpAllData := make([]*EdbInfoSearchData, 0)
  457. tmpAllData = append(tmpAllData, realPredictEdbInfoData...)
  458. tmpAllData = append(tmpAllData, predictEdbInfoData...)
  459. lenTmpAllData := len(tmpAllData)
  460. if lenTmpAllData > 0 {
  461. lastDataValue = tmpAllData[lenTmpAllData-1].Value
  462. }
  463. predictEdbInfoData = GetChartPredictEdbInfoDataListByRule1(predictEdbConf.PredictEdbInfoId, lastDataValue, startDate, dataEndTime, frequency, predictEdbInfoData, existMap)
  464. tmpMaxValue = lastDataValue
  465. tmpMinValue = lastDataValue
  466. case 2: //2:固定值
  467. tmpValDecimal, tmpErr := decimal.NewFromString(predictEdbConf.Value)
  468. if tmpErr != nil {
  469. err = tmpErr
  470. return
  471. }
  472. dataValue, _ := tmpValDecimal.Float64()
  473. predictEdbInfoData = GetChartPredictEdbInfoDataListByRule1(predictEdbConf.PredictEdbInfoId, dataValue, startDate, dataEndTime, frequency, predictEdbInfoData, existMap)
  474. tmpMaxValue = dataValue
  475. tmpMinValue = dataValue
  476. case 3: //3:同比
  477. tmpValDecimal, tmpErr := decimal.NewFromString(predictEdbConf.Value)
  478. if tmpErr != nil {
  479. err = tmpErr
  480. return
  481. }
  482. tbValue, _ := tmpValDecimal.Float64()
  483. predictEdbInfoData, tmpMinValue, tmpMaxValue = GetChartPredictEdbInfoDataListByRuleTb(predictEdbConf.PredictEdbInfoId, tbValue, startDate, dataEndTime, frequency, realPredictEdbInfoData, predictEdbInfoData, existMap)
  484. case 4: //4:同差
  485. tmpValDecimal, tmpErr := decimal.NewFromString(predictEdbConf.Value)
  486. if tmpErr != nil {
  487. err = tmpErr
  488. return
  489. }
  490. tcValue, _ := tmpValDecimal.Float64()
  491. predictEdbInfoData, tmpMinValue, tmpMaxValue = GetChartPredictEdbInfoDataListByRuleTc(predictEdbConf.PredictEdbInfoId, tcValue, startDate, dataEndTime, frequency, realPredictEdbInfoData, predictEdbInfoData, existMap)
  492. case 5: //5:环比
  493. tmpValDecimal, tmpErr := decimal.NewFromString(predictEdbConf.Value)
  494. if tmpErr != nil {
  495. err = tmpErr
  496. return
  497. }
  498. hbValue, _ := tmpValDecimal.Float64()
  499. predictEdbInfoData, tmpMinValue, tmpMaxValue = GetChartPredictEdbInfoDataListByRuleHb(predictEdbConf.PredictEdbInfoId, hbValue, startDate, dataEndTime, frequency, realPredictEdbInfoData, predictEdbInfoData, existMap)
  500. case 6: //6:环差
  501. tmpValDecimal, tmpErr := decimal.NewFromString(predictEdbConf.Value)
  502. if tmpErr != nil {
  503. err = tmpErr
  504. return
  505. }
  506. hcValue, _ := tmpValDecimal.Float64()
  507. predictEdbInfoData, tmpMinValue, tmpMaxValue = GetChartPredictEdbInfoDataListByRuleHc(predictEdbConf.PredictEdbInfoId, hcValue, startDate, dataEndTime, frequency, realPredictEdbInfoData, predictEdbInfoData, existMap)
  508. case 7: //7:N期移动均值
  509. nValue, tmpErr := strconv.Atoi(predictEdbConf.Value)
  510. if tmpErr != nil {
  511. err = tmpErr
  512. return
  513. }
  514. predictEdbInfoData, tmpMinValue, tmpMaxValue = GetChartPredictEdbInfoDataListByRuleNMoveMeanValue(predictEdbConf.PredictEdbInfoId, nValue, startDate, dataEndTime, frequency, realPredictEdbInfoData, predictEdbInfoData, existMap)
  515. case 8: //8:N期段线性外推值
  516. nValue, tmpErr := strconv.Atoi(predictEdbConf.Value)
  517. if tmpErr != nil {
  518. err = tmpErr
  519. return
  520. }
  521. predictEdbInfoData, tmpMinValue, tmpMaxValue, err = GetChartPredictEdbInfoDataListByRuleNLinearRegression(predictEdbConf.PredictEdbInfoId, nValue, startDate, dataEndTime, frequency, realPredictEdbInfoData, predictEdbInfoData, existMap)
  522. if err != nil {
  523. return
  524. }
  525. case 9: //9:动态环差”预测规则;
  526. predictEdbInfoData, tmpMinValue, tmpMaxValue = GetChartPredictEdbInfoDataListByRuleTrendsHC(predictEdbConf.PredictEdbInfoId, predictEdbConf.ConfigId, startDate, dataEndTime, frequency, realPredictEdbInfoData, predictEdbInfoData, existMap)
  527. case 10: //10:根据 给定终值后插值 规则获取预测数据
  528. tmpValDecimal, tmpErr := decimal.NewFromString(predictEdbConf.Value)
  529. if tmpErr != nil {
  530. err = tmpErr
  531. return
  532. }
  533. finalValue, _ := tmpValDecimal.Float64()
  534. predictEdbInfoData, tmpMinValue, tmpMaxValue = GetChartPredictEdbInfoDataListByRuleFinalValueHc(predictEdbConf.PredictEdbInfoId, finalValue, startDate, dataEndTime, frequency, realPredictEdbInfoData, predictEdbInfoData, existMap)
  535. }
  536. //startDate = dataEndTime.AddDate(0, 0, 1)
  537. if startDate.Before(dataEndTime) {
  538. startDate = dataEndTime
  539. }
  540. if tmpMinValue < minValue {
  541. minValue = tmpMinValue
  542. }
  543. if tmpMaxValue < maxValue {
  544. maxValue = tmpMaxValue
  545. }
  546. }
  547. return
  548. }
  549. // GetPredictEdbDayList 获取预测指标日期列表
  550. func getPredictEdbDayList(startDate, endDate time.Time, frequency string) (dayList []time.Time) {
  551. //if !utils.InArrayByStr([]string{"日度", "周度", "月度"}, frequency)
  552. switch frequency {
  553. case "日度":
  554. for currDate := startDate.AddDate(0, 0, 1); currDate.Before(endDate) || currDate.Equal(endDate); currDate = currDate.AddDate(0, 0, 1) {
  555. //周六、日排除
  556. if currDate.Weekday() == time.Sunday || currDate.Weekday() == time.Saturday {
  557. continue
  558. }
  559. dayList = append(dayList, currDate)
  560. }
  561. case "周度":
  562. //nextDate := startDate.AddDate(0, 0, 7)
  563. for currDate := startDate.AddDate(0, 0, 7); currDate.Before(endDate) || currDate.Equal(endDate); currDate = currDate.AddDate(0, 0, 7) {
  564. dayList = append(dayList, currDate)
  565. }
  566. case "月度":
  567. for currDate := startDate; currDate.Before(endDate) || currDate.Equal(endDate); {
  568. currDate = time.Date(currDate.Year(), currDate.Month(), 1, 0, 0, 0, 0, time.Now().Location()).AddDate(0, 2, -1)
  569. if !currDate.After(endDate) {
  570. dayList = append(dayList, currDate)
  571. }
  572. }
  573. }
  574. return
  575. }
  576. // GetPredictDataListByPredictEdbInfoId 根据预测指标id获取预测指标的数据,order:1升序,其余值为降序
  577. func GetPredictDataListByPredictEdbInfoId(edbInfoId, order int, startDate string) (dataList []*EdbInfoSearchData, sourceEdbInfoItem *EdbInfo, predictEdbConf *PredictEdbConf, err error, errMsg string) {
  578. edbInfo, err := GetEdbInfoById(edbInfoId)
  579. if err != nil {
  580. errMsg = `获取预测指标信息失败`
  581. return
  582. }
  583. return GetPredictDataListByPredictEdbInfo(edbInfo, order, startDate)
  584. }
  585. // GetPredictDataListByPredictEdbInfo 根据预测指标信息获取预测指标的数据,order:1升序,其余值为降序
  586. func GetPredictDataListByPredictEdbInfo(edbInfo *EdbInfo, order int, startDate string) (dataList []*EdbInfoSearchData, sourceEdbInfoItem *EdbInfo, predictEdbConf *PredictEdbConf, err error, errMsg string) {
  587. // 查找该预测指标配置
  588. predictEdbConfList, err := GetPredictEdbConfListById(edbInfo.EdbInfoId)
  589. if err != nil && err.Error() != utils.ErrNoRow() {
  590. errMsg = "获取预测指标配置信息失败"
  591. return
  592. }
  593. if len(predictEdbConfList) == 0 {
  594. errMsg = "获取预测指标配置信息失败"
  595. err = errors.New(errMsg)
  596. return
  597. }
  598. predictEdbConf = predictEdbConfList[0]
  599. // 来源指标
  600. sourceEdbInfoItem, err = GetEdbInfoById(predictEdbConf.SourceEdbInfoId)
  601. if err != nil {
  602. if err.Error() == utils.ErrNoRow() {
  603. errMsg = "找不到来源指标信息"
  604. err = errors.New(errMsg)
  605. }
  606. return
  607. }
  608. allDataList := make([]*EdbInfoSearchData, 0)
  609. //获取指标数据(实际已生成)
  610. var condition string
  611. var pars []interface{}
  612. condition += " AND edb_info_id=? "
  613. pars = append(pars, sourceEdbInfoItem.EdbInfoId)
  614. if startDate != "" {
  615. condition += " AND data_time>=? "
  616. pars = append(pars, startDate)
  617. }
  618. dataList, err = GetEdbDataListAll(condition, pars, sourceEdbInfoItem.Source, order)
  619. if err != nil {
  620. return
  621. }
  622. // 如果选择了日期,那么需要筛选所有的数据,用于未来指标的生成
  623. if startDate != `` {
  624. allDataList, err = GetEdbDataListAll(" AND edb_info_id=? ", []interface{}{sourceEdbInfoItem.EdbInfoId}, sourceEdbInfoItem.Source, 1)
  625. if err != nil {
  626. return
  627. }
  628. } else {
  629. allDataList = dataList
  630. }
  631. // 获取预测指标未来的数据
  632. predictDataList := make([]*EdbInfoSearchData, 0)
  633. endDateStr := edbInfo.EndDate //预测指标的结束日期
  634. //predictDataList, err = GetChartPredictEdbInfoDataList(*predictEdbConf, sourceEdbInfoItem.LatestDate, sourceEdbInfoItem.LatestValue, endDateStr, edbInfo.Frequency, order)
  635. var predictMinValue, predictMaxValue float64
  636. predictDataList, predictMinValue, predictMaxValue, err = GetChartPredictEdbInfoDataListByConfList(predictEdbConfList, startDate, sourceEdbInfoItem.LatestDate, endDateStr, edbInfo.Frequency, allDataList)
  637. if err != nil {
  638. return
  639. }
  640. dataList = append(dataList, predictDataList...)
  641. if len(predictDataList) > 0 {
  642. // 如果最小值 大于 预测值,那么将预测值作为最小值数据返回
  643. if edbInfo.MinValue > predictMinValue {
  644. edbInfo.MinValue = predictMinValue
  645. }
  646. // 如果最大值 小于 预测值,那么将预测值作为最大值数据返回
  647. if edbInfo.MaxValue < predictMaxValue {
  648. edbInfo.MaxValue = predictMaxValue
  649. }
  650. }
  651. return
  652. }
  653. // GetPredictEdbDataListAll 获取该预测指标所有的数据 ,order:1升序,其余值为降序
  654. func GetPredictEdbDataListAll(edbInfo *EdbInfo, order int) (items []*EdbInfoSearchData, err error) {
  655. if edbInfo.Source == utils.DATA_SOURCE_PREDICT { //普通的预测指标是没有入库数据的,直接往配置里面获取
  656. items, _, _, err, _ = GetPredictDataListByPredictEdbInfo(edbInfo, 1, "")
  657. } else {
  658. var condition string
  659. var pars []interface{}
  660. condition += " AND edb_info_id=? "
  661. pars = append(pars, edbInfo.EdbInfoId)
  662. items, err = GetEdbDataListAll(condition, pars, edbInfo.Source, order)
  663. }
  664. return
  665. }
  666. // GetPredictEdbDataListAllByStartDate 根据开始日期获取该预测指标所有的数据 ,order:1升序,其余值为降序
  667. func GetPredictEdbDataListAllByStartDate(edbInfo *EdbInfo, order int, startDate string) (items []*EdbInfoSearchData, err error) {
  668. if edbInfo.Source == utils.DATA_SOURCE_PREDICT { //普通的预测指标是没有入库数据的,直接往配置里面获取
  669. items, _, _, err, _ = GetPredictDataListByPredictEdbInfo(edbInfo, 1, startDate)
  670. } else {
  671. var condition string
  672. var pars []interface{}
  673. condition += " AND edb_info_id=? "
  674. pars = append(pars, edbInfo.EdbInfoId)
  675. if startDate != "" {
  676. condition += " AND data_time>=? "
  677. pars = append(pars, startDate)
  678. }
  679. items, err = GetEdbDataListAll(condition, pars, edbInfo.Source, order)
  680. }
  681. return
  682. }
  683. // ModifyPredictEdbInfoMaxAndMinInfo 修改预测指标的最新数据信息
  684. func ModifyPredictEdbInfoMaxAndMinInfo(edbInfoId int, item *EdbInfoMaxAndMinInfo) (err error) {
  685. o := orm.NewOrm()
  686. 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=? `
  687. _, err = o.Raw(sql, item.MinDate, item.MaxDate, item.MinValue, item.MaxValue, item.LatestDate, item.LatestValue, edbInfoId).Exec()
  688. return
  689. }
  690. // ModifyCalculateEdbInfo 修改计算指标信息
  691. func ModifyCalculateEdbInfo(edbName, frequency, unit, calculateFormula string, classifyId, edbInfoId int) (err error) {
  692. o := orm.NewOrm()
  693. sql := ` UPDATE edb_info
  694. SET
  695. edb_name =?,
  696. edb_name_source =?,
  697. frequency = ?,
  698. unit = ?,
  699. classify_id = ?,
  700. calculate_formula=?,
  701. modify_time = NOW()
  702. WHERE edb_info_id = ?`
  703. _, err = o.Raw(sql, edbName, edbName, frequency, unit, classifyId, calculateFormula, edbInfoId).Exec()
  704. return
  705. }
  706. func GetEdbInfoItemByCondition(condition string, pars []interface{}) (item *EdbInfoList, err error) {
  707. o := orm.NewOrm()
  708. sql := ` SELECT * FROM edb_info WHERE 1=1 `
  709. if condition != "" {
  710. sql += condition
  711. }
  712. err = o.Raw(sql, pars).QueryRow(&item)
  713. return
  714. }