12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001 |
- package excel
- import (
- "encoding/json"
- "errors"
- "eta_gn/eta_api/models"
- "eta_gn/eta_api/models/data_manage"
- "eta_gn/eta_api/models/data_manage/excel"
- "eta_gn/eta_api/models/data_manage/excel/request"
- "eta_gn/eta_api/models/data_manage/excel/response"
- "eta_gn/eta_api/models/system"
- "eta_gn/eta_api/services/data"
- "eta_gn/eta_api/services/data/data_manage_permission"
- "eta_gn/eta_api/utils"
- "fmt"
- "sort"
- "strings"
- "time"
- "github.com/xuri/excelize/v2"
- )
- // GetExcelDetailInfoByExcelInfoId 根据表格id获取表格详情
- func GetExcelDetailInfoByExcelInfoId(excelInfoId, sysUserId int, lang string) (excelDetail response.ExcelInfoDetail, errMsg string, err error) {
- errMsg = `获取失败`
- //获取eta表格信息
- excelInfo, err := excel.GetExcelInfoById(excelInfoId)
- if err != nil {
- err = errors.New("获取ETA表格信息失败,Err:" + err.Error())
- if utils.IsErrNoRow(err) {
- errMsg = "ETA表格被删除,请刷新页面"
- err = errors.New("ETA表格被删除,请刷新页面,Err:" + err.Error())
- }
- return
- }
- return formatExcelInfo2Detail(excelInfo, sysUserId, lang)
- }
- // GetExcelDetailInfoByUnicode 根据表格编码获取表格详情
- func GetExcelDetailInfoByUnicode(unicode string, sysUserId int, lang string) (excelDetail response.ExcelInfoDetail, errMsg string, err error) {
- errMsg = `获取失败`
- // 获取eta表格信息
- excelInfo, err := excel.GetExcelInfoByUnicode(unicode)
- if err != nil {
- err = errors.New("获取ETA表格信息失败,Err:" + err.Error())
- if utils.IsErrNoRow(err) {
- errMsg = "ETA表格被删除,请刷新页面"
- err = errors.New("ETA表格被删除,请刷新页面,Err:" + err.Error())
- }
- return
- }
- return formatExcelInfo2Detail(excelInfo, sysUserId, lang)
- }
- func formatExcelInfo2Detail(excelInfo *excel.ExcelInfo, sysUserId int, lang string) (excelDetail response.ExcelInfoDetail, errMsg string, err error) {
- checkExcelInfo := excelInfo
- if excelInfo.Source == utils.BALANCE_TABLE {
- checkExcelInfoId := excelInfo.ExcelInfoId
- if excelInfo.BalanceType == 1 {
- checkExcelInfoId = excelInfo.RelExcelInfoId
- } else {
- if excelInfo.ParentId > 0 {
- checkExcelInfoId = excelInfo.ParentId
- }
- }
- if checkExcelInfoId != excelInfo.ExcelInfoId {
- checkExcelInfo, err = excel.GetExcelInfoById(checkExcelInfoId)
- if err != nil {
- errMsg = "获取平衡表格信息失败"
- err = errors.New("获取平衡表格信息失败,Err:" + err.Error())
- return
- }
- }
- }
- // 数据权限
- haveOperaAuth, err := data_manage_permission.CheckExcelPermissionByExcelInfoId(checkExcelInfo.ExcelInfoId, checkExcelInfo.ExcelClassifyId, checkExcelInfo.IsJoinPermission, sysUserId)
- if err != nil {
- err = errors.New("获取表格权限信息失败,Err" + err.Error())
- return
- }
- excelDetail = response.ExcelInfoDetail{
- ExcelInfoId: excelInfo.ExcelInfoId,
- Source: excelInfo.Source,
- ExcelType: excelInfo.ExcelType,
- ExcelName: excelInfo.ExcelName,
- UniqueCode: excelInfo.UniqueCode,
- ExcelClassifyId: excelInfo.ExcelClassifyId,
- SysUserId: excelInfo.SysUserId,
- SysUserRealName: excelInfo.SysUserRealName,
- Content: excelInfo.Content,
- ExcelImage: excelInfo.ExcelImage,
- FileUrl: excelInfo.FileUrl,
- Sort: excelInfo.Sort,
- IsDelete: excelInfo.IsDelete,
- ModifyTime: excelInfo.ModifyTime,
- CreateTime: excelInfo.CreateTime,
- TableData: nil,
- HaveOperaAuth: haveOperaAuth,
- ParentId: excelInfo.ParentId,
- BalanceType: excelInfo.BalanceType,
- UpdateUserId: excelInfo.UpdateUserId,
- UpdateUserRealName: excelInfo.UpdateUserRealName,
- RelExcelInfoId: excelInfo.RelExcelInfoId,
- SourcesFrom: excelInfo.SourcesFrom,
- }
- // 无权限,不需要返回数据
- if !haveOperaAuth {
- return
- }
- switch excelInfo.Source {
- case utils.TIME_TABLE: // 时间序列表格
- var tableDataConfig TableDataConfig
- err = json.Unmarshal([]byte(excelDetail.Content), &tableDataConfig)
- if err != nil {
- err = errors.New("表格json转结构体失败,Err:" + err.Error())
- return
- }
- result, tmpErr := GetDataByTableDataConfig(tableDataConfig)
- if tmpErr != nil {
- err = errors.New("获取最新的表格数据失败,Err:" + tmpErr.Error())
- return
- }
- result = SetExcelByDecimalConfig(result, tableDataConfig.DecimalConfig)
- if len(result.EdbInfoIdList) > 0 {
- classifyIdList := make([]int, 0)
- for _, v := range result.Data {
- classifyIdList = append(classifyIdList, v.ClassifyId)
- }
- classifyMap := make(map[int]*data_manage.EdbClassify)
- classifyList, tmpErr := data_manage.GetEdbClassifyByIdList(classifyIdList)
- if tmpErr != nil {
- err = errors.New("获取分类列表失败,Err:" + tmpErr.Error())
- return
- }
- for _, v := range classifyList {
- classifyMap[v.ClassifyId] = v
- }
- // 获取所有有权限的指标和分类
- permissionEdbIdList, permissionClassifyIdList, tmpErr := data_manage_permission.GetUserEdbAndClassifyPermissionList(sysUserId, 0, 0)
- if tmpErr != nil {
- err = errors.New("获取所有有权限的指标和分类失败,Err:" + tmpErr.Error())
- return
- }
- for i, v := range result.Data {
- if currClassify, ok := classifyMap[v.ClassifyId]; ok {
- result.Data[i].HaveOperaAuth = data_manage_permission.CheckEdbPermissionByPermissionIdList(v.IsJoinPermission, currClassify.IsJoinPermission, v.EdbInfoId, v.ClassifyId, permissionEdbIdList, permissionClassifyIdList)
- }
- }
- }
- excelDetail.TableData = result
- case utils.MIXED_TABLE, utils.BALANCE_TABLE: // 混合表格 平衡表
- var result request.MixedTableReq
- err = json.Unmarshal([]byte(excelDetail.Content), &result)
- if err != nil {
- err = errors.New("表格json转结构体失败,Err:" + err.Error())
- return
- }
- newData, tmpErr, tmpErrMsg := GetMixedTableCellData(result, lang)
- if tmpErr != nil {
- errMsg = "获取失败"
- if tmpErrMsg != `` {
- errMsg = tmpErrMsg
- }
- err = errors.New("获取最新的数据失败,Err:" + tmpErr.Error())
- return
- }
- result.Data = newData
- excelDetail.TableData = result
- }
- if excelDetail.Source == utils.BALANCE_TABLE {
- excelDetail.Button = GetBalanceExcelInfoOpButton(sysUserId, checkExcelInfo.SysUserId, excelDetail.HaveOperaAuth, checkExcelInfo.ExcelInfoId)
- }
- return
- }
- // SetExcelByDecimalConfig 设置表格的小数位
- func SetExcelByDecimalConfig(tableData request.TableDataReq, config []request.DecimalConfig) request.TableDataReq {
- excelData := tableData.Data
- edbInfoIndex := make(map[int]int)
- dateIndex := make(map[string]int)
- for i, v := range excelData {
- edbInfoIndex[v.EdbInfoId] = i
- }
- for i, v := range excelData[0].Data {
- dateIndex[v.DataTime] = i
- }
- for _, conf := range config {
- if conf.Col > 0 {
- if v, ok := edbInfoIndex[conf.Col]; ok {
- excelData[v].Decimal = conf.Decimal
- for i := 0; i < len(excelData[v].Data); i++ {
- excelData[v].Data[i].Decimal = conf.Decimal
- }
- }
- }
- if conf.Row != "" {
- if v, ok := dateIndex[conf.Row]; ok {
- for i := 0; i < len(excelData); i++ {
- excelData[i].Data[v].Decimal = conf.Decimal
- }
- }
- }
- }
- tableData.Data = excelData
- tableData.DecimalConfig = config
- return tableData
- }
- // GetExcelInfoOpButton 获取ETA表格的操作权限
- func GetExcelInfoOpButton(sysUser *system.Admin, belongUserId, source int, haveOperaAuth bool) (button excel.ExcelInfoDetailButton) {
- // 如果没有数据权限,那么直接返回
- if !haveOperaAuth {
- return
- }
- //非管理员角色查看其他用户创建的表格,可刷新、另存为、下载表格;
- button.RefreshButton = true
- button.CopyButton = true
- button.DownloadButton = true
- // 1、本用户创建的表格,可编辑、刷新、另存为、下载、删除,删除需二次确认;
- // 2、管理员角色对所有表格有如上权限;
- // 3、在线excel所有人都能编辑
- if sysUser.RoleTypeCode == utils.ROLE_TYPE_CODE_ADMIN || sysUser.RoleTypeCode == utils.ROLE_TYPE_CODE_FICC_ADMIN || sysUser.AdminId == belongUserId || source == utils.EXCEL_DEFAULT {
- button.OpButton = true
- button.DeleteButton = true
- }
- // 自定义分析
- if source == utils.CUSTOM_ANALYSIS_TABLE {
- if sysUser.RoleTypeCode == utils.ROLE_TYPE_CODE_ADMIN || sysUser.RoleTypeCode == utils.ROLE_TYPE_CODE_FICC_ADMIN || sysUser.AdminId == belongUserId {
- button.OpEdbButton = true // 生成、查看指标按钮
- button.RefreshEdbButton = true // 刷新指标按钮
- }
- }
- return
- }
- // GetFirstEdbDataList 获取第一列的数据
- func GetFirstEdbDataList(edbInfo *data_manage.EdbInfo, num int, manualDateList []string, decimal int) (resultDataList []request.ManualDataReq, err error) {
- var dataList []*data_manage.EdbDataList
- resultDataList = make([]request.ManualDataReq, 0)
- switch edbInfo.EdbInfoType {
- case 0:
- dataList, err = data_manage.GetEdbDataList(edbInfo.Source, edbInfo.SubSource, edbInfo.EdbInfoId, ``, ``)
- case 1:
- _, dataList, _, _, err, _ = data.GetPredictDataListByPredictEdbInfoId(edbInfo.EdbInfoId, ``, ``, false)
- default:
- err = errors.New(fmt.Sprint("获取失败,指标类型异常", edbInfo.EdbInfoType))
- }
- if err != nil {
- return
- }
- // 获取需要的期数
- lenData := len(dataList)
- if lenData <= 0 {
- return
- }
- tmpManualDateNum := 0 // 手工数据的期数
- lenManualDate := len(manualDateList)
- if lenManualDate > 0 {
- sortDateList := manualDateList
- baseDateList := utils.StrArr{}
- baseDateList = append(baseDateList, sortDateList...)
- sort.Sort(baseDateList)
- sortDateList = append([]string{}, baseDateList...)
- lastData := dataList[lenData-1]
- lastDataDate, tmpErr := time.ParseInLocation(utils.FormatDate, lastData.DataTime, time.Local)
- if tmpErr != nil {
- err = tmpErr
- return
- }
- // 遍历倒序后的日期,然后匹配在实际数据之后日期的个数
- for _, tmpDateStr := range sortDateList {
- tmpDate, tmpErr := time.ParseInLocation(utils.FormatDate, tmpDateStr, time.Local)
- if tmpErr != nil {
- err = tmpErr
- return
- }
- if tmpDate.After(lastDataDate) {
- tmpManualDateNum++
- continue
- }
- break
- }
- }
- // 需要的期数减去手工数据的期数,这才是A列指标需要的数据
- num = num - tmpManualDateNum
- if num > lenData {
- num = lenData
- }
- latestDateTime, _ := time.ParseInLocation(utils.FormatDate, edbInfo.LatestDate, time.Local)
- for i := 1; i <= num; i++ {
- dataTime, _ := time.ParseInLocation(utils.FormatDate, dataList[lenData-i].DataTime, time.Local)
- dataType := 1
- // 如果是预测指标,且当前值的日期,晚于实际日期,那么是预测值
- if edbInfo.EdbInfoType == 1 && dataTime.After(latestDateTime) {
- dataType = 5
- }
- resultDataList = append(resultDataList, request.ManualDataReq{
- DataType: dataType,
- DataTime: dataList[lenData-i].DataTime,
- ShowValue: fmt.Sprint(dataList[lenData-i].Value),
- Value: fmt.Sprint(dataList[lenData-i].Value),
- Decimal: decimal,
- DataTimeType: 1,
- })
- }
- return
- }
- // PadFirstEdbDataList 补齐第一列的数据
- func PadFirstEdbDataList(resultDataList []request.ManualDataReq, dateList []string, sortType string, decimal int) []request.ManualDataReq {
- originDataNum := len(resultDataList)
- requsetDateNum := len(dateList)
- if originDataNum >= requsetDateNum {
- return resultDataList
- }
- padNum := requsetDateNum - originDataNum
- if sortType == "asc" {
- for i := 0; i < padNum; i++ {
- resultDataList = append(resultDataList, request.ManualDataReq{
- DataType: 0,
- DataTime: dateList[originDataNum+i],
- Decimal: decimal,
- ShowValue: ``,
- Value: ``,
- DataTimeType: 1,
- })
- }
- } else {
- var tmpDateList []request.ManualDataReq
- for i := padNum - 1; i <= 0; i-- {
- tmpDateList = append(tmpDateList, request.ManualDataReq{
- DataType: 0,
- DataTime: dateList[originDataNum+i],
- Decimal: decimal,
- ShowValue: ``,
- Value: ``,
- DataTimeType: 1,
- })
- resultDataList = append(tmpDateList, resultDataList...)
- }
- }
- return resultDataList
- }
- // GetOtherEdbDataListFollowDate 获取其他列的数据,并设置日期的小数位数
- func GetOtherEdbDataListFollowDate(edbInfo *data_manage.EdbInfo, dateList []string, dateDecimal map[string]int) (resultDataList []request.ManualDataReq, err error) {
- lenDate := len(dateList)
- if lenDate <= 0 {
- return
- }
- sortDateList := dateList
- baseDateList := utils.StrArr{}
- baseDateList = append(baseDateList, sortDateList...)
- sort.Sort(baseDateList)
- sortDateList = append([]string{}, baseDateList...)
- endDateTime, err := time.ParseInLocation(utils.FormatDate, sortDateList[0], time.Local)
- if err != nil {
- return
- }
- firstDateTime, err := time.ParseInLocation(utils.FormatDate, sortDateList[lenDate-1], time.Local)
- if err != nil {
- return
- }
- var dataList []*data_manage.EdbDataList
- switch edbInfo.EdbInfoType {
- case 0:
- dataList, err = data_manage.GetEdbDataList(edbInfo.Source, edbInfo.SubSource, edbInfo.EdbInfoId, ``, ``)
- case 1:
- _, dataList, _, _, err, _ = data.GetPredictDataListByPredictEdbInfoId(edbInfo.EdbInfoId, ``, ``, false)
- default:
- err = errors.New(fmt.Sprint("获取失败,指标类型异常", edbInfo.EdbInfoType))
- }
- if err != nil {
- return
- }
- // 获取日期内的数据(包含开始日期前一个日期,以及 结束日期后一个日期,目的为了做空日期时的 插值法兼容)
- baseDataList := make([]*data_manage.EdbDataList, 0)
- var lastData *data_manage.EdbDataList
- var isInsert bool
- for _, data := range dataList {
- tmpDate := data.DataTime
- tmpDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, tmpDate, time.Local)
- if tmpErr != nil {
- err = tmpErr
- return
- }
- if tmpDateTime.Before(firstDateTime) {
- lastData = data
- continue
- }
- // 如果是第一次写入数据
- if !isInsert && lastData != nil {
- baseDataList = append(baseDataList, lastData)
- }
- if tmpDateTime.After(endDateTime) {
- baseDataList = append(baseDataList, data)
- break
- }
- baseDataList = append(baseDataList, data)
- isInsert = true
- }
- // 实际数据的日期map
- realValMap := make(map[string]string)
- for _, v := range baseDataList {
- realValMap[v.DataTime] = v.DataTime
- }
- // 插值法处理
- handleDataMap := make(map[string]float64)
- err = data.HandleDataByLinearRegression(baseDataList, handleDataMap)
- if err != nil {
- return
- }
- latestDateTime, _ := time.ParseInLocation(utils.FormatDate, edbInfo.LatestDate, time.Local)
- // 对于不存在的数据做补充
- for _, date := range sortDateList {
- dataType := 1
- if _, ok := realValMap[date]; !ok {
- dataType = 2
- } else {
- dataTime, _ := time.ParseInLocation(utils.FormatDate, date, time.Local)
- // 如果是预测指标,且当前值的日期,晚于实际日期,那么是预测值
- if edbInfo.EdbInfoType == 1 && dataTime.After(latestDateTime) {
- dataType = 5
- }
- }
- var value, showValue string
- if tmpVal, ok := handleDataMap[date]; ok {
- value = fmt.Sprint(tmpVal)
- showValue = value
- } else {
- dataType = 3
- }
- tmpData := request.ManualDataReq{
- DataType: dataType,
- DataTime: date,
- ShowValue: showValue,
- Value: value,
- }
- if v, ok := dateDecimal[date]; ok {
- tmpData.Decimal = v
- } else {
- tmpData.Decimal = -1
- }
- resultDataList = append(resultDataList, tmpData)
- }
- return
- }
- // GetOtherEdbDataList 获取其他列的数据
- func GetOtherEdbDataList(edbInfo *data_manage.EdbInfo, dateList []string, decimal int) (resultDataList []request.ManualDataReq, err error) {
- lenDate := len(dateList)
- if lenDate <= 0 {
- return
- }
- sortDateList := dateList
- baseDateList := utils.StrArr{}
- baseDateList = append(baseDateList, sortDateList...)
- sort.Sort(baseDateList)
- sortDateList = append([]string{}, baseDateList...)
- endDateTime, err := time.ParseInLocation(utils.FormatDate, sortDateList[0], time.Local)
- if err != nil {
- return
- }
- firstDateTime, err := time.ParseInLocation(utils.FormatDate, sortDateList[lenDate-1], time.Local)
- if err != nil {
- return
- }
- var dataList []*data_manage.EdbDataList
- switch edbInfo.EdbInfoType {
- case 0:
- dataList, err = data_manage.GetEdbDataList(edbInfo.Source, edbInfo.SubSource, edbInfo.EdbInfoId, ``, ``)
- case 1:
- _, dataList, _, _, err, _ = data.GetPredictDataListByPredictEdbInfoId(edbInfo.EdbInfoId, ``, ``, false)
- default:
- err = errors.New(fmt.Sprint("获取失败,指标类型异常", edbInfo.EdbInfoType))
- }
- if err != nil {
- return
- }
- // 获取日期内的数据(包含开始日期前一个日期,以及 结束日期后一个日期,目的为了做空日期时的 插值法兼容)
- baseDataList := make([]*data_manage.EdbDataList, 0)
- var lastData *data_manage.EdbDataList
- var isInsert bool
- for _, data := range dataList {
- tmpDate := data.DataTime
- tmpDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, tmpDate, time.Local)
- if tmpErr != nil {
- err = tmpErr
- return
- }
- if tmpDateTime.Before(firstDateTime) {
- lastData = data
- continue
- }
- // 如果是第一次写入数据
- if !isInsert && lastData != nil {
- baseDataList = append(baseDataList, lastData)
- }
- if tmpDateTime.After(endDateTime) {
- baseDataList = append(baseDataList, data)
- break
- }
- baseDataList = append(baseDataList, data)
- isInsert = true
- }
- // 实际数据的日期map
- realValMap := make(map[string]string)
- for _, v := range baseDataList {
- realValMap[v.DataTime] = v.DataTime
- }
- // 插值法处理
- handleDataMap := make(map[string]float64)
- err = data.HandleDataByLinearRegression(baseDataList, handleDataMap)
- if err != nil {
- return
- }
- latestDateTime, _ := time.ParseInLocation(utils.FormatDate, edbInfo.LatestDate, time.Local)
- // 对于不存在的数据做补充
- for _, date := range sortDateList {
- dataType := 1
- if _, ok := realValMap[date]; !ok {
- dataType = 2
- } else {
- dataTime, _ := time.ParseInLocation(utils.FormatDate, date, time.Local)
- // 如果是预测指标,且当前值的日期,晚于实际日期,那么是预测值
- if edbInfo.EdbInfoType == 1 && dataTime.After(latestDateTime) {
- dataType = 5
- }
- }
- var value, showValue string
- if tmpVal, ok := handleDataMap[date]; ok {
- value = fmt.Sprint(tmpVal)
- showValue = value
- } else {
- dataType = 3
- }
- resultDataList = append(resultDataList, request.ManualDataReq{
- DataType: dataType,
- DataTime: date,
- Decimal: decimal,
- ShowValue: showValue,
- Value: value,
- })
- }
- return
- }
- // GetFirstHistoryEdbDataList 获取指标的历史的数据
- func GetFirstHistoryEdbDataList(edbInfo *data_manage.EdbInfo, num int, endDate string, decimal int) (resultDataList []request.ManualDataReq, err error) {
- endDateTime, err := time.ParseInLocation(utils.FormatDate, endDate, time.Local)
- if err != nil {
- return
- }
- var dataList []*data_manage.EdbDataList
- switch edbInfo.EdbInfoType {
- case 0:
- dataList, err = data_manage.GetEdbDataList(edbInfo.Source, edbInfo.SubSource, edbInfo.EdbInfoId, ``, endDate)
- case 1:
- _, dataList, _, _, err, _ = data.GetPredictDataListByPredictEdbInfoId(edbInfo.EdbInfoId, ``, endDate, true)
- default:
- err = errors.New(fmt.Sprint("获取失败,指标类型异常", edbInfo.EdbInfoType))
- }
- if err != nil {
- return
- }
- // 获取需要的期数
- lenData := len(dataList)
- if lenData <= 0 {
- return
- }
- lastData := dataList[lenData-1]
- lastDataDateTime, err := time.ParseInLocation(utils.FormatDate, lastData.DataTime, time.Local)
- if err != nil {
- return
- }
- if endDateTime.Equal(lastDataDateTime) || lastDataDateTime.After(endDateTime) {
- dataList = dataList[:lenData-1]
- lenData = len(dataList)
- }
- if num > lenData {
- num = lenData
- }
- latestDateTime, _ := time.ParseInLocation(utils.FormatDate, edbInfo.LatestDate, time.Local)
- for i := 1; i <= num; i++ {
- dataTime, _ := time.ParseInLocation(utils.FormatDate, dataList[lenData-i].DataTime, time.Local)
- dataType := 1
- // 如果是预测指标,且当前值的日期,晚于实际日期,那么是预测值
- if edbInfo.EdbInfoType == 1 && dataTime.After(latestDateTime) {
- dataType = 5
- }
- resultDataList = append(resultDataList, request.ManualDataReq{
- DataType: dataType,
- DataTime: dataList[lenData-i].DataTime,
- Decimal: decimal,
- ShowValue: fmt.Sprint(dataList[lenData-i].Value),
- Value: fmt.Sprint(dataList[lenData-i].Value),
- })
- }
- return
- }
- // GetEdbIdsFromExcelCodes 获取表格中的指标IDs
- func GetEdbIdsFromExcelCodes(excelCodes []string, sysUserId int, lang string) (edbIds []int, err error) {
- edbIds = make([]int, 0)
- edbIdExist := make(map[int]bool)
- for _, v := range excelCodes {
- // 表格详情
- detail, msg, e := GetExcelDetailInfoByUnicode(v, sysUserId, lang)
- if e != nil {
- err = fmt.Errorf("GetExcelDetailInfoByExcelInfoId err: %s, errMsg: %s", e.Error(), msg)
- return
- }
- // 自定义表格
- if detail.Source == utils.TIME_TABLE {
- jsonByte, e := json.Marshal(detail.TableData)
- if e != nil {
- err = fmt.Errorf("JSON格式化自定义表格数据失败, Err: %s", e.Error())
- return
- }
- var tableData request.TableDataReq
- if e = json.Unmarshal(jsonByte, &tableData); e != nil {
- err = fmt.Errorf("解析自定义表格数据失败, Err: %s", e.Error())
- return
- }
- for _, tv := range tableData.EdbInfoIdList {
- if edbIdExist[tv] {
- continue
- }
- edbIdExist[tv] = true
- edbIds = append(edbIds, tv)
- }
- }
- // 混合表格
- if detail.Source == utils.MIXED_TABLE {
- jsonByte, e := json.Marshal(detail.TableData)
- if e != nil {
- err = fmt.Errorf("JSON格式化混合表格数据失败, Err: %s", e.Error())
- return
- }
- var tableData request.MixedTableReq
- if e = json.Unmarshal(jsonByte, &tableData); e != nil {
- err = fmt.Errorf("解析混合表格数据失败, Err: %s", e.Error())
- return
- }
- if len(tableData.Data) > 0 {
- for _, td := range tableData.Data {
- for _, tv := range td {
- if tv.EdbInfoId > 0 && !edbIdExist[tv.EdbInfoId] {
- edbIdExist[tv.EdbInfoId] = true
- edbIds = append(edbIds, tv.EdbInfoId)
- }
- }
- }
- }
- }
- }
- return
- }
- // GetExcelEdbBatchRefreshKey 获取批量刷新表格指标缓存key
- func GetExcelEdbBatchRefreshKey(source string, primaryId, subId int) string {
- if source == `` {
- return ``
- }
- return fmt.Sprint("batch_refresh_excel_edb:", source, ":", primaryId, ":", subId)
- }
- // GetEdbSourceByEdbInfoIdList 获取关联指标的来源
- func GetEdbSourceByEdbInfoIdList(edbInfoIdList []int) (sourceNameList, sourceNameEnList []string, err error) {
- sourceNameList = make([]string, 0)
- sourceNameEnList = make([]string, 0)
- sourceMap := make(map[int]string)
- edbInfoList, tmpErr := data_manage.GetEdbInfoByIdList(edbInfoIdList)
- if tmpErr != nil {
- err = tmpErr
- return
- }
- for _, v := range edbInfoList {
- // 指标类型:1:基础指标,2:计算指标
- if v.EdbType == 2 {
- baseEdbInfoArr, _, _ := data_manage.GetRefreshEdbInfoFromBase(v.EdbInfoId, v.Source)
- for _, baseEdbInfo := range baseEdbInfoArr {
- if baseEdbInfo.EdbInfoType == 0 { //普通指标才参与,预测指标不参与
- sourceMap[baseEdbInfo.Source] = baseEdbInfo.SourceName
- }
- }
- } else {
- sourceMap[v.Source] = v.SourceName
- }
- }
- for source, sourceName := range sourceMap {
- if utils.InArrayByInt([]int{utils.DATA_SOURCE_MANUAL, utils.DATA_SOURCE_MYSTEEL_CHEMICAL}, source) {
- continue
- }
- sourceNameList = append(sourceNameList, sourceName)
- sourceNameEn, ok := utils.DataSourceEnMap[source]
- if !ok {
- sourceNameEn = sourceName
- }
- sourceNameEnList = append(sourceNameEnList, sourceNameEn)
- }
- //sourceNameList = append(sourceNameList, utils.ChartDefaultNameCn)
- //sourceNameEnList = append(sourceNameEnList, utils.ChartDefaultNameEn)
- // 图表来源
- conf, e := models.GetBusinessConf()
- if e != nil {
- return
- }
- if conf[models.BusinessConfCompanyName] != "" {
- sourceNameList = append(sourceNameList, conf[models.BusinessConfCompanyName])
- sourceNameEnList = append(sourceNameEnList, conf[models.BusinessConfCompanyName])
- }
- return
- }
- // GetCustomAnalysisOpButton 获取自定义分析按钮权限
- func GetCustomAnalysisOpButton(sysUser *system.Admin, belongUserId int, permissionType []int) (button excel.ExcelInfoDetailButton) {
- // 管理员/所属人所有权限
- if sysUser.RoleTypeCode == utils.ROLE_TYPE_CODE_ADMIN || sysUser.RoleTypeCode == utils.ROLE_TYPE_CODE_FICC_ADMIN || sysUser.AdminId == belongUserId {
- button.RefreshButton = true
- button.CopyButton = true
- button.DownloadButton = true
- button.OpEdbButton = true
- button.RefreshEdbButton = true
- button.OpButton = true
- button.DeleteButton = true
- return
- }
- // 非管理员-被分享人
- for _, v := range permissionType {
- if v == 1 {
- button.RefreshButton = true
- button.CopyButton = true
- button.DownloadButton = true
- }
- if v == 2 {
- button.OpEdbButton = true
- button.RefreshEdbButton = true
- button.OpButton = true
- button.CopyButton = true
- button.DownloadButton = true
- }
- }
- return
- }
- func parseExcelScopeCoord(scopeList []string) (scopeCoord string, err error) {
- if len(scopeList) == 2 {
- x1, y1, er := excelize.CellNameToCoordinates(scopeList[0])
- if er != nil {
- return "", er
- }
- x2, y2, er := excelize.CellNameToCoordinates(scopeList[1])
- if er != nil {
- return "", er
- }
- scopeCoord = fmt.Sprintf("%d,%d,%d,%d", x1, y1, x2, y2)
- }
- if len(scopeList) == 1 {
- x1, y1, er := excelize.CellNameToCoordinates(scopeList[0])
- if er != nil {
- return "", er
- }
- scopeCoord = fmt.Sprintf("%d,%d", x1, y1)
- }
- return
- }
- func ExcelRuleFormat(req *request.ExcelRuleMappingReq, lang string) (res *excel.ExcelInfoRuleMapping, err error) {
- res = new(excel.ExcelInfoRuleMapping)
- if req.RuleType == 5 {
- switch req.LeftValue {
- // 今天
- case "today":
- res.LeftValueBack = req.LeftValue
- // 明天
- case "tomorrow":
- res.LeftValueBack = req.LeftValue
- // 最近7天
- case "last7days":
- res.LeftValueBack = req.LeftValue
- // 上周
- case "lastweek":
- res.LeftValueBack = req.LeftValue
- // 本周
- case "thisweek":
- res.LeftValueBack = req.LeftValue
- // 下周
- case "nextweek":
- res.LeftValueBack = req.LeftValue
- // 上月
- case "lastmonth":
- res.LeftValueBack = req.LeftValue
- // 本月
- case "thismonth":
- res.LeftValueBack = req.LeftValue
- // 下月
- case "nextmonth":
- res.LeftValueBack = req.LeftValue
- default:
- err = errors.New("发生日期规则错误")
- return
- }
- }
- // 格式化条件值
- switch req.LeftValueType {
- // 坐标
- case 2:
- x, y, err := excelize.CellNameToCoordinates(req.LeftValue)
- if err != nil {
- return nil, err
- }
- res.LeftValue = req.LeftValue
- res.LeftValueBack = fmt.Sprintf("%d,%d", x, y)
- default:
- res.LeftValue = req.LeftValue
- res.LeftValueBack = req.LeftValue
- }
- switch req.RightValueType {
- // 坐标
- case 2:
- x, y, err := excelize.CellNameToCoordinates(req.RightValue)
- if err != nil {
- return nil, err
- }
- res.RightValue = req.RightValue
- res.RightValueBack = fmt.Sprintf("%d,%d", x, y)
- default:
- res.RightValue = req.RightValue
- res.RightValueBack = req.RightValue
- }
- if res.LeftValueBack == "" {
- res.LeftValueBack = req.LeftValue
- }
- if res.RightValueBack == "" {
- res.RightValueBack = req.RightValue
- }
- res.RuleType = req.RuleType
- res.ExcelInfoId = req.ExcelInfoId
- res.ExcelInfoRuleMappingId = req.ExcelRuleMappingId
- res.LeftValueType = req.LeftValueType
- res.RightValueType = req.RightValueType
- res.FontColor = req.FontColor
- res.BackgroundColor = req.BackgroundColor
- res.Remark = req.Remark
- res.RemarkEn = req.Remark
- res.ScopeShow = req.Scope
- scopeList := strings.Split(req.Scope, ":")
- res.ScopeCoord, err = parseExcelScopeCoord(scopeList)
- return
- }
- func AddExcelRule(req *request.ExcelRuleMappingReq, lang string) (err error) {
- excelRule, err := ExcelRuleFormat(req, lang)
- if err != nil {
- return
- }
- if excelRule.ExcelInfoRuleMappingId != 0 {
- return errors.New("规则已存在")
- }
- excelRule.CreateTime = time.Now()
- _, err = excelRule.Insert()
- return
- }
- func ModifyExcelRule(req *request.ExcelRuleMappingReq, lang string) (err error) {
- excelInfo, err := excel.GetExcelRuleMappingById(req.ExcelRuleMappingId)
- if err != nil {
- return
- }
- editExcelInfo, err := ExcelRuleFormat(req, lang)
- if err != nil {
- return
- }
- var updateCols []string
- if excelInfo.LeftValue != editExcelInfo.LeftValue {
- updateCols = append(updateCols, "LeftValue")
- updateCols = append(updateCols, "LeftValueBack")
- }
- if excelInfo.LeftValueType != editExcelInfo.LeftValueType {
- updateCols = append(updateCols, "LeftValueType")
- }
- if excelInfo.RightValue != editExcelInfo.RightValue {
- updateCols = append(updateCols, "RightValue")
- updateCols = append(updateCols, "RightValueBack")
- }
- if excelInfo.RightValueType != editExcelInfo.RightValueType {
- updateCols = append(updateCols, "RightValueType")
- }
- if excelInfo.FontColor != editExcelInfo.FontColor {
- updateCols = append(updateCols, "FontColor")
- }
- if excelInfo.BackgroundColor != editExcelInfo.BackgroundColor {
- updateCols = append(updateCols, "BackgroundColor")
- }
- if excelInfo.Remark != editExcelInfo.Remark {
- updateCols = append(updateCols, "Remark")
- }
- if excelInfo.RemarkEn != editExcelInfo.RemarkEn {
- updateCols = append(updateCols, "RemarkEn")
- }
- if excelInfo.ScopeShow != editExcelInfo.ScopeShow {
- updateCols = append(updateCols, "ScopeCoord")
- updateCols = append(updateCols, "ScopeShow")
- }
- if len(updateCols) > 0 {
- err = editExcelInfo.Update(updateCols)
- }
- return
- }
- // GetExcelRuleList 获取规则列表
- func GetExcelRuleList(excelInfoId int) (resp *response.ExcelRuleListResp, err error) {
- resp = new(response.ExcelRuleListResp)
- excelInfoList, err := excel.GetExcelRuleMappingByExcelInfoId(excelInfoId)
- if err != nil {
- return
- }
- for _, v := range excelInfoList {
- v.BackgroundColor = strings.TrimSpace(v.BackgroundColor)
- v.FontColor = strings.TrimSpace(v.FontColor)
- resp.List = append(resp.List, v)
- }
- resp.List = excelInfoList
- return
- }
- // GetExcelRuleDetail 获取规则列表
- func GetExcelRuleDetail(excelInfoMappingId int) (resp *excel.ExcelInfoRuleMappingView, err error) {
- resp = new(excel.ExcelInfoRuleMappingView)
- excelInfoDetail, err := excel.GetExcelRuleMappingById(excelInfoMappingId)
- if err != nil {
- return
- }
- resp = excelInfoDetail
- return
- }
|