chart_info.go 95 KB

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