edb_info.go 52 KB

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