123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996 |
- 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
- }
- 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
- }
|