12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097 |
- package services
- import (
- "encoding/json"
- "fmt"
- "hongze/mysteel_watch/global"
- "hongze/mysteel_watch/models/index"
- "hongze/mysteel_watch/utils"
- "strconv"
- "strings"
- "sync"
- "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")
- }
- //年度
- func IndexYearMerge() {
- frequency := "年度"
- fileName := "year" + ".xlsx" //保存的文件名称
- yearIndexFilePath := utils.IndexMsergeSaveDir + fileName
- var err error
- indexObj := new(index.BaseFromMysteelChemicalIndex)
- yearList, err := indexObj.GetIndexByFrequency(frequency)
- if err != nil {
- fmt.Println("GetIndexByFrequency Err:" + err.Error())
- return
- }
- if len(yearList) <= 0 {
- return
- }
- commentResult, err := GetIndexComment(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 := utils.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()
- }
- //处理excel数据
- }
- //季度
- func IndexSeasonMerge() {
- frequency := "季度"
- fileName := "season" + ".xlsx" //保存的文件名称
- seasonIndexFilePath := utils.IndexMsergeSaveDir + fileName
- var err error
- indexObj := new(index.BaseFromMysteelChemicalIndex)
- seasonList, err := indexObj.GetIndexByFrequency(frequency)
- if err != nil {
- fmt.Println("GetIndexByFrequency Err:" + err.Error())
- return
- }
- if len(seasonList) <= 0 {
- return
- }
- commentResult, err := GetIndexComment(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 := utils.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()
- }
- //处理excel数据
- }
- //月度
- //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 := utils.IndexMsergeSaveDir + 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 := utils.IndexMsergeSaveDir + 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 := utils.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 := utils.IndexMsergeSaveDir + 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 := utils.IndexMsergeSaveDir + 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 := utils.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 := utils.IndexMsergeSaveDir + 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 := utils.IndexMsergeSaveDir + 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 := utils.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)
- // 获取未合并的指标总数量
- 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 := utils.IndexMsergeSaveDir + 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 := utils.IndexMsergeSaveDir + 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 := indexObj.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 := GetIndexComment(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 := utils.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)
- }
- 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)
- 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
- }
- 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"`
- }
- func DataAnalysis(filePath string) {
- runMode := "debug"
- //runMode = "release"
- fmt.Println("filePath:", filePath)
- time.Sleep(1 * time.Second)
- if !utils.FileIsExist(filePath) {
- fmt.Println("filePath is not exist:" + filePath)
- return
- }
- //读取文件内容
- global.LOG.Info("WatchFile:" + filePath)
- f, err := excelize.OpenFile(filePath)
- global.LOG.Info("OpenFile:" + filePath)
- if err != nil {
- fmt.Println("OpenFile:" + filePath + ",Err:" + err.Error())
- return
- }
- defer func() {
- if err := f.Close(); err != nil {
- fmt.Println("FileClose Err:" + err.Error())
- return
- }
- }()
- indexObj := new(index.BaseFromMysteelChemicalIndex)
- var wg = sync.WaitGroup{}
- wg.Add(1)
- go func() {
- sheetList := f.GetSheetList()
- for _, sv := range sheetList {
- rows, err := f.GetRows(sv)
- if err != nil {
- fmt.Println("f.GetRows:err:" + err.Error())
- return
- }
- var nameArr []string
- unitArr := make([]string, 0)
- sourceArr := make([]string, 0)
- codeArr := make([]string, 0)
- frequencyArr := make([]string, 0)
- dateArr := make([]string, 0)
- describeArr := make([]string, 0)
- endDateArr := make([]string, 0)
- updateArr := make([]string, 0)
- indexDateArr := make([]string, 0)
- dataMap := make(map[string][]string)
- var nameLen int
- for rk, row := range rows {
- if rk == 0 {
- continue
- }
- if rk == 1 {
- for ck, colCell := range row {
- if ck >= 1 {
- nameArr = append(nameArr, colCell)
- }
- }
- nameLen = len(nameArr)
- unitArr = make([]string, nameLen)
- sourceArr = make([]string, nameLen)
- codeArr = make([]string, nameLen)
- frequencyArr = make([]string, nameLen)
- dateArr = make([]string, nameLen)
- describeArr = make([]string, nameLen)
- endDateArr = make([]string, nameLen)
- updateArr = make([]string, nameLen)
- } else if rk == 2 {
- for ck, colCell := range row {
- if ck >= 1 {
- unitArr[ck-1] = colCell
- }
- }
- } else if rk == 3 {
- for ck, colCell := range row {
- if ck >= 1 {
- //sourceArr = append(sourceArr, colCell)
- sourceArr[ck-1] = colCell
- }
- }
- } else if rk == 4 {
- for ck, colCell := range row {
- if ck >= 1 {
- //codeArr = append(codeArr, colCell)
- codeArr[ck-1] = colCell
- }
- }
- } else if rk == 5 {
- for ck, colCell := range row {
- if ck >= 1 {
- //frequencyArr = append(frequencyArr, colCell)
- frequencyArr[ck-1] = colCell
- }
- }
- } else if rk == 6 {
- for ck, colCell := range row {
- if ck >= 1 {
- //dateArr = append(dateArr, colCell)
- dateArr[ck-1] = colCell
- }
- }
- } else if rk == 7 {
- for ck, colCell := range row {
- if ck >= 1 {
- //describeArr = append(describeArr, colCell)
- describeArr[ck-1] = colCell
- }
- }
- } else if rk == 8 {
- for ck, colCell := range row {
- if ck >= 1 {
- //endDateArr = append(endDateArr, colCell)
- endDateArr[ck-1] = colCell
- }
- }
- } else if rk == 9 {
- for ck, colCell := range row {
- if ck >= 1 {
- //updateArr = append(updateArr, colCell)
- updateArr[ck-1] = colCell
- }
- }
- } else {
- var date string
- dataArr := make([]string, nameLen)
- for ck, colCell := range row {
- if ck == 0 {
- date = colCell
- indexDateArr = append(indexDateArr, date)
- } else {
- if colCell != "" {
- dataArr[ck-1] = colCell
- }
- }
- }
- dataMap[date] = dataArr
- }
- }
- var indexId int64
- existDataMap := make(map[string]string)
- for k, v := range nameArr {
- indexName := v
- indexCode := codeArr[k]
- unit := unitArr[k]
- source := sourceArr[k]
- describe := describeArr[k]
- dateStr := dateArr[k]
- frequency := frequencyArr[k]
- //判断指标是否存在
- var isAdd int
- item, err := indexObj.GetIndexItem(runMode, indexCode)
- if err != nil {
- if err.Error() == "record not found" {
- isAdd = 1
- } else {
- isAdd = -1
- fmt.Println("GetIndexItem Err:" + err.Error())
- return
- }
- }
- if item != nil && item.BaseFromMysteelChemicalIndexId > 0 {
- fmt.Println("item:", item)
- isAdd = 2
- } else {
- isAdd = 1
- }
- var startDate, endDate string
- startEndArr := strings.Split(dateStr, "~")
- if len(startEndArr) >= 2 {
- startDate = startEndArr[0]
- endDate = startEndArr[1]
- }
- if !strings.Contains(frequency, "度") {
- frequency = frequency + "度"
- }
- indexItem := new(index.BaseFromMysteelChemicalIndex)
- if isAdd == 1 {
- indexItem.IndexCode = indexCode
- indexItem.IndexName = indexName
- indexItem.Unit = unit
- indexItem.Source = source
- indexItem.Describe = describe
- indexItem.StartDate, _ = time.ParseInLocation(utils.FormatDate, startDate, time.Local)
- indexItem.EndDate, _ = time.ParseInLocation(utils.FormatDate, endDate, time.Local)
- indexItem.Frequency = frequency
- indexItem.FilePath = filePath
- indexItem.MergeFilePath = filePath
- err = indexItem.Add(runMode)
- if err != nil {
- fmt.Println("indexItem.add err:" + err.Error())
- return
- }
- indexId = indexObj.BaseFromMysteelChemicalIndexId
- } else if isAdd == 2 {
- indexItem.IndexCode = indexCode
- indexItem.IndexName = indexName
- indexItem.Unit = unit
- indexItem.Source = source
- indexItem.Describe = describe
- indexItem.StartDate, _ = time.ParseInLocation(utils.FormatDate, startDate, time.Local)
- indexItem.EndDate, _ = time.ParseInLocation(utils.FormatDate, endDate, time.Local)
- indexItem.Frequency = frequency
- indexItem.FilePath = filePath
- indexItem.ModifyTime = time.Now()
- indexId = item.BaseFromMysteelChemicalIndexId
- //修改数据
- updateColsArr := make([]string, 0)
- updateColsArr = append(updateColsArr, "index_name")
- updateColsArr = append(updateColsArr, "unit")
- updateColsArr = append(updateColsArr, "source")
- updateColsArr = append(updateColsArr, "frequency")
- updateColsArr = append(updateColsArr, "start_date")
- updateColsArr = append(updateColsArr, "end_date")
- updateColsArr = append(updateColsArr, "describe")
- updateColsArr = append(updateColsArr, "end_date")
- updateColsArr = append(updateColsArr, "modify_time")
- updateColsArr = append(updateColsArr, "file_path")
- err = indexItem.Update(runMode, updateColsArr)
- if err != nil {
- fmt.Println("indexObj.Update err:" + err.Error())
- return
- }
- dataObj := new(index.BaseFromMysteelChemicalData)
- //获取已存在的所有数据
- dataList, err := dataObj.GetIndexDataList(runMode, indexCode)
- if err != nil {
- fmt.Println("GetIndexDataList Err:" + err.Error())
- return
- }
- fmt.Println("dataListLen:", len(dataList))
- for _, v := range dataList {
- dateStr := v.DataTime.Format(utils.FormatDate)
- existDataMap[dateStr] = v.Value
- }
- }
- dataList := make([]index.BaseFromMysteelChemicalData, 0)
- for _, dv := range indexDateArr {
- dataArr := dataMap[dv]
- dataVal := dataArr[k]
- updateDate := updateArr[k]
- if dataVal != "" {
- if _, ok := existDataMap[dv]; !ok {
- dateTime, err := time.ParseInLocation(utils.FormatDate, dv, time.Local)
- if err != nil {
- fmt.Println("time.ParseInLocation Err:" + err.Error())
- return
- }
- dataItem := new(index.BaseFromMysteelChemicalData)
- dataItem.BaseFromMysteelChemicalIndexId = indexId
- dataItem.IndexCode = indexCode
- dataItem.DataTime = dateTime
- dataItem.Value = dataVal
- dataItem.UpdateDate = updateDate
- dataItem.CreateTime = time.Now()
- dataItem.ModifyTime = time.Now()
- dataList = append(dataList, *dataItem)
- }
- }
- }
- if len(dataList) > 0 {
- dataObj := new(index.BaseFromMysteelChemicalData)
- err = dataObj.Add(runMode, dataList)
- if err != nil {
- fmt.Println("dataObj.Add() Err:" + err.Error())
- return
- }
- }
- }
- }
- wg.Done()
- }()
- wg.Wait()
- }
|