excel_op.go 8.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305
  1. package excel
  2. import (
  3. "errors"
  4. excelModel "eta/eta_api/models/data_manage/excel"
  5. "eta/eta_api/models/system"
  6. "eta/eta_api/services"
  7. "eta/eta_api/services/alarm_msg"
  8. "eta/eta_api/services/data/data_manage_permission"
  9. excel "eta/eta_api/services/excel"
  10. "eta/eta_api/utils"
  11. "fmt"
  12. "os"
  13. "strconv"
  14. "time"
  15. )
  16. // Delete excel删除
  17. func Delete(excelInfo *excelModel.ExcelInfo, sysUser *system.Admin) (err error, errMsg string, isSendEmail bool) {
  18. isSendEmail = true
  19. // 操作权限校验
  20. {
  21. // 数据权限
  22. haveOperaAuth, tmpErr := data_manage_permission.CheckExcelPermissionByExcelInfoId(excelInfo.ExcelInfoId, excelInfo.ExcelClassifyId, excelInfo.IsJoinPermission, sysUser.AdminId)
  23. if err != nil {
  24. errMsg = "获取ETA表格权限失败"
  25. err = errors.New("获取ETA表格权限失败,Err:" + tmpErr.Error())
  26. return
  27. }
  28. button := GetExcelInfoOpButton(sysUser, excelInfo.SysUserId, excelInfo.Source, haveOperaAuth)
  29. if !button.DeleteButton {
  30. errMsg = "无操作权限"
  31. err = errors.New(errMsg)
  32. isSendEmail = false
  33. return
  34. }
  35. }
  36. // 自定义分析,需要做这个指标关联的校验
  37. if excelInfo.Source == utils.CUSTOM_ANALYSIS_TABLE {
  38. list, tmpErr := excelModel.GetExcelEdbMappingByExcelInfoId(excelInfo.ExcelInfoId)
  39. if tmpErr != nil {
  40. errMsg = `获取关联的指标信息失败`
  41. err = tmpErr
  42. return
  43. }
  44. if len(list) > 0 {
  45. errMsg = "已关联指标,不可删除!"
  46. err = errors.New(errMsg)
  47. isSendEmail = false
  48. return
  49. }
  50. }
  51. // todo 如果父级删除是否删除子表和静态表,同时删除图表和图表数据
  52. // 标记删除
  53. excelInfo.IsDelete = 1
  54. excelInfo.ModifyTime = time.Now()
  55. err = excelInfo.Update([]string{"IsDelete", "ModifyTime"})
  56. return
  57. }
  58. // Copy 复制excel
  59. func Copy(oldExcelInfoId, excelClassifyId int, excelName string, sysUser *system.Admin) (excelInfo *excelModel.ExcelInfo, err error, errMsg string, isSendEmail bool) {
  60. isSendEmail = true
  61. excelName = utils.TrimLRStr(excelName)
  62. excelClassify, err := excelModel.GetExcelClassifyById(excelClassifyId)
  63. if err != nil {
  64. if err.Error() == utils.ErrNoRow() {
  65. errMsg = "分类不存在"
  66. err = errors.New(errMsg)
  67. isSendEmail = false
  68. return
  69. }
  70. errMsg = "获取分类信息失败"
  71. return
  72. }
  73. if excelClassify == nil {
  74. errMsg = "分类不存在"
  75. err = errors.New(errMsg)
  76. isSendEmail = false
  77. return
  78. }
  79. // 获取原ETA表格信息
  80. oldExcelInfo, err := excelModel.GetExcelInfoById(oldExcelInfoId)
  81. if err != nil {
  82. errMsg = "获取ETA表格失败"
  83. return
  84. }
  85. // 操作权限校验
  86. {
  87. // 数据权限
  88. haveOperaAuth, tmpErr := data_manage_permission.CheckExcelPermissionByExcelInfoId(oldExcelInfo.ExcelInfoId, oldExcelInfo.ExcelClassifyId, oldExcelInfo.IsJoinPermission, sysUser.AdminId)
  89. if err != nil {
  90. errMsg = "获取ETA表格权限失败"
  91. err = errors.New("获取ETA表格权限失败,Err:" + tmpErr.Error())
  92. return
  93. }
  94. button := GetExcelInfoOpButton(sysUser, oldExcelInfo.SysUserId, oldExcelInfo.Source, haveOperaAuth)
  95. if !button.CopyButton {
  96. errMsg = "无操作权限"
  97. err = errors.New(errMsg)
  98. isSendEmail = false
  99. return
  100. }
  101. }
  102. // 检验分类下是否存在该表格名称
  103. {
  104. var condition string
  105. var pars []interface{}
  106. condition += " AND excel_classify_id=? "
  107. pars = append(pars, excelClassifyId)
  108. condition += " AND excel_name=? "
  109. pars = append(pars, excelName)
  110. count, tmpErr := excelModel.GetExcelInfoCountByCondition(condition, pars)
  111. if tmpErr != nil {
  112. errMsg = "判断表格名称是否存在失败"
  113. err = tmpErr
  114. return
  115. }
  116. if count > 0 {
  117. errMsg = "表格名称已存在,请重新填写表格名称"
  118. err = errors.New(errMsg)
  119. isSendEmail = false
  120. return
  121. }
  122. }
  123. // 表格信息
  124. timestamp := strconv.FormatInt(time.Now().UnixNano(), 10)
  125. excelInfo = &excelModel.ExcelInfo{
  126. //ExcelInfoId: 0,
  127. ExcelName: excelName,
  128. Source: oldExcelInfo.Source,
  129. ExcelType: oldExcelInfo.ExcelType,
  130. UniqueCode: utils.MD5(utils.EXCEL_DATA_PREFIX + "_" + timestamp),
  131. ExcelClassifyId: excelClassifyId,
  132. SysUserId: sysUser.AdminId,
  133. SysUserRealName: sysUser.RealName,
  134. Content: oldExcelInfo.Content,
  135. ExcelImage: oldExcelInfo.ExcelImage,
  136. FileUrl: oldExcelInfo.FileUrl,
  137. Sort: 0,
  138. IsDelete: 0,
  139. ModifyTime: time.Now(),
  140. CreateTime: time.Now(),
  141. }
  142. // 如果不是自定义分析,那么直接加主表就好了
  143. if excelInfo.Source != utils.CUSTOM_ANALYSIS_TABLE {
  144. // 获取excel与指标的关系表
  145. list, tmpErr := excelModel.GetAllExcelEdbMappingByExcelInfoId(excelInfo.ExcelInfoId)
  146. if tmpErr != nil {
  147. errMsg = "获取失败"
  148. err = tmpErr
  149. return
  150. }
  151. for k, v := range list {
  152. v.ExcelEdbMappingId = 0
  153. v.ExcelInfoId = 0
  154. list[k] = v
  155. }
  156. var childExcel *excelModel.ExcelInfo
  157. err = excelModel.AddExcelInfo(excelInfo, list, childExcel)
  158. if err != nil {
  159. errMsg = "保存失败"
  160. }
  161. return
  162. }
  163. // 自定义分析,需要有额外信息
  164. addSheetList := make([]excelModel.AddExcelSheetParams, 0)
  165. // 获取所有的sheet页
  166. oldSheetItemList, err := excelModel.GetAllSheetList(oldExcelInfo.ExcelInfoId)
  167. if err != nil {
  168. errMsg = `获取sheet页失败`
  169. return
  170. }
  171. // 获取所有的sheet页的sheet数据
  172. sheetCellDataMapList := make(map[int][]*excelModel.ExcelSheetData)
  173. {
  174. dataList, tmpErr := excelModel.GetAllSheetDataListByExcelInfoId(oldExcelInfo.ExcelInfoId)
  175. if tmpErr != nil {
  176. errMsg = `获取sheet页的单元格数据失败`
  177. err = tmpErr
  178. return
  179. }
  180. for _, cellData := range dataList {
  181. sheetDataList, ok := sheetCellDataMapList[cellData.ExcelSheetId]
  182. if !ok {
  183. sheetDataList = make([]*excelModel.ExcelSheetData, 0)
  184. }
  185. sheetCellDataMapList[cellData.ExcelSheetId] = append(sheetDataList, cellData)
  186. }
  187. }
  188. // sheet处理
  189. for _, sheetInfo := range oldSheetItemList {
  190. addSheetItem := excelModel.AddExcelSheetParams{
  191. ExcelSheetId: 0,
  192. ExcelInfoId: 0,
  193. SheetName: sheetInfo.SheetName,
  194. Sort: sheetInfo.Sort,
  195. Config: sheetInfo.Config,
  196. CalcChain: sheetInfo.CalcChain,
  197. }
  198. sheetDataList, ok := sheetCellDataMapList[sheetInfo.ExcelSheetId]
  199. if ok {
  200. for i, sheetData := range sheetDataList {
  201. sheetData.ExcelDataId = 0
  202. sheetData.ExcelSheetId = 0
  203. sheetData.ExcelInfoId = 0
  204. sheetDataList[i] = sheetData
  205. }
  206. }
  207. addSheetItem.DataList = sheetDataList
  208. addSheetList = append(addSheetList, addSheetItem)
  209. }
  210. // 添加表格
  211. err = excelModel.AddExcelInfoAndSheet(excelInfo, addSheetList)
  212. return
  213. }
  214. // UpdateExcelInfoFileUrl 更新excel表格的下载地址
  215. func UpdateExcelInfoFileUrl(excelInfo *excelModel.ExcelInfo) {
  216. var err error
  217. defer func() {
  218. if err != nil {
  219. go alarm_msg.SendAlarmMsg(fmt.Sprintf("更新excel表格的下载地址失败,表格id:%d;表格名称:%s; ERR:%s", excelInfo.ExcelInfoId, excelInfo.ExcelName, err), 3)
  220. utils.FileLog.Info(fmt.Sprintf("更新excel表格的下载地址失败,表格id:%d;表格名称:%s; ERR:%s", excelInfo.ExcelInfoId, excelInfo.ExcelName, err), 3)
  221. }
  222. }()
  223. fileName := excelInfo.ExcelName + "_" + excelInfo.UniqueCode + ".xlsx"
  224. var downloadFilePath string // excel文件下载地址
  225. switch excelInfo.Source {
  226. case utils.EXCEL_DEFAULT: // 自定义表格
  227. luckySheetData, tmpErr := excel.GetLuckySheetData(excelInfo.Content)
  228. if tmpErr != nil {
  229. err = tmpErr
  230. fmt.Println("err:", err)
  231. return
  232. }
  233. //_, err = luckySheetData.GetTableDataByLuckySheetDataStr()
  234. downloadFilePath, err = luckySheetData.ToExcel()
  235. case utils.CUSTOM_ANALYSIS_TABLE: // 自定义分析表格
  236. downloadFilePath, err, _ = GenerateExcelCustomAnalysisExcel(excelInfo)
  237. }
  238. if err != nil {
  239. fmt.Println("err:", err)
  240. return
  241. }
  242. defer func() {
  243. _ = os.Remove(downloadFilePath)
  244. }()
  245. var resourceUrl string
  246. //上传到阿里云
  247. //if utils.ObjectStorageClient == "minio" {
  248. // resourceUrl, err = services.UploadImgToMinIo(fileName, downloadFilePath)
  249. //} else {
  250. // resourceUrl, err = services.UploadAliyunV2(fileName, downloadFilePath)
  251. //}
  252. //if err != nil {
  253. // return
  254. //}
  255. ossClient := services.NewOssClient()
  256. if ossClient == nil {
  257. err = fmt.Errorf("初始化OSS服务失败")
  258. return
  259. }
  260. resourceUrl, err = ossClient.UploadFile(fileName, downloadFilePath, "")
  261. if err != nil {
  262. err = fmt.Errorf("文件上传失败, Err: %s", err.Error())
  263. return
  264. }
  265. excelInfo.FileUrl = resourceUrl
  266. err = excelInfo.Update([]string{"FileUrl"})
  267. }