excel_info.go 8.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299
  1. package controllers
  2. import (
  3. "encoding/json"
  4. "eta/eta_chart_lib/models"
  5. "eta/eta_chart_lib/models/request"
  6. "eta/eta_chart_lib/models/response"
  7. "eta/eta_chart_lib/services/data"
  8. excel2 "eta/eta_chart_lib/services/data/excel"
  9. "eta/eta_chart_lib/services/excel"
  10. "eta/eta_chart_lib/utils"
  11. "github.com/shopspring/decimal"
  12. "time"
  13. )
  14. // ExcelInfoController excel表格
  15. type ExcelInfoController struct {
  16. BaseAuthController
  17. }
  18. // GetTableDetail
  19. // @Title 获取表格表详情
  20. // @Description 获取表格表详情接口
  21. // @Param UniqueCode query string true "表格表唯一编码,如果是管理后台访问,传固定字符串:7c69b590249049942070ae9dcd5bf6dc"
  22. // @Param FromScene query int true "场景来源,1:智能研报,2:研报列表;3:英文研报;4:中文PPT;5:英文PPT"
  23. // @Success 200 {object} response.ExcelTableDetailResp
  24. // @router /detail [get]
  25. func (this *ExcelInfoController) GetTableDetail() {
  26. br := new(models.BaseResponse).Init()
  27. defer func() {
  28. this.Data["json"] = br
  29. this.ServeJSON()
  30. }()
  31. uniqueCode := this.GetString("UniqueCode")
  32. if uniqueCode == "" {
  33. br.Msg = "参数错误"
  34. br.ErrMsg = "参数错误,uniqueCode is empty"
  35. return
  36. }
  37. fromScene, _ := this.GetInt("FromScene", 0)
  38. key := utils.HZ_CHART_LIB_EXCEL_TABLE_DETAIL + ":" + uniqueCode
  39. resp := new(response.ExcelTableDetailResp)
  40. //判断是否有缓存
  41. //if utils.Re == nil {
  42. // if utils.Re == nil && utils.Rc.IsExist(key) {
  43. // if data, err1 := utils.Rc.RedisBytes(key); err1 == nil {
  44. // err := json.Unmarshal(data, &resp)
  45. // if err == nil && resp != nil {
  46. // // 配置项
  47. // config := response.ExcelTableDetailConfigResp{
  48. // FontSize: 9,
  49. // }
  50. // // 获取配置的字体大小
  51. // confName := models.FromSceneMap[fromScene]
  52. // if confName != `` {
  53. // busConf, err := models.GetBusinessConfByKey(confName)
  54. // if err == nil {
  55. // sizeDeci, err := decimal.NewFromString(busConf.ConfVal)
  56. // if err == nil {
  57. // config.FontSize = int(sizeDeci.IntPart())
  58. // }
  59. // }
  60. // }
  61. // resp.Config = config
  62. //
  63. // br.Ret = 200
  64. // br.Success = true
  65. // br.Msg = "获取成功"
  66. // br.Data = resp
  67. // fmt.Println("source redis")
  68. // return
  69. // }
  70. // }
  71. // }
  72. //}
  73. excelInfo, err := models.GetExcelInfoByUniqueCode(uniqueCode)
  74. if err != nil {
  75. if err.Error() == utils.ErrNoRow() {
  76. br.Msg = "该表格已被删除,请刷新页面"
  77. br.ErrMsg = "该表格已被删除,请刷新页面,Err:" + err.Error()
  78. return
  79. }
  80. br.Msg = "获取失败"
  81. br.ErrMsg = "获取表格表信息失败,Err:" + err.Error()
  82. return
  83. }
  84. var tableData excel.TableData
  85. switch excelInfo.Source {
  86. case utils.EXCEL_DEFAULT:
  87. luckySheetData, err := excel.GetLuckySheetData(excelInfo.Content)
  88. if err != nil {
  89. br.Msg = "获取失败"
  90. br.ErrMsg = "获取excel数据失败,Err:" + err.Error()
  91. return
  92. }
  93. tableData, err = luckySheetData.GetTableDataByLuckySheetDataStr()
  94. if err != nil {
  95. br.Msg = "获取失败"
  96. br.ErrMsg = "转换成table失败,Err:" + err.Error()
  97. return
  98. }
  99. case utils.TIME_TABLE:
  100. var tableDataConfig excel2.TableDataConfig
  101. err = json.Unmarshal([]byte(excelInfo.Content), &tableDataConfig)
  102. if err != nil {
  103. br.Msg = "获取失败"
  104. br.ErrMsg = "表格json转结构体失败,Err:" + err.Error()
  105. return
  106. }
  107. result, err := excel2.GetDataByTableDataConfig(tableDataConfig)
  108. if err != nil {
  109. br.Msg = "获取失败"
  110. br.ErrMsg = "获取最新的表格数据失败,Err:" + err.Error()
  111. return
  112. }
  113. tableData, err = excel.GetTableDataByCustomData(excelInfo.ExcelType, result, this.Lang)
  114. if err != nil {
  115. br.Msg = "获取失败"
  116. br.ErrMsg = "转换成table失败,Err:" + err.Error()
  117. return
  118. }
  119. case utils.MIXED_TABLE:
  120. var result request.MixedTableReq
  121. err = json.Unmarshal([]byte(excelInfo.Content), &result)
  122. if err != nil {
  123. br.Msg = "获取失败"
  124. br.ErrMsg = "表格json转结构体失败,Err:" + err.Error()
  125. return
  126. }
  127. newResult, tmpErr, tmpErrMsg := excel2.GetMixedTableCellData(result, this.Lang)
  128. if tmpErr != nil {
  129. br.Msg = "获取失败"
  130. if tmpErrMsg != `` {
  131. br.Msg = tmpErrMsg
  132. }
  133. br.ErrMsg = "获取最新的数据失败,Err:" + tmpErr.Error()
  134. return
  135. }
  136. tableData, err = excel.GetTableDataByMixedTableData(newResult)
  137. if err != nil {
  138. br.Msg = "获取失败"
  139. br.ErrMsg = "转换成table失败,Err:" + err.Error()
  140. return
  141. }
  142. }
  143. tableData = excel.HandleTableCell(tableData)
  144. config := response.ExcelTableDetailConfigResp{
  145. FontSize: 9,
  146. }
  147. // 获取配置的字体大小
  148. confName := models.FromSceneMap[fromScene]
  149. if confName != `` {
  150. busConf, err := models.GetBusinessConfByKey(confName)
  151. if err == nil {
  152. sizeDeci, err := decimal.NewFromString(busConf.ConfVal)
  153. if err == nil {
  154. config.FontSize = int(sizeDeci.IntPart())
  155. }
  156. }
  157. }
  158. resp.TableInfo = tableData
  159. resp.Source = excelInfo.Source
  160. resp.ExcelType = excelInfo.ExcelType
  161. resp.ExcelName = excelInfo.ExcelName
  162. resp.ExcelImage = excelInfo.ExcelImage
  163. resp.UniqueCode = excelInfo.UniqueCode
  164. resp.Config = config
  165. if utils.Re == nil {
  166. data, _ := json.Marshal(resp)
  167. utils.Rc.Put(key, data, 5*time.Minute)
  168. }
  169. br.Ret = 200
  170. br.Success = true
  171. br.Msg = "获取成功"
  172. br.Data = resp
  173. }
  174. // Refresh
  175. // @Title 刷新
  176. // @Description 刷新接口
  177. // @Param UniqueCode query string true "表格id"
  178. // @router /refresh [get]
  179. func (this *ExcelInfoController) Refresh() {
  180. br := new(models.BaseResponse).Init()
  181. defer func() {
  182. if br.ErrMsg == "" {
  183. br.IsSendEmail = false
  184. }
  185. this.Data["json"] = br
  186. this.ServeJSON()
  187. }()
  188. uniqueCode := this.GetString("UniqueCode")
  189. if uniqueCode == "" {
  190. br.Msg = "参数错误"
  191. br.ErrMsg = "参数错误, UniqueCode is empty"
  192. return
  193. }
  194. key := utils.HZ_CHART_LIB_EXCEL_TABLE_DETAIL_REFRESH + ":" + uniqueCode
  195. ok := utils.Rc.SetNX(key, 1, time.Second*300)
  196. if !ok {
  197. br.Ret = 200
  198. br.Success = true
  199. br.Msg = "表格刷新中"
  200. return
  201. }
  202. defer func() {
  203. _ = utils.Rc.Delete(key)
  204. }()
  205. // 获取数据详情
  206. excelDetail, errMsg, err := excel2.GetExcelDetailInfoByExcelInfoId(uniqueCode, this.Lang)
  207. if err != nil {
  208. if err.Error() == utils.ErrNoRow() {
  209. br.Msg = "该表格已被删除,请刷新页面"
  210. br.ErrMsg = "该表格已被删除,请刷新页面,Err:" + err.Error()
  211. return
  212. }
  213. br.Msg = errMsg
  214. br.ErrMsg = "获取表格表信息失败,Err:" + err.Error()
  215. return
  216. }
  217. // 数据刷新(只有自定义表格有刷新)
  218. if excelDetail.Source == utils.TIME_TABLE {
  219. jsonByte, e := json.Marshal(excelDetail.TableData)
  220. if e != nil {
  221. br.Msg = "自定义表格数据获取失败"
  222. br.ErrMsg = "自定义表格数据获取失败,转json失败,Err:" + e.Error()
  223. return
  224. }
  225. var tableData request.TableDataReq
  226. if e = json.Unmarshal(jsonByte, &tableData); e != nil {
  227. br.Msg = "自定义表格数据获取失败"
  228. br.ErrMsg = "自定义表格数据获取失败,json转结构体失败,Err:" + e.Error()
  229. return
  230. }
  231. if len(tableData.EdbInfoIdList) > 0 {
  232. if e, _ = data.EdbInfoRefreshAllFromBase(tableData.EdbInfoIdList, false); e != nil {
  233. br.Msg = "刷新失败"
  234. br.ErrMsg = "刷新失败,Err:" + e.Error()
  235. return
  236. }
  237. }
  238. }
  239. // 数据刷新-混合表格
  240. if excelDetail.Source == utils.MIXED_TABLE {
  241. jsonByte, e := json.Marshal(excelDetail.TableData)
  242. if e != nil {
  243. br.Msg = "刷新失败"
  244. br.ErrMsg = "JSON格式化混合表格数据失败, Err: " + e.Error()
  245. return
  246. }
  247. var tableData request.MixedTableReq
  248. if e = json.Unmarshal(jsonByte, &tableData); e != nil {
  249. br.Msg = "刷新失败"
  250. br.ErrMsg = "解析混合表格数据失败, Err: " + e.Error()
  251. return
  252. }
  253. edbInfoIds := make([]int, 0)
  254. edbInfoIdExist := make(map[int]bool)
  255. if len(tableData.Data) > 0 {
  256. for _, t := range tableData.Data {
  257. for _, v := range t {
  258. if v.EdbInfoId > 0 && !edbInfoIdExist[v.EdbInfoId] {
  259. edbInfoIdExist[v.EdbInfoId] = true
  260. edbInfoIds = append(edbInfoIds, v.EdbInfoId)
  261. }
  262. }
  263. }
  264. }
  265. if len(edbInfoIds) > 0 {
  266. if e, _ = data.EdbInfoRefreshAllFromBase(edbInfoIds, false); e != nil {
  267. br.Msg = "刷新失败"
  268. br.ErrMsg = "刷新混合表格数据失败, Err: " + e.Error()
  269. return
  270. }
  271. }
  272. }
  273. // 清除缓存
  274. detailKey := utils.HZ_CHART_LIB_EXCEL_TABLE_DETAIL + ":" + excelDetail.UniqueCode
  275. if utils.Re == nil {
  276. _ = utils.Rc.Delete(detailKey)
  277. }
  278. br.Ret = 200
  279. br.Success = true
  280. br.Msg = "刷新成功"
  281. }