excel_info.go 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709
  1. package excel
  2. import (
  3. "encoding/json"
  4. "errors"
  5. "eta/eta_api/models/data_manage"
  6. "eta/eta_api/models/data_manage/excel"
  7. "eta/eta_api/models/data_manage/excel/request"
  8. "eta/eta_api/models/data_manage/excel/response"
  9. "eta/eta_api/models/system"
  10. "eta/eta_api/services/alarm_msg"
  11. "eta/eta_api/services/data"
  12. "eta/eta_api/services/data/data_manage_permission"
  13. "eta/eta_api/utils"
  14. "fmt"
  15. "sort"
  16. "time"
  17. )
  18. // GetExcelDetailInfoByExcelInfoId 根据表格id获取表格详情
  19. func GetExcelDetailInfoByExcelInfoId(excelInfoId, sysUserId int, lang string) (excelDetail response.ExcelInfoDetail, errMsg string, err error) {
  20. errMsg = `获取失败`
  21. //获取eta表格信息
  22. excelInfo, err := excel.GetExcelInfoById(excelInfoId)
  23. if err != nil {
  24. err = errors.New("获取ETA表格信息失败,Err:" + err.Error())
  25. if err.Error() == utils.ErrNoRow() {
  26. errMsg = "ETA表格被删除,请刷新页面"
  27. err = errors.New("ETA表格被删除,请刷新页面,Err:" + err.Error())
  28. }
  29. return
  30. }
  31. return formatExcelInfo2Detail(excelInfo, sysUserId, lang)
  32. }
  33. // GetExcelDetailInfoByUnicode 根据表格编码获取表格详情
  34. func GetExcelDetailInfoByUnicode(unicode string, sysUserId int, lang string) (excelDetail response.ExcelInfoDetail, errMsg string, err error) {
  35. errMsg = `获取失败`
  36. // 获取eta表格信息
  37. excelInfo, err := excel.GetExcelInfoByUnicode(unicode)
  38. if err != nil {
  39. err = errors.New("获取ETA表格信息失败,Err:" + err.Error())
  40. if err.Error() == utils.ErrNoRow() {
  41. errMsg = "ETA表格被删除,请刷新页面"
  42. err = errors.New("ETA表格被删除,请刷新页面,Err:" + err.Error())
  43. }
  44. return
  45. }
  46. return formatExcelInfo2Detail(excelInfo, sysUserId, lang)
  47. }
  48. func formatExcelInfo2Detail(excelInfo *excel.ExcelInfo, sysUserId int, lang string) (excelDetail response.ExcelInfoDetail, errMsg string, err error) {
  49. checkExcelInfo := excelInfo
  50. if excelInfo.Source == utils.BALANCE_TABLE {
  51. checkExcelInfoId := excelInfo.ExcelInfoId
  52. if excelInfo.BalanceType == 1 {
  53. checkExcelInfoId = excelInfo.RelExcelInfoId
  54. } else {
  55. if excelInfo.ParentId > 0 {
  56. checkExcelInfoId = excelInfo.ParentId
  57. }
  58. }
  59. if checkExcelInfoId != excelInfo.ExcelInfoId {
  60. checkExcelInfo, err = excel.GetExcelInfoById(checkExcelInfoId)
  61. if err != nil {
  62. errMsg = "获取平衡表格信息失败"
  63. err = errors.New("获取平衡表格信息失败,Err:" + err.Error())
  64. return
  65. }
  66. }
  67. }
  68. // 数据权限
  69. haveOperaAuth, err := data_manage_permission.CheckExcelPermissionByExcelInfoId(checkExcelInfo.ExcelInfoId, checkExcelInfo.ExcelClassifyId, checkExcelInfo.IsJoinPermission, sysUserId)
  70. if err != nil {
  71. err = errors.New("获取表格权限信息失败,Err" + err.Error())
  72. return
  73. }
  74. excelDetail = response.ExcelInfoDetail{
  75. ExcelInfoId: excelInfo.ExcelInfoId,
  76. Source: excelInfo.Source,
  77. ExcelType: excelInfo.ExcelType,
  78. ExcelName: excelInfo.ExcelName,
  79. UniqueCode: excelInfo.UniqueCode,
  80. ExcelClassifyId: excelInfo.ExcelClassifyId,
  81. SysUserId: excelInfo.SysUserId,
  82. SysUserRealName: excelInfo.SysUserRealName,
  83. Content: excelInfo.Content,
  84. ExcelImage: excelInfo.ExcelImage,
  85. FileUrl: excelInfo.FileUrl,
  86. Sort: excelInfo.Sort,
  87. IsDelete: excelInfo.IsDelete,
  88. ModifyTime: excelInfo.ModifyTime,
  89. CreateTime: excelInfo.CreateTime,
  90. TableData: nil,
  91. HaveOperaAuth: haveOperaAuth,
  92. ParentId: excelInfo.ParentId,
  93. BalanceType: excelInfo.BalanceType,
  94. UpdateUserId: excelInfo.UpdateUserId,
  95. UpdateUserRealName: excelInfo.UpdateUserRealName,
  96. RelExcelInfoId: excelInfo.RelExcelInfoId,
  97. }
  98. // 无权限,不需要返回数据
  99. if !haveOperaAuth {
  100. return
  101. }
  102. switch excelInfo.Source {
  103. case utils.TIME_TABLE: // 时间序列表格
  104. var tableDataConfig TableDataConfig
  105. err = json.Unmarshal([]byte(excelDetail.Content), &tableDataConfig)
  106. if err != nil {
  107. err = errors.New("表格json转结构体失败,Err:" + err.Error())
  108. return
  109. }
  110. result, tmpErr := GetDataByTableDataConfig(tableDataConfig)
  111. if tmpErr != nil {
  112. err = errors.New("获取最新的表格数据失败,Err:" + tmpErr.Error())
  113. return
  114. }
  115. if len(result.EdbInfoIdList) > 0 {
  116. classifyIdList := make([]int, 0)
  117. for _, v := range result.Data {
  118. classifyIdList = append(classifyIdList, v.ClassifyId)
  119. }
  120. classifyMap := make(map[int]*data_manage.EdbClassify)
  121. classifyList, tmpErr := data_manage.GetEdbClassifyByIdList(classifyIdList)
  122. if tmpErr != nil {
  123. err = errors.New("获取分类列表失败,Err:" + tmpErr.Error())
  124. return
  125. }
  126. for _, v := range classifyList {
  127. classifyMap[v.ClassifyId] = v
  128. }
  129. // 获取所有有权限的指标和分类
  130. permissionEdbIdList, permissionClassifyIdList, tmpErr := data_manage_permission.GetUserEdbAndClassifyPermissionList(sysUserId, 0, 0)
  131. if err != nil {
  132. err = errors.New("获取所有有权限的指标和分类失败,Err:" + tmpErr.Error())
  133. return
  134. }
  135. for i, v := range result.Data {
  136. if currClassify, ok := classifyMap[v.ClassifyId]; ok {
  137. result.Data[i].HaveOperaAuth = data_manage_permission.CheckEdbPermissionByPermissionIdList(v.IsJoinPermission, currClassify.IsJoinPermission, v.EdbInfoId, v.ClassifyId, permissionEdbIdList, permissionClassifyIdList)
  138. }
  139. }
  140. }
  141. excelDetail.TableData = result
  142. case utils.MIXED_TABLE, utils.BALANCE_TABLE: // 混合表格 平衡表
  143. var result request.MixedTableReq
  144. err = json.Unmarshal([]byte(excelDetail.Content), &result)
  145. if err != nil {
  146. err = errors.New("表格json转结构体失败,Err:" + err.Error())
  147. return
  148. }
  149. newData, tmpErr, tmpErrMsg := GetMixedTableCellData(result, lang)
  150. if tmpErr != nil {
  151. errMsg = "获取失败"
  152. if tmpErrMsg != `` {
  153. errMsg = tmpErrMsg
  154. }
  155. err = errors.New("获取最新的数据失败,Err:" + tmpErr.Error())
  156. return
  157. }
  158. result.Data = newData
  159. excelDetail.TableData = result
  160. }
  161. if excelDetail.Source == utils.BALANCE_TABLE {
  162. excelDetail.Button = GetBalanceExcelInfoOpButton(sysUserId, checkExcelInfo.SysUserId, excelDetail.HaveOperaAuth, checkExcelInfo.ExcelInfoId)
  163. }
  164. return
  165. }
  166. // GetExcelInfoOpButton 获取ETA表格的操作权限
  167. func GetExcelInfoOpButton(sysUser *system.Admin, belongUserId, source int, haveOperaAuth bool) (button excel.ExcelInfoDetailButton) {
  168. // 如果没有数据权限,那么直接返回
  169. if !haveOperaAuth {
  170. return
  171. }
  172. //非管理员角色查看其他用户创建的表格,可刷新、另存为、下载表格;
  173. button.RefreshButton = true
  174. button.CopyButton = true
  175. button.DownloadButton = true
  176. // 1、本用户创建的表格,可编辑、刷新、另存为、下载、删除,删除需二次确认;
  177. // 2、管理员角色对所有表格有如上权限;
  178. // 3、在线excel所有人都能编辑
  179. if sysUser.RoleTypeCode == utils.ROLE_TYPE_CODE_ADMIN || sysUser.RoleTypeCode == utils.ROLE_TYPE_CODE_FICC_ADMIN || sysUser.AdminId == belongUserId || source == utils.EXCEL_DEFAULT {
  180. button.OpButton = true
  181. button.DeleteButton = true
  182. }
  183. // 自定义分析
  184. if source == utils.CUSTOM_ANALYSIS_TABLE {
  185. if sysUser.RoleTypeCode == utils.ROLE_TYPE_CODE_ADMIN || sysUser.RoleTypeCode == utils.ROLE_TYPE_CODE_FICC_ADMIN || sysUser.AdminId == belongUserId {
  186. button.OpEdbButton = true // 生成、查看指标按钮
  187. button.RefreshEdbButton = true // 刷新指标按钮
  188. }
  189. }
  190. return
  191. }
  192. // GetFirstEdbDataList 获取第一列的数据
  193. func GetFirstEdbDataList(edbInfo *data_manage.EdbInfo, num int, manualDateList []string) (resultDataList []request.ManualDataReq, err error) {
  194. var dataList []*data_manage.EdbDataList
  195. switch edbInfo.EdbInfoType {
  196. case 0:
  197. dataList, err = data_manage.GetEdbDataList(edbInfo.Source, edbInfo.SubSource, edbInfo.EdbInfoId, ``, ``)
  198. case 1:
  199. _, dataList, _, _, err, _ = data.GetPredictDataListByPredictEdbInfoId(edbInfo.EdbInfoId, ``, ``, false)
  200. default:
  201. err = errors.New(fmt.Sprint("获取失败,指标类型异常", edbInfo.EdbInfoType))
  202. }
  203. if err != nil {
  204. return
  205. }
  206. // 获取需要的期数
  207. lenData := len(dataList)
  208. if lenData <= 0 {
  209. return
  210. }
  211. tmpManualDateNum := 0 // 手工数据的期数
  212. lenManualDate := len(manualDateList)
  213. if lenManualDate > 0 {
  214. sortDateList := manualDateList
  215. baseDateList := utils.StrArr{}
  216. baseDateList = append(baseDateList, sortDateList...)
  217. sort.Sort(baseDateList)
  218. sortDateList = append([]string{}, baseDateList...)
  219. lastData := dataList[lenData-1]
  220. lastDataDate, tmpErr := time.ParseInLocation(utils.FormatDate, lastData.DataTime, time.Local)
  221. if tmpErr != nil {
  222. err = tmpErr
  223. return
  224. }
  225. // 遍历倒序后的日期,然后匹配在实际数据之后日期的个数
  226. for _, tmpDateStr := range sortDateList {
  227. tmpDate, tmpErr := time.ParseInLocation(utils.FormatDate, tmpDateStr, time.Local)
  228. if tmpErr != nil {
  229. err = tmpErr
  230. return
  231. }
  232. if tmpDate.After(lastDataDate) {
  233. tmpManualDateNum++
  234. continue
  235. }
  236. break
  237. }
  238. }
  239. // 需要的期数减去手工数据的期数,这才是A列指标需要的数据
  240. num = num - tmpManualDateNum
  241. if num > lenData {
  242. num = lenData
  243. }
  244. latestDateTime, _ := time.ParseInLocation(utils.FormatDate, edbInfo.LatestDate, time.Local)
  245. for i := 1; i <= num; i++ {
  246. dataTime, _ := time.ParseInLocation(utils.FormatDate, dataList[lenData-i].DataTime, time.Local)
  247. dataType := 1
  248. // 如果是预测指标,且当前值的日期,晚于实际日期,那么是预测值
  249. if edbInfo.EdbInfoType == 1 && dataTime.After(latestDateTime) {
  250. dataType = 5
  251. }
  252. resultDataList = append(resultDataList, request.ManualDataReq{
  253. DataType: dataType,
  254. DataTime: dataList[lenData-i].DataTime,
  255. ShowValue: fmt.Sprint(dataList[lenData-i].Value),
  256. Value: fmt.Sprint(dataList[lenData-i].Value),
  257. DataTimeType: 1,
  258. })
  259. }
  260. return
  261. }
  262. // GetOtherEdbDataList 获取其他列的数据
  263. func GetOtherEdbDataList(edbInfo *data_manage.EdbInfo, dateList []string) (resultDataList []request.ManualDataReq, err error) {
  264. lenDate := len(dateList)
  265. if lenDate <= 0 {
  266. return
  267. }
  268. sortDateList := dateList
  269. baseDateList := utils.StrArr{}
  270. baseDateList = append(baseDateList, sortDateList...)
  271. sort.Sort(baseDateList)
  272. sortDateList = append([]string{}, baseDateList...)
  273. endDateTime, err := time.ParseInLocation(utils.FormatDate, sortDateList[0], time.Local)
  274. if err != nil {
  275. return
  276. }
  277. firstDateTime, err := time.ParseInLocation(utils.FormatDate, sortDateList[lenDate-1], time.Local)
  278. if err != nil {
  279. return
  280. }
  281. var dataList []*data_manage.EdbDataList
  282. switch edbInfo.EdbInfoType {
  283. case 0:
  284. dataList, err = data_manage.GetEdbDataList(edbInfo.Source, edbInfo.SubSource, edbInfo.EdbInfoId, ``, ``)
  285. case 1:
  286. _, dataList, _, _, err, _ = data.GetPredictDataListByPredictEdbInfoId(edbInfo.EdbInfoId, ``, ``, false)
  287. default:
  288. err = errors.New(fmt.Sprint("获取失败,指标类型异常", edbInfo.EdbInfoType))
  289. }
  290. if err != nil {
  291. return
  292. }
  293. // 获取日期内的数据(包含开始日期前一个日期,以及 结束日期后一个日期,目的为了做空日期时的 插值法兼容)
  294. baseDataList := make([]*data_manage.EdbDataList, 0)
  295. var lastData *data_manage.EdbDataList
  296. var isInsert bool
  297. for _, data := range dataList {
  298. tmpDate := data.DataTime
  299. tmpDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, tmpDate, time.Local)
  300. if tmpErr != nil {
  301. err = tmpErr
  302. return
  303. }
  304. if tmpDateTime.Before(firstDateTime) {
  305. lastData = data
  306. continue
  307. }
  308. // 如果是第一次写入数据
  309. if !isInsert && lastData != nil {
  310. baseDataList = append(baseDataList, lastData)
  311. }
  312. if tmpDateTime.After(endDateTime) {
  313. baseDataList = append(baseDataList, data)
  314. break
  315. }
  316. baseDataList = append(baseDataList, data)
  317. isInsert = true
  318. }
  319. // 实际数据的日期map
  320. realValMap := make(map[string]string)
  321. for _, v := range baseDataList {
  322. realValMap[v.DataTime] = v.DataTime
  323. }
  324. // 插值法处理
  325. handleDataMap := make(map[string]float64)
  326. err = data.HandleDataByLinearRegression(baseDataList, handleDataMap)
  327. if err != nil {
  328. return
  329. }
  330. latestDateTime, _ := time.ParseInLocation(utils.FormatDate, edbInfo.LatestDate, time.Local)
  331. // 对于不存在的数据做补充
  332. for _, date := range sortDateList {
  333. dataType := 1
  334. if _, ok := realValMap[date]; !ok {
  335. dataType = 2
  336. } else {
  337. dataTime, _ := time.ParseInLocation(utils.FormatDate, date, time.Local)
  338. // 如果是预测指标,且当前值的日期,晚于实际日期,那么是预测值
  339. if edbInfo.EdbInfoType == 1 && dataTime.After(latestDateTime) {
  340. dataType = 5
  341. }
  342. }
  343. var value, showValue string
  344. if tmpVal, ok := handleDataMap[date]; ok {
  345. value = fmt.Sprint(tmpVal)
  346. showValue = value
  347. } else {
  348. dataType = 3
  349. }
  350. resultDataList = append(resultDataList, request.ManualDataReq{
  351. DataType: dataType,
  352. DataTime: date,
  353. ShowValue: showValue,
  354. Value: value,
  355. })
  356. }
  357. return
  358. }
  359. // GetFirstHistoryEdbDataList 获取指标的历史的数据
  360. func GetFirstHistoryEdbDataList(edbInfo *data_manage.EdbInfo, num int, endDate string) (resultDataList []request.ManualDataReq, err error) {
  361. endDateTime, err := time.ParseInLocation(utils.FormatDate, endDate, time.Local)
  362. if err != nil {
  363. return
  364. }
  365. var dataList []*data_manage.EdbDataList
  366. switch edbInfo.EdbInfoType {
  367. case 0:
  368. dataList, err = data_manage.GetEdbDataList(edbInfo.Source, edbInfo.SubSource, edbInfo.EdbInfoId, ``, endDate)
  369. case 1:
  370. _, dataList, _, _, err, _ = data.GetPredictDataListByPredictEdbInfoId(edbInfo.EdbInfoId, ``, endDate, true)
  371. default:
  372. err = errors.New(fmt.Sprint("获取失败,指标类型异常", edbInfo.EdbInfoType))
  373. }
  374. if err != nil {
  375. return
  376. }
  377. // 获取需要的期数
  378. lenData := len(dataList)
  379. if lenData <= 0 {
  380. return
  381. }
  382. lastData := dataList[lenData-1]
  383. lastDataDateTime, err := time.ParseInLocation(utils.FormatDate, lastData.DataTime, time.Local)
  384. if err != nil {
  385. return
  386. }
  387. if endDateTime.Equal(lastDataDateTime) || lastDataDateTime.After(endDateTime) {
  388. dataList = dataList[:lenData-1]
  389. lenData = len(dataList)
  390. }
  391. if num > lenData {
  392. num = lenData
  393. }
  394. latestDateTime, _ := time.ParseInLocation(utils.FormatDate, edbInfo.LatestDate, time.Local)
  395. for i := 1; i <= num; i++ {
  396. dataTime, _ := time.ParseInLocation(utils.FormatDate, dataList[lenData-i].DataTime, time.Local)
  397. dataType := 1
  398. // 如果是预测指标,且当前值的日期,晚于实际日期,那么是预测值
  399. if edbInfo.EdbInfoType == 1 && dataTime.After(latestDateTime) {
  400. dataType = 5
  401. }
  402. resultDataList = append(resultDataList, request.ManualDataReq{
  403. DataType: dataType,
  404. DataTime: dataList[lenData-i].DataTime,
  405. ShowValue: fmt.Sprint(dataList[lenData-i].Value),
  406. Value: fmt.Sprint(dataList[lenData-i].Value),
  407. })
  408. }
  409. return
  410. }
  411. // GetEdbIdsFromExcelCodes 获取表格中的指标IDs
  412. func GetEdbIdsFromExcelCodes(excelCodes []string, sysUserId int, lang string) (edbIds []int, err error) {
  413. edbIds = make([]int, 0)
  414. edbIdExist := make(map[int]bool)
  415. for _, v := range excelCodes {
  416. // 表格详情
  417. detail, msg, e := GetExcelDetailInfoByUnicode(v, sysUserId, lang)
  418. if e != nil {
  419. err = fmt.Errorf("GetExcelDetailInfoByExcelInfoId err: %s, errMsg: %s", e.Error(), msg)
  420. return
  421. }
  422. // 自定义表格
  423. if detail.Source == utils.TIME_TABLE {
  424. jsonByte, e := json.Marshal(detail.TableData)
  425. if e != nil {
  426. err = fmt.Errorf("JSON格式化自定义表格数据失败, Err: %s", e.Error())
  427. return
  428. }
  429. var tableData request.TableDataReq
  430. if e = json.Unmarshal(jsonByte, &tableData); e != nil {
  431. err = fmt.Errorf("解析自定义表格数据失败, Err: %s", e.Error())
  432. return
  433. }
  434. for _, tv := range tableData.EdbInfoIdList {
  435. if edbIdExist[tv] {
  436. continue
  437. }
  438. edbIdExist[tv] = true
  439. edbIds = append(edbIds, tv)
  440. }
  441. }
  442. // 混合表格
  443. if detail.Source == utils.MIXED_TABLE {
  444. jsonByte, e := json.Marshal(detail.TableData)
  445. if e != nil {
  446. err = fmt.Errorf("JSON格式化混合表格数据失败, Err: %s", e.Error())
  447. return
  448. }
  449. var tableData request.MixedTableReq
  450. if e = json.Unmarshal(jsonByte, &tableData); e != nil {
  451. err = fmt.Errorf("解析混合表格数据失败, Err: %s", e.Error())
  452. return
  453. }
  454. if len(tableData.Data) > 0 {
  455. for _, td := range tableData.Data {
  456. for _, tv := range td {
  457. if tv.EdbInfoId > 0 && !edbIdExist[tv.EdbInfoId] {
  458. edbIdExist[tv.EdbInfoId] = true
  459. edbIds = append(edbIds, tv.EdbInfoId)
  460. }
  461. }
  462. }
  463. }
  464. }
  465. }
  466. return
  467. }
  468. // GetExcelEdbBatchRefreshKey 获取批量刷新表格指标缓存key
  469. func GetExcelEdbBatchRefreshKey(source string, reportId, chapterId int) string {
  470. if source == `` {
  471. return ``
  472. }
  473. return fmt.Sprint("batch_refresh_excel_edb:", source, ":", reportId, ":", chapterId)
  474. }
  475. // ReplaceEdbInExcel 替换表格中的指标
  476. func ReplaceEdbInExcel(oldEdbInfo, newEdbInfo *data_manage.EdbInfo) (err error) {
  477. defer func() {
  478. if err != nil {
  479. go alarm_msg.SendAlarmMsg("替换表格中的指标失败提醒,errmsg:"+err.Error(), 3)
  480. }
  481. }()
  482. //查询和指标相关的 表格
  483. mappingList, err := excel.GetExcelEdbMappingByEdbInfoId(oldEdbInfo.EdbInfoId)
  484. if err != nil {
  485. err = fmt.Errorf("查询和指标相关的表格失败,错误:%s", err.Error())
  486. return
  487. }
  488. updateList := make([]*excel.ExcelInfo, 0)
  489. // 循环列表,根据表格类型单独处理
  490. for _, excelMapping := range mappingList {
  491. //查询和指标相关的混合表格
  492. excelInfo, tmpErr := excel.GetExcelInfoById(excelMapping.ExcelInfoId)
  493. if tmpErr != nil {
  494. err = fmt.Errorf("查询和指标相关的混合表格失败,错误:%s", tmpErr.Error())
  495. return
  496. }
  497. // 清除缓存
  498. key := utils.HZ_CHART_LIB_EXCEL_TABLE_DETAIL + ":" + excelInfo.UniqueCode
  499. if utils.Re == nil {
  500. _ = utils.Rc.Delete(key)
  501. }
  502. // 根据表格类型,调用不同的处理函数
  503. switch excelMapping.Source {
  504. case utils.TIME_TABLE: // 时间序列表格
  505. // 替换余额表格中的指标
  506. newExcelInfo, e := replaceEdbInTimeExcel(oldEdbInfo, newEdbInfo, excelInfo)
  507. if e != nil {
  508. err = fmt.Errorf("替换余额表格中的指标失败,错误:%s", e.Error())
  509. return
  510. }
  511. updateList = append(updateList, newExcelInfo)
  512. case utils.MIXED_TABLE, utils.BALANCE_TABLE:
  513. // 替换余额表格中的指标
  514. newExcelInfo, e := replaceEdbInBalanceExcel(oldEdbInfo, newEdbInfo, excelInfo)
  515. if e != nil {
  516. err = fmt.Errorf("替换余额表格中的指标失败,错误:%s", e.Error())
  517. return
  518. }
  519. updateList = append(updateList, newExcelInfo)
  520. default:
  521. // 其他表格类型的处理逻辑
  522. }
  523. }
  524. err = excel.ReplaceEdbInExcel(oldEdbInfo.EdbInfoId, newEdbInfo.EdbInfoId, updateList)
  525. if err != nil {
  526. err = fmt.Errorf("替换表格中的指标失败,错误:%s", err.Error())
  527. return
  528. }
  529. //todo 是否需要刷新表格中的指标数据
  530. return
  531. }
  532. func replaceEdbInBalanceExcel(oldEdbInfo, newEdbInfo *data_manage.EdbInfo, excelInfo *excel.ExcelInfo) (newExcelInfo *excel.ExcelInfo, err error) {
  533. newExcelInfo = excelInfo
  534. var mixedTableReq request.MixedTableReq
  535. err = json.Unmarshal([]byte(excelInfo.Content), &mixedTableReq)
  536. if err != nil {
  537. err = fmt.Errorf("表格json转结构体失败,Err:" + err.Error())
  538. return
  539. }
  540. // 处理data
  541. configList := mixedTableReq.Data
  542. for ck, rowList := range configList {
  543. for rk, cell := range rowList {
  544. switch cell.DataType {
  545. case request.EdbDT: // 指标信息
  546. if cell.EdbInfoId == oldEdbInfo.EdbInfoId {
  547. //更换成新指标ID
  548. configList[ck][rk].EdbInfoId = newEdbInfo.EdbInfoId
  549. }
  550. case request.InsertDataDT, request.PopInsertDataDT: // 插值、弹框插值
  551. if cell.EdbInfoId == oldEdbInfo.EdbInfoId {
  552. //更换成新指标ID
  553. configList[ck][rk].EdbInfoId = newEdbInfo.EdbInfoId
  554. }
  555. case request.InsertEdbCalculateDataDT: // 插入指标计算公式生成的值
  556. var config request.CalculateConf
  557. err = json.Unmarshal([]byte(cell.Value), &config)
  558. if err != nil {
  559. return
  560. }
  561. if cell.EdbInfoId == oldEdbInfo.EdbInfoId {
  562. //更换成新指标ID
  563. configList[ck][rk].EdbInfoId = newEdbInfo.EdbInfoId
  564. }
  565. if config.EdbInfoId == oldEdbInfo.EdbInfoId {
  566. config.EdbInfoId = newEdbInfo.EdbInfoId
  567. var configStr []byte
  568. configStr, err = json.Marshal(config)
  569. if err != nil {
  570. return
  571. }
  572. configList[ck][rk].Value = string(configStr)
  573. }
  574. case request.DateDT: // 日期类型
  575. // 指标日期类型的单元格需要额外将指标id取出来
  576. if cell.DataTimeType == request.EdbDateDT {
  577. var config request.EdbDateConf
  578. err = json.Unmarshal([]byte(cell.Value), &config)
  579. if err != nil {
  580. return
  581. }
  582. if config.EdbInfoId == oldEdbInfo.EdbInfoId {
  583. config.EdbInfoId = newEdbInfo.EdbInfoId
  584. var configStr []byte
  585. configStr, err = json.Marshal(config)
  586. if err != nil {
  587. return
  588. }
  589. configList[ck][rk].Value = string(configStr)
  590. }
  591. }
  592. }
  593. }
  594. }
  595. mixedTableReq.Data = configList
  596. var newContentByte []byte
  597. newContentByte, err = json.Marshal(mixedTableReq)
  598. if err != nil {
  599. return
  600. }
  601. // 生成的新内容替换原先的旧内容
  602. excelInfo.Content = string(newContentByte)
  603. newExcelInfo = excelInfo
  604. return
  605. }
  606. func replaceEdbInTimeExcel(oldEdbInfo, newEdbInfo *data_manage.EdbInfo, excelInfo *excel.ExcelInfo) (newExcelInfo *excel.ExcelInfo, err error) {
  607. newExcelInfo = excelInfo
  608. var tableDataConfig TableDataConfig
  609. err = json.Unmarshal([]byte(excelInfo.Content), &tableDataConfig)
  610. if err != nil {
  611. err = errors.New("表格json转结构体失败,Err:" + err.Error())
  612. return
  613. }
  614. if len(tableDataConfig.EdbInfoIdList) <= 0 {
  615. return
  616. }
  617. // 实际期数没有的情况下,直接返回吧
  618. if tableDataConfig.Num <= 0 {
  619. return
  620. }
  621. // 先处理edbInfoList
  622. for k, id := range tableDataConfig.EdbInfoIdList {
  623. if id == oldEdbInfo.EdbInfoId {
  624. tableDataConfig.EdbInfoIdList[k] = newEdbInfo.EdbInfoId
  625. }
  626. }
  627. // 先处理tableEdbInfoList
  628. for k, tableEdbInfo := range tableDataConfig.TableEdbInfoList {
  629. if tableEdbInfo.EdbInfoId == oldEdbInfo.EdbInfoId {
  630. tableDataConfig.TableEdbInfoList[k].EdbInfoId = newEdbInfo.EdbInfoId
  631. }
  632. }
  633. var newContentByte []byte
  634. newContentByte, err = json.Marshal(tableDataConfig)
  635. if err != nil {
  636. return
  637. }
  638. // 生成的新内容替换原先的旧内容
  639. excelInfo.Content = string(newContentByte)
  640. newExcelInfo = excelInfo
  641. return
  642. }