mysteel_watch/services/index_merge_v2.go

package services

import (
	"fmt"
	"github.com/rdlucklib/rdluck_tools/paging"
	"github.com/xuri/excelize/v2"
	"hongze/mysteel_watch/global"
	"hongze/mysteel_watch/utils"
	"hongze/mysteel_watch/watch"
	"os"
	"strconv"
	"time"
)

// Merge 合并指标
func MergeV2() {
	fmt.Println("MergeV2 start")
	/*
		年底，季度，存放在一个excel中
		月度30个指标存一个excel
		周度25个指标存放一个excel
		日度20个指标存一个excel
	*/
	//年度->38
	//IndexYearMerge()
	////季度->5
	//IndexSeasonMerge()
	////月度->86
	//IndexMonthMerge()
	////周度->292
	//IndexWeekMerge()
	//日度
	IndexDayMergeV2()
	//周度
	IndexWeekMergeV2()
	fmt.Println("MergeV2 end")
}

// IndexDayMerge 日度指标合并
func IndexDayMergeV2() {
	frequency := "日度"
	filePre := "day"
	pageSize := 150 //每个excel表需要的指标数
	startDate := time.Now().AddDate(0, 0, -7).Format(utils.FormatDate)
	endDate := time.Now().Format(utils.FormatDate)
	err := indexMergeV2(frequency, startDate, endDate, filePre, pageSize)
	if err != nil {
		fmt.Println("合并日度指标失败，err:", err)
	}
}

// IndexDayMerge 日度指标合并
func IndexWeekMergeV2() {
	frequency := "周度"
	filePre := "week"
	pageSize := 150 //每个excel表需要的指标数
	startDate := time.Now().AddDate(0, 0, -49).Format(utils.FormatDate)
	endDate := utils.GetNowWeekMonday().Format(utils.FormatDate)
	err := indexMergeV2(frequency, startDate, endDate, filePre, pageSize)
	if err != nil {
		fmt.Println("合并日度指标失败，err:", err)
	}
}

// indexMerge 指标合并
// @params frequency string 频度
// @params filePre string 文件名前缀
// @params pageSize int 每个excel表中的指标数量
func indexMergeV2(frequency, startDate, endDate, filePre string, pageSize int) (err error) {

	firstIndexCount := pageSize //第一个添加的excel表需要的指标数

	frequencyReq := watch.GetMaxFileIndexReq{
		Frequency: frequency,
	}
	// 获取未合并的指标总数量
	total, err := watch.GetIndexCountByFrequencyV2(frequencyReq)
	if err != nil {
		fmt.Println("GetIndexCountByFrequencyV2 Err:" + err.Error())
		return
	}
	//未合并的指标总数量小于等于0，说明没有需要合并的指标
	if total <= 0 {
		return
	}

	fileName := filePre + ".xlsx" //保存的文件名称
	templatePath := global.CONFIG.Serve.IndexSaveDir + "index_template.xlsx"
	templateFile, tmpErr := excelize.OpenFile(templatePath)
	if tmpErr != nil {
		fmt.Println("打开文件失败,Err:" + tmpErr.Error())
		err = tmpErr
		return
	}

	mergeIndexFilePath := global.CONFIG.Serve.IndexMergeSaveDir + fileName

	if utils.FileIsExist(mergeIndexFilePath) { //修改文件
		err = os.Remove(mergeIndexFilePath)
		if err != nil {
			fmt.Println("os.Remove Err:" + err.Error() + "; mergeIndexFilePath:" + mergeIndexFilePath)
			return err
		}
	}

	// 除去第一个excel表的剩余需要添加的指标数量
	nextTotal := int(total) - firstIndexCount
	// 总文件数
	totalPage := paging.PageCount(nextTotal, pageSize)
	totalPage = totalPage + 1 //需要加上除去在外面的指标（第一张excel）
	fmt.Println("total:", total)
	fmt.Println("totalPage:", totalPage)
	sheetMap := make(map[string]string)
	for i := 1; i <= totalPage; i++ { //sheet
		startSize := paging.StartIndex(i, pageSize)
		// 当下需要合并到excel的指标
		pageList, tmpErr := watch.GetIndexPageByFrequency(frequency, endDate, startSize, pageSize)
		if tmpErr != nil {
			fmt.Println("GetIndexLimitByFrequency Err:" + tmpErr.Error())
			err = tmpErr
			return
		}

		commentResult, tmpErr := GetIndexCommentV3(mergeIndexFilePath, startDate, pageList)
		if tmpErr != nil {
			fmt.Println("GetIndexCommentV2 Err:" + tmpErr.Error())
			err = tmpErr
			return
		}

		fmt.Println("Sheet" + strconv.Itoa(i))

		sheetName := "Sheet" + strconv.Itoa(i)
		sheetMap[sheetName] = sheetName
		if i > 1 {
			templateFile.NewSheet(sheetName)
		}

		templateFile.DeleteComment(sheetName, "A1")
		templateFile.AddComment(sheetName, "A1", commentResult)

		//刷新数据
		//MysteelChemicalRefresh(mergeIndexFilePath)
		//AddIndexRefreshToLpush(mergeIndexFilePath)
	}

	sheetList := templateFile.GetSheetList()
	for _, v := range sheetList {
		if _, ok := sheetMap[v]; !ok {
			templateFile.DeleteSheet(v)
		}
	}

	if err = templateFile.SaveAs(mergeIndexFilePath); err != nil {
		fmt.Println("保存失败,Err:" + err.Error())
		templateFile.Close()
		return
	}
	templateFile.Close()
	time.Sleep(1 * time.Second)
	return
}