1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060 |
- package services
- import (
- "encoding/json"
- "eta/mysteel_watch/global"
- "eta/mysteel_watch/models/index"
- "eta/mysteel_watch/utils"
- "eta/mysteel_watch/watch"
- "fmt"
- "os"
- "strconv"
- "strings"
- "time"
- "github.com/rdlucklib/rdluck_tools/paging"
- "github.com/xuri/excelize/v2"
- )
- // Merge 合并指标
- func Merge() {
- fmt.Println("merge start")
- /*
- 年底,季度,存放在一个excel中
- 月度30个指标存一个excel
- 周度25个指标存放一个excel
- 日度20个指标存一个excel
- */
- //年度->38
- IndexYearMerge()
- //季度->5
- IndexSeasonMerge()
- //月度->86
- IndexMonthMerge()
- //周度->292
- IndexWeekMerge()
- //日度->114
- IndexDayMerge()
- //merge_file_path
- fmt.Println("merge end")
- }
- // IndexYearMerge 年度
- func IndexYearMerge() {
- frequency := "年度"
- fileName := "year" + ".xlsx" //保存的文件名称
- yearIndexFilePath := global.CONFIG.Serve.IndexMergeSaveDir + fileName
- var err error
- //indexObj := new(index.BaseFromMysteelChemicalIndex)
- //yearList, err := indexObj.GetIndexByFrequency(frequency)
- yearList, err := watch.GetIndexByFrequency(frequency)
- if err != nil {
- fmt.Println("GetIndexByFrequency Err:" + err.Error())
- return
- }
- if len(yearList) <= 0 {
- return
- }
- //commentResult, err := GetIndexComment(yearIndexFilePath, yearList, 1)
- commentResult, err := GetIndexCommentV2(yearIndexFilePath, yearList, 1)
- if err != nil {
- fmt.Println("GetIndexComment Err:" + err.Error())
- return
- }
- if utils.FileIsExist(yearIndexFilePath) { //修改文件
- fileObj, err := excelize.OpenFile(yearIndexFilePath)
- if err != nil {
- fmt.Println("打开文件失败,Err:" + err.Error())
- return
- }
- fileObj.DeleteComment("Sheet1", "A1")
- fileObj.AddComment("Sheet1", "A1", commentResult)
- if err := fileObj.SaveAs(yearIndexFilePath); err != nil {
- fmt.Println("保存失败,Err:" + err.Error())
- fileObj.Close()
- return
- }
- fileObj.Close()
- } else { //新增文件
- templatePath := global.CONFIG.Serve.IndexSaveDir + "index_template.xlsx"
- templateFile, err := excelize.OpenFile(templatePath)
- if err != nil {
- fmt.Println("打开文件失败,Err:" + err.Error())
- return
- }
- templateFile.DeleteComment("Sheet1", "A1")
- templateFile.AddComment("Sheet1", "A1", commentResult)
- if err := templateFile.SaveAs(yearIndexFilePath); err != nil {
- fmt.Println("保存失败,Err:" + err.Error())
- templateFile.Close()
- return
- }
- templateFile.Close()
- }
- //刷新数据
- //MysteelChemicalRefresh(yearIndexFilePath)
- AddIndexRefreshToLPush(yearIndexFilePath)
- }
- // IndexSeasonMerge 季度
- func IndexSeasonMerge() {
- frequency := "季度"
- fileName := "season" + ".xlsx" //保存的文件名称
- seasonIndexFilePath := global.CONFIG.Serve.IndexMergeSaveDir + fileName
- var err error
- //indexObj := new(index.BaseFromMysteelChemicalIndex)
- //seasonList, err := indexObj.GetIndexByFrequency(frequency)
- seasonList, err := watch.GetIndexByFrequency(frequency)
- if err != nil {
- fmt.Println("GetIndexByFrequency Err:" + err.Error())
- return
- }
- if len(seasonList) <= 0 {
- return
- }
- //commentResult, err := GetIndexComment(seasonIndexFilePath, seasonList, 1)
- commentResult, err := GetIndexCommentV2(seasonIndexFilePath, seasonList, 1)
- if err != nil {
- fmt.Println("GetIndexComment Err:" + err.Error())
- return
- }
- if utils.FileIsExist(seasonIndexFilePath) { //修改文件
- fileObj, err := excelize.OpenFile(seasonIndexFilePath)
- if err != nil {
- fmt.Println("打开文件失败,Err:" + err.Error())
- return
- }
- fileObj.DeleteComment("Sheet1", "A1")
- fileObj.AddComment("Sheet1", "A1", commentResult)
- if err := fileObj.SaveAs(seasonIndexFilePath); err != nil {
- fmt.Println("保存失败,Err:" + err.Error())
- fileObj.Close()
- return
- }
- fileObj.Close()
- } else { //新增文件
- templatePath := global.CONFIG.Serve.IndexSaveDir + "index_template.xlsx"
- templateFile, err := excelize.OpenFile(templatePath)
- if err != nil {
- fmt.Println("打开文件失败,Err:" + err.Error())
- return
- }
- templateFile.DeleteComment("Sheet1", "A1")
- templateFile.AddComment("Sheet1", "A1", commentResult)
- if err := templateFile.SaveAs(seasonIndexFilePath); err != nil {
- fmt.Println("保存失败,Err:" + err.Error())
- templateFile.Close()
- return
- }
- templateFile.Close()
- }
- //刷新数据
- //MysteelChemicalRefresh(seasonIndexFilePath)
- AddIndexRefreshToLPush(seasonIndexFilePath)
- }
- //月度
- //func IndexMonthMerge() {
- // frequency := "月度"
- // filePre := "month"
- // pageSize := 30
- // var err error
- //
- // indexObj := new(index.BaseFromMysteelChemicalIndex)
- // total, err := indexObj.GetIndexByFrequencyCount(frequency)
- // if err != nil {
- // fmt.Println("GetIndexByFrequencyCount Err:" + err.Error())
- // return
- // }
- //
- // maxFileIndexItem, err := indexObj.GetMaxFileIndex(frequency)
- // if err != nil {
- // fmt.Println("GetMaxFileIndex Err:" + err.Error())
- // return
- // }
- //
- // maxFileIndex := maxFileIndexItem.FileIndex
- // if maxFileIndex <= 0 {
- // maxFileIndex = 1
- // }
- //
- // fileName := filePre + "_" + strconv.Itoa(maxFileIndex) + ".xlsx" //保存的文件名称
- // monthIndexFilePath := global.CONFIG.Serve.IndexMergeSaveDir + fileName
- //
- // indexTotal, err := indexObj.GetIndexCountByMergeFilePath(monthIndexFilePath)
- // if err != nil {
- // fmt.Println("GetIndexCountByMergeFilePath Err:" + err.Error())
- // return
- // }
- //
- // if (indexTotal > int64(pageSize)/2) && maxFileIndex != 1 {
- // maxFileIndex += 1
- // }
- //
- // fmt.Println("maxFileIndex:", maxFileIndex)
- // totalPage := paging.PageCount(int(total), pageSize)
- // fmt.Println("total:", total)
- // fmt.Println("totalPage:", totalPage)
- // for i := 1; i <= totalPage; i++ {
- //
- // offset := (i - 1) * pageSize
- //
- // fileName = filePre + "_" + strconv.Itoa(maxFileIndex) + ".xlsx" //保存的文件名称
- // monthIndexFilePath := global.CONFIG.Serve.IndexMergeSaveDir + fileName
- // fmt.Println(monthIndexFilePath)
- //
- // fmt.Println(offset, pageSize)
- // pageList, err := indexObj.GetIndexByFrequencyPage(frequency, offset, pageSize)
- // if err != nil {
- // fmt.Println("GetIndexByFrequency Err:" + err.Error())
- // return
- // }
- //
- // commentResult, err := GetIndexComment(monthIndexFilePath, pageList, maxFileIndex)
- // if err != nil {
- // fmt.Println("GetIndexComment Err:" + err.Error())
- // return
- // }
- // if utils.FileIsExist(monthIndexFilePath) { //修改文件
- // fileObj, err := excelize.OpenFile(monthIndexFilePath)
- // if err != nil {
- // fmt.Println("打开文件失败,Err:" + err.Error())
- // return
- // }
- //
- // fileObj.DeleteComment("Sheet1", "A1")
- // fileObj.AddComment("Sheet1", "A1", commentResult)
- // if err := fileObj.SaveAs(monthIndexFilePath); err != nil {
- // fmt.Println("保存失败,Err:" + err.Error())
- // fileObj.Close()
- // return
- // }
- // fileObj.Close()
- // } else { //新增文件
- // templatePath := global.CONFIG.Serve.IndexSaveDir + "index_template.xlsx"
- // templateFile, err := excelize.OpenFile(templatePath)
- // if err != nil {
- // fmt.Println("打开文件失败,Err:" + err.Error())
- // return
- // }
- // templateFile.DeleteComment("Sheet1", "A1")
- // templateFile.AddComment("Sheet1", "A1", commentResult)
- // if err := templateFile.SaveAs(monthIndexFilePath); err != nil {
- // fmt.Println("保存失败,Err:" + err.Error())
- // templateFile.Close()
- // return
- // }
- // templateFile.Close()
- // }
- // maxFileIndex += 1
- // time.Sleep(1 * time.Second)
- // }
- //}
- func IndexMonthMerge() {
- frequency := "月度"
- filePre := "month"
- pageSize := 30
- err := indexMerge(frequency, filePre, pageSize)
- if err != nil {
- fmt.Println("合并月度指标失败,err:", err)
- }
- }
- //日度
- //func IndexDayMerge() {
- // frequency := "日度"
- // filePre := "day"
- // pageSize := 30
- // var err error
- //
- // indexObj := new(index.BaseFromMysteelChemicalIndex)
- // total, err := indexObj.GetIndexByFrequencyCount(frequency)
- // if err != nil {
- // fmt.Println("GetIndexByFrequencyCount Err:" + err.Error())
- // return
- // }
- //
- // maxFileIndexItem, err := indexObj.GetMaxFileIndex(frequency)
- // if err != nil {
- // fmt.Println("GetMaxFileIndex Err:" + err.Error())
- // return
- // }
- //
- // maxFileIndex := maxFileIndexItem.FileIndex
- // if maxFileIndex <= 0 {
- // maxFileIndex = 1
- // }
- //
- // fileName := filePre + "_" + strconv.Itoa(maxFileIndex) + ".xlsx" //保存的文件名称
- // monthIndexFilePath := global.CONFIG.Serve.IndexMergeSaveDir + fileName
- //
- // indexTotal, err := indexObj.GetIndexCountByMergeFilePath(monthIndexFilePath)
- // if err != nil {
- // fmt.Println("GetIndexCountByMergeFilePath Err:" + err.Error())
- // return
- // }
- //
- // if (indexTotal > int64(pageSize)/2) && maxFileIndex != 1 {
- // maxFileIndex += 1
- // }
- //
- // fmt.Println("maxFileIndex:", maxFileIndex)
- // totalPage := paging.PageCount(int(total), pageSize)
- // fmt.Println("total:", total)
- // fmt.Println("totalPage:", totalPage)
- // for i := 1; i <= totalPage; i++ {
- //
- // offset := (i - 1) * pageSize
- //
- // fileName = filePre + "_" + strconv.Itoa(maxFileIndex) + ".xlsx" //保存的文件名称
- // monthIndexFilePath := global.CONFIG.Serve.IndexMergeSaveDir + fileName
- // fmt.Println(monthIndexFilePath)
- //
- // fmt.Println(offset, pageSize)
- // pageList, err := indexObj.GetIndexByFrequencyPage(frequency, offset, pageSize)
- // if err != nil {
- // fmt.Println("GetIndexByFrequency Err:" + err.Error())
- // return
- // }
- //
- // commentResult, err := GetIndexComment(monthIndexFilePath, pageList, maxFileIndex)
- // if err != nil {
- // fmt.Println("GetIndexComment Err:" + err.Error())
- // return
- // }
- // if utils.FileIsExist(monthIndexFilePath) { //修改文件
- // fileObj, err := excelize.OpenFile(monthIndexFilePath)
- // if err != nil {
- // fmt.Println("打开文件失败,Err:" + err.Error())
- // return
- // }
- //
- // fileObj.DeleteComment("Sheet1", "A1")
- // fileObj.AddComment("Sheet1", "A1", commentResult)
- // if err := fileObj.SaveAs(monthIndexFilePath); err != nil {
- // fmt.Println("保存失败,Err:" + err.Error())
- // fileObj.Close()
- // return
- // }
- // fileObj.Close()
- // } else { //新增文件
- // templatePath := global.CONFIG.Serve.IndexSaveDir + "index_template.xlsx"
- // templateFile, err := excelize.OpenFile(templatePath)
- // if err != nil {
- // fmt.Println("打开文件失败,Err:" + err.Error())
- // return
- // }
- // templateFile.DeleteComment("Sheet1", "A1")
- // templateFile.AddComment("Sheet1", "A1", commentResult)
- // if err := templateFile.SaveAs(monthIndexFilePath); err != nil {
- // fmt.Println("保存失败,Err:" + err.Error())
- // templateFile.Close()
- // return
- // }
- // templateFile.Close()
- // }
- // maxFileIndex += 1
- // time.Sleep(1 * time.Second)
- // }
- //}
- // IndexDayMerge 日度指标合并
- func IndexDayMerge() {
- frequency := "日度"
- filePre := "day"
- pageSize := 30 //每个excel表需要的指标数
- err := indexMerge(frequency, filePre, pageSize)
- if err != nil {
- fmt.Println("合并日度指标失败,err:", err)
- }
- }
- // IndexWeekMerge 周度指标合并
- func IndexWeekMerge() {
- frequency := "周度"
- filePre := "week"
- pageSize := 30 //每个excel表需要的指标数
- err := indexMerge(frequency, filePre, pageSize)
- if err != nil {
- fmt.Println("合并周度指标失败,err:", err)
- }
- }
- //func IndexWeekMerge() {
- // frequency := "周度"
- // filePre := "week"
- // pageSize := 30 //每个excel表需要的指标数
- // firstIndexCount := pageSize //第一个添加的excel表需要的指标数
- // var err error
- //
- // indexObj := new(index.BaseFromMysteelChemicalIndex)
- //
- // // 获取未合并的指标总数量
- // total, err := indexObj.GetNoMergeIndexByFrequencyCount(frequency)
- // if err != nil {
- // fmt.Println("GetIndexByFrequencyCount Err:" + err.Error())
- // return
- // }
- // //未合并的指标总数量小于等于0,说明没有需要合并的指标
- // if total <=0{
- // return
- // }
- //
- // // 获取最大的文件编号下标
- // maxFileIndexItem, err := indexObj.GetMaxFileIndex(frequency)
- // if err != nil {
- // fmt.Println("GetMaxFileIndex Err:" + err.Error())
- // return
- // }
- //
- // maxFileIndex := maxFileIndexItem.FileIndex
- // if maxFileIndex <= 0 {
- // maxFileIndex = 1
- // }
- //
- // fileName := filePre + "_" + strconv.Itoa(maxFileIndex) + ".xlsx" //保存的文件名称
- // maxMergeIndexFilePath := global.CONFIG.Serve.IndexMergeSaveDir + fileName
- //
- // // 根据文件下标获取该文件下标已经存在多少的指标
- // indexTotal, err := indexObj.GetIndexCountByMergeFilePath(maxMergeIndexFilePath)
- // if err != nil {
- // fmt.Println("GetIndexCountByMergeFilePath Err:" + err.Error())
- // return
- // }
- //
- // // 如果最近的excel表的指标数量 少于 每页需要的指标数
- // if int(indexTotal) < pageSize {
- // // 最近的excel表的待添加指标数 = 每页需要的指标数 - 已经添加的指标
- // firstIndexCount = pageSize - int(indexTotal)
- // }
- //
- // fmt.Println("maxFileIndex:", maxFileIndex)
- //
- // // 除去第一个excel表的剩余需要添加的指标数量
- // nextTotal := int(total)-firstIndexCount
- // // 总文件数
- // totalPage := paging.PageCount(nextTotal, pageSize)
- // totalPage = totalPage+1 //需要加上除去在外面的指标(第一张excel)
- // fmt.Println("total:", total)
- // fmt.Println("totalPage:", totalPage)
- // for i := 1; i <= totalPage; i++ {
- // //offset := (i - 1) * pageSize
- //
- // fileName = filePre + "_" + strconv.Itoa(maxFileIndex) + ".xlsx" //保存的文件名称
- // mergeIndexFilePath := global.CONFIG.Serve.IndexMergeSaveDir + fileName
- // fmt.Println(mergeIndexFilePath)
- //
- // var list []*index.BaseFromMysteelChemicalIndex
- // // 需要查询的指标数
- // size := pageSize
- // if i == 1{
- // size = firstIndexCount // 第一页需要合并的新指标数
- //
- // // 第一页需要的指标列表
- // tmpList ,tmpErr := indexObj.GetIndexByFrequencyListByMergeFilePath(frequency,mergeIndexFilePath)
- // if tmpErr != nil{
- // fmt.Println("GetIndexByFrequencyListByMergeFilePath Err:" + err.Error())
- // return
- // }
- // list = tmpList
- // }
- // fmt.Println("需要查询",size,"个指标")
- //
- // //fmt.Println(offset, pageSize)
- //
- // // 当下需要合并到excel的指标
- // pageList, err := indexObj.GetNoMergeIndexByFrequencyPage(frequency, size)
- // if err != nil {
- // fmt.Println("GetIndexByFrequency Err:" + err.Error())
- // return
- // }
- //
- // if len(list)>0{
- // pageList = append(pageList,list...)
- // }
- //
- // commentResult, err := GetIndexComment(mergeIndexFilePath, pageList, maxFileIndex)
- // if err != nil {
- // fmt.Println("GetIndexComment Err:" + err.Error())
- // return
- // }
- // if utils.FileIsExist(mergeIndexFilePath) { //修改文件
- // fileObj, err := excelize.OpenFile(mergeIndexFilePath)
- // if err != nil {
- // fmt.Println("打开文件失败,Err:" + err.Error())
- // return
- // }
- //
- // fileObj.DeleteComment("Sheet1", "A1")
- // fileObj.AddComment("Sheet1", "A1", commentResult)
- // if err := fileObj.SaveAs(mergeIndexFilePath); err != nil {
- // fmt.Println("保存失败,Err:" + err.Error())
- // fileObj.Close()
- // return
- // }
- // fileObj.Close()
- // } else { //新增文件
- // templatePath := global.CONFIG.Serve.IndexSaveDir + "index_template.xlsx"
- // templateFile, err := excelize.OpenFile(templatePath)
- // if err != nil {
- // fmt.Println("打开文件失败,Err:" + err.Error())
- // return
- // }
- // templateFile.DeleteComment("Sheet1", "A1")
- // templateFile.AddComment("Sheet1", "A1", commentResult)
- // if err := templateFile.SaveAs(mergeIndexFilePath); err != nil {
- // fmt.Println("保存失败,Err:" + err.Error())
- // templateFile.Close()
- // return
- // }
- // templateFile.Close()
- // }
- // maxFileIndex += 1
- // time.Sleep(1 * time.Second)
- // }
- //}
- // indexMerge 指标合并
- // @params frequency string 频度
- // @params filePre string 文件名前缀
- // @params pageSize int 每个excel表中的指标数量
- func indexMerge(frequency, filePre string, pageSize int) (err error) {
- firstIndexCount := pageSize //第一个添加的excel表需要的指标数
- //indexObj := new(index.BaseFromMysteelChemicalIndex)
- frequencyReq := watch.GetMaxFileIndexReq{
- Frequency: frequency,
- }
- // 获取未合并的指标总数量
- total, err := watch.GetNoMergeIndexByFrequencyCount(frequencyReq)
- if err != nil {
- fmt.Println("GetIndexByFrequencyCount Err:" + err.Error())
- return
- }
- //未合并的指标总数量小于等于0,说明没有需要合并的指标
- if total <= 0 {
- return
- }
- // 获取最大的文件编号下标
- maxFileIndexItem, err := watch.GetMaxFileIndex(frequencyReq)
- maxFileIndex := maxFileIndexItem.FileIndex
- if maxFileIndex <= 0 {
- maxFileIndex = 1
- }
- fileName := filePre + "_" + strconv.Itoa(maxFileIndex) + ".xlsx" //保存的文件名称
- maxMergeIndexFilePath := global.CONFIG.Serve.IndexMergeSaveDir + fileName
- // 根据文件下标获取该文件下标已经存在多少的指标
- //indexTotal, err := indexObj.GetIndexCountByMergeFilePath(maxMergeIndexFilePath)
- indexTotal, err := watch.GetIndexCountByMergeFilePath(maxMergeIndexFilePath)
- if err != nil {
- fmt.Println("GetIndexCountByMergeFilePath Err:" + err.Error())
- return
- }
- fmt.Println("当前待合并的文件名:", maxMergeIndexFilePath)
- fmt.Println("当前文件中指标的最大数量:", indexTotal)
- fmt.Println("maxFileIndex:", maxFileIndex)
- // 如果最近的excel表的指标数量 少于 每页需要的指标数
- if int(indexTotal) < pageSize {
- // 最近的excel表的待添加指标数 = 每页需要的指标数 - 已经添加的指标
- firstIndexCount = pageSize - int(indexTotal)
- } else {
- firstIndexCount = pageSize
- maxFileIndex++
- }
- fmt.Println("maxFileIndex2:", maxFileIndex)
- // 除去第一个excel表的剩余需要添加的指标数量
- nextTotal := int(total) - firstIndexCount
- // 总文件数
- totalPage := paging.PageCount(nextTotal, pageSize)
- totalPage = totalPage + 1 //需要加上除去在外面的指标(第一张excel)
- fmt.Println("total:", total)
- fmt.Println("totalPage:", totalPage)
- for i := 1; i <= totalPage; i++ {
- //offset := (i - 1) * pageSize
- fileName = filePre + "_" + strconv.Itoa(maxFileIndex) + ".xlsx" //保存的文件名称
- mergeIndexFilePath := global.CONFIG.Serve.IndexMergeSaveDir + fileName
- fmt.Println(mergeIndexFilePath)
- //var list []*index.BaseFromMysteelChemicalIndex
- // 需要查询的指标数
- size := pageSize
- if i == 1 {
- size = firstIndexCount // 第一页需要合并的新指标数
- // 第一页需要的指标列表
- //tmpList, tmpErr := indexObj.GetIndexByFrequencyListByMergeFilePath(frequency, mergeIndexFilePath)
- //if tmpErr != nil {
- // fmt.Println("GetIndexByFrequencyListByMergeFilePath Err:" + tmpErr.Error())
- // err = tmpErr
- // return
- //}
- //list = tmpList
- }
- fmt.Println("需要查询", size, "个指标")
- //fmt.Println(offset, pageSize)
- // 当下需要合并到excel的指标
- pageList, tmpErr := watch.GetNoMergeIndexByFrequencyPage(frequency, size)
- if tmpErr != nil {
- fmt.Println("GetIndexByFrequency Err:" + tmpErr.Error())
- err = tmpErr
- return
- }
- //if len(list) > 0 {
- // pageList = append(pageList, list...)
- //}
- commentResult, tmpErr := GetIndexCommentV2(mergeIndexFilePath, pageList, maxFileIndex)
- if tmpErr != nil {
- fmt.Println("GetIndexComment Err:" + tmpErr.Error())
- err = tmpErr
- return
- }
- if utils.FileIsExist(mergeIndexFilePath) { //修改文件
- fileObj, tmpErr := excelize.OpenFile(mergeIndexFilePath)
- if tmpErr != nil {
- fmt.Println("打开文件失败,Err:" + tmpErr.Error())
- err = tmpErr
- return
- }
- fileObj.DeleteComment("Sheet1", "A1")
- fileObj.AddComment("Sheet1", "A1", commentResult)
- if err = fileObj.SaveAs(mergeIndexFilePath); err != nil {
- fmt.Println("保存失败,Err:" + err.Error())
- fileObj.Close()
- return
- }
- fileObj.Close()
- } else { //新增文件
- templatePath := global.CONFIG.Serve.IndexSaveDir + "index_template.xlsx"
- templateFile, tmpErr := excelize.OpenFile(templatePath)
- if tmpErr != nil {
- fmt.Println("打开文件失败,Err:" + tmpErr.Error())
- err = tmpErr
- return
- }
- templateFile.DeleteComment("Sheet1", "A1")
- templateFile.AddComment("Sheet1", "A1", commentResult)
- if err = templateFile.SaveAs(mergeIndexFilePath); err != nil {
- fmt.Println("保存失败,Err:" + err.Error())
- templateFile.Close()
- return
- }
- templateFile.Close()
- }
- maxFileIndex += 1
- time.Sleep(1 * time.Second)
- //刷新数据
- //MysteelChemicalRefresh(mergeIndexFilePath)
- //AddIndexRefreshToLPush(mergeIndexFilePath)
- }
- return
- }
- //func GetIndexComment(indexFilePath string, list []*index.BaseFromMysteelChemicalIndex, fileIndex int) (commentResult string, err error) {
- // runMode := "release"
- // indexInfo := new(IndexObj)
- // if utils.FileIsExist(indexFilePath) { //文件存在
- // fmt.Println("utils.FileIsExist")
- // getCommentStr := GetComment(indexFilePath)
- // fmt.Println(getCommentStr)
- //
- // err = json.Unmarshal([]byte(getCommentStr), &indexInfo)
- // if err != nil {
- // fmt.Println("json.Unmarshal err:" + err.Error())
- // return
- // }
- // } else { //文件不存在,
- // indexInfo.BlankValue = "0"
- // indexInfo.CanMark = true
- // indexInfo.ChartLineType = "0"
- // indexInfo.DateBlock = 0
- // indexInfo.DateBlockCount = 1
- // indexInfo.DateFormat = 0
- // indexInfo.DateTimeTag = "637973605613980000"
- // indexInfo.EndDate = ""
- // indexInfo.ExportType = 0
- // indexInfo.HasDescription = true
- // indexInfo.HasEmptyRows = false
- // indexInfo.HasFrequency = true
- // indexInfo.HasIndexID = true
- // indexInfo.HasLastDate = true
- // indexInfo.HasSourceName = true
- // indexInfo.HasTimeInterval = true
- // indexInfo.HasUnit = true
- // indexInfo.HasUpdateDate = true
- // indexInfo.IsCreateChart = false
- // indexInfo.IsDataSort = true
- // indexInfo.IsNewSheet = false
- // indexInfo.IsNewWorkbook = false
- // indexInfo.Position = "A1"
- // indexInfo.ShowBlankLines = false
- // indexInfo.StartDate = ""
- // indexInfo.Transpose = false
- // indexInfo.UpdateMode = 1
- // indexInfo.LookModel.IsLast = false
- // indexInfo.LookModel.LookValue = 0
- // indexInfo.LookModel.LookType = 0
- // indexInfo.Ver = 3
- // }
- // modelsList := make([]IndexModels, 0)
- // startDate := "1990-01-01"
- //
- // indexObj := new(index.BaseFromMysteelChemicalIndex)
- //
- // modelsList = append(modelsList, indexInfo.Models...)
- //
- // for k, v := range list {
- // fmt.Println(k, v)
- // item := new(IndexModels)
- // item.DataFormat = 0
- // if v.IndexName == "" {
- // item.DataStartDate = startDate
- // } else {
- // item.DataStartDate = v.StartDate.Format(utils.FormatDate)
- // }
- // item.DefineName = ""
- // item.DefineUnit = ""
- // item.DisplayIndexCode = v.IndexCode
- // item.IndexCode = v.IndexCode
- // item.IndexFormula = v.IndexCode
- // item.PointValue = 0
- // item.UnionStart = ""
- // modelsList = append(modelsList, *item)
- //
- // indexObj.IndexCode = v.IndexCode
- // indexObj.MergeFilePath = indexFilePath
- // indexObj.FileIndex = fileIndex
- //
- // updateColsArr := make([]string, 0)
- // updateColsArr = append(updateColsArr, "merge_file_path")
- // updateColsArr = append(updateColsArr, "file_index")
- //
- // err = indexObj.Update(runMode, updateColsArr)
- // if err != nil {
- // fmt.Println("indexObj.Update err:" + err.Error())
- // return
- // }
- // }
- //
- // indexInfo.Models = modelsList
- //
- // indexStr, err := json.Marshal(indexInfo)
- // if err != nil {
- // fmt.Println("json.Marshal err:" + err.Error())
- // return
- // }
- // text := string(indexStr)
- // text = strings.Trim(text, "{")
- //
- // commentMap := make(map[string]interface{})
- // commentMap["author"] = "{"
- // commentMap["text"] = text
- // //commentMap["text"] = commentItem
- //
- // commentJson, err := json.Marshal(commentMap)
- // if err != nil {
- // fmt.Println("json.Marshal err:" + err.Error())
- // }
- // commentResult = string(commentJson)
- // return
- //}
- func GetIndexCommentV2(indexFilePath string, list []index.BaseFromMysteelChemicalIndex, fileIndex int) (commentResult string, err error) {
- fmt.Println("GetIndexCommentV2 start")
- fmt.Println(indexFilePath)
- fmt.Println("GetIndexCommentV2 end")
- //runMode := "release"
- indexInfo := new(IndexObj)
- if utils.FileIsExist(indexFilePath) { //文件存在
- //fmt.Println("utils.FileIsExist")
- //getCommentStr := GetComment(indexFilePath)
- ////fmt.Println(getCommentStr)
- //
- //err = json.Unmarshal([]byte(getCommentStr), &indexInfo)
- //if err != nil {
- // fmt.Println("json.Unmarshal err:" + err.Error())
- // return
- //}
- err = os.Remove(indexFilePath)
- if err != nil {
- fmt.Println("os.Remove Err:" + err.Error() + ";indexFilePath:" + indexFilePath)
- return
- }
- }
- indexInfo.BlankValue = "0"
- indexInfo.CanMark = true
- indexInfo.ChartLineType = "0"
- indexInfo.DateBlock = 0
- indexInfo.DateBlockCount = 1
- indexInfo.DateFormat = 0
- indexInfo.DateTimeTag = "637973605613980000"
- indexInfo.EndDate = ""
- indexInfo.ExportType = 0
- indexInfo.HasDescription = true
- indexInfo.HasEmptyRows = false
- indexInfo.HasFrequency = true
- indexInfo.HasIndexID = true
- indexInfo.HasLastDate = true
- indexInfo.HasSourceName = true
- indexInfo.HasTimeInterval = true
- indexInfo.HasUnit = true
- indexInfo.HasUpdateDate = true
- indexInfo.IsCreateChart = false
- indexInfo.IsDataSort = true
- indexInfo.IsNewSheet = false
- indexInfo.IsNewWorkbook = false
- indexInfo.Position = "A1"
- indexInfo.ShowBlankLines = false
- indexInfo.StartDate = time.Now().AddDate(0, 0, -6).Format(utils.FormatDate)
- indexInfo.Transpose = false
- indexInfo.UpdateMode = 1
- indexInfo.LookModel.IsLast = false
- indexInfo.LookModel.LookValue = 0
- indexInfo.LookModel.LookType = 0
- indexInfo.Ver = 3
- modelsList := make([]IndexModels, 0)
- startDate := "1990-01-01"
- updateIndexes := make([]watch.UpdateIndexReq, 0)
- updateCols := []string{"merge_file_path", "file_index"}
- modelsList = append(modelsList, indexInfo.Models...)
- for _, v := range list {
- //fmt.Println(k, v)
- item := new(IndexModels)
- item.DataFormat = 0
- if v.IndexName == "" {
- item.DataStartDate = startDate
- } else {
- item.DataStartDate = v.EndDate.AddDate(0, 0, -7).Format(utils.FormatDate)
- }
- item.DefineName = ""
- item.DefineUnit = ""
- item.DisplayIndexCode = v.IndexCode
- item.IndexCode = v.IndexCode
- item.IndexFormula = v.IndexCode
- item.PointValue = 0
- item.UnionStart = ""
- modelsList = append(modelsList, *item)
- // 待更新的指标
- var req watch.UpdateIndexReq
- req.Item = new(index.BaseFromMysteelChemicalIndex)
- req.Item.BaseFromMysteelChemicalIndexId = v.BaseFromMysteelChemicalIndexId
- req.Item.IndexCode = v.IndexCode
- req.Item.MergeFilePath = indexFilePath
- req.Item.FileIndex = fileIndex
- req.UpdateCols = updateCols
- updateIndexes = append(updateIndexes, req)
- }
- // 批量更新Index
- if len(updateIndexes) > 0 {
- e := watch.MultiUpdateIndex(updateIndexes)
- if e != nil {
- fmt.Println("UpdateIndexByIndexCode err: " + e.Error())
- return
- }
- }
- indexInfo.Models = modelsList
- indexStr, err := json.Marshal(indexInfo)
- if err != nil {
- fmt.Println("json.Marshal err:" + err.Error())
- return
- }
- text := string(indexStr)
- text = strings.Trim(text, "{")
- commentMap := make(map[string]interface{})
- commentMap["author"] = "{"
- commentMap["text"] = text
- //commentMap["text"] = commentItem
- commentJson, err := json.Marshal(commentMap)
- if err != nil {
- fmt.Println("json.Marshal err:" + err.Error())
- }
- commentResult = string(commentJson)
- return
- }
- func GetIndexCommentV3(indexFilePath, StartDate string, list []index.BaseFromMysteelChemicalIndex) (commentResult string, err error) {
- indexInfo := new(IndexObj)
- indexInfo.BlankValue = "0"
- indexInfo.CanMark = true
- indexInfo.ChartLineType = "0"
- indexInfo.DateBlock = 0
- indexInfo.DateBlockCount = 1
- indexInfo.DateFormat = 0
- indexInfo.DateTimeTag = "637973605613980000"
- indexInfo.EndDate = ""
- indexInfo.ExportType = 0
- indexInfo.HasDescription = true
- indexInfo.HasEmptyRows = false
- indexInfo.HasFrequency = true
- indexInfo.HasIndexID = true
- indexInfo.HasLastDate = true
- indexInfo.HasSourceName = true
- indexInfo.HasTimeInterval = true
- indexInfo.HasUnit = true
- indexInfo.HasUpdateDate = true
- indexInfo.IsCreateChart = false
- indexInfo.IsDataSort = true
- indexInfo.IsNewSheet = false
- indexInfo.IsNewWorkbook = false
- indexInfo.Position = "A1"
- indexInfo.ShowBlankLines = false
- indexInfo.StartDate = StartDate // 这个是定义数据开始刷新的日期
- indexInfo.Transpose = false
- indexInfo.UpdateMode = 1
- indexInfo.LookModel.IsLast = false
- indexInfo.LookModel.LookValue = 0
- indexInfo.LookModel.LookType = 0
- indexInfo.Ver = 3
- modelsList := make([]IndexModels, 0)
- startDate := "1990-01-01"
- updateIndexes := make([]watch.UpdateIndexReq, 0)
- updateCols := []string{"merge_file_path", "file_index"}
- modelsList = append(modelsList, indexInfo.Models...)
- for _, v := range list {
- //fmt.Println(k, v)
- item := new(IndexModels)
- item.DataFormat = 0
- if v.IndexName == "" {
- item.DataStartDate = startDate // 数据的开始日期
- } else {
- item.DataStartDate = v.StartDate.Format(utils.FormatDate) // 数据的开始日期
- }
- item.DefineName = ""
- item.DefineUnit = ""
- item.DisplayIndexCode = v.IndexCode
- item.IndexCode = v.IndexCode
- item.IndexFormula = v.IndexCode
- item.PointValue = 0
- item.UnionStart = ""
- modelsList = append(modelsList, *item)
- // 待更新的指标
- var req watch.UpdateIndexReq
- req.Item = new(index.BaseFromMysteelChemicalIndex)
- req.Item.BaseFromMysteelChemicalIndexId = v.BaseFromMysteelChemicalIndexId
- req.Item.IndexCode = v.IndexCode
- req.Item.MergeFilePath = indexFilePath
- req.UpdateCols = updateCols
- updateIndexes = append(updateIndexes, req)
- }
- // 批量更新Index
- if len(updateIndexes) > 0 {
- e := watch.MultiUpdateIndex(updateIndexes)
- if e != nil {
- fmt.Println("UpdateIndexByIndexCode err: " + e.Error())
- return
- }
- }
- indexInfo.Models = modelsList
- indexStr, err := json.Marshal(indexInfo)
- if err != nil {
- fmt.Println("json.Marshal err:" + err.Error())
- return
- }
- text := string(indexStr)
- text = strings.Trim(text, "{")
- commentMap := make(map[string]interface{})
- commentMap["author"] = "{"
- commentMap["text"] = text
- //commentMap["text"] = commentItem
- commentJson, err := json.Marshal(commentMap)
- if err != nil {
- fmt.Println("json.Marshal err:" + err.Error())
- }
- commentResult = string(commentJson)
- return
- }
- type IndexObj struct {
- BlankValue string `json:"BlankValue"`
- CanMark bool `json:"CanMark"`
- ChartLineType string `json:"ChartLineType"`
- DateBlock int64 `json:"DateBlock"`
- DateBlockCount int64 `json:"DateBlockCount"`
- DateFormat int64 `json:"DateFormat"`
- DateTimeTag string `json:"DateTimeTag"`
- EndDate string `json:"EndDate"`
- ExportType int64 `json:"ExportType"`
- HasDescription bool `json:"HasDescription"`
- HasEmptyRows bool `json:"HasEmptyRows"`
- HasFrequency bool `json:"HasFrequency"`
- HasIndexID bool `json:"HasIndexID"`
- HasLastDate bool `json:"HasLastDate"`
- HasSourceName bool `json:"HasSourceName"`
- HasTimeInterval bool `json:"HasTimeInterval"`
- HasUnit bool `json:"HasUnit"`
- HasUpdateDate bool `json:"HasUpdateDate"`
- IsCreateChart bool `json:"IsCreateChart"`
- IsDataSort bool `json:"IsDataSort"`
- IsNewSheet bool `json:"IsNewSheet"`
- IsNewWorkbook bool `json:"IsNewWorkbook"`
- Models []IndexModels `json:"Models"`
- Position string `json:"Position"`
- RangeData string `json:"RangeData"`
- ShowBlankLines bool `json:"ShowBlankLines"`
- StartDate string `json:"StartDate"`
- Transpose bool `json:"Transpose"`
- UpdateMode int64 `json:"UpdateMode"`
- LookModel struct {
- IsLast bool `json:"IsLast"`
- LookValue int64 `json:"LookValue"`
- LookType int64 `json:"lookType"`
- } `json:"lookModel"`
- Ver int64 `json:"ver"`
- }
- type IndexModels struct {
- DataFormat int64 `json:"DataFormat"`
- DataStartDate string `json:"DataStartDate"`
- DefineName string `json:"DefineName"`
- DefineUnit string `json:"DefineUnit"`
- DisplayIndexCode string `json:"DisplayIndexCode"`
- IndexCode string `json:"IndexCode"`
- IndexFormula string `json:"IndexFormula"`
- PointValue int64 `json:"PointValue"`
- UnionStart string `json:"UnionStart"`
- }
|