mysteel_watch/services/index.go

package services

import (
	"encoding/json"
	"fmt"
	"github.com/xuri/excelize/v2"
	"hongze/mysteel_watch/global"
	"hongze/mysteel_watch/models/index"
	"hongze/mysteel_watch/services/alarm_msg"
	"hongze/mysteel_watch/utils"
	"os"
	"strconv"
	"strings"
	"time"
)

func IndexCreateCheck() (err error) {
	fmt.Println("IndexCreateCheck")
	indexObj := new(index.BaseFromMysteelChemicalIndex)
	list, err := indexObj.GetIndexCreate()
	if err != nil {
		return
	}
	fmt.Println("listLen:", len(list))
	if len(list) <= 0 {
		return nil
	}
	var indexCode []string
	for _, v := range list {
		indexCode = append(indexCode, v.IndexCode)
	}
	indexCodeStr := strings.Join(indexCode, ";")
	fmt.Println("indexCodeStr:" + indexCodeStr)
	go alarm_msg.SendAlarmMsg(utils.APPNAME+" 存在指标数据未生成:"+indexCodeStr, 3)
	for _, v := range list {
		//err := IndexCreate(v)
		//if err != nil {
		//	go alarm_msg.SendAlarmMsg(utils.APPNAME+" 指标数据未生成检测失败:"+err.Error(), 3)
		//}
		MysteelChemicalRefresh(v.FilePath)
	}
	return nil
}

func IndexCreate(item *index.BaseFromMysteelChemicalIndex) (err error) {
	item.UpdateWeek = utils.GetUpdateWeekEn(item.UpdateWeek)
	global.LOG.Info("task IndexCreate:" + time.Now().Format(utils.FormatDateTime))

	if utils.FileIsExist(item.FilePath) {
		os.Remove(item.FilePath)
	}
	runMode := "release"
	//fileName := req.IndexName + "_" + req.IndexCode + ".xlsx"
	var fileName string
	if item.UpdateWeek != "" {
		fileName = item.IndexCode + "_" + item.UpdateWeek + "_" + runMode + ".xlsx" //保存的文件名称
	} else {
		fileName = item.IndexCode + "_" + runMode + ".xlsx" //保存的文件名称
	}
	filePath := utils.IndexSaveDir + fileName

	templatePath := utils.IndexSaveDir + "index_template.xlsx"
	templateFile, err := excelize.OpenFile(templatePath)
	if err != nil {
		return
	}
	defer func() {
		templateFile.Close()
	}()

	startDate := "1990-01-01"
	commentStr := `"BlankValue":"0","CanMark":true,"ChartLineType":"0","DateBlock":0,"DateBlockCount":1,"DateFormat":0,"DateTimeTag":"","EndDate":"","ExportType":0,"HasDescription":true,"HasEmptyRows":false,"HasFrequency":true,"HasIndexID":true,"HasLastDate":true,"HasSourceName":true,"HasTimeInterval":true,"HasUnit":true,"HasUpdateDate":true,"IsCreateChart":false,"IsDataSort":true,"IsNewSheet":false,"IsNewWorkbook":false,"Models":[{"DataFormat":0,"DataStartDate":"` + startDate + `","DefineName":"","DefineUnit":"","DisplayIndexCode":"` + item.IndexCode + `","IndexCode":"` + item.IndexCode + `","IndexFormula":"` + item.IndexCode + `","PointValue":0,"UnionStart":""}],"Position":"A1","RangeData":"A2:B280","ShowBlankLines":false,"StartDate":"","Transpose":false,"UpdateMode":0,"lookModel":{"IsLast":false,"LookValue":0,"lookType":0},"ver":3}
`
	commentMap := make(map[string]interface{})
	commentMap["author"] = "{"
	commentMap["text"] = commentStr
	//commentMap["text"] = commentItem

	commentJson, err := json.Marshal(commentMap)
	if err != nil {
		fmt.Println("json.Marshal err:" + err.Error())
		return err
	}

	fmt.Println("commentJson")
	fmt.Println(string(commentJson))
	templateFile.DeleteComment("Sheet1", "A1")
	templateFile.AddComment("Sheet1", "A1", string(commentJson))
	if err := templateFile.SaveAs(filePath); err != nil {
		fmt.Println(err)
		return err
	}
	time.Sleep(1 * time.Minute)
	return
}

func IndexRefreshAll() {
	fmt.Println("IndexCreateCheck")
	indexObj := new(index.BaseFromMysteelChemicalIndex)
	list, err := indexObj.GetIndexRefreshAll()
	if err != nil {
		fmt.Println("GetIndexRefreshAll Err:" + err.Error())
		return
	}
	fmt.Println("listLen:", len(list))
	if len(list) <= 0 {
		return
	}
	nowDate := time.Now().Format(utils.FormatDate)
	monthLastDay := utils.GetNowMonthLastDay()
	monthLastDayStr := monthLastDay.Format(utils.FormatDate)
	nowWeek := time.Now().Weekday().String()
	nowWeekZn := utils.GetWeekZn(nowWeek)
	yearLast := strconv.Itoa(time.Now().Year()) + "12-31"
	for _, v := range list {
		if v.Frequency == "周度" && v.UpdateWeek == nowWeekZn && v.UpdateTime == "" && v.UpdateTime2 == "" {
			MysteelChemicalRefresh(v.FilePath)
		} else if (v.Frequency == "月度" || v.Frequency == "旬度") && nowDate == monthLastDayStr {
			MysteelChemicalRefresh(v.FilePath)
		} else if v.Frequency == "年度" && nowDate == yearLast {
			MysteelChemicalRefresh(v.FilePath)
		} else {
			MysteelChemicalRefresh(v.FilePath)
		}
	}
	return
}


func IndexRefreshWeek() {
	nowWeek := time.Now().Weekday().String()
	nowWeekZn := utils.GetWeekZn(nowWeek)

	indexObj := new(index.BaseFromMysteelChemicalIndex)
	list, err := indexObj.GetIndexRefreshWeek(nowWeek)
	if err != nil {
		fmt.Println("GetIndexRefreshWeek Err:" + err.Error())
		return
	}
	fmt.Println("listLen:", len(list))
	if len(list) <= 0 {
		return
	}
	nowDate := time.Now().Format(utils.FormatDate)
	monthLastDay := utils.GetNowMonthLastDay()
	monthLastDayStr := monthLastDay.Format(utils.FormatDate)

	yearLast := strconv.Itoa(time.Now().Year()) + "12-31"
	for _, v := range list {
		if v.Frequency == "周度" && v.UpdateWeek == nowWeekZn && v.UpdateTime == "" && v.UpdateTime2 == "" {
			MysteelChemicalRefresh(v.FilePath)
		} else if (v.Frequency == "月度" || v.Frequency == "旬度") && nowDate == monthLastDayStr {
			MysteelChemicalRefresh(v.FilePath)
		} else if v.Frequency == "年度" && nowDate == yearLast {
			MysteelChemicalRefresh(v.FilePath)
		} else {
			MysteelChemicalRefresh(v.FilePath)
		}
	}
	return
}