eta_api/services/data/excel/excel_op.go

package excel

import (
	"errors"
	excelModel "eta/eta_api/models/data_manage/excel"
	"eta/eta_api/models/system"
	"eta/eta_api/services"
	"eta/eta_api/services/alarm_msg"
	excel "eta/eta_api/services/excel"
	"eta/eta_api/utils"
	"fmt"
	"os"
	"strconv"
	"time"
)

// Delete excel删除
func Delete(excelInfo *excelModel.ExcelInfo, sysUser *system.Admin) (err error, errMsg string, isSendEmail bool) {
	isSendEmail = true

	// 操作权限校验
	{
		button := GetExcelInfoOpButton(sysUser, excelInfo.SysUserId, excelInfo.Source)
		if !button.DeleteButton {
			errMsg = "无操作权限"
			err = errors.New(errMsg)
			isSendEmail = false
			return
		}
	}

	// 自定义分析，需要做这个指标关联的校验
	if excelInfo.Source == utils.CUSTOM_ANALYSIS_TABLE {
		list, tmpErr := excelModel.GetExcelEdbMappingByExcelInfoId(excelInfo.ExcelInfoId)
		if tmpErr != nil {
			errMsg = `获取关联的指标信息失败`
			err = tmpErr
			return
		}

		if len(list) > 0 {
			errMsg = "已关联指标，不可删除！"
			err = errors.New(errMsg)
			isSendEmail = false
			return
		}
	}

	// 标记删除
	excelInfo.IsDelete = 1
	excelInfo.ModifyTime = time.Now()
	err = excelInfo.Update([]string{"IsDelete", "ModifyTime"})

	return
}

// Copy 复制excel
func Copy(oldExcelInfoId, excelClassifyId int, excelName string, sysUser *system.Admin) (excelInfo *excelModel.ExcelInfo, err error, errMsg string, isSendEmail bool) {
	isSendEmail = true

	excelName = utils.TrimLRStr(excelName)

	excelClassify, err := excelModel.GetExcelClassifyById(excelClassifyId)
	if err != nil {
		if err.Error() == utils.ErrNoRow() {
			errMsg = "分类不存在"
			err = errors.New(errMsg)
			isSendEmail = false
			return
		}
		errMsg = "获取分类信息失败"
		return
	}
	if excelClassify == nil {
		errMsg = "分类不存在"
		err = errors.New(errMsg)
		isSendEmail = false
		return
	}

	// 获取原ETA表格信息
	oldExcelInfo, err := excelModel.GetExcelInfoById(oldExcelInfoId)
	if err != nil {
		errMsg = "获取ETA表格失败"
		return
	}

	// 操作权限校验
	{
		button := GetExcelInfoOpButton(sysUser, oldExcelInfo.SysUserId, oldExcelInfo.Source)
		if !button.CopyButton {
			errMsg = "无操作权限"
			err = errors.New(errMsg)
			isSendEmail = false
			return
		}
	}

	// 检验分类下是否存在该表格名称
	{
		var condition string
		var pars []interface{}
		condition += " AND excel_classify_id=? "
		pars = append(pars, excelClassifyId)

		condition += " AND excel_name=? "
		pars = append(pars, excelName)

		count, tmpErr := excelModel.GetExcelInfoCountByCondition(condition, pars)
		if tmpErr != nil {
			errMsg = "判断表格名称是否存在失败"
			err = tmpErr
			return
		}
		if count > 0 {
			errMsg = "表格名称已存在，请重新填写表格名称"
			err = errors.New(errMsg)
			isSendEmail = false
			return
		}
	}

	// 表格信息
	timestamp := strconv.FormatInt(time.Now().UnixNano(), 10)
	excelInfo = &excelModel.ExcelInfo{
		//ExcelInfoId:     0,
		ExcelName:       excelName,
		Source:          oldExcelInfo.Source,
		ExcelType:       oldExcelInfo.ExcelType,
		UniqueCode:      utils.MD5(utils.EXCEL_DATA_PREFIX + "_" + timestamp),
		ExcelClassifyId: excelClassifyId,
		SysUserId:       sysUser.AdminId,
		SysUserRealName: sysUser.RealName,
		Content:         oldExcelInfo.Content,
		ExcelImage:      oldExcelInfo.ExcelImage,
		FileUrl:         oldExcelInfo.FileUrl,
		Sort:            0,
		IsDelete:        0,
		ModifyTime:      time.Now(),
		CreateTime:      time.Now(),
	}

	// 如果不是自定义分析，那么直接加主表就好了
	if excelInfo.Source != utils.CUSTOM_ANALYSIS_TABLE {
		err = excelModel.AddExcelInfo(excelInfo)
		if err != nil {
			errMsg = "保存失败"
		}

		return
	}

	// 自定义分析，需要有额外信息
	addSheetList := make([]excelModel.AddExcelSheetParams, 0)

	// 获取所有的sheet页
	oldSheetItemList, err := excelModel.GetAllSheetList(oldExcelInfo.ExcelInfoId)
	if err != nil {
		errMsg = `获取sheet页失败`
		return
	}

	// 获取所有的sheet页的sheet数据
	sheetCellDataMapList := make(map[int][]*excelModel.ExcelSheetData)
	{
		dataList, tmpErr := excelModel.GetAllSheetDataListByExcelInfoId(oldExcelInfo.ExcelInfoId)
		if tmpErr != nil {
			errMsg = `获取sheet页的单元格数据失败`
			err = tmpErr
			return
		}

		for _, cellData := range dataList {
			sheetDataList, ok := sheetCellDataMapList[cellData.ExcelSheetId]
			if !ok {
				sheetDataList = make([]*excelModel.ExcelSheetData, 0)
			}
			sheetCellDataMapList[cellData.ExcelSheetId] = append(sheetDataList, cellData)
		}
	}

	// sheet处理
	for _, sheetInfo := range oldSheetItemList {
		addSheetItem := excelModel.AddExcelSheetParams{
			ExcelSheetId: 0,
			ExcelInfoId:  0,
			SheetName:    sheetInfo.SheetName,
			Sort:         sheetInfo.Sort,
			Config:       sheetInfo.Config,
			CalcChain:    sheetInfo.CalcChain,
		}

		sheetDataList, ok := sheetCellDataMapList[sheetInfo.ExcelSheetId]
		if ok {
			for i, sheetData := range sheetDataList {
				sheetData.ExcelDataId = 0
				sheetData.ExcelSheetId = 0
				sheetData.ExcelInfoId = 0
				sheetDataList[i] = sheetData
			}
		}

		addSheetItem.DataList = sheetDataList

		addSheetList = append(addSheetList, addSheetItem)
	}

	// 添加表格
	err = excelModel.AddExcelInfoAndSheet(excelInfo, addSheetList)

	return
}

// UpdateExcelInfoFileUrl 更新excel表格的下载地址
func UpdateExcelInfoFileUrl(excelInfo *excelModel.ExcelInfo) {
	var err error
	defer func() {
		if err != nil {
			go alarm_msg.SendAlarmMsg(fmt.Sprintf("更新excel表格的下载地址失败，表格id：%d；表格名称：%s； ERR:%s", excelInfo.ExcelInfoId, excelInfo.ExcelName, err), 3)
		}
	}()
	fileName := excelInfo.ExcelName + "_" + excelInfo.UniqueCode + ".xlsx"

	var downloadFilePath string // excel文件下载地址

	switch excelInfo.Source {
	case utils.EXCEL_DEFAULT: // 自定义表格
		luckySheetData, err := excel.GetLuckySheetData(excelInfo.Content)
		if err != nil {
			fmt.Println("err:", err)
			return
		}
		//_, err = luckySheetData.GetTableDataByLuckySheetDataStr()
		downloadFilePath, err = luckySheetData.ToExcel()
	case utils.CUSTOM_ANALYSIS_TABLE: // 自定义分析表格
		downloadFilePath, err, _ = GenerateExcelCustomAnalysisExcel(excelInfo)
	}

	if err != nil {
		fmt.Println("err:", err)
		return
	}
	defer func() {
		_ = os.Remove(downloadFilePath)
	}()

	//上传到阿里云
	resourceUrl, err := services.UploadAliyunV2(fileName, downloadFilePath)
	if err != nil {
		return
	}
	excelInfo.FileUrl = resourceUrl
	err = excelInfo.Update([]string{"FileUrl"})
}