chart_info.go 94 KB

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