chart_info.go 94 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611261226132614261526162617261826192620262126222623262426252626262726282629263026312632263326342635263626372638263926402641264226432644264526462647264826492650265126522653265426552656265726582659266026612662266326642665266626672668266926702671267226732674267526762677267826792680268126822683268426852686268726882689269026912692269326942695269626972698269927002701270227032704270527062707270827092710271127122713271427152716271727182719272027212722272327242725272627272728272927302731273227332734273527362737273827392740274127422743274427452746274727482749275027512752275327542755275627572758275927602761276227632764276527662767276827692770277127722773277427752776277727782779278027812782278327842785278627872788278927902791279227932794279527962797279827992800280128022803280428052806280728082809281028112812281328142815281628172818281928202821282228232824282528262827282828292830283128322833283428352836283728382839284028412842284328442845284628472848284928502851285228532854285528562857285828592860286128622863286428652866286728682869287028712872287328742875287628772878287928802881288228832884288528862887288828892890289128922893289428952896289728982899290029012902290329042905290629072908290929102911291229132914291529162917291829192920292129222923292429252926292729282929293029312932293329342935293629372938293929402941294229432944294529462947294829492950295129522953295429552956295729582959296029612962296329642965296629672968296929702971297229732974297529762977297829792980298129822983298429852986298729882989299029912992299329942995299629972998299930003001300230033004300530063007300830093010301130123013301430153016301730183019302030213022
  1. package range_analysis
  2. import (
  3. "encoding/json"
  4. "eta/eta_api/controllers"
  5. "eta/eta_api/models"
  6. "eta/eta_api/models/data_manage"
  7. "eta/eta_api/models/data_manage/correlation/request"
  8. "eta/eta_api/models/system"
  9. "eta/eta_api/services/data"
  10. "eta/eta_api/services/data/data_manage_permission"
  11. rangeServ "eta/eta_api/services/data/range_analysis"
  12. "eta/eta_api/utils"
  13. "fmt"
  14. "github.com/rdlucklib/rdluck_tools/paging"
  15. "sort"
  16. "strconv"
  17. "strings"
  18. "time"
  19. )
  20. // RangeChartChartInfoController 区间计算图表管理
  21. type RangeChartChartInfoController struct {
  22. controllers.BaseAuthController
  23. }
  24. // Preview
  25. // @Title 区间计算图表-预览数据
  26. // @Description 区间计算图表-获取预览数据
  27. // @Param StartDate query string true "自定义开始日期"
  28. // @Param EndDate query string true "自定义结束日期"
  29. // @Param LeadValue query int true "领先值"
  30. // @Param LeadUnit query string true "领先单位:年,天,月,季,周"
  31. // @Success 200 {object} data_manage.ChartEdbInfoDetailResp
  32. // @router /chart_info/preview [post]
  33. func (this *RangeChartChartInfoController) Preview() {
  34. br := new(models.BaseResponse).Init()
  35. defer func() {
  36. this.Data["json"] = br
  37. this.ServeJSON()
  38. }()
  39. sysUser := this.SysUser
  40. if sysUser == nil {
  41. br.Msg = "请登录"
  42. br.ErrMsg = "请登录,SysUser Is Empty"
  43. br.Ret = 408
  44. return
  45. }
  46. var req data_manage.ChartRangeAnalysisPreviewReq
  47. if err := json.Unmarshal(this.Ctx.Input.RequestBody, &req); err != nil {
  48. br.Msg = "参数解析异常!"
  49. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  50. return
  51. }
  52. // 基础校验
  53. if req.ExtraConf.EdbInfoMode == 1 && req.EdbInfoId == 0 {
  54. br.Msg = "请选择指标"
  55. return
  56. }
  57. // todo 校验时间区间配置
  58. switch req.ExtraConf.DateRangeType {
  59. case 0:
  60. case 1:
  61. if len(req.ExtraConf.ManualDateConf) == 0 {
  62. br.Msg = "请选择时间区间"
  63. return
  64. }
  65. // 先按开始时间排序
  66. sort.Sort(data_manage.ChartRangeAnalysisManualDateConfList(req.ExtraConf.ManualDateConf))
  67. // 校验日期
  68. // 1.如果截止时间小于指标的截止日期,需要重置为指标的截止日期
  69. // 2.时间区间不能重叠
  70. for i := 1; i < len(req.ExtraConf.ManualDateConf); i++ {
  71. start1, err := time.Parse(utils.FormatDate, req.ExtraConf.ManualDateConf[i-1].EndDate)
  72. if err != nil {
  73. br.Msg = "截止日期格式有误"
  74. return
  75. }
  76. start2, err := time.Parse(utils.FormatDate, req.ExtraConf.ManualDateConf[i].EndDate)
  77. if err != nil {
  78. br.Msg = "截止日期格式有误"
  79. return
  80. }
  81. start3, err := time.Parse(utils.FormatDate, req.ExtraConf.ManualDateConf[i].StartDate)
  82. if err != nil {
  83. br.Msg = "截止日期格式有误"
  84. return
  85. }
  86. // 如果当前区间的开始时间小于等于前一个区间的结束时间,则存在重叠
  87. if !start2.After(start1) || start3.Before(start1) {
  88. br.Msg = "日期区间存在重叠"
  89. return
  90. }
  91. }
  92. //如果截止时间大于指标的截止日期,需要重置为指标的截止日期
  93. case 2:
  94. if req.ExtraConf.YearDateConf.StartDay == "" || req.ExtraConf.YearDateConf.EndDay == "" {
  95. br.Msg = "请选择时间区间"
  96. return
  97. }
  98. if _, e := time.Parse(utils.FormatMonthDay, req.ExtraConf.YearDateConf.StartDay); e != nil {
  99. br.Msg = "开始日期格式有误"
  100. return
  101. }
  102. if _, e := time.Parse(utils.FormatMonthDay, req.ExtraConf.YearDateConf.EndDay); e != nil {
  103. br.Msg = "结束日期格式有误"
  104. return
  105. }
  106. }
  107. // 获取指标信息
  108. edbInfoMapping, e := data_manage.GetChartEdbMappingByEdbInfoId(req.EdbInfoId)
  109. if e != nil {
  110. br.Msg = "获取失败"
  111. br.ErrMsg = "获取区间计算图表, A指标mapping信息失败, Err:" + e.Error()
  112. return
  113. }
  114. chartInfo := new(data_manage.ChartInfoView)
  115. chartInfo.ChartType = 9
  116. //chartInfo.CorrelationLeadUnit = req.LeadUnit
  117. // 获取图表x轴y轴
  118. xEdbIdValue, yDataList, e := rangeServ.GetChartDataByEdbInfo(edbInfoMappingA, edbInfoMappingB, req.LeadValue, req.LeadUnit, req.StartDate, req.EndDate, "")
  119. if e != nil {
  120. br.Msg = "获取失败"
  121. br.ErrMsg = "获取区间计算图表, 图表计算值失败, Err:" + e.Error()
  122. return
  123. }
  124. // 完善指标信息
  125. edbList, e := rangeServ.GetChartEdbInfoFormat(chartInfo.ChartInfoId, edbInfoMappingA, edbInfoMappingB)
  126. if e != nil {
  127. br.Msg = "获取失败"
  128. br.ErrMsg = "获取区间计算图表, 完善指标信息失败, Err:" + e.Error()
  129. return
  130. }
  131. correlationInfo := new(data_manage.CorrelationInfo)
  132. correlationInfo.LeadValue = req.LeadValue
  133. correlationInfo.LeadUnit = req.LeadUnit
  134. correlationInfo.StartDate = req.StartDate
  135. correlationInfo.EndDate = req.EndDate
  136. correlationInfo.LeadValue = req.LeadValue
  137. correlationInfo.EdbInfoIdFirst = edbInfoMappingA.EdbInfoId
  138. correlationInfo.EdbInfoIdSecond = edbInfoMappingB.EdbInfoId
  139. resp := new(data_manage.ChartInfoDetailResp)
  140. resp.ChartInfo = chartInfo
  141. resp.EdbInfoList = edbList
  142. resp.XEdbIdValue = xEdbIdValue
  143. resp.YDataList = yDataList
  144. resp.DataResp = correlationInfo
  145. br.Data = resp
  146. br.Ret = 200
  147. br.Success = true
  148. br.Msg = "获取成功"
  149. }
  150. // Add
  151. // @Title 新增图表接口
  152. // @Description 新增图表接口
  153. // @Param request body data_manage.AddChartInfoReq true "type json string"
  154. // @Success 200 {object} data_manage.AddChartInfoResp
  155. // @router /chart_info/add [post]
  156. func (this *RangeChartChartInfoController) Add() {
  157. br := new(models.BaseResponse).Init()
  158. defer func() {
  159. this.Data["json"] = br
  160. this.ServeJSON()
  161. }()
  162. sysUser := this.SysUser
  163. if sysUser == nil {
  164. br.Msg = "请登录"
  165. br.ErrMsg = "请登录,SysUser Is Empty"
  166. br.Ret = 408
  167. return
  168. }
  169. deleteCache := true
  170. cacheKey := "CACHE_CHART_INFO_ADD_" + strconv.Itoa(sysUser.AdminId)
  171. defer func() {
  172. if deleteCache {
  173. _ = utils.Rc.Delete(cacheKey)
  174. }
  175. }()
  176. if !utils.Rc.SetNX(cacheKey, 1, 30*time.Second) {
  177. deleteCache = false
  178. br.Msg = "系统处理中,请稍后重试!"
  179. br.ErrMsg = "系统处理中,请稍后重试!" + sysUser.RealName + ";data:" + string(this.Ctx.Input.RequestBody)
  180. return
  181. }
  182. var req data_manage.AddChartInfoReq
  183. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  184. if err != nil {
  185. br.Msg = "参数解析异常!"
  186. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  187. return
  188. }
  189. chartInfo, err, errMsg, isSendEmail := rangeServ.AddChartInfo(req, utils.CHART_SOURCE_CORRELATION, sysUser, this.Lang)
  190. if err != nil {
  191. br.Msg = "保存失败"
  192. if errMsg != `` {
  193. br.Msg = errMsg
  194. }
  195. br.ErrMsg = err.Error()
  196. br.IsSendEmail = isSendEmail
  197. return
  198. }
  199. //新增操作日志
  200. {
  201. chartLog := new(data_manage.ChartInfoLog)
  202. chartLog.ChartInfoId = chartInfo.ChartInfoId
  203. chartLog.ChartName = req.ChartName
  204. chartLog.ChartClassifyId = req.ChartClassifyId
  205. chartLog.SysUserId = sysUser.AdminId
  206. chartLog.SysUserRealName = sysUser.RealName
  207. chartLog.UniqueCode = chartInfo.UniqueCode
  208. chartLog.CreateTime = time.Now()
  209. chartLog.Content = string(this.Ctx.Input.RequestBody)
  210. chartLog.Status = "新增区间计算图表"
  211. chartLog.Method = this.Ctx.Input.URI()
  212. go data_manage.AddChartInfoLog(chartLog)
  213. }
  214. resp := new(data_manage.AddChartInfoResp)
  215. resp.ChartInfoId = chartInfo.ChartInfoId
  216. resp.UniqueCode = chartInfo.UniqueCode
  217. resp.ChartType = chartInfo.ChartType
  218. br.Ret = 200
  219. br.Success = true
  220. br.Msg = "保存成功"
  221. br.Data = resp
  222. br.IsAddLog = true
  223. }
  224. // Edit
  225. // @Title 编辑图表接口
  226. // @Description 编辑图表接口
  227. // @Param request body data_manage.EditChartInfoReq true "type json string"
  228. // @Success Ret=200 保存成功
  229. // @router /chart_info/edit [post]
  230. func (this *RangeChartChartInfoController) Edit() {
  231. br := new(models.BaseResponse).Init()
  232. defer func() {
  233. this.Data["json"] = br
  234. this.ServeJSON()
  235. }()
  236. sysUser := this.SysUser
  237. if sysUser == nil {
  238. br.Msg = "请登录"
  239. br.ErrMsg = "请登录,SysUser Is Empty"
  240. br.Ret = 408
  241. return
  242. }
  243. var req data_manage.EditChartInfoReq
  244. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  245. if err != nil {
  246. br.Msg = "参数解析异常!"
  247. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  248. return
  249. }
  250. chartItem, err, errMsg, isSendEmail := rangeServ.EditChartInfo(req, sysUser, this.Lang)
  251. if err != nil {
  252. br.Msg = "保存失败"
  253. if errMsg != `` {
  254. br.Msg = errMsg
  255. }
  256. br.ErrMsg = err.Error()
  257. br.IsSendEmail = isSendEmail
  258. return
  259. }
  260. resp := new(data_manage.AddChartInfoResp)
  261. resp.ChartInfoId = chartItem.ChartInfoId
  262. resp.UniqueCode = chartItem.UniqueCode
  263. resp.ChartType = req.ChartType
  264. //新增操作日志
  265. {
  266. chartLog := new(data_manage.ChartInfoLog)
  267. chartLog.ChartName = chartItem.ChartName
  268. chartLog.ChartInfoId = req.ChartInfoId
  269. chartLog.ChartClassifyId = chartItem.ChartClassifyId
  270. chartLog.SysUserId = sysUser.AdminId
  271. chartLog.SysUserRealName = sysUser.RealName
  272. chartLog.UniqueCode = chartItem.UniqueCode
  273. chartLog.CreateTime = time.Now()
  274. chartLog.Content = string(this.Ctx.Input.RequestBody)
  275. chartLog.Status = "编辑区间计算图表"
  276. chartLog.Method = this.Ctx.Input.URL()
  277. go data_manage.AddChartInfoLog(chartLog)
  278. }
  279. br.Ret = 200
  280. br.Success = true
  281. br.Msg = "保存成功"
  282. br.Data = resp
  283. br.IsAddLog = true
  284. }
  285. // Move
  286. // @Title 移动图表接口
  287. // @Description 移动图表接口
  288. // @Success 200 {object} data_manage.MoveChartInfoReq
  289. // @router /chart_info/move [post]
  290. func (this *RangeChartChartInfoController) Move() {
  291. br := new(models.BaseResponse).Init()
  292. defer func() {
  293. this.Data["json"] = br
  294. this.ServeJSON()
  295. }()
  296. sysUser := this.SysUser
  297. if sysUser == nil {
  298. br.Msg = "请登录"
  299. br.ErrMsg = "请登录,SysUser Is Empty"
  300. br.Ret = 408
  301. return
  302. }
  303. var req data_manage.MoveChartInfoReq
  304. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  305. if err != nil {
  306. br.Msg = "参数解析异常!"
  307. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  308. return
  309. }
  310. if req.ChartInfoId <= 0 {
  311. br.Msg = "参数错误"
  312. br.ErrMsg = "图表id小于等于0"
  313. return
  314. }
  315. if req.ChartClassifyId <= 0 {
  316. br.Msg = "请选择分类"
  317. return
  318. }
  319. //判断分类是否存在
  320. count, _ := data_manage.GetChartClassifyCountById(req.ChartClassifyId)
  321. if count <= 0 {
  322. br.Msg = "分类已被删除,不可移动,请刷新页面"
  323. return
  324. }
  325. chartInfo, err := data_manage.GetChartInfoById(req.ChartInfoId)
  326. if err != nil {
  327. br.Msg = "移动失败"
  328. br.ErrMsg = "获取图表信息失败,Err:" + err.Error()
  329. return
  330. }
  331. if !utils.InArrayByInt([]int{utils.CHART_SOURCE_CORRELATION, utils.CHART_SOURCE_ROLLING_CORRELATION}, chartInfo.Source) {
  332. br.Msg = "图表异常"
  333. br.ErrMsg = "分类异常,不是区间计算的图表"
  334. return
  335. }
  336. //如果改变了分类,那么移动该图表数据
  337. if chartInfo.ChartClassifyId != req.ChartClassifyId {
  338. //查询需要修改的分类下是否存在同一个图表名称
  339. tmpChartInfo, tmpErr := data_manage.GetChartInfoByClassifyIdAndName(req.ChartClassifyId, chartInfo.ChartName)
  340. if tmpErr != nil && tmpErr.Error() != utils.ErrNoRow() {
  341. br.Msg = "移动失败"
  342. br.ErrMsg = "移动失败,Err:" + tmpErr.Error()
  343. return
  344. }
  345. if tmpChartInfo != nil {
  346. br.Msg = "移动失败,同一个分类下图表名称不允许重复"
  347. br.ErrMsg = "移动失败,同一个分类下图表名称不允许重复"
  348. return
  349. }
  350. err = data_manage.MoveChartInfo(req.ChartInfoId, req.ChartClassifyId)
  351. if err != nil {
  352. br.Msg = "移动失败"
  353. br.ErrMsg = "移动失败,Err:" + err.Error()
  354. return
  355. }
  356. }
  357. //移动排序
  358. updateCol := make([]string, 0)
  359. //如果有传入 上一个兄弟节点分类id
  360. if req.PrevChartInfoId > 0 {
  361. prevChartInfo, err := data_manage.GetChartInfoById(req.PrevChartInfoId)
  362. if err != nil {
  363. br.Msg = "移动失败"
  364. br.ErrMsg = "获取上一个兄弟节点分类信息失败,Err:" + err.Error()
  365. return
  366. }
  367. //如果是移动在两个兄弟节点之间
  368. if req.NextChartInfoId > 0 {
  369. //下一个兄弟节点
  370. nextChartInfo, err := data_manage.GetChartInfoById(req.NextChartInfoId)
  371. if err != nil {
  372. br.Msg = "移动失败"
  373. br.ErrMsg = "获取下一个兄弟节点分类信息失败,Err:" + err.Error()
  374. return
  375. }
  376. //如果上一个兄弟与下一个兄弟的排序权重是一致的,那么需要将下一个兄弟(以及下个兄弟的同样排序权重)的排序权重+2,自己变成上一个兄弟的排序权重+1
  377. if prevChartInfo.Sort == nextChartInfo.Sort || prevChartInfo.Sort == chartInfo.Sort {
  378. //变更兄弟节点的排序
  379. updateSortStr := `sort + 2`
  380. _ = data_manage.UpdateChartInfoSortByClassifyId(prevChartInfo.ChartClassifyId, prevChartInfo.Sort, prevChartInfo.ChartInfoId, []int{chartInfo.Source}, updateSortStr)
  381. } else {
  382. //如果下一个兄弟的排序权重正好是上个兄弟节点的下一层,那么需要再加一层了
  383. if nextChartInfo.Sort-prevChartInfo.Sort == 1 {
  384. //变更兄弟节点的排序
  385. updateSortStr := `sort + 1`
  386. _ = data_manage.UpdateChartInfoSortByClassifyId(prevChartInfo.ChartClassifyId, prevChartInfo.Sort, prevChartInfo.ChartInfoId, []int{chartInfo.Source}, updateSortStr)
  387. }
  388. }
  389. }
  390. chartInfo.Sort = prevChartInfo.Sort + 1
  391. chartInfo.ModifyTime = time.Now()
  392. updateCol = append(updateCol, "Sort", "ModifyTime")
  393. } else {
  394. firstClassify, err := data_manage.GetFirstChartInfoByClassifyId(req.ChartClassifyId)
  395. if err != nil && err.Error() != utils.ErrNoRow() {
  396. br.Msg = "移动失败"
  397. br.ErrMsg = "获取获取当前父级分类下的排序第一条的分类信息失败,Err:" + err.Error()
  398. return
  399. }
  400. //如果该分类下存在其他分类,且第一个其他分类的排序等于0,那么需要调整排序
  401. if firstClassify != nil && firstClassify.Sort == 0 {
  402. updateSortStr := ` sort + 1 `
  403. _ = data_manage.UpdateChartInfoSortByClassifyId(firstClassify.ChartClassifyId, 0, firstClassify.ChartInfoId-1, []int{chartInfo.Source}, updateSortStr)
  404. }
  405. chartInfo.Sort = 0 //那就是排在第一位
  406. chartInfo.ModifyTime = time.Now()
  407. updateCol = append(updateCol, "Sort", "ModifyTime")
  408. }
  409. //更新
  410. if len(updateCol) > 0 {
  411. err = chartInfo.Update(updateCol)
  412. if err != nil {
  413. br.Msg = "移动失败"
  414. br.ErrMsg = "修改失败,Err:" + err.Error()
  415. return
  416. }
  417. }
  418. if err != nil {
  419. br.Msg = "移动失败"
  420. br.ErrMsg = "修改失败,Err:" + err.Error()
  421. return
  422. }
  423. //添加es数据
  424. go data.EsAddOrEditChartInfo(req.ChartInfoId)
  425. //修改my eta es数据
  426. go data.EsAddOrEditMyChartInfoByChartInfoId(req.ChartInfoId)
  427. //新增操作日志
  428. {
  429. chartLog := new(data_manage.ChartInfoLog)
  430. chartLog.ChartName = chartInfo.ChartName
  431. chartLog.ChartInfoId = req.ChartInfoId
  432. chartLog.ChartClassifyId = chartInfo.ChartClassifyId
  433. chartLog.SysUserId = sysUser.AdminId
  434. chartLog.SysUserRealName = sysUser.RealName
  435. chartLog.UniqueCode = chartInfo.UniqueCode
  436. chartLog.CreateTime = time.Now()
  437. chartLog.Content = string(this.Ctx.Input.RequestBody)
  438. chartLog.Status = "移动区间计算图表"
  439. chartLog.Method = this.Ctx.Input.URL()
  440. go data_manage.AddChartInfoLog(chartLog)
  441. }
  442. br.Ret = 200
  443. br.Success = true
  444. br.Msg = "移动成功"
  445. }
  446. // List
  447. // @Title 区间计算图表列表接口
  448. // @Description 区间计算图表列表接口
  449. // @Param PageSize query int true "每页数据条数"
  450. // @Param CurrentIndex query int true "当前页页码,从1开始"
  451. // @Param ChartClassifyId query int true "分类id"
  452. // @Param Keyword query string true "搜索关键词"
  453. // @Param IsShowMe query bool true "是否只看我的,true、false"
  454. // @Param Source query int true "图表类型,3:区间计算,4:滚动区间计算"
  455. // @Success 200 {object} data_manage.ChartListResp
  456. // @router /chart_info/list [get]
  457. func (this *RangeChartChartInfoController) List() {
  458. br := new(models.BaseResponse).Init()
  459. defer func() {
  460. this.Data["json"] = br
  461. this.ServeJSON()
  462. }()
  463. sysUser := this.SysUser
  464. if sysUser == nil {
  465. br.Msg = "请登录"
  466. br.ErrMsg = "请登录,SysUser Is Empty"
  467. br.Ret = 408
  468. return
  469. }
  470. chartClassifyId, _ := this.GetInt("ChartClassifyId")
  471. pageSize, _ := this.GetInt("PageSize")
  472. currentIndex, _ := this.GetInt("CurrentIndex")
  473. keyword := this.GetString("KeyWord")
  474. var total int
  475. page := paging.GetPaging(currentIndex, pageSize, total)
  476. var startSize int
  477. if pageSize <= 0 {
  478. pageSize = utils.PageSize20
  479. }
  480. if currentIndex <= 0 {
  481. currentIndex = 1
  482. }
  483. startSize = paging.StartIndex(currentIndex, pageSize)
  484. source, _ := this.GetInt("Source")
  485. if source <= 0 {
  486. source = utils.CHART_SOURCE_CORRELATION
  487. }
  488. var condition string
  489. var pars []interface{}
  490. // 普通图表
  491. condition += ` AND source = ? `
  492. pars = append(pars, source)
  493. if chartClassifyId > 0 {
  494. chartClassifyId, err := data_manage.GetChartClassify(chartClassifyId)
  495. if err != nil && err.Error() != utils.ErrNoRow() {
  496. br.Msg = "获取图表信息失败"
  497. br.ErrMsg = "获取信息失败,GetChartClassify,Err:" + err.Error()
  498. return
  499. }
  500. condition += " AND chart_classify_id IN(" + chartClassifyId + ") "
  501. }
  502. if keyword != "" {
  503. condition += ` AND ( chart_name LIKE '%` + keyword + `%' )`
  504. }
  505. //只看我的
  506. isShowMe, _ := this.GetBool("IsShowMe")
  507. if isShowMe {
  508. condition += ` AND sys_user_id = ? `
  509. pars = append(pars, sysUser.AdminId)
  510. }
  511. // 获取当前账号的不可见指标
  512. noPermissionChartIdList := make([]int, 0)
  513. {
  514. obj := data_manage.EdbInfoNoPermissionAdmin{}
  515. confList, err := obj.GetAllChartListByAdminId(this.SysUser.AdminId)
  516. if err != nil && err.Error() != utils.ErrNoRow() {
  517. br.Msg = "获取失败"
  518. br.ErrMsg = "获取不可见指标配置数据失败,Err:" + err.Error()
  519. return
  520. }
  521. for _, v := range confList {
  522. noPermissionChartIdList = append(noPermissionChartIdList, v.ChartInfoId)
  523. }
  524. }
  525. lenNoPermissionChartIdList := len(noPermissionChartIdList)
  526. if lenNoPermissionChartIdList > 0 {
  527. condition += ` AND chart_info_id not in (` + utils.GetOrmInReplace(lenNoPermissionChartIdList) + `) `
  528. pars = append(pars, noPermissionChartIdList)
  529. }
  530. //获取图表信息
  531. list, err := data_manage.GetChartListByCondition(condition, pars, startSize, pageSize)
  532. if err != nil && err.Error() != utils.ErrNoRow() {
  533. br.Success = true
  534. br.Msg = "获取图表信息失败"
  535. br.ErrMsg = "获取图表信息失败,Err:" + err.Error()
  536. return
  537. }
  538. myChartList, err := data_manage.GetMyChartListByAdminId(sysUser.AdminId)
  539. if err != nil && err.Error() != utils.ErrNoRow() {
  540. br.Msg = "获取图表信息失败"
  541. br.ErrMsg = "获取我的图表信息失败,Err:" + err.Error()
  542. return
  543. }
  544. myChartMap := make(map[int]*data_manage.MyChartView)
  545. for _, v := range myChartList {
  546. myChartMap[v.ChartInfoId] = v
  547. }
  548. listLen := len(list)
  549. chartEdbMap := make(map[int][]*data_manage.ChartEdbInfoMapping)
  550. if listLen > 0 {
  551. chartInfoIds := ""
  552. for _, v := range list {
  553. chartInfoIds += strconv.Itoa(v.ChartInfoId) + ","
  554. }
  555. if chartInfoIds != "" {
  556. chartInfoIds = strings.Trim(chartInfoIds, ",")
  557. //判断是否需要展示英文标识
  558. edbList, e := data_manage.GetChartEdbMappingListByChartInfoIds(chartInfoIds)
  559. if e != nil {
  560. br.Msg = "获取失败"
  561. br.ErrMsg = "获取图表,指标信息失败,Err:" + e.Error()
  562. return
  563. }
  564. for _, v := range edbList {
  565. chartEdbMap[v.ChartInfoId] = append(chartEdbMap[v.ChartInfoId], v)
  566. }
  567. }
  568. }
  569. for i := 0; i < listLen; i++ {
  570. //判断是否需要展示英文标识
  571. if edbTmpList, ok := chartEdbMap[list[i].ChartInfoId]; ok {
  572. list[i].IsEnChart = data.CheckIsEnChart(list[i].ChartNameEn, edbTmpList, list[i].Source, list[i].ChartType)
  573. }
  574. if existItem, ok := myChartMap[list[i].ChartInfoId]; ok {
  575. list[i].IsAdd = true
  576. list[i].MyChartId = existItem.MyChartId
  577. list[i].MyChartClassifyId = existItem.MyChartClassifyId
  578. }
  579. }
  580. resp := new(data_manage.ChartListResp)
  581. if list == nil || len(list) <= 0 || (err != nil && err.Error() == utils.ErrNoRow()) {
  582. items := make([]*data_manage.ChartInfoView, 0)
  583. resp.Paging = page
  584. resp.List = items
  585. br.Ret = 200
  586. br.Success = true
  587. br.Msg = "获取成功"
  588. return
  589. }
  590. dataCount, err := data_manage.GetChartListCountByCondition(condition, pars)
  591. if err != nil && err.Error() != utils.ErrNoRow() {
  592. br.Msg = "获取指标信息失败"
  593. br.ErrMsg = "获取指标数据总数失败,Err:" + err.Error()
  594. return
  595. }
  596. page = paging.GetPaging(currentIndex, pageSize, dataCount)
  597. resp.Paging = page
  598. resp.List = list
  599. br.Ret = 200
  600. br.Success = true
  601. br.Msg = "获取成功"
  602. br.Data = resp
  603. }
  604. // Detail
  605. // @Title 获取图表详情
  606. // @Description 获取图表详情接口
  607. // @Param ChartInfoId query int true "图表id"
  608. // @Param DateType query int true "日期类型:1:00年至今,2:10年至今,3:15年至今,4:年初至今,5:自定义时间"
  609. // @Param StartDate query string true "自定义开始日期"
  610. // @Param EndDate query string true "自定义结束日期"
  611. // @Param Calendar query string true "公历/农历"
  612. // @Param SeasonStartDate query string true "季节性图开始日期"
  613. // @Param SeasonEndDate query string true "季节性图结束日期"
  614. // @Param EdbInfoId query string true "指标ID,多个用英文逗号隔开"
  615. // @Param ChartType query int true "生成样式:1:曲线图,2:季节性图"
  616. // @Success 200 {object} data_manage.ChartInfoDetailResp
  617. // @router /chart_info/detail [get]
  618. func (this *RangeChartChartInfoController) Detail() {
  619. br := new(models.BaseResponse).Init()
  620. defer func() {
  621. this.Data["json"] = br
  622. this.ServeJSON()
  623. }()
  624. sysUser := this.SysUser
  625. if sysUser == nil {
  626. br.Msg = "请登录"
  627. br.ErrMsg = "请登录,SysUser Is Empty"
  628. br.Ret = 408
  629. return
  630. }
  631. chartInfoId, _ := this.GetInt("ChartInfoId")
  632. if chartInfoId <= 0 {
  633. br.Msg = "参数有误"
  634. return
  635. }
  636. var err error
  637. chartInfo := new(data_manage.ChartInfoView)
  638. chartInfo, err = data_manage.GetChartInfoViewById(chartInfoId)
  639. if err != nil {
  640. if err.Error() == utils.ErrNoRow() {
  641. br.Msg = "图被删除,请刷新页面"
  642. br.ErrMsg = "图被删除,请刷新页面,Err:" + err.Error()
  643. return
  644. }
  645. br.Msg = "获取失败"
  646. br.ErrMsg = "获取图表信息失败,Err:" + err.Error()
  647. return
  648. }
  649. // 获取主题样式
  650. chartTheme, err := data.GetChartThemeConfig(chartInfo.ChartThemeId, utils.CHART_SOURCE_DEFAULT, 1)
  651. if err != nil {
  652. br.Msg = "获取失败"
  653. br.ErrMsg = "获取主题信息失败,Err:" + err.Error()
  654. return
  655. }
  656. chartInfo.ChartThemeStyle = chartTheme.Config
  657. chartInfo.ChartThemeId = chartTheme.ChartThemeId
  658. // 区间计算图表信息
  659. correlationChart := new(data_manage.ChartInfoCorrelation)
  660. if e := correlationChart.GetItemById(chartInfoId); e != nil {
  661. br.Msg = "获取失败"
  662. br.ErrMsg = "获取图表区间计算信息失败, Err: " + e.Error()
  663. return
  664. }
  665. // 获取指标信息
  666. edbInfoMappingA, e := data_manage.GetChartEdbMappingByEdbInfoId(correlationChart.EdbInfoIdFirst)
  667. if e != nil {
  668. br.Msg = "获取失败"
  669. br.ErrMsg = "获取区间计算图表, A指标mapping信息失败, Err:" + e.Error()
  670. return
  671. }
  672. edbInfoMappingB := new(data_manage.ChartEdbInfoMapping)
  673. // 非多因子
  674. if correlationChart.AnalysisMode != 1 {
  675. edbInfoMappingB, e = data_manage.GetChartEdbMappingByEdbInfoId(correlationChart.EdbInfoIdSecond)
  676. if e != nil {
  677. br.Msg = "获取失败"
  678. br.ErrMsg = "获取区间计算图表, B指标mapping信息失败, Err:" + e.Error()
  679. return
  680. }
  681. }
  682. var dataResp interface{} // 绘图数据返回(目前是滚动区间计算的图)
  683. var xEdbIdValue []int
  684. var yDataList []data_manage.YData
  685. if correlationChart.AnalysisMode != 1 {
  686. switch chartInfo.Source {
  687. case utils.CHART_SOURCE_CORRELATION: // 区间计算图
  688. moveUnitDays, ok := utils.FrequencyDaysMap[correlationChart.CalculateUnit]
  689. if !ok {
  690. br.Msg = "错误的分析周期"
  691. br.IsSendEmail = false
  692. return
  693. }
  694. startDate := time.Now().AddDate(0, 0, -correlationChart.CalculateValue*moveUnitDays).Format(utils.FormatDate)
  695. endDate := time.Now().Format(utils.FormatDate)
  696. xEdbIdValue, yDataList, e = rangeServ.GetChartDataByEdbInfo(edbInfoMappingA, edbInfoMappingB, correlationChart.LeadValue, correlationChart.LeadUnit, startDate, endDate, chartInfo.ExtraConfig)
  697. if e != nil {
  698. br.Msg = "获取失败"
  699. br.ErrMsg = "获取区间计算图表, 图表计算值失败, Err:" + e.Error()
  700. return
  701. }
  702. case utils.CHART_SOURCE_ROLLING_CORRELATION: // 滚动区间计算图
  703. startDate, endDate := utils.GetDateByDateType(correlationChart.DateType, correlationChart.StartDate.Format(utils.FormatDate), correlationChart.EndDate.Format(utils.FormatDate))
  704. dataResp, e = rangeServ.GetRollingCorrelationChartDataByEdbInfo(edbInfoMappingA, edbInfoMappingB, correlationChart.LeadValue, correlationChart.LeadUnit, correlationChart.CalculateValue, correlationChart.CalculateUnit, startDate, endDate, chartInfo.ChartName, chartInfo.ChartNameEn)
  705. if e != nil {
  706. br.Msg = "获取失败"
  707. br.ErrMsg = "获取滚动区间计算图表, 图表计算值失败, Err:" + e.Error()
  708. return
  709. }
  710. }
  711. } else {
  712. xEdbIdValue, yDataList, e = rangeServ.GetFactorChartDataByChartId(chartInfoId, chartInfo.ExtraConfig)
  713. if e != nil {
  714. br.Msg = "获取失败"
  715. br.ErrMsg = "获取区间计算图表, 图表计算值失败, Err:" + e.Error()
  716. return
  717. }
  718. }
  719. // 完善指标信息
  720. edbList := make([]*data_manage.ChartEdbInfoMapping, 0)
  721. if correlationChart.AnalysisMode != 1 {
  722. edbList, e = rangeServ.GetChartEdbInfoFormat(chartInfo.ChartInfoId, edbInfoMappingA, edbInfoMappingB)
  723. if e != nil {
  724. br.Msg = "获取失败"
  725. br.ErrMsg = "获取区间计算图表, 完善指标信息失败, Err:" + e.Error()
  726. return
  727. }
  728. } else {
  729. // 多因子指标, 获取图表引用到的系列指标(去重后)
  730. edbList, e = data_manage.GetChartEdbMappingListByIdList([]int{chartInfoId})
  731. if e != nil {
  732. err = fmt.Errorf("获取图表引用系列指标失败, err: %v", e)
  733. return
  734. }
  735. }
  736. correlationInfo := new(data_manage.CorrelationInfo)
  737. correlationInfo.LeadValue = correlationChart.LeadValue
  738. correlationInfo.LeadUnit = correlationChart.LeadUnit
  739. correlationInfo.CalculateValue = correlationChart.CalculateValue
  740. correlationInfo.CalculateUnit = correlationChart.CalculateUnit
  741. correlationInfo.StartDate = correlationChart.StartDate.Format(utils.FormatDate)
  742. correlationInfo.EndDate = correlationChart.EndDate.Format(utils.FormatDate)
  743. correlationInfo.LeadValue = correlationChart.LeadValue
  744. correlationInfo.EdbInfoIdFirst = correlationChart.EdbInfoIdFirst
  745. correlationInfo.EdbInfoIdSecond = correlationChart.EdbInfoIdSecond
  746. correlationInfo.AnalysisMode = correlationChart.AnalysisMode
  747. // 判断是否加入我的图库
  748. if chartInfoId > 0 && chartInfo != nil {
  749. {
  750. var myChartCondition string
  751. var myChartPars []interface{}
  752. myChartCondition += ` AND a.admin_id=? `
  753. myChartPars = append(myChartPars, sysUser.AdminId)
  754. myChartCondition += ` AND a.chart_info_id=? `
  755. myChartPars = append(myChartPars, chartInfo.ChartInfoId)
  756. myChartList, err := data_manage.GetMyChartByCondition(myChartCondition, myChartPars)
  757. if err != nil && err.Error() != utils.ErrNoRow() {
  758. br.Msg = "获取失败"
  759. br.ErrMsg = "获取我的图表信息失败,GetMyChartByCondition,Err:" + err.Error()
  760. return
  761. }
  762. if myChartList != nil && len(myChartList) > 0 {
  763. chartInfo.IsAdd = true
  764. chartInfo.MyChartId = myChartList[0].MyChartId
  765. chartInfo.MyChartClassifyId = myChartList[0].MyChartClassifyId
  766. }
  767. }
  768. }
  769. //图表操作权限
  770. chartInfo.IsEdit = data.CheckOpChartPermission(sysUser, chartInfo.SysUserId, true)
  771. //判断是否需要展示英文标识
  772. chartInfo.IsEnChart = data.CheckIsEnChart(chartInfo.ChartNameEn, edbList[0:1], chartInfo.Source, chartInfo.ChartType)
  773. chartInfo.UnitEn = edbInfoMappingA.UnitEn
  774. sourceNameList, sourceNameEnList := data.GetEdbSourceByEdbInfoIdList(edbList)
  775. chartInfo.ChartSource = strings.Join(sourceNameList, ",")
  776. chartInfo.ChartSourceEn = strings.Join(sourceNameEnList, ",")
  777. isSaveAs := true
  778. if chartInfo.Source == utils.CHART_SOURCE_ROLLING_CORRELATION {
  779. isSaveAs = false
  780. }
  781. // 另存为
  782. chartInfo.Button = data_manage.ChartViewButton{
  783. IsEdit: chartInfo.IsEdit,
  784. IsEnChart: chartInfo.IsEnChart,
  785. IsAdd: chartInfo.IsAdd,
  786. IsCopy: isSaveAs,
  787. IsSetName: chartInfo.IsSetName,
  788. }
  789. // 指标权限
  790. {
  791. edbClassifyPermissionMap := make(map[int]data_manage_permission.EdbClassifyPermission)
  792. classifyMap := make(map[int]*data_manage.EdbClassify)
  793. // 分类
  794. {
  795. classifyIdList := make([]int, 0)
  796. for _, v := range edbList {
  797. classifyIdList = append(classifyIdList, v.ClassifyId)
  798. }
  799. classifyList, tmpErr := data_manage.GetEdbClassifyByIdList(classifyIdList)
  800. if tmpErr != nil {
  801. br.Msg = "获取失败"
  802. br.ErrMsg = "获取分类列表失败,Err:" + tmpErr.Error()
  803. return
  804. }
  805. for _, v := range classifyList {
  806. classifyMap[v.ClassifyId] = v
  807. }
  808. }
  809. // 遍历到校验map
  810. for _, v := range edbList {
  811. edbClassifyPermissionMap[v.EdbInfoId] = data_manage_permission.EdbClassifyPermission{
  812. ClassifyId: v.ClassifyId,
  813. IsJoinPermission: v.IsJoinPermission,
  814. EdbInfoId: v.EdbInfoId,
  815. }
  816. }
  817. // 获取所有有权限的指标和分类
  818. permissionEdbIdList, permissionClassifyIdList, err := data_manage_permission.GetUserEdbAndClassifyPermissionList(sysUser.AdminId, 0, 0)
  819. if err != nil {
  820. br.Msg = "获取失败"
  821. br.ErrMsg = "获取所有有权限的指标和分类失败,Err:" + err.Error()
  822. return
  823. }
  824. for _, v := range edbList {
  825. // 数据权限
  826. edbItem, ok := edbClassifyPermissionMap[v.EdbInfoId]
  827. if !ok {
  828. continue
  829. }
  830. if currClassify, ok := classifyMap[edbItem.ClassifyId]; ok {
  831. v.HaveOperaAuth = data_manage_permission.CheckEdbPermissionByPermissionIdList(edbItem.IsJoinPermission, currClassify.IsJoinPermission, edbItem.EdbInfoId, edbItem.ClassifyId, permissionEdbIdList, permissionClassifyIdList)
  832. }
  833. }
  834. }
  835. // 图表当前分类的分类树
  836. classifyLevels := make([]string, 0)
  837. {
  838. list, e := data_manage.GetChartClassifyAllBySource(utils.CHART_SOURCE_CORRELATION)
  839. if e != nil {
  840. br.Msg = "获取失败"
  841. br.ErrMsg = fmt.Sprintf("获取图表分类失败, Err: %v", e)
  842. return
  843. }
  844. parents := data.GetChartClassifyParentRecursive(list, chartInfo.ChartClassifyId)
  845. sort.Slice(parents, func(i, j int) bool {
  846. return parents[i].Level < parents[i].Level
  847. })
  848. for _, v := range parents {
  849. classifyLevels = append(classifyLevels, v.UniqueCode)
  850. }
  851. }
  852. resp := new(data_manage.ChartInfoDetailResp)
  853. resp.ChartInfo = chartInfo
  854. resp.EdbInfoList = edbList
  855. resp.XEdbIdValue = xEdbIdValue
  856. resp.YDataList = yDataList
  857. resp.CorrelationChartInfo = correlationInfo
  858. resp.DataResp = dataResp
  859. resp.ClassifyLevels = classifyLevels
  860. br.Ret = 200
  861. br.Success = true
  862. br.Msg = "获取成功"
  863. br.Data = resp
  864. }
  865. // DetailFromUniqueCode
  866. // @Title 根据编码获取图表详情
  867. // @Description 根据编码获取图表详情接口
  868. // @Param UniqueCode query int true "图表唯一编码,如果是管理后台访问,传固定字符串:7c69b590249049942070ae9dcd5bf6dc"
  869. // @Param IsCache query bool true "是否走缓存,默认false"
  870. // @Success 200 {object} data_manage.ChartInfoDetailFromUniqueCodeResp
  871. // @router /chart_info/detail/from_unique_code [get]
  872. func (this *RangeChartChartInfoController) DetailFromUniqueCode() {
  873. br := new(models.BaseResponse).Init()
  874. defer func() {
  875. this.Data["json"] = br
  876. this.ServeJSON()
  877. }()
  878. sysUser := this.SysUser
  879. if sysUser == nil {
  880. br.Msg = "请登录"
  881. br.ErrMsg = "请登录,SysUser Is Empty"
  882. br.Ret = 408
  883. return
  884. }
  885. adminId := sysUser.AdminId
  886. uniqueCode := this.GetString("UniqueCode")
  887. if uniqueCode == "" {
  888. br.Msg = "参数错误"
  889. br.ErrMsg = "参数错误,uniqueCode is empty"
  890. return
  891. }
  892. //是否走缓存
  893. isCache, _ := this.GetBool("IsCache")
  894. resp := new(data_manage.ChartInfoDetailFromUniqueCodeResp)
  895. status := true
  896. chartInfo, err := data_manage.GetChartInfoViewByUniqueCode(uniqueCode)
  897. if err != nil {
  898. if err.Error() == utils.ErrNoRow() {
  899. status = false
  900. } else {
  901. br.Msg = "获取失败"
  902. br.ErrMsg = "获取图表信息失败,Err:" + err.Error()
  903. return
  904. }
  905. }
  906. if chartInfo == nil {
  907. status = false
  908. }
  909. if !status {
  910. endInfoList := make([]*data_manage.ChartEdbInfoMapping, 0)
  911. resp.EdbInfoList = endInfoList
  912. resp.ChartInfo = chartInfo
  913. resp.Status = false
  914. br.Ret = 200
  915. br.Success = true
  916. br.Msg = "获取成功"
  917. br.Data = resp
  918. return
  919. }
  920. //判断是否存在缓存,如果存在缓存,那么直接从缓存中获取
  921. key := data.GetChartInfoDataKey(chartInfo.ChartInfoId)
  922. if utils.Re == nil && isCache {
  923. if utils.Re == nil && utils.Rc.IsExist(key) {
  924. if d, e := utils.Rc.RedisBytes(key); e == nil {
  925. err := json.Unmarshal(d, &resp)
  926. if err == nil && resp != nil {
  927. // 这里跟当前用户相关的信息重新查询写入resp, 不使用缓存中的
  928. var myCond string
  929. var myPars []interface{}
  930. myCond += ` AND a.admin_id=? `
  931. myPars = append(myPars, adminId)
  932. myCond += ` AND a.chart_info_id=? `
  933. myPars = append(myPars, chartInfo.ChartInfoId)
  934. myList, err := data_manage.GetMyChartByCondition(myCond, myPars)
  935. if err != nil && err.Error() != utils.ErrNoRow() {
  936. br.Msg = "获取失败"
  937. br.ErrMsg = "获取我的图表信息失败,GetMyChartByCondition,Err:" + err.Error()
  938. return
  939. }
  940. resp.ChartInfo.IsAdd = false
  941. resp.ChartInfo.MyChartId = 0
  942. resp.ChartInfo.MyChartClassifyId = ""
  943. if myList != nil && len(myList) > 0 {
  944. resp.ChartInfo.IsAdd = true
  945. resp.ChartInfo.MyChartId = myList[0].MyChartId
  946. resp.ChartInfo.MyChartClassifyId = myList[0].MyChartClassifyId
  947. }
  948. br.Ret = 200
  949. br.Success = true
  950. br.Msg = "获取成功"
  951. br.Data = resp
  952. return
  953. }
  954. }
  955. }
  956. }
  957. resp, isOk, msg, errMsg := GetChartInfoDetailFromUniqueCode(chartInfo, isCache, sysUser)
  958. if !isOk {
  959. br.Msg = msg
  960. br.ErrMsg = errMsg
  961. return
  962. }
  963. if !isOk {
  964. br.Msg = msg
  965. br.ErrMsg = errMsg
  966. return
  967. }
  968. br.Ret = 200
  969. br.Success = true
  970. br.Msg = "获取成功"
  971. br.Data = resp
  972. }
  973. // Refresh
  974. // @Title 图表刷新接口
  975. // @Description 图表刷新接口
  976. // @Param ChartInfoId query int true "图表id"
  977. // @Param UniqueCode query string true "唯一code"
  978. // @Success Ret=200 刷新成功
  979. // @router /chart_info/refresh [get]
  980. func (this *RangeChartChartInfoController) Refresh() {
  981. br := new(models.BaseResponse).Init()
  982. defer func() {
  983. this.Data["json"] = br
  984. this.ServeJSON()
  985. }()
  986. sysUser := this.SysUser
  987. if sysUser == nil {
  988. br.Msg = "请登录"
  989. br.ErrMsg = "请登录,SysUser Is Empty"
  990. br.Ret = 408
  991. return
  992. }
  993. chartInfoId, _ := this.GetInt("ChartInfoId")
  994. uniqueCode := this.GetString("UniqueCode")
  995. if chartInfoId <= 0 && uniqueCode == `` {
  996. br.Msg = "参数错误"
  997. br.ErrMsg = "参数错误:chartInfoId:" + strconv.Itoa(chartInfoId) + ",UniqueCode:" + uniqueCode
  998. return
  999. }
  1000. var chartInfo *data_manage.ChartInfo
  1001. var err error
  1002. if chartInfoId > 0 {
  1003. chartInfo, err = data_manage.GetChartInfoById(chartInfoId)
  1004. } else {
  1005. chartInfo, err = data_manage.GetChartInfoByUniqueCode(uniqueCode)
  1006. }
  1007. if err != nil {
  1008. if err.Error() == utils.ErrNoRow() {
  1009. br.Msg = "图表已被删除,无需刷新"
  1010. br.ErrMsg = "获取指标信息失败,Err:" + err.Error()
  1011. return
  1012. }
  1013. br.Msg = "刷新失败"
  1014. br.ErrMsg = "获取图表信息失败,Err:" + err.Error()
  1015. return
  1016. }
  1017. // 刷新区间计算图表
  1018. isAsync, e := rangeServ.ChartInfoRefresh(chartInfo.ChartInfoId, chartInfo.UniqueCode)
  1019. if e != nil {
  1020. br.Msg = "刷新失败"
  1021. br.ErrMsg = "刷新区间计算图表失败, Err:" + err.Error()
  1022. return
  1023. }
  1024. // 多因子区间计算异步刷新, 前端提示
  1025. if isAsync {
  1026. br.Ret = 200
  1027. br.Success = true
  1028. br.Msg = "刷新时间较长, 请10分钟后查看"
  1029. return
  1030. }
  1031. //清除图表缓存
  1032. {
  1033. key := utils.HZ_CHART_LIB_DETAIL + chartInfo.UniqueCode
  1034. _ = utils.Rc.Delete(key)
  1035. }
  1036. br.Ret = 200
  1037. br.Success = true
  1038. br.Msg = "刷新成功"
  1039. }
  1040. // Copy
  1041. // @Title 复制并新增图表接口
  1042. // @Description 新增图表接口
  1043. // @Param request body data_manage.CopyAddChartInfoReq true "type json string"
  1044. // @Success 200 {object} data_manage.AddChartInfoResp
  1045. // @router /chart_info/copy [post]
  1046. func (this *RangeChartChartInfoController) Copy() {
  1047. br := new(models.BaseResponse).Init()
  1048. defer func() {
  1049. if br.ErrMsg == "" {
  1050. br.IsSendEmail = false
  1051. }
  1052. this.Data["json"] = br
  1053. this.ServeJSON()
  1054. }()
  1055. sysUser := this.SysUser
  1056. if sysUser == nil {
  1057. br.Msg = "请登录"
  1058. br.ErrMsg = "请登录,SysUser Is Empty"
  1059. br.Ret = 408
  1060. return
  1061. }
  1062. var req data_manage.CopyAddChartInfoReq
  1063. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  1064. if err != nil {
  1065. br.Msg = "参数解析异常!"
  1066. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  1067. return
  1068. }
  1069. if req.ChartInfoId <= 0 {
  1070. br.Msg = "参数有误"
  1071. br.ErrMsg = fmt.Sprintf("参数有误, ChartInfoId: %d", req.ChartInfoId)
  1072. return
  1073. }
  1074. req.ChartName = strings.TrimSpace(req.ChartName)
  1075. if req.ChartName == "" {
  1076. br.Msg = "请输入图表名称"
  1077. return
  1078. }
  1079. if req.ChartClassifyId <= 0 {
  1080. br.Msg = "请选择图表分类"
  1081. return
  1082. }
  1083. deleteCache := true
  1084. cacheKey := "CACHE_CHART_INFO_ADD_" + strconv.Itoa(sysUser.AdminId)
  1085. defer func() {
  1086. if deleteCache {
  1087. utils.Rc.Delete(cacheKey)
  1088. }
  1089. }()
  1090. if !utils.Rc.SetNX(cacheKey, 1, 30*time.Second) {
  1091. deleteCache = false
  1092. br.Msg = "系统处理中,请稍后重试!"
  1093. br.ErrMsg = "系统处理中,请稍后重试!" + sysUser.RealName + ";data:" + string(this.Ctx.Input.RequestBody)
  1094. return
  1095. }
  1096. chartSource := utils.CHART_SOURCE_CORRELATION
  1097. // 校验分类、图表名称
  1098. {
  1099. var cond string
  1100. var pars []interface{}
  1101. switch this.Lang {
  1102. case utils.EnLangVersion:
  1103. cond += " AND chart_name_en = ? AND source = ? "
  1104. default:
  1105. cond += " AND chart_name = ? AND source = ? "
  1106. }
  1107. pars = append(pars, req.ChartName, chartSource)
  1108. count, e := data_manage.GetChartInfoCountByCondition(cond, pars)
  1109. if e != nil {
  1110. br.Msg = "保存失败"
  1111. br.ErrMsg = fmt.Sprintf("获取同名图表失败, Err: %v", e)
  1112. return
  1113. }
  1114. if count > 0 {
  1115. br.Msg = "图表名称已存在, 请重新填写"
  1116. return
  1117. }
  1118. _, e = data_manage.GetChartClassifyById(req.ChartClassifyId)
  1119. if e != nil {
  1120. if e.Error() == utils.ErrNoRow() {
  1121. br.Msg = "分类不存在"
  1122. return
  1123. }
  1124. br.Msg = "保存失败"
  1125. br.ErrMsg = fmt.Sprintf("获取图表分类失败, Err: %v", e)
  1126. return
  1127. }
  1128. }
  1129. // 图表信息
  1130. originChart, e := data_manage.GetChartInfoById(req.ChartInfoId)
  1131. if e != nil {
  1132. if e.Error() == utils.ErrNoRow() {
  1133. br.Msg = "原图表不存在"
  1134. return
  1135. }
  1136. br.Msg = "保存失败"
  1137. br.ErrMsg = fmt.Sprintf("获取原图表信息失败, Err: %v", e)
  1138. return
  1139. }
  1140. if originChart.Source == utils.CHART_SOURCE_ROLLING_CORRELATION {
  1141. br.Msg = `滚动区间计算图不支持另存为`
  1142. br.IsSendEmail = false
  1143. return
  1144. }
  1145. // 区间计算图
  1146. originCorrelate := new(data_manage.ChartInfoCorrelation)
  1147. if e = originCorrelate.GetItemById(req.ChartInfoId); e != nil {
  1148. if e.Error() == utils.ErrNoRow() {
  1149. br.Msg = "原区间计算图表不存在"
  1150. return
  1151. }
  1152. br.Msg = "保存失败"
  1153. br.ErrMsg = fmt.Sprintf("获取原区间计算图表信息失败, Err: %v", e)
  1154. return
  1155. }
  1156. // 普通区间计算图表
  1157. chartInfo := new(data_manage.ChartInfo)
  1158. if originCorrelate.AnalysisMode != 1 {
  1159. multipleGraphConfigChartMapping, err := data_manage.GetMultipleGraphConfigChartMappingByChartId(req.ChartInfoId)
  1160. if err != nil && err.Error() != utils.ErrNoRow() {
  1161. br.Msg = `保存失败`
  1162. br.ErrMsg = "获取配置与图表的关联关系失败,ERR:" + err.Error()
  1163. return
  1164. }
  1165. correlationChartInfoReq := data_manage.CorrelationChartInfoReq{
  1166. LeadValue: originCorrelate.LeadValue,
  1167. LeadUnit: originCorrelate.LeadUnit,
  1168. CalculateValue: originCorrelate.CalculateValue,
  1169. CalculateUnit: originCorrelate.CalculateUnit,
  1170. BaseCalculateUnit: originCorrelate.BaseCalculateUnit,
  1171. BaseCalculateValue: originCorrelate.BaseCalculateValue,
  1172. StartDate: originCorrelate.StartDate.Format(utils.FormatDate),
  1173. EndDate: originCorrelate.EndDate.Format(utils.FormatDate),
  1174. EdbInfoIdList: []data_manage.CorrelationChartInfoEdbItemReq{
  1175. {
  1176. EdbInfoId: originCorrelate.EdbInfoIdFirst,
  1177. Name: "",
  1178. NameEn: "",
  1179. }, {
  1180. EdbInfoId: originCorrelate.EdbInfoIdSecond,
  1181. Name: "",
  1182. NameEn: "",
  1183. },
  1184. },
  1185. }
  1186. newChart, err, errMsg, isSendEmail := rangeServ.CopyChartInfo(multipleGraphConfigChartMapping.MultipleGraphConfigId, req.ChartClassifyId, req.ChartName, correlationChartInfoReq, originChart, sysUser, this.Lang)
  1187. chartInfo = newChart
  1188. if err != nil {
  1189. br.Msg = "保存失败"
  1190. if errMsg != `` {
  1191. br.Msg = errMsg
  1192. }
  1193. br.ErrMsg = err.Error()
  1194. br.IsSendEmail = isSendEmail
  1195. return
  1196. }
  1197. }
  1198. // 多因子区间计算图表
  1199. if originCorrelate.AnalysisMode == 1 {
  1200. chartInfo.ChartName = req.ChartName
  1201. chartInfo.ChartNameEn = req.ChartName
  1202. chartInfo.ChartClassifyId = req.ChartClassifyId
  1203. chartInfo.SysUserId = sysUser.AdminId
  1204. chartInfo.SysUserRealName = sysUser.RealName
  1205. chartInfo.CreateTime = time.Now()
  1206. chartInfo.ModifyTime = time.Now()
  1207. timestamp := strconv.FormatInt(time.Now().UnixNano(), 10)
  1208. chartInfo.UniqueCode = utils.MD5(utils.CHART_PREFIX + "_" + timestamp)
  1209. chartInfo.ChartType = originChart.ChartType
  1210. chartInfo.Calendar = originChart.Calendar
  1211. chartInfo.DateType = originChart.DateType
  1212. chartInfo.StartDate = originChart.StartDate
  1213. chartInfo.EndDate = originChart.EndDate
  1214. chartInfo.SeasonStartDate = originChart.StartDate
  1215. chartInfo.SeasonEndDate = originChart.EndDate
  1216. chartInfo.Disabled = originChart.Disabled
  1217. chartInfo.Source = originChart.Source
  1218. chartInfo.ChartThemeId = originChart.ChartThemeId
  1219. chartInfo.ExtraConfig = originChart.ExtraConfig
  1220. chartInfo.SourcesFrom = originChart.SourcesFrom
  1221. chartInfo.Instructions = originChart.Instructions
  1222. chartInfo.MarkersLines = originChart.MarkersLines
  1223. chartInfo.MarkersAreas = originChart.MarkersAreas
  1224. // 区间计算图
  1225. chartCorrelate := new(data_manage.ChartInfoCorrelation)
  1226. chartCorrelate.LeadValue = originCorrelate.LeadValue
  1227. chartCorrelate.LeadUnit = originCorrelate.LeadUnit
  1228. chartCorrelate.CalculateValue = originCorrelate.CalculateValue
  1229. chartCorrelate.CalculateUnit = originCorrelate.CalculateUnit
  1230. chartCorrelate.EdbInfoIdFirst = originCorrelate.EdbInfoIdFirst
  1231. chartCorrelate.AnalysisMode = 1
  1232. chartCorrelate.CreateTime = time.Now().Local()
  1233. chartCorrelate.ModifyTime = time.Now().Local()
  1234. // 图表指标关联
  1235. edbMappings := make([]*data_manage.ChartEdbMapping, 0)
  1236. {
  1237. mappings, e := data_manage.GetChartMappingList(req.ChartInfoId)
  1238. if e != nil {
  1239. br.Msg = "保存失败"
  1240. br.ErrMsg = fmt.Sprintf("获取图表指标关联失败, Err: %v", e)
  1241. return
  1242. }
  1243. for _, v := range mappings {
  1244. m := new(data_manage.ChartEdbMapping)
  1245. m.EdbInfoId = v.EdbInfoId
  1246. edbTimestamp := strconv.FormatInt(time.Now().UnixNano(), 10)
  1247. m.UniqueCode = utils.MD5(utils.CHART_PREFIX + "_" + edbTimestamp + "_" + strconv.Itoa(v.EdbInfoId))
  1248. m.IsOrder = true
  1249. m.IsAxis = 1
  1250. m.EdbInfoType = 1
  1251. m.Source = utils.CHART_SOURCE_CORRELATION
  1252. m.CreateTime = time.Now()
  1253. m.ModifyTime = time.Now()
  1254. edbMappings = append(edbMappings, m)
  1255. }
  1256. }
  1257. // 指标系列-图表关联
  1258. seriesIds := make([]int, 0)
  1259. originChartMappings := make([]*data_manage.FactorEdbSeriesChartMapping, 0)
  1260. {
  1261. ob := new(data_manage.FactorEdbSeriesChartMapping)
  1262. cond := fmt.Sprintf(" AND %s = ?", ob.Cols().ChartInfoId)
  1263. pars := make([]interface{}, 0)
  1264. pars = append(pars, req.ChartInfoId)
  1265. list, e := ob.GetItemsByCondition(cond, pars, []string{}, "")
  1266. if e != nil {
  1267. br.Msg = "保存失败"
  1268. br.ErrMsg = fmt.Sprintf("获取指标系列-图表关联失败, Err: %v", e)
  1269. return
  1270. }
  1271. for _, v := range list {
  1272. //seriesMappings = append(seriesMappings, &data_manage.FactorEdbSeriesChartMapping{
  1273. // FactorEdbSeriesId: v.FactorEdbSeriesId,
  1274. // ChartInfoId: req.ChartInfoId,
  1275. // EdbInfoId: v.EdbInfoId,
  1276. // Source: v.Source,
  1277. // CreateTime: time.Now().Local(),
  1278. // ModifyTime: time.Now().Local(),
  1279. //})
  1280. if !utils.InArrayByInt(seriesIds, v.FactorEdbSeriesId) {
  1281. seriesIds = append(seriesIds, v.FactorEdbSeriesId)
  1282. }
  1283. }
  1284. originChartMappings = list
  1285. }
  1286. if len(seriesIds) == 0 {
  1287. br.Msg = "保存失败"
  1288. br.ErrMsg = "源图表系列有误"
  1289. return
  1290. }
  1291. // 注意此处要复制一份系列及系列关联指标, 不可共用被复制图表的系列
  1292. originSeries := make([]*data_manage.FactorEdbSeries, 0)
  1293. {
  1294. ob := new(data_manage.FactorEdbSeries)
  1295. cond := fmt.Sprintf(" AND %s IN (%s)", ob.Cols().PrimaryId, utils.GetOrmInReplace(len(seriesIds)))
  1296. pars := make([]interface{}, 0)
  1297. pars = append(pars, seriesIds)
  1298. list, e := ob.GetItemsByCondition(cond, pars, []string{}, "")
  1299. if e != nil {
  1300. br.Msg = "保存失败"
  1301. br.ErrMsg = fmt.Sprintf("获取源图表系列失败, %v", e)
  1302. return
  1303. }
  1304. originSeries = list
  1305. }
  1306. originEdbSeriesMapping := make([]*data_manage.FactorEdbSeriesMapping, 0)
  1307. {
  1308. ob := new(data_manage.FactorEdbSeriesMapping)
  1309. cond := fmt.Sprintf(" AND %s IN (%s)", ob.Cols().FactorEdbSeriesId, utils.GetOrmInReplace(len(seriesIds)))
  1310. pars := make([]interface{}, 0)
  1311. pars = append(pars, seriesIds)
  1312. list, e := ob.GetItemsByCondition(cond, pars, []string{}, "")
  1313. if e != nil {
  1314. br.Msg = "保存失败"
  1315. br.ErrMsg = fmt.Sprintf("获取源图表系列指标失败, %v", e)
  1316. return
  1317. }
  1318. originEdbSeriesMapping = list
  1319. }
  1320. // 新增图表/区间计算图表/图表指标关联/指标系列图表关联
  1321. chartInfoId, seriesIdMap, e := data_manage.CreateMultiFactorCorrelationChartAndEdb(chartInfo, edbMappings, chartCorrelate, originChartMappings, true, originSeries, originEdbSeriesMapping)
  1322. if e != nil {
  1323. br.Msg = "保存失败"
  1324. br.ErrMsg = fmt.Sprintf("新增多因子区间计算图表失败, Err: %v", e)
  1325. return
  1326. }
  1327. // 如果原图表有设置图例-那么替换图例设置中的系列ID
  1328. if originChart.ExtraConfig != "" {
  1329. conf := new(data_manage.CorrelationChartInfoExtraConfig)
  1330. e = json.Unmarshal([]byte(originChart.ExtraConfig), &conf)
  1331. if e != nil {
  1332. br.Msg = "保存失败"
  1333. br.ErrMsg = fmt.Sprintf("解析图表额外配置失败, Err: %v", e)
  1334. return
  1335. }
  1336. for _, v := range conf.LegendConfig {
  1337. v.SeriesId = seriesIdMap[v.SeriesId]
  1338. }
  1339. b, e := json.Marshal(conf)
  1340. if e != nil {
  1341. br.Msg = "保存失败"
  1342. br.ErrMsg = fmt.Sprintf("图表来源JSON格式化失败, Err: %v", e)
  1343. return
  1344. }
  1345. chartInfo.ExtraConfig = string(b)
  1346. if e = chartInfo.Update([]string{"ExtraConfig"}); e != nil {
  1347. br.Msg = "保存失败"
  1348. br.ErrMsg = fmt.Sprintf("更新图例信息失败, %v", e)
  1349. return
  1350. }
  1351. }
  1352. go data.EsAddOrEditChartInfo(chartInfoId)
  1353. }
  1354. // 新增操作日志
  1355. {
  1356. chartLog := new(data_manage.ChartInfoLog)
  1357. chartLog.ChartInfoId = chartInfo.ChartInfoId
  1358. chartLog.ChartName = req.ChartName
  1359. chartLog.ChartClassifyId = req.ChartClassifyId
  1360. chartLog.SysUserId = sysUser.AdminId
  1361. chartLog.SysUserRealName = sysUser.RealName
  1362. chartLog.UniqueCode = chartInfo.UniqueCode
  1363. chartLog.CreateTime = time.Now()
  1364. chartLog.Content = string(this.Ctx.Input.RequestBody)
  1365. chartLog.Status = "复制区间计算图表"
  1366. chartLog.Method = this.Ctx.Input.URI()
  1367. go data_manage.AddChartInfoLog(chartLog)
  1368. }
  1369. br.Ret = 200
  1370. br.Success = true
  1371. br.Msg = "保存成功"
  1372. br.Data = data_manage.AddChartInfoResp{
  1373. ChartInfoId: chartInfo.ChartInfoId,
  1374. UniqueCode: chartInfo.UniqueCode,
  1375. ChartType: chartInfo.ChartType,
  1376. }
  1377. br.IsAddLog = true
  1378. }
  1379. // GetChartInfoDetailFromUniqueCode 根据编码获取图表详情
  1380. func GetChartInfoDetailFromUniqueCode(chartInfo *data_manage.ChartInfoView, isCache bool, sysUser *system.Admin) (resp *data_manage.ChartInfoDetailFromUniqueCodeResp, isOk bool, msg, errMsg string) {
  1381. resp = new(data_manage.ChartInfoDetailFromUniqueCodeResp)
  1382. adminId := sysUser.AdminId
  1383. // 指标数据map
  1384. edbClassifyPermissionMap := make(map[int]data_manage_permission.EdbClassifyPermission)
  1385. defer func() {
  1386. if isOk {
  1387. // 这里跟当前用户相关的信息重新查询写入resp, 不使用缓存中的
  1388. {
  1389. //判断是否加入我的图库
  1390. var myChartCondition string
  1391. var myChartPars []interface{}
  1392. myChartCondition += ` AND a.admin_id=? `
  1393. myChartPars = append(myChartPars, adminId)
  1394. myChartCondition += ` AND a.chart_info_id=? `
  1395. myChartPars = append(myChartPars, chartInfo.ChartInfoId)
  1396. myChartList, err := data_manage.GetMyChartByCondition(myChartCondition, myChartPars)
  1397. if err != nil && err.Error() != utils.ErrNoRow() {
  1398. msg = "获取失败"
  1399. errMsg = "获取我的图表信息失败,GetMyChartByCondition,Err:" + err.Error()
  1400. return
  1401. }
  1402. if myChartList != nil && len(myChartList) > 0 {
  1403. chartInfo.IsAdd = true
  1404. chartInfo.MyChartId = myChartList[0].MyChartId
  1405. chartInfo.MyChartClassifyId = myChartList[0].MyChartClassifyId
  1406. }
  1407. }
  1408. resp.ChartInfo.HaveOperaAuth = true
  1409. // 指标权限
  1410. {
  1411. classifyMap := make(map[int]*data_manage.EdbClassify)
  1412. // 分类
  1413. {
  1414. classifyIdList := make([]int, 0)
  1415. for _, v := range resp.EdbInfoList {
  1416. classifyIdList = append(classifyIdList, v.ClassifyId)
  1417. }
  1418. classifyList, tmpErr := data_manage.GetEdbClassifyByIdList(classifyIdList)
  1419. if tmpErr != nil {
  1420. errMsg = "获取分类列表失败,Err:" + tmpErr.Error()
  1421. return
  1422. }
  1423. for _, v := range classifyList {
  1424. classifyMap[v.ClassifyId] = v
  1425. }
  1426. }
  1427. // 指标
  1428. if len(edbClassifyPermissionMap) < 0 {
  1429. edbInfoIdList := make([]int, 0)
  1430. for _, v := range resp.EdbInfoList {
  1431. edbInfoIdList = append(edbInfoIdList, v.EdbInfoId)
  1432. }
  1433. edbInfoList, tmpErr := data_manage.GetEdbInfoByIdList(edbInfoIdList)
  1434. if tmpErr != nil {
  1435. errMsg = "获取指标列表失败,Err:" + tmpErr.Error()
  1436. return
  1437. }
  1438. for _, v := range edbInfoList {
  1439. edbClassifyPermissionMap[v.EdbInfoId] = data_manage_permission.EdbClassifyPermission{
  1440. ClassifyId: v.ClassifyId,
  1441. IsJoinPermission: v.IsJoinPermission,
  1442. EdbInfoId: v.EdbInfoId,
  1443. }
  1444. }
  1445. }
  1446. // 获取所有有权限的指标和分类
  1447. permissionEdbIdList, permissionClassifyIdList, err := data_manage_permission.GetUserEdbAndClassifyPermissionList(sysUser.AdminId, 0, 0)
  1448. if err != nil {
  1449. errMsg = "获取所有有权限的指标和分类失败,Err:" + err.Error()
  1450. return
  1451. }
  1452. for _, v := range resp.EdbInfoList {
  1453. // 数据权限
  1454. edbItem, ok := edbClassifyPermissionMap[v.EdbInfoId]
  1455. if !ok {
  1456. continue
  1457. }
  1458. if currClassify, ok := classifyMap[edbItem.ClassifyId]; ok {
  1459. v.HaveOperaAuth = data_manage_permission.CheckEdbPermissionByPermissionIdList(edbItem.IsJoinPermission, currClassify.IsJoinPermission, edbItem.EdbInfoId, edbItem.ClassifyId, permissionEdbIdList, permissionClassifyIdList)
  1460. }
  1461. }
  1462. }
  1463. }
  1464. }()
  1465. //判断是否存在缓存,如果存在缓存,那么直接从缓存中获取
  1466. key := data.GetChartInfoDataKey(chartInfo.ChartInfoId)
  1467. if utils.Re == nil && isCache {
  1468. if utils.Re == nil && utils.Rc.IsExist(key) {
  1469. if chartData, err1 := utils.Rc.RedisBytes(key); err1 == nil {
  1470. err := json.Unmarshal(chartData, &resp)
  1471. if err != nil || resp == nil {
  1472. return
  1473. }
  1474. isOk = true
  1475. fmt.Println("source redis")
  1476. return
  1477. }
  1478. }
  1479. }
  1480. // 获取主题样式
  1481. chartTheme, err := data.GetChartThemeConfig(chartInfo.ChartThemeId, utils.CHART_SOURCE_DEFAULT, 1)
  1482. if err != nil {
  1483. msg = "获取失败"
  1484. errMsg = "获取主题信息失败,Err:" + err.Error()
  1485. return
  1486. }
  1487. chartInfo.ChartThemeStyle = chartTheme.Config
  1488. chartInfo.ChartThemeId = chartTheme.ChartThemeId
  1489. chartInfoId := chartInfo.ChartInfoId
  1490. // 区间计算图表信息
  1491. correlationChart := new(data_manage.ChartInfoCorrelation)
  1492. if e := correlationChart.GetItemById(chartInfoId); e != nil {
  1493. msg = "获取失败"
  1494. errMsg = "获取图表区间计算信息失败, Err:" + e.Error()
  1495. return
  1496. }
  1497. // 获取指标信息
  1498. edbInfoMappingA, e := data_manage.GetChartEdbMappingByEdbInfoId(correlationChart.EdbInfoIdFirst)
  1499. if e != nil {
  1500. msg = "获取失败"
  1501. errMsg = "获取区间计算图表, A指标mapping信息失败, Err:" + e.Error()
  1502. return
  1503. }
  1504. edbInfoMappingB := new(data_manage.ChartEdbInfoMapping)
  1505. // 非多因子
  1506. if correlationChart.AnalysisMode != 1 {
  1507. edbInfoMappingB, e = data_manage.GetChartEdbMappingByEdbInfoId(correlationChart.EdbInfoIdSecond)
  1508. if e != nil {
  1509. msg = "获取失败"
  1510. errMsg = "获取区间计算图表, B指标mapping信息失败, Err:" + e.Error()
  1511. return
  1512. }
  1513. }
  1514. var dataResp interface{} // 绘图数据返回(目前是滚动区间计算的图)
  1515. var xEdbIdValue []int
  1516. var yDataList []data_manage.YData
  1517. if correlationChart.AnalysisMode != 1 {
  1518. switch chartInfo.Source {
  1519. case utils.CHART_SOURCE_CORRELATION: // 区间计算图
  1520. moveUnitDays, ok := utils.FrequencyDaysMap[correlationChart.CalculateUnit]
  1521. if !ok {
  1522. msg = "错误的分析周期"
  1523. errMsg = "区间计算图表数据有误"
  1524. return
  1525. }
  1526. startDate := time.Now().AddDate(0, 0, -correlationChart.CalculateValue*moveUnitDays).Format(utils.FormatDate)
  1527. endDate := time.Now().Format(utils.FormatDate)
  1528. xEdbIdValue, yDataList, e = rangeServ.GetChartDataByEdbInfo(edbInfoMappingA, edbInfoMappingB, correlationChart.LeadValue, correlationChart.LeadUnit, startDate, endDate, chartInfo.ExtraConfig)
  1529. if e != nil {
  1530. msg = "获取失败"
  1531. errMsg = "获取区间计算图表, 图表计算值失败, Err:" + e.Error()
  1532. return
  1533. }
  1534. case utils.CHART_SOURCE_ROLLING_CORRELATION: // 滚动区间计算图
  1535. startDate, endDate := utils.GetDateByDateType(correlationChart.DateType, correlationChart.StartDate.Format(utils.FormatDate), correlationChart.EndDate.Format(utils.FormatDate))
  1536. dataResp, e = rangeServ.GetRollingCorrelationChartDataByEdbInfo(edbInfoMappingA, edbInfoMappingB, correlationChart.LeadValue, correlationChart.LeadUnit, correlationChart.CalculateValue, correlationChart.CalculateUnit, startDate, endDate, chartInfo.ChartName, chartInfo.ChartNameEn)
  1537. if e != nil {
  1538. msg = "获取失败"
  1539. errMsg = "获取滚动区间计算图表, 图表计算值失败, Err:" + e.Error()
  1540. return
  1541. }
  1542. }
  1543. } else {
  1544. xEdbIdValue, yDataList, e = rangeServ.GetFactorChartDataByChartId(chartInfoId, chartInfo.ExtraConfig)
  1545. if e != nil {
  1546. msg = "获取失败"
  1547. errMsg = "获取区间计算图表, 图表计算值失败, Err:" + e.Error()
  1548. return
  1549. }
  1550. }
  1551. // 完善指标信息
  1552. edbList := make([]*data_manage.ChartEdbInfoMapping, 0)
  1553. if correlationChart.AnalysisMode != 1 {
  1554. edbList, e = rangeServ.GetChartEdbInfoFormat(chartInfo.ChartInfoId, edbInfoMappingA, edbInfoMappingB)
  1555. if e != nil {
  1556. msg = "获取失败"
  1557. errMsg = "获取区间计算图表, 完善指标信息失败, Err:" + e.Error()
  1558. return
  1559. }
  1560. } else {
  1561. // 多因子指标, 获取图表引用到的系列指标(去重后)
  1562. edbList, e = data_manage.GetChartEdbMappingListByIdList([]int{chartInfoId})
  1563. if e != nil {
  1564. msg = "获取失败"
  1565. errMsg = "获取图表引用系列指标失败, Err:" + e.Error()
  1566. return
  1567. }
  1568. }
  1569. correlationInfo := new(data_manage.CorrelationInfo)
  1570. correlationInfo.LeadValue = correlationChart.LeadValue
  1571. correlationInfo.LeadUnit = correlationChart.LeadUnit
  1572. correlationInfo.CalculateValue = correlationChart.CalculateValue
  1573. correlationInfo.CalculateUnit = correlationChart.CalculateUnit
  1574. correlationInfo.StartDate = correlationChart.StartDate.Format(utils.FormatDate)
  1575. correlationInfo.EndDate = correlationChart.EndDate.Format(utils.FormatDate)
  1576. correlationInfo.LeadValue = correlationChart.LeadValue
  1577. correlationInfo.EdbInfoIdFirst = correlationChart.EdbInfoIdFirst
  1578. correlationInfo.EdbInfoIdSecond = correlationChart.EdbInfoIdSecond
  1579. correlationInfo.AnalysisMode = correlationChart.AnalysisMode
  1580. if chartInfoId > 0 && chartInfo != nil {
  1581. //判断是否加入我的图库
  1582. {
  1583. var myChartCondition string
  1584. var myChartPars []interface{}
  1585. myChartCondition += ` AND a.admin_id=? `
  1586. myChartPars = append(myChartPars, sysUser.AdminId)
  1587. myChartCondition += ` AND a.chart_info_id=? `
  1588. myChartPars = append(myChartPars, chartInfo.ChartInfoId)
  1589. myChartList, err := data_manage.GetMyChartByCondition(myChartCondition, myChartPars)
  1590. if err != nil && err.Error() != utils.ErrNoRow() {
  1591. msg = "获取失败"
  1592. errMsg = "获取我的图表信息失败,GetMyChartByCondition,Err:" + err.Error()
  1593. return
  1594. }
  1595. if myChartList != nil && len(myChartList) > 0 {
  1596. chartInfo.IsAdd = true
  1597. chartInfo.MyChartId = myChartList[0].MyChartId
  1598. chartInfo.MyChartClassifyId = myChartList[0].MyChartClassifyId
  1599. }
  1600. }
  1601. }
  1602. //图表操作权限
  1603. chartInfo.IsEdit = data.CheckOpChartPermission(sysUser, chartInfo.SysUserId, true)
  1604. //判断是否需要展示英文标识
  1605. chartInfo.IsEnChart = data.CheckIsEnChart(chartInfo.ChartNameEn, edbList[0:1], chartInfo.Source, chartInfo.ChartType)
  1606. chartInfo.UnitEn = edbInfoMappingA.UnitEn
  1607. isSaveAs := true
  1608. if chartInfo.Source == utils.CHART_SOURCE_ROLLING_CORRELATION {
  1609. isSaveAs = false
  1610. }
  1611. // 另存为
  1612. chartInfo.Button = data_manage.ChartViewButton{
  1613. IsEdit: chartInfo.IsEdit,
  1614. IsEnChart: chartInfo.IsEnChart,
  1615. IsAdd: chartInfo.IsAdd,
  1616. IsCopy: isSaveAs,
  1617. IsSetName: chartInfo.IsSetName,
  1618. }
  1619. // 图表的指标来源
  1620. sourceNameList, sourceNameEnList := data.GetEdbSourceByEdbInfoIdList(edbList)
  1621. chartInfo.ChartSource = strings.Join(sourceNameList, ",")
  1622. chartInfo.ChartSourceEn = strings.Join(sourceNameEnList, ",")
  1623. resp.ChartInfo = chartInfo
  1624. resp.EdbInfoList = edbList
  1625. resp.XEdbIdValue = xEdbIdValue
  1626. resp.YDataList = yDataList
  1627. resp.CorrelationChartInfo = correlationInfo
  1628. resp.DataResp = dataResp
  1629. resp.Status = true
  1630. // 遍历到校验map
  1631. for _, v := range edbList {
  1632. edbClassifyPermissionMap[v.EdbInfoId] = data_manage_permission.EdbClassifyPermission{
  1633. ClassifyId: v.ClassifyId,
  1634. IsJoinPermission: v.IsJoinPermission,
  1635. EdbInfoId: v.EdbInfoId,
  1636. }
  1637. }
  1638. // 将数据加入缓存
  1639. if utils.Re == nil {
  1640. d, _ := json.Marshal(resp)
  1641. _ = utils.Rc.Put(key, d, 2*time.Hour)
  1642. }
  1643. isOk = true
  1644. return
  1645. }
  1646. // EnInfoEdit
  1647. // @Title 编辑图表英文信息接口
  1648. // @Description 编辑图表英文信息接口
  1649. // @Param request body data_manage.EditChartEnInfoReq true "type json string"
  1650. // @Success Ret=200 编辑成功
  1651. // @router /chart_info/en/edit [post]
  1652. func (this *RangeChartChartInfoController) EnInfoEdit() {
  1653. br := new(models.BaseResponse).Init()
  1654. defer func() {
  1655. this.Data["json"] = br
  1656. this.ServeJSON()
  1657. }()
  1658. sysUser := this.SysUser
  1659. if sysUser == nil {
  1660. br.Msg = "请登录"
  1661. br.ErrMsg = "请登录,SysUser Is Empty"
  1662. br.Ret = 408
  1663. return
  1664. }
  1665. var req request.EditChartEnInfoReq
  1666. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  1667. if err != nil {
  1668. br.Msg = "参数解析异常!"
  1669. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  1670. return
  1671. }
  1672. req.ChartNameEn = strings.Trim(req.ChartNameEn, " ")
  1673. if req.ChartInfoId <= 0 {
  1674. br.Msg = "请选择图表"
  1675. return
  1676. }
  1677. if req.ChartNameEn == "" {
  1678. br.Msg = "请输入英文图表名称"
  1679. return
  1680. }
  1681. chartItem, _, tips, e := rangeServ.GetChartAndCorrelationInfo(req.ChartInfoId)
  1682. if e != nil {
  1683. if tips != "" {
  1684. br.Msg = tips
  1685. return
  1686. }
  1687. br.Msg = "操作失败"
  1688. br.ErrMsg = "获取图表信息失败, Err: " + e.Error()
  1689. return
  1690. }
  1691. var condition string
  1692. var pars []interface{}
  1693. condition += " AND chart_info_id <> ? AND chart_name_en = ? AND source = ?"
  1694. pars = append(pars, req.ChartInfoId, req.ChartNameEn, utils.CHART_SOURCE_CORRELATION)
  1695. existItem, err := data_manage.GetChartInfoByCondition(condition, pars)
  1696. if err != nil {
  1697. if err.Error() != utils.ErrNoRow() {
  1698. br.Msg = "判断英文图表名称是否存在失败"
  1699. br.ErrMsg = "判断英文图表名称是否存在失败,Err:" + err.Error()
  1700. return
  1701. }
  1702. }
  1703. if err == nil && existItem.ChartInfoId > 0 {
  1704. br.Msg = existItem.ChartName + ":" + req.ChartNameEn + "图表名称已存在"
  1705. return
  1706. }
  1707. chartItem.ChartNameEn = req.ChartNameEn
  1708. chartItem.ModifyTime = time.Now().Local()
  1709. if e := chartItem.Update([]string{"ChartNameEn", "ModifyTime"}); e != nil {
  1710. br.Msg = "操作失败"
  1711. br.ErrMsg = "更新图表信息失败, Err: " + e.Error()
  1712. return
  1713. }
  1714. //添加es数据
  1715. go data.EsAddOrEditChartInfo(chartItem.ChartInfoId)
  1716. //修改my eta es数据
  1717. go data.EsAddOrEditMyChartInfoByChartInfoId(chartItem.ChartInfoId)
  1718. //指标 修改es信息
  1719. //go data.AddOrEditEdbInfoToEs(edbInfo.EdbInfoId)
  1720. //新增操作日志
  1721. {
  1722. chartLog := new(data_manage.ChartInfoLog)
  1723. chartLog.ChartName = chartItem.ChartName
  1724. chartLog.ChartInfoId = req.ChartInfoId
  1725. chartLog.ChartClassifyId = chartItem.ChartClassifyId
  1726. chartLog.SysUserId = sysUser.AdminId
  1727. chartLog.SysUserRealName = sysUser.RealName
  1728. chartLog.UniqueCode = chartItem.UniqueCode
  1729. chartLog.CreateTime = time.Now()
  1730. chartLog.Content = string(this.Ctx.Input.RequestBody)
  1731. chartLog.Status = "编辑区间计算图表英文信息"
  1732. chartLog.Method = this.Ctx.Input.URL()
  1733. go data_manage.AddChartInfoLog(chartLog)
  1734. }
  1735. // 清除缓存
  1736. if utils.Re == nil && utils.Rc != nil {
  1737. _ = utils.Rc.Delete(utils.HZ_CHART_LIB_DETAIL + chartItem.UniqueCode) //图表分享链接缓存
  1738. _ = utils.Rc.Delete(data.GetChartInfoDataKey(req.ChartInfoId))
  1739. }
  1740. br.Ret = 200
  1741. br.Success = true
  1742. br.Msg = "编辑成功"
  1743. br.IsAddLog = true
  1744. }
  1745. // SearchByEs
  1746. // @Title 图表模糊搜索(从es获取)
  1747. // @Description 图表模糊搜索(从es获取)
  1748. // @Param Keyword query string true "图表名称"
  1749. // @Param IsShowMe query bool true "是否只看我的,true、false"
  1750. // @Param Source query int true "来源,3:区间计算,4:滚动区间计算,默认0:全部"
  1751. // @Success 200 {object} data_manage.ChartInfo
  1752. // @router /chart_info/search_by_es [get]
  1753. func (this *RangeChartChartInfoController) SearchByEs() {
  1754. br := new(models.BaseResponse).Init()
  1755. defer func() {
  1756. this.Data["json"] = br
  1757. this.ServeJSON()
  1758. }()
  1759. sysUser := this.SysUser
  1760. if sysUser == nil {
  1761. br.Msg = "请登录"
  1762. br.ErrMsg = "请登录,SysUser Is Empty"
  1763. br.Ret = 408
  1764. return
  1765. }
  1766. pageSize, _ := this.GetInt("PageSize")
  1767. currentIndex, _ := this.GetInt("CurrentIndex")
  1768. var startSize int
  1769. if pageSize <= 0 {
  1770. pageSize = utils.PageSize20
  1771. }
  1772. if currentIndex <= 0 {
  1773. currentIndex = 1
  1774. }
  1775. startSize = paging.StartIndex(currentIndex, pageSize)
  1776. keyword := this.GetString("Keyword")
  1777. //只看我的
  1778. isShowMe, _ := this.GetBool("IsShowMe")
  1779. showSysId := 0
  1780. if isShowMe {
  1781. showSysId = sysUser.AdminId
  1782. }
  1783. source, _ := this.GetInt("Source")
  1784. sourceList := make([]int, 0)
  1785. if source <= 0 {
  1786. sourceList = append(sourceList, utils.CHART_SOURCE_CORRELATION, utils.CHART_SOURCE_ROLLING_CORRELATION)
  1787. } else {
  1788. sourceList = append(sourceList, source)
  1789. }
  1790. var searchList []*data_manage.ChartInfo
  1791. var total int64
  1792. var err error
  1793. // 获取当前账号的不可见指标
  1794. noPermissionChartIdList := make([]int, 0)
  1795. {
  1796. obj := data_manage.EdbInfoNoPermissionAdmin{}
  1797. confList, err := obj.GetAllChartListByAdminId(this.SysUser.AdminId)
  1798. if err != nil && err.Error() != utils.ErrNoRow() {
  1799. br.Msg = "获取失败"
  1800. br.ErrMsg = "获取不可见指标配置数据失败,Err:" + err.Error()
  1801. return
  1802. }
  1803. for _, v := range confList {
  1804. noPermissionChartIdList = append(noPermissionChartIdList, v.ChartInfoId)
  1805. }
  1806. }
  1807. if keyword != "" {
  1808. searchList, total, err = data.EsSearchChartInfo(keyword, showSysId, sourceList, noPermissionChartIdList, startSize, pageSize)
  1809. } else {
  1810. total, searchList, err = data_manage.ChartInfoSearchByEmptyKeyWord(showSysId, sourceList, noPermissionChartIdList, startSize, pageSize)
  1811. if err != nil && err.Error() != utils.ErrNoRow() {
  1812. br.Msg = "获取失败"
  1813. br.ErrMsg = "获取图表信息失败,Err:" + err.Error()
  1814. return
  1815. }
  1816. }
  1817. finalList := make([]*data_manage.ChartInfoMore, 0)
  1818. if len(searchList) > 0 {
  1819. chartInfoIds := ""
  1820. chartEdbMap := make(map[int][]*data_manage.ChartEdbInfoMapping)
  1821. for _, v := range searchList {
  1822. chartInfoIds += strconv.Itoa(v.ChartInfoId) + ","
  1823. }
  1824. if chartInfoIds != "" {
  1825. chartInfoIds = strings.Trim(chartInfoIds, ",")
  1826. //判断是否需要展示英文标识
  1827. edbList, e := data_manage.GetChartEdbMappingListByChartInfoIds(chartInfoIds)
  1828. if e != nil {
  1829. br.Msg = "获取失败"
  1830. br.ErrMsg = "获取图表,指标信息失败,Err:" + e.Error()
  1831. return
  1832. }
  1833. for _, v := range edbList {
  1834. chartEdbMap[v.ChartInfoId] = append(chartEdbMap[v.ChartInfoId], v)
  1835. }
  1836. }
  1837. for _, v := range searchList {
  1838. tmp := new(data_manage.ChartInfoMore)
  1839. tmp.ChartInfo = *v
  1840. // 图表数据权限
  1841. tmp.HaveOperaAuth = true
  1842. //判断是否需要展示英文标识
  1843. if edbTmpList, ok := chartEdbMap[v.ChartInfoId]; ok {
  1844. tmp.IsEnChart = data.CheckIsEnChart(v.ChartNameEn, edbTmpList, v.Source, v.ChartType)
  1845. }
  1846. finalList = append(finalList, tmp)
  1847. }
  1848. }
  1849. //新增搜索词记录
  1850. {
  1851. searchKeyword := new(data_manage.SearchKeyword)
  1852. searchKeyword.KeyWord = keyword
  1853. searchKeyword.CreateTime = time.Now()
  1854. go data_manage.AddSearchKeyword(searchKeyword)
  1855. }
  1856. page := paging.GetPaging(currentIndex, pageSize, int(total))
  1857. resp := data_manage.ChartInfoListByEsResp{
  1858. Paging: page,
  1859. List: finalList,
  1860. }
  1861. br.Ret = 200
  1862. br.Success = true
  1863. br.Msg = "获取成功"
  1864. br.Data = resp
  1865. }
  1866. // Newest
  1867. // @Title 获取最新图表接口
  1868. // @Description 获取最新图表接口
  1869. // @Success 200 {object} data_manage.ChartInfoList
  1870. // @router /chart_info/newest [get]
  1871. func (this *RangeChartChartInfoController) Newest() {
  1872. br := new(models.BaseResponse).Init()
  1873. defer func() {
  1874. this.Data["json"] = br
  1875. this.ServeJSON()
  1876. }()
  1877. item, err := data_manage.GetChartInfoByNewest(utils.CHART_SOURCE_CORRELATION)
  1878. if err != nil && err.Error() != utils.ErrNoRow() {
  1879. br.Msg = "获取最新图表失败"
  1880. br.ErrMsg = "获取最新图表失败,Err:" + err.Error()
  1881. return
  1882. }
  1883. br.Ret = 200
  1884. br.Success = true
  1885. br.Msg = "获取成功"
  1886. br.Data = item
  1887. }
  1888. // BaseInfoEdit
  1889. // @Title 编辑图表基础信息接口
  1890. // @Description 编辑图表基础信息接口
  1891. // @Param request body data_manage.EditChartInfoBaseReq true "type json string"
  1892. // @Success Ret=200 编辑成功
  1893. // @router /chart_info/base/edit [post]
  1894. func (this *RangeChartChartInfoController) BaseInfoEdit() {
  1895. br := new(models.BaseResponse).Init()
  1896. defer func() {
  1897. this.Data["json"] = br
  1898. this.ServeJSON()
  1899. }()
  1900. sysUser := this.SysUser
  1901. if sysUser == nil {
  1902. br.Msg = "请登录"
  1903. br.ErrMsg = "请登录,SysUser Is Empty"
  1904. br.Ret = 408
  1905. return
  1906. }
  1907. var req request.EditChartInfoBaseReq
  1908. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  1909. if err != nil {
  1910. br.Msg = "参数解析异常!"
  1911. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  1912. return
  1913. }
  1914. req.ChartName = strings.Trim(req.ChartName, " ")
  1915. if req.ChartInfoId <= 0 {
  1916. br.Msg = "请选择图表"
  1917. return
  1918. }
  1919. if req.ChartName == "" {
  1920. br.Msg = "请输入图表名称"
  1921. return
  1922. }
  1923. chartItem, _, tips, e := rangeServ.GetChartAndCorrelationInfo(req.ChartInfoId)
  1924. if e != nil {
  1925. if tips != "" {
  1926. br.Msg = tips
  1927. return
  1928. }
  1929. br.Msg = "操作失败"
  1930. br.ErrMsg = "获取图表信息失败, Err: " + e.Error()
  1931. return
  1932. }
  1933. var condition string
  1934. var pars []interface{}
  1935. condition += " AND chart_info_id <> ? AND source = ? "
  1936. pars = append(pars, req.ChartInfoId, utils.CHART_SOURCE_CORRELATION)
  1937. switch this.Lang {
  1938. case utils.EnLangVersion:
  1939. condition += " AND chart_name_en = ? "
  1940. default:
  1941. condition += " AND chart_name = ? "
  1942. }
  1943. pars = append(pars, req.ChartName)
  1944. existItem, err := data_manage.GetChartInfoByCondition(condition, pars)
  1945. if err != nil {
  1946. if err.Error() != utils.ErrNoRow() {
  1947. br.Msg = "判断英文图表名称是否存在失败"
  1948. br.ErrMsg = "判断英文图表名称是否存在失败,Err:" + err.Error()
  1949. return
  1950. }
  1951. }
  1952. if err == nil && existItem.ChartInfoId > 0 {
  1953. br.Msg = existItem.ChartName + ":" + req.ChartName + "图表名称已存在"
  1954. return
  1955. }
  1956. switch this.Lang {
  1957. case utils.EnLangVersion:
  1958. chartItem.ChartNameEn = req.ChartName
  1959. default:
  1960. chartItem.ChartName = req.ChartName
  1961. }
  1962. chartItem.ModifyTime = time.Now().Local()
  1963. if e := chartItem.Update([]string{"ChartName", "ChartNameEn", "ModifyTime"}); e != nil {
  1964. br.Msg = "操作失败"
  1965. br.ErrMsg = "更新图表信息失败, Err: " + e.Error()
  1966. return
  1967. }
  1968. //添加es数据
  1969. go data.EsAddOrEditChartInfo(chartItem.ChartInfoId)
  1970. //修改my eta es数据
  1971. go data.EsAddOrEditMyChartInfoByChartInfoId(chartItem.ChartInfoId)
  1972. //新增操作日志
  1973. {
  1974. chartLog := new(data_manage.ChartInfoLog)
  1975. chartLog.ChartName = chartItem.ChartName
  1976. chartLog.ChartInfoId = req.ChartInfoId
  1977. chartLog.ChartClassifyId = chartItem.ChartClassifyId
  1978. chartLog.SysUserId = sysUser.AdminId
  1979. chartLog.SysUserRealName = sysUser.RealName
  1980. chartLog.UniqueCode = chartItem.UniqueCode
  1981. chartLog.CreateTime = time.Now()
  1982. chartLog.Content = string(this.Ctx.Input.RequestBody)
  1983. chartLog.Status = "编辑区间计算图表基础信息"
  1984. chartLog.Method = this.Ctx.Input.URL()
  1985. go data_manage.AddChartInfoLog(chartLog)
  1986. }
  1987. // 清除缓存
  1988. if utils.Re == nil && utils.Rc != nil {
  1989. _ = utils.Rc.Delete(utils.HZ_CHART_LIB_DETAIL + chartItem.UniqueCode) //图表分享链接缓存
  1990. _ = utils.Rc.Delete(data.GetChartInfoDataKey(req.ChartInfoId))
  1991. }
  1992. br.Ret = 200
  1993. br.Success = true
  1994. br.Msg = "编辑成功"
  1995. br.IsAddLog = true
  1996. }
  1997. // MultiFactorAdd
  1998. // @Title 多因子图表-新增
  1999. // @Description 多因子图表-新增
  2000. // @Param request body request.CorrelationChartMultiFactorSaveReq true "type json string"
  2001. // @Success Ret=200 返回图表id
  2002. // @router /chart_info/multi_factor/add [post]
  2003. func (this *RangeChartChartInfoController) MultiFactorAdd() {
  2004. br := new(models.BaseResponse).Init()
  2005. defer func() {
  2006. if br.ErrMsg == "" {
  2007. br.IsSendEmail = false
  2008. }
  2009. this.Data["json"] = br
  2010. this.ServeJSON()
  2011. }()
  2012. sysUser := this.SysUser
  2013. if sysUser == nil {
  2014. br.Msg = "请登录"
  2015. br.ErrMsg = "请登录,SysUser Is Empty"
  2016. br.Ret = 408
  2017. return
  2018. }
  2019. var req request.CorrelationChartMultiFactorSaveReq
  2020. if e := json.Unmarshal(this.Ctx.Input.RequestBody, &req); e != nil {
  2021. br.Msg = "参数解析异常!"
  2022. br.ErrMsg = fmt.Sprintf("参数解析失败, Err: %v", e)
  2023. return
  2024. }
  2025. // 此方法只加多因子图表
  2026. if req.AnalysisMode != 1 {
  2027. br.Msg = "分析模式有误"
  2028. br.ErrMsg = fmt.Sprintf("分析模式有误, Mode: %d", req.AnalysisMode)
  2029. return
  2030. }
  2031. if req.BaseEdbInfoId <= 0 {
  2032. br.Msg = "请选择标的指标"
  2033. return
  2034. }
  2035. if len(req.FactorCorrelation.SeriesIds) == 0 {
  2036. br.Msg = "请选择因子指标系列"
  2037. return
  2038. }
  2039. if len(req.FactorCorrelation.SeriesEdb) == 0 {
  2040. br.Msg = "请选择因子指标"
  2041. return
  2042. }
  2043. if req.FactorCorrelation.LeadValue <= 0 {
  2044. br.Msg = "分析周期不允许设置为负数或0"
  2045. return
  2046. }
  2047. if req.FactorCorrelation.LeadUnit == "" {
  2048. br.Msg = "请选择分析周期频度"
  2049. return
  2050. }
  2051. leadUnitDays, ok := utils.FrequencyDaysMap[req.FactorCorrelation.LeadUnit]
  2052. if !ok {
  2053. br.Msg = "错误的分析周期频度"
  2054. br.ErrMsg = fmt.Sprintf("分析周期频度有误: %s", req.FactorCorrelation.LeadUnit)
  2055. return
  2056. }
  2057. if req.FactorCorrelation.CalculateUnit == "" {
  2058. br.Msg = "请选择计算窗口频度"
  2059. return
  2060. }
  2061. calculateUnitDays, ok := utils.FrequencyDaysMap[req.FactorCorrelation.CalculateUnit]
  2062. if !ok {
  2063. br.Msg = "错误的计算窗口频度"
  2064. br.ErrMsg = fmt.Sprintf("计算窗口频度有误: %s", req.FactorCorrelation.CalculateUnit)
  2065. return
  2066. }
  2067. leadDays := 2 * req.FactorCorrelation.LeadValue * leadUnitDays
  2068. calculateDays := req.FactorCorrelation.CalculateValue * calculateUnitDays
  2069. if calculateDays < leadDays {
  2070. br.Msg = "计算窗口必须≥2*分析周期"
  2071. return
  2072. }
  2073. req.ChartName = strings.TrimSpace(req.ChartName)
  2074. if req.ChartName == "" {
  2075. br.Msg = "请输入图表名称"
  2076. return
  2077. }
  2078. if req.ClassifyId <= 0 {
  2079. br.Msg = "请选择图表分类"
  2080. return
  2081. }
  2082. cacheKey := "CACHE_CORRELATION_CHART_INFO_ADD_" + strconv.Itoa(sysUser.AdminId)
  2083. if !utils.Rc.SetNX(cacheKey, 1, 30*time.Second) {
  2084. br.Msg = "系统处理中, 请稍后重试"
  2085. return
  2086. }
  2087. defer func() {
  2088. _ = utils.Rc.Delete(cacheKey)
  2089. }()
  2090. chartType := 9
  2091. chartSource := utils.CHART_SOURCE_CORRELATION
  2092. startDate := time.Now().AddDate(0, 0, -calculateDays).Format(utils.FormatDate)
  2093. endDate := time.Now().Format(utils.FormatDate)
  2094. mappingEdbIds := make([]int, 0) // 该图表关联的指标(去除系列中重复指标之后的)
  2095. {
  2096. exists := make(map[int]bool)
  2097. for _, v := range req.FactorCorrelation.SeriesEdb {
  2098. if !exists[v.EdbInfoId] {
  2099. mappingEdbIds = append(mappingEdbIds, v.EdbInfoId)
  2100. }
  2101. }
  2102. }
  2103. // 校验分类、图表名称
  2104. {
  2105. var cond string
  2106. var pars []interface{}
  2107. switch this.Lang {
  2108. case utils.EnLangVersion:
  2109. cond += " AND chart_name_en = ? AND source = ? "
  2110. default:
  2111. cond += " AND chart_name = ? AND source = ? "
  2112. }
  2113. pars = append(pars, req.ChartName, chartSource)
  2114. count, e := data_manage.GetChartInfoCountByCondition(cond, pars)
  2115. if e != nil {
  2116. br.Msg = "保存失败"
  2117. br.ErrMsg = fmt.Sprintf("获取同名图表失败, Err: %v", e)
  2118. return
  2119. }
  2120. if count > 0 {
  2121. br.Msg = "图表名称已存在, 请重新填写"
  2122. return
  2123. }
  2124. _, e = data_manage.GetChartClassifyById(req.ClassifyId)
  2125. if e != nil {
  2126. if e.Error() == utils.ErrNoRow() {
  2127. br.Msg = "分类不存在"
  2128. return
  2129. }
  2130. br.Msg = "保存失败"
  2131. br.ErrMsg = fmt.Sprintf("获取图表分类失败, Err: %v", e)
  2132. return
  2133. }
  2134. }
  2135. // 图表来源、图例设置
  2136. var sourceFrom, extraConfig string
  2137. if req.SourcesFrom != nil {
  2138. b, e := json.Marshal(req.SourcesFrom)
  2139. if e != nil {
  2140. br.Msg = "保存失败"
  2141. br.ErrMsg = fmt.Sprintf("图表来源JSON格式化失败, Err: %v", e)
  2142. return
  2143. }
  2144. sourceFrom = string(b)
  2145. }
  2146. if req.ExtraConfig != nil {
  2147. b, e := json.Marshal(req.ExtraConfig)
  2148. if e != nil {
  2149. br.Msg = "保存失败"
  2150. br.ErrMsg = fmt.Sprintf("图表来源JSON格式化失败, Err: %v", e)
  2151. return
  2152. }
  2153. extraConfig = string(b)
  2154. }
  2155. // 图表信息
  2156. chartInfo := new(data_manage.ChartInfo)
  2157. chartInfo.ChartName = req.ChartName
  2158. chartInfo.ChartNameEn = req.ChartName
  2159. chartInfo.ChartClassifyId = req.ClassifyId
  2160. chartInfo.SysUserId = sysUser.AdminId
  2161. chartInfo.SysUserRealName = sysUser.RealName
  2162. chartInfo.CreateTime = time.Now()
  2163. chartInfo.ModifyTime = time.Now()
  2164. chartInfo.IsSetName = 0
  2165. timestamp := strconv.FormatInt(time.Now().UnixNano(), 10)
  2166. chartInfo.UniqueCode = utils.MD5(utils.CHART_PREFIX + "_" + timestamp)
  2167. chartInfo.ChartType = chartType
  2168. chartInfo.Calendar = "公历"
  2169. chartInfo.DateType = 6
  2170. chartInfo.StartDate = startDate
  2171. chartInfo.EndDate = endDate
  2172. chartInfo.SeasonStartDate = startDate
  2173. chartInfo.SeasonEndDate = endDate
  2174. chartInfo.Disabled = data.CheckIsDisableChart(mappingEdbIds)
  2175. chartInfo.Source = chartSource
  2176. chartInfo.ExtraConfig = extraConfig
  2177. chartInfo.SourcesFrom = sourceFrom
  2178. // 区间计算图
  2179. chartCorrelate := new(data_manage.ChartInfoCorrelation)
  2180. chartCorrelate.LeadValue = req.FactorCorrelation.LeadValue
  2181. chartCorrelate.LeadUnit = req.FactorCorrelation.LeadUnit
  2182. chartCorrelate.CalculateValue = req.FactorCorrelation.CalculateValue
  2183. chartCorrelate.CalculateUnit = req.FactorCorrelation.CalculateUnit
  2184. chartCorrelate.EdbInfoIdFirst = req.BaseEdbInfoId
  2185. chartCorrelate.AnalysisMode = 1
  2186. chartCorrelate.CreateTime = time.Now().Local()
  2187. chartCorrelate.ModifyTime = time.Now().Local()
  2188. // 图表指标关联
  2189. edbMappings := make([]*data_manage.ChartEdbMapping, 0)
  2190. for _, v := range mappingEdbIds {
  2191. m := new(data_manage.ChartEdbMapping)
  2192. m.EdbInfoId = v
  2193. m.CreateTime = time.Now()
  2194. m.ModifyTime = time.Now()
  2195. edbTimestamp := strconv.FormatInt(time.Now().UnixNano(), 10)
  2196. m.UniqueCode = utils.MD5(utils.CHART_PREFIX + "_" + edbTimestamp + "_" + strconv.Itoa(v))
  2197. m.IsOrder = true
  2198. m.IsAxis = 1
  2199. m.EdbInfoType = 1
  2200. m.Source = utils.CHART_SOURCE_CORRELATION
  2201. edbMappings = append(edbMappings, m)
  2202. }
  2203. // 标记引用的指标
  2204. seriesIds := req.FactorCorrelation.SeriesIds
  2205. edbUsed := make(map[string]bool)
  2206. for _, v := range req.FactorCorrelation.SeriesEdb {
  2207. k := fmt.Sprintf("%d-%d", v.SeriesId, v.EdbInfoId)
  2208. edbUsed[k] = true
  2209. }
  2210. // 指标系列-图表关联
  2211. chartMappings := make([]*data_manage.FactorEdbSeriesChartMapping, 0)
  2212. {
  2213. ob := new(data_manage.FactorEdbSeriesChartMapping)
  2214. cond := fmt.Sprintf(" AND %s IN (%s)", ob.Cols().FactorEdbSeriesId, utils.GetOrmInReplace(len(seriesIds)))
  2215. pars := make([]interface{}, 0)
  2216. pars = append(pars, seriesIds)
  2217. items, e := ob.GetItemsByCondition(cond, pars, []string{}, "")
  2218. if e != nil {
  2219. br.Msg = "保存失败"
  2220. br.ErrMsg = fmt.Sprintf("获取指标系列图表关联失败, Err: %v", e)
  2221. return
  2222. }
  2223. chartMappings = items
  2224. }
  2225. for _, v := range chartMappings {
  2226. v.Source = chartSource
  2227. k := fmt.Sprintf("%d-%d", v.FactorEdbSeriesId, v.EdbInfoId)
  2228. v.EdbUsed = 0 // 先重置一下
  2229. if edbUsed[k] {
  2230. v.EdbUsed = 1
  2231. }
  2232. v.ModifyTime = time.Now().Local()
  2233. }
  2234. // 另存为
  2235. originSeries := make([]*data_manage.FactorEdbSeries, 0)
  2236. originEdbSeriesMapping := make([]*data_manage.FactorEdbSeriesMapping, 0)
  2237. if req.SaveAs {
  2238. // 注意此处要复制一份系列及系列关联指标, 不可共用被复制图表的系列
  2239. {
  2240. ob := new(data_manage.FactorEdbSeries)
  2241. cond := fmt.Sprintf(" AND %s IN (%s)", ob.Cols().PrimaryId, utils.GetOrmInReplace(len(seriesIds)))
  2242. pars := make([]interface{}, 0)
  2243. pars = append(pars, seriesIds)
  2244. list, e := ob.GetItemsByCondition(cond, pars, []string{}, "")
  2245. if e != nil {
  2246. br.Msg = "保存失败"
  2247. br.ErrMsg = fmt.Sprintf("获取源图表系列失败, %v", e)
  2248. return
  2249. }
  2250. originSeries = list
  2251. }
  2252. {
  2253. ob := new(data_manage.FactorEdbSeriesMapping)
  2254. cond := fmt.Sprintf(" AND %s IN (%s)", ob.Cols().FactorEdbSeriesId, utils.GetOrmInReplace(len(seriesIds)))
  2255. pars := make([]interface{}, 0)
  2256. pars = append(pars, seriesIds)
  2257. list, e := ob.GetItemsByCondition(cond, pars, []string{}, "")
  2258. if e != nil {
  2259. br.Msg = "保存失败"
  2260. br.ErrMsg = fmt.Sprintf("获取源图表系列指标失败, %v", e)
  2261. return
  2262. }
  2263. originEdbSeriesMapping = list
  2264. }
  2265. }
  2266. // 新增图表/区间计算图表/图表指标关联/指标系列图表关联
  2267. chartInfoId, seriesIdMap, e := data_manage.CreateMultiFactorCorrelationChartAndEdb(chartInfo, edbMappings, chartCorrelate, chartMappings, req.SaveAs, originSeries, originEdbSeriesMapping)
  2268. if e != nil {
  2269. br.Msg = "保存失败"
  2270. br.ErrMsg = fmt.Sprintf("新增多因子区间计算图表失败, Err: %v", e)
  2271. return
  2272. }
  2273. // 另存为-替换图例设置中的系列ID
  2274. if req.SaveAs {
  2275. if req.ExtraConfig != nil {
  2276. for _, v := range req.ExtraConfig.LegendConfig {
  2277. v.SeriesId = seriesIdMap[v.SeriesId]
  2278. }
  2279. b, e := json.Marshal(req.ExtraConfig)
  2280. if e != nil {
  2281. br.Msg = "保存失败"
  2282. br.ErrMsg = fmt.Sprintf("图表来源JSON格式化失败, Err: %v", e)
  2283. return
  2284. }
  2285. chartInfo.ExtraConfig = string(b)
  2286. if e = chartInfo.Update([]string{"ExtraConfig"}); e != nil {
  2287. br.Msg = "保存失败"
  2288. br.ErrMsg = fmt.Sprintf("更新图例信息失败, %v", e)
  2289. return
  2290. }
  2291. }
  2292. }
  2293. go data.EsAddOrEditChartInfo(chartInfoId)
  2294. // 操作日志
  2295. {
  2296. chartLog := new(data_manage.ChartInfoLog)
  2297. chartLog.ChartInfoId = chartInfo.ChartInfoId
  2298. chartLog.ChartName = req.ChartName
  2299. chartLog.ChartClassifyId = req.ClassifyId
  2300. chartLog.SysUserId = sysUser.AdminId
  2301. chartLog.SysUserRealName = sysUser.RealName
  2302. chartLog.UniqueCode = chartInfo.UniqueCode
  2303. chartLog.CreateTime = time.Now()
  2304. chartLog.Content = string(this.Ctx.Input.RequestBody)
  2305. chartLog.Status = "新增多因子区间计算图表"
  2306. chartLog.Method = this.Ctx.Input.URI()
  2307. go data_manage.AddChartInfoLog(chartLog)
  2308. }
  2309. resp := new(data_manage.AddChartInfoResp)
  2310. resp.ChartInfoId = chartInfo.ChartInfoId
  2311. resp.UniqueCode = chartInfo.UniqueCode
  2312. resp.ChartType = chartInfo.ChartType
  2313. resp.ClassifyId = req.ClassifyId
  2314. br.Data = resp
  2315. br.Ret = 200
  2316. br.Success = true
  2317. br.Msg = "保存成功"
  2318. br.IsAddLog = true
  2319. }
  2320. // MultiFactorEdit
  2321. // @Title 多因子图表-编辑
  2322. // @Description 多因子图表-编辑
  2323. // @Param request body request.CorrelationChartMultiFactorSaveReq true "type json string"
  2324. // @Success Ret=200 返回图表id
  2325. // @router /chart_info/multi_factor/edit [post]
  2326. func (this *RangeChartChartInfoController) MultiFactorEdit() {
  2327. br := new(models.BaseResponse).Init()
  2328. defer func() {
  2329. if br.ErrMsg == "" {
  2330. br.IsSendEmail = false
  2331. }
  2332. this.Data["json"] = br
  2333. this.ServeJSON()
  2334. }()
  2335. sysUser := this.SysUser
  2336. if sysUser == nil {
  2337. br.Msg = "请登录"
  2338. br.ErrMsg = "请登录,SysUser Is Empty"
  2339. br.Ret = 408
  2340. return
  2341. }
  2342. var req request.CorrelationChartMultiFactorSaveReq
  2343. if e := json.Unmarshal(this.Ctx.Input.RequestBody, &req); e != nil {
  2344. br.Msg = "参数解析异常!"
  2345. br.ErrMsg = fmt.Sprintf("参数解析失败, Err: %v", e)
  2346. return
  2347. }
  2348. if req.ChartInfoId <= 0 {
  2349. br.Msg = "参数有误"
  2350. br.ErrMsg = fmt.Sprintf("参数有误, ChartInfoId: %d", req.ChartInfoId)
  2351. return
  2352. }
  2353. // 此方法只加多因子图表
  2354. if req.AnalysisMode != 1 {
  2355. br.Msg = "分析模式有误"
  2356. br.ErrMsg = fmt.Sprintf("分析模式有误, Mode: %d", req.AnalysisMode)
  2357. return
  2358. }
  2359. if req.BaseEdbInfoId <= 0 {
  2360. br.Msg = "请选择标的指标"
  2361. return
  2362. }
  2363. if len(req.FactorCorrelation.SeriesIds) == 0 {
  2364. br.Msg = "请选择因子指标系列"
  2365. return
  2366. }
  2367. if len(req.FactorCorrelation.SeriesEdb) == 0 {
  2368. br.Msg = "请选择因子指标"
  2369. return
  2370. }
  2371. if req.FactorCorrelation.LeadValue <= 0 {
  2372. br.Msg = "分析周期不允许设置为负数或0"
  2373. return
  2374. }
  2375. if req.FactorCorrelation.LeadUnit == "" {
  2376. br.Msg = "请选择分析周期频度"
  2377. return
  2378. }
  2379. leadUnitDays, ok := utils.FrequencyDaysMap[req.FactorCorrelation.LeadUnit]
  2380. if !ok {
  2381. br.Msg = "错误的分析周期频度"
  2382. br.ErrMsg = fmt.Sprintf("分析周期频度有误: %s", req.FactorCorrelation.LeadUnit)
  2383. return
  2384. }
  2385. if req.FactorCorrelation.CalculateUnit == "" {
  2386. br.Msg = "请选择计算窗口频度"
  2387. return
  2388. }
  2389. calculateUnitDays, ok := utils.FrequencyDaysMap[req.FactorCorrelation.CalculateUnit]
  2390. if !ok {
  2391. br.Msg = "错误的计算窗口频度"
  2392. br.ErrMsg = fmt.Sprintf("计算窗口频度有误: %s", req.FactorCorrelation.CalculateUnit)
  2393. return
  2394. }
  2395. leadDays := 2 * req.FactorCorrelation.LeadValue * leadUnitDays
  2396. calculateDays := req.FactorCorrelation.CalculateValue * calculateUnitDays
  2397. if calculateDays < leadDays {
  2398. br.Msg = "计算窗口必须≥2*分析周期"
  2399. return
  2400. }
  2401. req.ChartName = strings.TrimSpace(req.ChartName)
  2402. if req.ChartName == "" {
  2403. br.Msg = "请输入图表名称"
  2404. return
  2405. }
  2406. if req.ClassifyId <= 0 {
  2407. br.Msg = "请选择图表分类"
  2408. return
  2409. }
  2410. cacheKey := "CACHE_CORRELATION_CHART_INFO_EDIT_" + strconv.Itoa(sysUser.AdminId)
  2411. if !utils.Rc.SetNX(cacheKey, 1, 30*time.Second) {
  2412. br.Msg = "系统处理中, 请稍后重试"
  2413. return
  2414. }
  2415. defer func() {
  2416. _ = utils.Rc.Delete(cacheKey)
  2417. }()
  2418. chartSource := utils.CHART_SOURCE_CORRELATION
  2419. startDate := time.Now().AddDate(0, 0, -calculateDays).Format(utils.FormatDate)
  2420. endDate := time.Now().Format(utils.FormatDate)
  2421. mappingEdbIds := make([]int, 0) // 该图表关联的指标(去除系列中重复指标之后的)
  2422. {
  2423. exists := make(map[int]bool)
  2424. for _, v := range req.FactorCorrelation.SeriesEdb {
  2425. if !exists[v.EdbInfoId] {
  2426. mappingEdbIds = append(mappingEdbIds, v.EdbInfoId)
  2427. }
  2428. }
  2429. }
  2430. chartInfo, e := data_manage.GetChartInfoById(req.ChartInfoId)
  2431. if e != nil {
  2432. if e.Error() == utils.ErrNoRow() {
  2433. br.Msg = "图表已被删除, 请刷新页面"
  2434. return
  2435. }
  2436. br.Msg = "保存失败"
  2437. br.ErrMsg = fmt.Sprintf("获取图表信息失败, Err: %v", e)
  2438. return
  2439. }
  2440. chartCorrelate := new(data_manage.ChartInfoCorrelation)
  2441. if e = chartCorrelate.GetItemById(chartInfo.ChartInfoId); e != nil {
  2442. br.Msg = "图表区间计算信息不存在"
  2443. br.ErrMsg = fmt.Sprintf("获取图表区间计算信息失败, Err: %v", e)
  2444. return
  2445. }
  2446. // 图表检验
  2447. {
  2448. if chartInfo.Source != chartSource {
  2449. br.Msg = "该图表不是区间计算图表"
  2450. return
  2451. }
  2452. // 图表操作权限
  2453. authOk := data.CheckOpChartPermission(sysUser, chartInfo.SysUserId, true)
  2454. if !authOk {
  2455. br.Msg = "没有该图表的操作权限"
  2456. return
  2457. }
  2458. // 图表名称、分类
  2459. var cond string
  2460. var pars []interface{}
  2461. switch this.Lang {
  2462. case utils.EnLangVersion:
  2463. cond += " AND chart_name_en = ? AND source = ? AND chart_info_id <> ? "
  2464. default:
  2465. cond += " AND chart_name = ? AND source = ? AND chart_info_id <> ? "
  2466. }
  2467. pars = append(pars, req.ChartName, chartSource, chartInfo.ChartInfoId)
  2468. count, e := data_manage.GetChartInfoCountByCondition(cond, pars)
  2469. if e != nil {
  2470. br.Msg = "保存失败"
  2471. br.ErrMsg = fmt.Sprintf("获取同名图表失败, Err: %v", e)
  2472. return
  2473. }
  2474. if count > 0 {
  2475. br.Msg = "图表名称已存在, 请重新填写"
  2476. return
  2477. }
  2478. _, e = data_manage.GetChartClassifyById(req.ClassifyId)
  2479. if e != nil {
  2480. if e.Error() == utils.ErrNoRow() {
  2481. br.Msg = "分类不存在"
  2482. return
  2483. }
  2484. br.Msg = "保存失败"
  2485. br.ErrMsg = fmt.Sprintf("获取图表分类失败, Err: %v", e)
  2486. return
  2487. }
  2488. }
  2489. // 图表来源、图例设置
  2490. var sourceFrom, extraConfig string
  2491. if req.SourcesFrom != nil {
  2492. b, e := json.Marshal(req.SourcesFrom)
  2493. if e != nil {
  2494. br.Msg = "保存失败"
  2495. br.ErrMsg = fmt.Sprintf("图表来源JSON格式化失败, Err: %v", e)
  2496. return
  2497. }
  2498. sourceFrom = string(b)
  2499. }
  2500. if req.ExtraConfig != nil {
  2501. b, e := json.Marshal(req.ExtraConfig)
  2502. if e != nil {
  2503. br.Msg = "保存失败"
  2504. br.ErrMsg = fmt.Sprintf("图表来源JSON格式化失败, Err: %v", e)
  2505. return
  2506. }
  2507. extraConfig = string(b)
  2508. }
  2509. // 图表信息
  2510. chartInfo.ChartName = req.ChartName
  2511. chartInfo.ChartNameEn = req.ChartName
  2512. chartInfo.ChartClassifyId = req.ClassifyId
  2513. chartInfo.ModifyTime = time.Now()
  2514. chartInfo.StartDate = startDate
  2515. chartInfo.EndDate = endDate
  2516. chartInfo.SeasonStartDate = startDate
  2517. chartInfo.SeasonEndDate = endDate
  2518. chartInfo.ExtraConfig = extraConfig
  2519. chartInfo.SourcesFrom = sourceFrom
  2520. chartInfo.Disabled = data.CheckIsDisableChart(mappingEdbIds)
  2521. chartUpdateCols := []string{"ChartName", "ChartNameEn", "ChartClassifyId", "ModifyTime", "StartDate", "EndDate", "SeasonStartDate", "SeasonEndDate", "ExtraConfig", "SourcesFrom", "Disabled"}
  2522. // 区间计算图
  2523. chartCorrelate.LeadValue = req.FactorCorrelation.LeadValue
  2524. chartCorrelate.LeadUnit = req.FactorCorrelation.LeadUnit
  2525. chartCorrelate.CalculateValue = req.FactorCorrelation.CalculateValue
  2526. chartCorrelate.CalculateUnit = req.FactorCorrelation.CalculateUnit
  2527. chartCorrelate.EdbInfoIdFirst = req.BaseEdbInfoId
  2528. chartCorrelate.ModifyTime = time.Now().Local()
  2529. correlateUpdateCols := []string{"LeadValue", "LeadUnit", "CalculateValue", "CalculateUnit", "EdbInfoIdFirst", "ModifyTime"}
  2530. // 图表指标关联
  2531. edbMappings := make([]*data_manage.ChartEdbMapping, 0)
  2532. for _, v := range mappingEdbIds {
  2533. m := new(data_manage.ChartEdbMapping)
  2534. m.EdbInfoId = v
  2535. m.CreateTime = time.Now()
  2536. m.ModifyTime = time.Now()
  2537. edbTimestamp := strconv.FormatInt(time.Now().UnixNano(), 10)
  2538. m.UniqueCode = utils.MD5(utils.CHART_PREFIX + "_" + edbTimestamp + "_" + strconv.Itoa(v))
  2539. m.IsOrder = true
  2540. m.IsAxis = 1
  2541. m.EdbInfoType = 1
  2542. m.Source = utils.CHART_SOURCE_CORRELATION
  2543. edbMappings = append(edbMappings, m)
  2544. }
  2545. seriesIds := req.FactorCorrelation.SeriesIds
  2546. edbUsed := make(map[string]bool)
  2547. for _, v := range req.FactorCorrelation.SeriesEdb {
  2548. k := fmt.Sprintf("%d-%d", v.SeriesId, v.EdbInfoId)
  2549. edbUsed[k] = true
  2550. }
  2551. // 指标系列-图表关联
  2552. chartMappings := make([]*data_manage.FactorEdbSeriesChartMapping, 0)
  2553. {
  2554. ob := new(data_manage.FactorEdbSeriesChartMapping)
  2555. cond := fmt.Sprintf(" AND %s IN (%s)", ob.Cols().FactorEdbSeriesId, utils.GetOrmInReplace(len(seriesIds)))
  2556. pars := make([]interface{}, 0)
  2557. pars = append(pars, seriesIds)
  2558. items, e := ob.GetItemsByCondition(cond, pars, []string{}, "")
  2559. if e != nil {
  2560. br.Msg = "保存失败"
  2561. br.ErrMsg = fmt.Sprintf("获取指标系列图表关联失败, Err: %v", e)
  2562. return
  2563. }
  2564. chartMappings = items
  2565. }
  2566. for _, v := range chartMappings {
  2567. k := fmt.Sprintf("%d-%d", v.FactorEdbSeriesId, v.EdbInfoId)
  2568. v.EdbUsed = 0
  2569. if edbUsed[k] {
  2570. v.EdbUsed = 1
  2571. }
  2572. v.ModifyTime = time.Now().Local()
  2573. }
  2574. // 更新图表/区间计算图表/图表指标关联/指标系列图表关联
  2575. e = data_manage.UpdateMultiFactorCorrelationChartAndEdb(chartInfo, edbMappings, chartCorrelate, chartMappings, chartUpdateCols, correlateUpdateCols)
  2576. if e != nil {
  2577. br.Msg = "保存失败"
  2578. br.ErrMsg = fmt.Sprintf("新增多因子区间计算图表失败, Err: %v", e)
  2579. return
  2580. }
  2581. // ES
  2582. go func() {
  2583. data.EsAddOrEditChartInfo(chartInfo.ChartInfoId)
  2584. data.EsAddOrEditMyChartInfoByChartInfoId(chartInfo.ChartInfoId)
  2585. }()
  2586. // 操作日志
  2587. {
  2588. chartLog := new(data_manage.ChartInfoLog)
  2589. chartLog.ChartInfoId = chartInfo.ChartInfoId
  2590. chartLog.ChartName = req.ChartName
  2591. chartLog.ChartClassifyId = req.ClassifyId
  2592. chartLog.SysUserId = sysUser.AdminId
  2593. chartLog.SysUserRealName = sysUser.RealName
  2594. chartLog.UniqueCode = chartInfo.UniqueCode
  2595. chartLog.CreateTime = time.Now()
  2596. chartLog.Content = string(this.Ctx.Input.RequestBody)
  2597. chartLog.Status = "编辑多因子区间计算图表"
  2598. chartLog.Method = this.Ctx.Input.URI()
  2599. go data_manage.AddChartInfoLog(chartLog)
  2600. }
  2601. resp := new(data_manage.AddChartInfoResp)
  2602. resp.ChartInfoId = chartInfo.ChartInfoId
  2603. resp.UniqueCode = chartInfo.UniqueCode
  2604. resp.ChartType = chartInfo.ChartType
  2605. resp.ClassifyId = req.ClassifyId
  2606. br.Data = resp
  2607. br.Ret = 200
  2608. br.Success = true
  2609. br.Msg = "保存成功"
  2610. br.IsAddLog = true
  2611. }
  2612. // MultiFactorDetail
  2613. // @Title 多因子图表-编辑页详情
  2614. // @Description 多因子图表-编辑页详情
  2615. // @Param UniqueCode query string true "图表唯一编码"
  2616. // @Success Ret=200 返回图表id
  2617. // @router /chart_info/multi_factor/detail [get]
  2618. func (this *RangeChartChartInfoController) MultiFactorDetail() {
  2619. br := new(models.BaseResponse).Init()
  2620. defer func() {
  2621. if br.ErrMsg == "" {
  2622. br.IsSendEmail = false
  2623. }
  2624. this.Data["json"] = br
  2625. this.ServeJSON()
  2626. }()
  2627. sysUser := this.SysUser
  2628. if sysUser == nil {
  2629. br.Msg = "请登录"
  2630. br.ErrMsg = "请登录,SysUser Is Empty"
  2631. br.Ret = 408
  2632. return
  2633. }
  2634. uniqueCode := this.GetString("UniqueCode")
  2635. if uniqueCode == "" {
  2636. br.Msg = "参数有误"
  2637. br.ErrMsg = fmt.Sprintf("参数有误, UniqueCode: %s", uniqueCode)
  2638. return
  2639. }
  2640. chartInfo, e := data_manage.GetChartInfoByUniqueCode(uniqueCode)
  2641. if e != nil {
  2642. if e.Error() == utils.ErrNoRow() {
  2643. br.Msg = "图表已被删除, 请刷新页面"
  2644. return
  2645. }
  2646. br.Msg = "获取失败"
  2647. br.ErrMsg = fmt.Sprintf("获取图表信息失败, Err: %v", e)
  2648. return
  2649. }
  2650. chartCorrelate := new(data_manage.ChartInfoCorrelation)
  2651. if e = chartCorrelate.GetItemById(chartInfo.ChartInfoId); e != nil {
  2652. br.Msg = "图表区间计算信息不存在"
  2653. br.ErrMsg = fmt.Sprintf("获取图表区间计算信息失败, Err: %v", e)
  2654. return
  2655. }
  2656. if chartCorrelate.AnalysisMode != 1 {
  2657. br.Msg = "图表分析模式有误"
  2658. br.ErrMsg = fmt.Sprintf("图表分析模式有误, Err: %v", e)
  2659. return
  2660. }
  2661. if chartCorrelate.EdbInfoIdFirst <= 0 {
  2662. br.Msg = "标的指标异常"
  2663. br.ErrMsg = fmt.Sprintf("标的指标ID异常, Err: %v", e)
  2664. return
  2665. }
  2666. resp := new(data_manage.FactorCorrelationEditDetail)
  2667. // 标的指标
  2668. baseEdbMapping := new(data_manage.ChartEdbInfoMapping)
  2669. {
  2670. mappings, e := data_manage.GetChartEdbMappingListByEdbInfoIdList([]int{chartCorrelate.EdbInfoIdFirst})
  2671. if e != nil {
  2672. br.Msg = "获取失败"
  2673. br.ErrMsg = fmt.Sprintf("获取标的指标信息异常, Err: %v", e)
  2674. return
  2675. }
  2676. if len(mappings) > 0 {
  2677. baseEdbMapping = mappings[0]
  2678. }
  2679. }
  2680. // 获取图表系列
  2681. chartMappings := make([]*data_manage.FactorEdbSeriesChartMapping, 0)
  2682. {
  2683. ob := new(data_manage.FactorEdbSeriesChartMapping)
  2684. cond := fmt.Sprintf(" AND %s = ?", ob.Cols().ChartInfoId)
  2685. pars := make([]interface{}, 0)
  2686. pars = append(pars, chartInfo.ChartInfoId)
  2687. list, e := ob.GetItemsByCondition(cond, pars, []string{}, "")
  2688. if e != nil {
  2689. br.Msg = "获取失败"
  2690. br.ErrMsg = fmt.Sprintf("获取图表关联系列列表失败, Err: %v", e)
  2691. return
  2692. }
  2693. chartMappings = list
  2694. }
  2695. if len(chartMappings) == 0 {
  2696. br.Msg = "因子指标系列异常"
  2697. br.ErrMsg = fmt.Sprintf("因子指标系列引用异常")
  2698. return
  2699. }
  2700. seriesIds := make([]int, 0)
  2701. seriesIdExist := make(map[int]bool)
  2702. edbInfoIds := make([]int, 0)
  2703. for _, v := range chartMappings {
  2704. if !seriesIdExist[v.FactorEdbSeriesId] {
  2705. seriesIds = append(seriesIds, v.FactorEdbSeriesId)
  2706. }
  2707. edbInfoIds = append(edbInfoIds, v.EdbInfoId)
  2708. }
  2709. if len(seriesIds) == 0 {
  2710. br.Msg = "因子指标系列异常"
  2711. br.ErrMsg = fmt.Sprintf("因子指标系列引用异常")
  2712. return
  2713. }
  2714. // 系列信息
  2715. seriesDetails := make([]*data_manage.FactorEdbSeriesDetail, 0)
  2716. {
  2717. seriesOb := new(data_manage.FactorEdbSeries)
  2718. cond := fmt.Sprintf(" AND %s IN (%s)", seriesOb.Cols().PrimaryId, utils.GetOrmInReplace(len(seriesIds)))
  2719. pars := make([]interface{}, 0)
  2720. pars = append(pars, seriesIds)
  2721. list, e := seriesOb.GetItemsByCondition(cond, pars, []string{}, fmt.Sprintf("%s ASC", seriesOb.Cols().PrimaryId))
  2722. if e != nil {
  2723. br.Msg = "获取失败"
  2724. br.ErrMsg = fmt.Sprintf("获取关联系列失败, Err: %v", e)
  2725. return
  2726. }
  2727. for _, v := range list {
  2728. seriesDetails = append(seriesDetails, &data_manage.FactorEdbSeriesDetail{
  2729. FactorEdbSeriesItem: v.Format2Item(),
  2730. EdbMappings: make([]*data_manage.FactorEdbSeriesMappingItem, 0),
  2731. })
  2732. }
  2733. }
  2734. // 区间计算配置
  2735. var correlateConf data_manage.CorrelationConfig
  2736. correlateConf.LeadUnit = chartCorrelate.LeadUnit
  2737. correlateConf.LeadValue = chartCorrelate.LeadValue
  2738. correlateConf.CalculateUnit = chartCorrelate.CalculateUnit
  2739. correlateConf.CalculateValue = chartCorrelate.CalculateValue
  2740. edbIdItem := make(map[int]*data_manage.EdbInfo)
  2741. edbItems, e := data_manage.GetEdbInfoByIdList(edbInfoIds)
  2742. if e != nil {
  2743. br.Msg = "获取失败"
  2744. br.ErrMsg = "获取因子指标失败, Err: " + e.Error()
  2745. return
  2746. }
  2747. for _, v := range edbItems {
  2748. edbIdItem[v.EdbInfoId] = v
  2749. }
  2750. // 区间计算矩阵
  2751. seriesEdb := make(map[int][]*data_manage.FactorEdbSeriesMappingItem)
  2752. var matrixItems []data_manage.FactorEdbSeriesCorrelationMatrixItem
  2753. for _, v := range chartMappings {
  2754. edbItem := edbIdItem[v.EdbInfoId]
  2755. if edbItem == nil {
  2756. continue
  2757. }
  2758. var item data_manage.FactorEdbSeriesCorrelationMatrixItem
  2759. item.SeriesId = v.FactorEdbSeriesId
  2760. item.EdbInfoId = edbItem.EdbInfoId
  2761. item.EdbCode = edbItem.EdbCode
  2762. item.EdbName = edbItem.EdbName
  2763. var values []data_manage.FactorEdbSeriesCorrelationMatrixValues
  2764. if v.CalculateData != "" {
  2765. if e = json.Unmarshal([]byte(v.CalculateData), &values); e != nil {
  2766. br.Msg = "获取失败"
  2767. br.ErrMsg = fmt.Sprintf("解析因子指标计算结果失败, Err: %v", e)
  2768. return
  2769. }
  2770. }
  2771. sort.Sort(data_manage.FactorEdbSeriesCorrelationMatrixOrder(values))
  2772. item.Values = values
  2773. if v.EdbUsed == 1 {
  2774. item.Used = true
  2775. }
  2776. item.SourceName = edbItem.SourceName
  2777. matrixItems = append(matrixItems, item)
  2778. // 系列关联的指标详情
  2779. if seriesEdb[v.FactorEdbSeriesId] == nil {
  2780. seriesEdb[v.FactorEdbSeriesId] = make([]*data_manage.FactorEdbSeriesMappingItem, 0)
  2781. }
  2782. seriesEdb[v.FactorEdbSeriesId] = append(seriesEdb[v.FactorEdbSeriesId], &data_manage.FactorEdbSeriesMappingItem{
  2783. SeriesId: v.FactorEdbSeriesId,
  2784. EdbInfoId: v.EdbInfoId,
  2785. EdbCode: edbItem.EdbCode,
  2786. EdbName: edbItem.EdbName,
  2787. EdbNameEn: edbItem.EdbNameEn,
  2788. })
  2789. }
  2790. // 系列关联的指标详情
  2791. for _, v := range seriesDetails {
  2792. v.EdbMappings = seriesEdb[v.SeriesId]
  2793. }
  2794. resp.ChartInfoId = chartInfo.ChartInfoId
  2795. resp.UniqueCode = chartInfo.UniqueCode
  2796. resp.BaseEdbInfo = baseEdbMapping
  2797. resp.EdbSeries = seriesDetails
  2798. resp.CorrelationConfig = correlateConf
  2799. resp.CorrelationMatrix = matrixItems
  2800. br.Data = resp
  2801. br.Ret = 200
  2802. br.Success = true
  2803. br.Msg = "获取成功"
  2804. }