base_from_smm.go 6.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254
  1. package controllers
  2. import (
  3. "encoding/json"
  4. "eta/eta_index_lib/logic"
  5. "eta/eta_index_lib/models"
  6. "eta/eta_index_lib/services"
  7. "eta/eta_index_lib/utils"
  8. "fmt"
  9. "strconv"
  10. "time"
  11. )
  12. // 有色
  13. type SmmController struct {
  14. BaseAuthController
  15. }
  16. // @Title 新增有色指标接口
  17. // @Description 新增有色指标接口
  18. // @Success 200 {object} models.AddEdbInfoReq
  19. // @router /add [post]
  20. func (this *SmmController) Add() {
  21. br := new(models.BaseResponse).Init()
  22. var cacheKey string
  23. defer func() {
  24. utils.Rc.Delete(cacheKey)
  25. this.Data["json"] = br
  26. this.ServeJSON()
  27. }()
  28. source := utils.DATA_SOURCE_YS
  29. var req models.AddEdbInfoReq
  30. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  31. if err != nil {
  32. br.Msg = "参数解析异常!"
  33. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  34. return
  35. }
  36. if req.EdbCode == "" {
  37. br.Msg = "请输入指标编码!"
  38. br.ErrMsg = "请输入指标编码,指标编码为空"
  39. return
  40. }
  41. cacheKey = utils.CACHE_EDB_DATA_ADD + strconv.Itoa(source) + "_" + req.EdbCode
  42. if !utils.Rc.IsExist(cacheKey) {
  43. utils.Rc.SetNX(cacheKey, 1, 1*time.Minute)
  44. err = models.AddEdbDataFromSmm(req.EdbCode)
  45. if err != nil {
  46. br.Msg = "获取指标信息失败!"
  47. br.ErrMsg = "获取指标信息失败 AddEdbDataFromSmm,Err:" + err.Error()
  48. return
  49. }
  50. br.Ret = 200
  51. br.Success = true
  52. br.Msg = "获取成功"
  53. } else {
  54. br.Ret = 501
  55. br.Success = true
  56. br.Msg = "系统处理中,请稍后重试"
  57. }
  58. }
  59. // @Title 刷新有色指标接口
  60. // @Description 刷新有色指标接口
  61. // @Success 200 {object} models.RefreshEdbInfoReq
  62. // @router /refresh [post]
  63. func (this *SmmController) Refresh() {
  64. br := new(models.BaseResponse).Init()
  65. var cacheKey string
  66. defer func() {
  67. this.Data["json"] = br
  68. this.ServeJSON()
  69. }()
  70. source := utils.DATA_SOURCE_YS
  71. var req models.RefreshEdbInfoReq
  72. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  73. if err != nil {
  74. br.Msg = "参数解析异常!"
  75. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  76. return
  77. }
  78. if req.EdbCode == "" {
  79. br.Msg = "请输入指标编码!"
  80. br.ErrMsg = "请输入指标编码,指标编码为空"
  81. return
  82. }
  83. if req.EdbInfoId < 0 {
  84. br.Msg = "请输入指标ID!"
  85. br.ErrMsg = "请输入指标ID"
  86. return
  87. }
  88. // 获取指标详情
  89. edbInfo, err := models.GetEdbInfoByEdbCode(source, req.EdbCode)
  90. if err != nil {
  91. br.Msg = "指标不存在!"
  92. br.ErrMsg = "指标不存在"
  93. return
  94. }
  95. cacheKey = utils.CACHE_EDB_DATA_REFRESH + strconv.Itoa(source) + "_" + req.EdbCode
  96. if utils.Rc.IsExist(cacheKey) {
  97. br.Ret = 501
  98. br.Success = true
  99. br.Msg = "系统处理中,请稍后重试"
  100. return
  101. }
  102. utils.Rc.SetNX(cacheKey, 1, 1*time.Minute)
  103. defer func() {
  104. utils.Rc.Delete(cacheKey)
  105. }()
  106. if req.EdbInfoId <= 0 {
  107. req.EdbInfoId = edbInfo.EdbInfoId
  108. }
  109. err = models.RefreshEdbDataFromSmm(req.EdbInfoId, req.EdbCode, req.StartDate)
  110. if err != nil && err.Error() != utils.ErrNoRow() {
  111. br.Msg = "刷新指标信息失败!"
  112. br.ErrMsg = "刷新指标信息失败 RefreshEdbDataFromSmm,Err:" + err.Error()
  113. return
  114. }
  115. // 更新指标最大最小值
  116. err, errMsg := models.UnifiedModifyEdbInfoMaxAndMinInfo(edbInfo)
  117. if err != nil {
  118. br.Msg = errMsg
  119. br.ErrMsg = err.Error()
  120. return
  121. }
  122. // 更新ES
  123. go logic.UpdateEs(edbInfo.EdbInfoId)
  124. br.Ret = 200
  125. br.Success = true
  126. br.Msg = "获取成功"
  127. }
  128. // HandleExcelData
  129. // @Title 处理有色指标的接口
  130. // @Description 处理有色指标的接口
  131. // @Success 200 string "操作成功"
  132. // @router /handle/excel_data [post]
  133. func (this *SmmController) HandleExcelData() {
  134. br := new(models.BaseResponse).Init()
  135. defer func() {
  136. this.Data["json"] = br
  137. this.ServeJSON()
  138. }()
  139. var req models.HandleSmmExcelDataReq
  140. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  141. if err != nil {
  142. br.Msg = "参数解析异常!"
  143. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  144. return
  145. }
  146. // 处理excel表数据
  147. e, errMsg := services.SmmIndexHandle(req.BaseFilePath, req.RenameFilePath, req.IndexName, req.IndexCode, req.Unit, req.Frequency, req.Source, req.ExcelDataMap, req.TerminalCode)
  148. if e != nil {
  149. br.Msg = "处理失败"
  150. br.ErrMsg = fmt.Sprintf("处理有色excel表数据失败, Err: %s, ErrMsg: %s", e.Error(), errMsg)
  151. return
  152. }
  153. br.Ret = 200
  154. br.Success = true
  155. br.Msg = "处理成功"
  156. }
  157. // AddIndexToDataSource
  158. // @Title 添加指标到数据源-SMM
  159. // @Description 添加指标到数据源-SMM
  160. // @Success 200 {object} models.AddSmmIndexToDataSourceResp
  161. // @router /add/index/to_data_source [post]
  162. func (this *SmmController) AddIndexToDataSource() {
  163. br := new(models.BaseResponse).Init()
  164. defer func() {
  165. this.Data["json"] = br
  166. this.ServeJSON()
  167. }()
  168. var req models.AddSmmIndexToDataSourceResp
  169. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  170. if err != nil {
  171. br.Msg = "参数解析异常!"
  172. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  173. return
  174. }
  175. smmIndexObj := new(models.BaseFromSmmIndex)
  176. indexItem, err := smmIndexObj.GetSmmIndexItem(req.EdbCode)
  177. var isAdd bool
  178. if err != nil {
  179. if err.Error() == utils.ErrNoRow() {
  180. isAdd = true
  181. } else {
  182. br.Msg = "判断指标是否存在失败"
  183. br.ErrMsg = "判断指标是否存在失败,Err:" + err.Error()
  184. return
  185. }
  186. }
  187. if indexItem == nil {
  188. isAdd = true
  189. }
  190. if isAdd {
  191. smmIndexObj.IndexCode = req.EdbCode
  192. smmIndexObj.ClassifyId = req.ClassifyId
  193. smmIndexObj.CreateTime = time.Now()
  194. smmIndexObj.ModifyTime = time.Now()
  195. err = smmIndexObj.Add()
  196. if err != nil {
  197. br.Msg = "新增指标失败"
  198. br.ErrMsg = "新增指标失败,Err:" + err.Error()
  199. return
  200. }
  201. }
  202. br.Ret = 200
  203. br.Success = true
  204. br.Msg = "新增成功"
  205. }
  206. // GetIndexDetailByCode
  207. // @Title 获取指标详情
  208. // @Description 获取指标详情
  209. // @Success 200 {object} models.GetIndexDetailReq
  210. // @router /index_detail/from_data_source [post]
  211. func (this *SmmController) GetIndexDetailFromDataSourceByCode() {
  212. br := new(models.BaseResponse).Init()
  213. defer func() {
  214. this.Data["json"] = br
  215. this.ServeJSON()
  216. }()
  217. var req models.GetIndexDetailReq
  218. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  219. if err != nil {
  220. br.Msg = "参数解析异常!"
  221. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  222. return
  223. }
  224. if req.IndexCode == "" {
  225. br.Msg = "参数有误"
  226. br.ErrMsg = "参数有误"
  227. return
  228. }
  229. indexObj := new(models.BaseFromSmmIndex)
  230. item, err := indexObj.GetIndexItem(req.IndexCode)
  231. if err != nil {
  232. br.Msg = "获取失败"
  233. br.ErrMsg = "获取指标失败, Err: " + err.Error()
  234. return
  235. }
  236. br.Data = item
  237. br.Ret = 200
  238. br.Success = true
  239. br.Msg = "获取成功"
  240. }