edb_info.go 42 KB

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