package data import ( "errors" "eta_gn/eta_api/models" "eta_gn/eta_api/models/data_manage" "eta_gn/eta_api/models/system" "eta_gn/eta_api/services/alarm_msg" "eta_gn/eta_api/utils" "fmt" "github.com/shopspring/decimal" "github.com/tealeg/xlsx" "strconv" "strings" "time" ) func GetManualSysUser(keyWord string) (list []*data_manage.ManualSysUser, err error) { //departmentId := 1 list = make([]*data_manage.ManualSysUser, 0) departmentItems, err := system.GetSysDepartmentAll() if err != nil { return list, err } for _, dv := range departmentItems { department := new(data_manage.ManualSysUser) department.ItemId = dv.DepartmentId * 10000 department.ItemName = dv.DepartmentName fmt.Println(department.ItemId, department.ItemName) //GetSysuserList var condition string var pars []interface{} if keyWord != "" { condition += ` AND (real_name LIKE ? OR admin_name LIKE ? OR mobile LIKE ? ) ` pars = utils.GetLikeKeywordPars(pars, keyWord, 3) } sysUsers, err := system.GetSysUserItems(condition, pars) if err != nil { return list, err } groups, err := system.GetSysGroupByDepartmentId(dv.DepartmentId) if err != nil { return list, err } dg := make([]*data_manage.ManualSysUser, 0) for _, v := range groups { group := new(data_manage.ManualSysUser) group.ItemId = v.DepartmentId * 100000 group.ItemName = v.GroupName for _, sv := range sysUsers { user := new(data_manage.ManualSysUser) user.ItemId = sv.AdminId user.ItemName = sv.RealName if sv.GroupId == v.GroupId { group.Children = append(group.Children, user) } } if len(group.Children) > 0 { dg = append(dg, group) } } if len(groups) <= 0 { group := new(data_manage.ManualSysUser) group.ItemId = dv.DepartmentId * 100000 group.ItemName = "无分组" for _, sv := range sysUsers { user := new(data_manage.ManualSysUser) user.ItemId = sv.AdminId user.ItemName = sv.RealName if sv.DepartmentId == dv.DepartmentId && sv.GroupId == 0 { group.Children = append(group.Children, user) } } if len(group.Children) > 0 { dg = append(dg, group) } } if len(dg) > 0 { department.Children = dg list = append(list, department) } } return } type ManualIndexSource2EdbReq struct { EdbCode string EdbName string Frequency string Unit string ClassifyId int AdminId int AdminRealName string } // ManualIndexSource2Edb // @Description: 新增手工数据源到指标库 // @author: Roc // @datetime 2024-07-26 13:23:19 // @param req ManualIndexSource2EdbReq // @param lang string // @return edb *data_manage.EdbInfo // @return err error // @return errMsg string // @return skip bool func ManualIndexSource2Edb(req ManualIndexSource2EdbReq, lang string) (edb *data_manage.EdbInfo, err error, errMsg string, skip bool) { if req.EdbCode == "" { err = fmt.Errorf("指标ID为空") return } defer func() { if err != nil { tips := fmt.Sprintf("ManualIndexSource2Edb新增失败, Err: %s", err.Error()) fmt.Println(tips) utils.FileLog.Info(tips) } }() source := utils.DATA_SOURCE_MANUAL // 是否已有指标数据 dataList, e := data_manage.GetEdbDataAllByEdbCode(req.EdbCode, source, utils.DATA_SUB_SOURCE_EDB, utils.EDB_DATA_LIMIT) if e != nil { err = fmt.Errorf("获取指标数据失败, Err: %s", e.Error()) return } // 新增指标数据 if len(dataList) == 0 { res, e := AddEdbData(source, req.EdbCode, req.Frequency) if e != nil { err = fmt.Errorf("index_lib: 新增指标数据失败, Err: %s", e.Error()) return } if res == nil { err = fmt.Errorf("index_lib: 新增指标数据失败, res nil") return } if res.Ret != 200 { err = fmt.Errorf("index_lib: 新增指标数据失败, Ret: %d", res.Ret) return } } // 是否新增过指标 exist, e := data_manage.GetEdbInfoByEdbCode(source, req.EdbCode) if e != nil && !utils.IsErrNoRow(e) { err = fmt.Errorf("获取指标是否存在失败, err: %s", e.Error()) return } if exist != nil { skip = true return } // 开始结束时间 var startDate, endDate string minMax, e := data_manage.GetEdbInfoMaxAndMinInfo(source, utils.DATA_SUB_SOURCE_EDB, req.EdbCode) if e != nil && !utils.IsErrNoRow(e) { err = fmt.Errorf("MinMax: 获取指标极值失败, err: %s", e.Error()) return } if minMax != nil { startDate = minMax.MinDate endDate = minMax.MaxDate } // 新增指标库 edbInfo, e, msg, _ := EdbInfoAdd(source, utils.DATA_SUB_SOURCE_EDB, req.ClassifyId, req.EdbCode, req.EdbName, req.Frequency, req.Unit, startDate, endDate, req.AdminId, req.AdminRealName, lang) if e != nil { errMsg = msg err = fmt.Errorf("EdbInfo: 新增指标失败, err: %s", e.Error()) return } edb = edbInfo // 新增es go AddOrEditEdbInfoToEs(edbInfo.EdbInfoId) return } // ImportManualData // @Description: 数据导入 // @author: Roc // @datetime 2024-08-01 11:27:21 // @param path string // @param sysUser *system.Admin // @return successCount int // @return failCount int // @return err error // @return errMsg string func ImportManualData(path string, sysUser *system.Admin) (successCount, failCount int, err error, errMsg string) { // 错误信息 errMsgList := make([]string, 0) // 操作记录 recordMap := make(map[string]string) defer func() { recordList := make([]*models.EdbinfoOpRecord, 0) for tradeCode, remark := range recordMap { recordList = append(recordList, &models.EdbinfoOpRecord{ TradeCode: tradeCode, Remark: remark, UserId: sysUser.AdminId, UserName: sysUser.RealName, CreateTime: time.Now(), }) // 修改最大最小值 go ModifyManualEdbMaxMinDate(tradeCode) } if len(recordList) > 0 { go func() { obj := models.EdbinfoOpRecord{} _ = obj.MulCreate(recordList) }() } // 错误信息记录 if len(errMsgList) > 0 { utils.FileLog.Info("导入失败, errMsgList: %v", strings.Join(errMsgList, "\n")) } }() errMsg = `导入失败` xlFile, err := xlsx.OpenFile(path) if err != nil { fmt.Println(err.Error()) return } if len(xlFile.Sheets) <= 0 { errMsg = "导入模板异常" err = errors.New(errMsg) return } //导入失败数据 failDataList := make([]*models.EdbdataImportFail, 0) var indexDataList []ImportManualIndexData // 指标名称列表 indexNameList := make([]string, 0) // 模板校验，然后处理成标准化格式 for _, sheet := range xlFile.Sheets { var tmpIndexDataList []ImportManualIndexData var tmpFailDataList []*models.EdbdataImportFail rowList := sheet.Rows if len(rowList) <= 0 { errMsg = sheet.Name + "页异常" err = errors.New(errMsg) return } templateType := 1 // 模板类型 minCellNum := 6 // 模板最小列数 headerCell := rowList[0].Cells // 确定模板 for _, v := range headerCell { if v.String() == "导入模板2/Import Template 2" { templateType = 2 minCellNum = 2 break } } // 如果小于最少列数，则报错 if len(headerCell) < minCellNum { errMsg = sheet.Name + "页模板异常" err = errors.New(errMsg) return } switch templateType { case 2: // 模板2需要走对应的取数逻辑 tmpIndexDataList, tmpFailDataList, err, errMsg = getDataByTemplate2(sheet, sysUser.AdminId) default: // 模板1需要走对应的取数逻辑 tmpIndexDataList, tmpFailDataList, err, errMsg = getDataByTemplate1(sheet, sysUser.AdminId) } indexDataList = append(indexDataList, tmpIndexDataList...) failDataList = append(failDataList, tmpFailDataList...) } indexDataListMap := make(map[string]ImportManualIndexData) for _, v := range indexDataList { indexData, ok := indexDataListMap[v.IndexName] if !ok { // 没有就赋值 indexDataListMap[v.IndexName] = v indexNameList = append(indexNameList, v.IndexName) continue } indexData.Unit = v.Unit indexData.ClassName = v.ClassName indexData.Frequency = v.Frequency indexData.DataMap = v.DataMap } if len(indexNameList) <= 0 { return } //超管账号可以查看分类下的所有频度数据 userId := sysUser.AdminId if sysUser.RoleTypeCode == utils.ROLE_TYPE_CODE_ADMIN { userId = 0 } //获取账户所拥有权限的分类id集合 classifyNameStrList, edbDataClassifyMap, err := GetEdbClassifyNameListByAdminId(int64(userId)) if err != nil { errMsg = "获取分类数据失败" return } //指标map targetMap := make(map[string]*models.Edbinfo) defer func() { for _, target := range targetMap { //结束后，清除掉对应的缓存 key := "import:edbinfo:data:" + target.TradeCode utils.Rc.Delete(key) //将该指标的code加入到 “手工数据导入后刷新” 缓存 if utils.Re == nil { err := utils.Rc.LPush(utils.CACHE_IMPORT_MANUAL_DATA, target.TradeCode) if err != nil { fmt.Println("CACHE_IMPORT_MANUAL_DATA LPush Err:" + err.Error()) } } } }() // 所有的指标数据map edbCodeDataMap := make(map[string]map[string]string) { // 获取指标信息 manualIndexList, tmpErr := models.GetEdbinfoListBySecNameList(indexNameList) if tmpErr != nil { err = tmpErr errMsg = "获取指标信息失败" return } tradeCodeList := make([]string, 0) for _, v := range manualIndexList { targetMap[v.SecName] = v tradeCodeList = append(tradeCodeList, v.TradeCode) recordMap[v.TradeCode] = "更新数据" } // 获取明细数据 dataList, tmpErr := models.GetTargetsDataListByCodeList(tradeCodeList) if tmpErr != nil { err = tmpErr errMsg = "获取指标明细数据失败" return } for _, tmpData := range dataList { _, ok := edbCodeDataMap[tmpData.TradeCode] if !ok { edbCodeDataMap[tmpData.TradeCode] = make(map[string]string) } edbCodeDataMap[tmpData.TradeCode][tmpData.Dt] = tmpData.Close } } addDataList := make([]*models.Edbdata, 0) for _, v := range indexDataListMap { if len(v.DataMap) <= 0 { continue } var tmpDate, tmpValue string for date, val := range v.DataMap { tmpDate = date tmpValue = val break } if !strings.Contains(strings.Join(classifyNameStrList, ","), v.ClassName) { failItem := &models.EdbdataImportFail{ ClassifyName: v.ClassName, CreateDate: tmpDate, SecName: v.IndexName, Close: tmpValue, Remark: "没有该品种分类权限", SysUserId: fmt.Sprint(sysUser.AdminId), Frequency: v.Frequency, Unit: v.Unit, } failDataList = append(failDataList, failItem) continue } //获取指标分类信息 classify, ok := edbDataClassifyMap[v.ClassName] if !ok { failItem := &models.EdbdataImportFail{ ClassifyName: v.ClassName, CreateDate: tmpDate, SecName: v.IndexName, Close: tmpValue, Remark: "指标分类不存在", SysUserId: fmt.Sprint(sysUser.AdminId), Frequency: v.Frequency, Unit: v.Unit, } failDataList = append(failDataList, failItem) continue } target, ok := targetMap[v.IndexName] if !ok { if v.Frequency == "" { failItem := &models.EdbdataImportFail{ ClassifyName: v.ClassName, CreateDate: tmpDate, SecName: v.IndexName, Close: tmpValue, Remark: "新增指标失败，频度字段为空", SysUserId: fmt.Sprint(sysUser.AdminId), Frequency: v.Frequency, Unit: v.Unit, } failDataList = append(failDataList, failItem) continue } if v.Unit == "" { failItem := &models.EdbdataImportFail{ ClassifyName: v.ClassName, CreateDate: tmpDate, SecName: v.IndexName, Close: tmpValue, Remark: "新增指标失败，单位字段为空", SysUserId: fmt.Sprint(sysUser.AdminId), Frequency: v.Frequency, Unit: v.Unit, } failDataList = append(failDataList, failItem) continue } tmpErr := AddEdbInfo(v.IndexName, v.Unit, v.Frequency, "", sysUser.Mobile, classify.ClassifyId, sysUser.AdminId, sysUser.RealName) if tmpErr != nil { fmt.Println("line 158") failItem := &models.EdbdataImportFail{ ClassifyName: v.ClassName, CreateDate: tmpDate, SecName: v.IndexName, Close: tmpValue, Remark: "新增指标失败", SysUserId: fmt.Sprint(sysUser.AdminId), Frequency: v.Frequency, Unit: v.Unit, } failDataList = append(failDataList, failItem) continue } tmpTarget, tmpErr := models.GetTargetBySecName(v.IndexName) target = tmpTarget targetMap[v.IndexName] = target recordMap[target.TradeCode] = "创建指标" } if target == nil { fmt.Println("指标不存在") failItem := &models.EdbdataImportFail{ ClassifyName: v.ClassName, CreateDate: tmpDate, SecName: v.IndexName, Close: tmpValue, Remark: "指标不存在", SysUserId: fmt.Sprint(sysUser.AdminId), Frequency: v.Frequency, Unit: v.Unit, } failDataList = append(failDataList, failItem) continue } //设置10分钟缓存，不允许其他地方删除 key := "import:edbinfo:data:" + target.TradeCode utils.Rc.SetNX(key, 1, time.Second*600) //更新指标信息 { updateCols := make([]string, 0) //更新指标分类 if target.ClassifyId <= 0 && classify.ClassifyId > 0 { target.ClassifyId = classify.ClassifyId updateCols = append(updateCols, "ClassifyId") } if target.Frequency != v.Frequency { target.Frequency = v.Frequency target.NoticeTime = "" updateCols = append(updateCols, "Frequency", "NoticeTime") } if target.Unit != v.Unit { target.Unit = v.Unit updateCols = append(updateCols, "Unit") } if len(updateCols) > 0 { _ = target.Update(updateCols) } } { // 判断指标数据列表是否已经存在 tmpDataMap, ok := edbCodeDataMap[target.TradeCode] if !ok { tmpDataMap = make(map[string]string) edbCodeDataMap[target.TradeCode] = tmpDataMap } for createDate, closeVal := range v.DataMap { //判断数据是否已经存在 tmpVal, ok := tmpDataMap[createDate] //数据已存在，进行更新操作 if ok { if tmpVal != closeVal { edbCodeDataMap[target.TradeCode][createDate] = closeVal tmpErr := models.ModifyTargetsDataByImport(target.TradeCode, createDate, closeVal) if tmpErr != nil { errMsgList = append(errMsgList, fmt.Sprintf("%s修改数据失败，日期：%s，值：%v，Err：%s", target.TradeCode, createDate, closeVal, tmpErr.Error())) } } } else { //数据不存在，进行新增操作 if target.TradeCode != "" && createDate != "" && closeVal != "" { addDataList = append(addDataList, &models.Edbdata{ TradeCode: target.TradeCode, Dt: createDate, Close: closeVal, ModifyTime: time.Now(), }) edbCodeDataMap[target.TradeCode][createDate] = closeVal } } successCount++ } } } // 批量添加明细数据 if len(addDataList) > 0 { tmpErr := models.OnlyMultiAddEdbdata(addDataList) if tmpErr != nil { fmt.Println("line 221") errMsgList = append(errMsgList, fmt.Sprintf("批量添加明细数据失败，Err：%s", tmpErr.Error())) } } // 失败数量 failCount = len(failDataList) //fmt.Println("failDataList:", len(failDataList)) if failCount > 0 { //先删除导入失败记录 _ = models.DelEdbDataImportFail(sysUser.AdminId) // 批量添加导入失败记录 err = models.MultiAddEdbdataImportFail(failDataList) if err != nil { go alarm_msg.SendAlarmMsg("导入数据 新增失败记录失败，Err："+err.Error(), 3) //go utils.SendEmail(utils.APPNAME+"失败提醒", "导入数据 新增失败记录失败:Err:"+err.Error(), utils.EmailSendToUsers) } { //错误信息字符串切片，最后作为发送邮件通知使用 failContents := make([]string, 0) for _, v := range failDataList { failContents = append(failContents, fmt.Sprint(v.SecName, "导入失败：", v.Remark)) } utils.FileLog.Info("导入数据 存在部分数据导入失败:" + strings.Join(failContents, "；")) //导入失败的话，最后统一邮件提醒就好啦，不需要每次都去提醒 go alarm_msg.SendAlarmMsg("导入数据 存在部分数据导入失败:"+strings.Join(failContents, "；"), 3) } } return } // ImportManualIndexData // @Description: excel模板后的内容 type ImportManualIndexData struct { IndexName string `description:"指标名称"` Unit string `description:"单位"` ClassName string `description:"所属品种"` Frequency string `description:"频度"` DataMap map[string]string `description:"时间数据"` } // getDataByTemplate1 // @Description: 根据模板1获取数据 // @author: Roc // @datetime 2024-07-24 16:17:45 // @param sheet *xlsx.Sheet // @return indexDataList []ImportManualIndexData // @return err error // @return errMsg string func getDataByTemplate1(sheet *xlsx.Sheet, sysUserId int) (indexDataList []ImportManualIndexData, failDataList []*models.EdbdataImportFail, err error, errMsg string) { fmt.Println("sheet name: ", sheet.Name) indexDataList = make([]ImportManualIndexData, 0) indexDataListMap := make(map[string]ImportManualIndexData, 0) failDataList = make([]*models.EdbdataImportFail, 0) //遍历行读取 maxRow := sheet.MaxRow fmt.Println("maxRow:", maxRow) // 表头信息 if maxRow <= 2 { errMsg = "模板异常1" err = errors.New(errMsg) return } // 表头校验 { headerRow := sheet.Row(1) cells := headerRow.Cells if len(cells) < 6 { errMsg = "导入文件异常，请下载最新导入模板文件" err = errors.New(errMsg) return } templateFail := false if cells[0].Value != "品种/Variety" { templateFail = true } if cells[1].Value != "指标名称/Indicator Name" { templateFail = true } if cells[2].Value != "指标日期/Indicator Date" { templateFail = true } if cells[3].Value != "值/Value" { templateFail = true } if cells[4].Value != "频度/Frequency" { templateFail = true } if cells[5].Value != "单位/Unit" { templateFail = true } if templateFail { errMsg = "导入文件异常，请下载最新导入模板文件" err = errors.New(errMsg) return } } for i := 2; i < maxRow; i++ { row := sheet.Row(i) cells := row.Cells lenCell := len(cells) // 过滤空白行 if lenCell <= 0 { continue } if lenCell < 6 { if cells[0].Value == `` { continue } errMsg = "导入文件异常，请下载最新导入模板文件" err = errors.New(errMsg) return } classifyName := strings.TrimSpace(cells[0].Value) //分类 if classifyName == "" { //过滤空白行 continue } // 指标名称 cell1 := cells[1].Value indexName := strings.TrimSpace(cell1) if indexName == "" { //过滤空白行 continue } //createDate := utils.ConvertToFormatDay(cell1) //录入日期 createDate := cells[2].Value //指标日期 frequency := strings.TrimSpace(cells[4].String()) //频度 unit := strings.TrimSpace(cells[5].String()) //单位 closeVal := cells[3].Value //值 if strings.Contains(closeVal, "#N/A") { continue } currDate, tmpErr := getExcelDate(createDate) if tmpErr != nil { failDataList = append(failDataList, &models.EdbdataImportFail{ //Id: 0, ClassifyName: classifyName, CreateDate: createDate, SecName: indexName, Close: closeVal, Remark: "日期格式异常", SysUserId: strconv.Itoa(sysUserId), Frequency: frequency, Unit: unit, }) continue } closeValFloat, tmpErr := cells[3].Float() //值 if tmpErr != nil { failDataList = append(failDataList, &models.EdbdataImportFail{ //Id: 0, ClassifyName: classifyName, CreateDate: currDate, SecName: indexName, Close: cells[3].Value, Remark: "值类型异常", SysUserId: strconv.Itoa(sysUserId), Frequency: frequency, Unit: unit, }) continue } newDecimal := decimal.NewFromFloat(closeValFloat) newDecimal.Round(4) closeVal = newDecimal.String() if strings.Contains(closeVal, "#N/A") { continue } _, ok := indexDataListMap[indexName] if !ok { indexDataListMap[indexName] = ImportManualIndexData{ IndexName: indexName, Unit: unit, ClassName: classifyName, Frequency: frequency, DataMap: make(map[string]string), } } indexDataListMap[indexName].DataMap[currDate] = closeVal } for _, v := range indexDataListMap { indexDataList = append(indexDataList, v) } return } // getDataByTemplate2 // @Description: 根据模板2获取数据 // @author: Roc // @datetime 2024-07-24 16:17:56 // @param sheet *xlsx.Sheet // @return indexDataList []ImportManualIndexData // @return err error // @return errMsg string func getDataByTemplate2(sheet *xlsx.Sheet, sysUserId int) (indexDataList []ImportManualIndexData, failDataList []*models.EdbdataImportFail, err error, errMsg string) { fmt.Println("sheet name: ", sheet.Name) indexDataList = make([]ImportManualIndexData, 0) failDataList = make([]*models.EdbdataImportFail, 0) //遍历行读取 maxRow := sheet.MaxRow fmt.Println("maxRow:", maxRow) varietyList := make([]string, 0) indexNameList := make([]string, 0) unitList := make([]string, 0) frequencyList := make([]string, 0) // make(map[指标下标]map[日期]值) indexDateValueMap := make(map[int]map[string]string) // 表头信息 if maxRow <= 4 { errMsg = "模板异常1" err = errors.New(errMsg) return } // 表头处理 for i := 1; i <= 4; i++ { row := sheet.Row(i) cells := row.Cells switch i { case 1: for k, v := range cells { if k == 0 { continue } varietyList = append(varietyList, strings.TrimSpace(v.String())) } case 2: for k, v := range cells { if k == 0 { continue } indexNameList = append(indexNameList, strings.TrimSpace(v.String())) } case 3: for k, v := range cells { if k == 0 { continue } unitList = append(unitList, strings.TrimSpace(v.String())) } case 4: for k, v := range cells { if k == 0 { continue } frequencyList = append(frequencyList, strings.TrimSpace(v.String())) } } } maxNameIndex := len(indexNameList) - 1 maxUnitIndex := len(unitList) - 1 maxClassNameIndex := len(varietyList) - 1 maxFrequencyIndex := len(frequencyList) - 1 // 数据处理 for i := 5; i < maxRow; i++ { row := sheet.Row(i) cells := row.Cells // 当前日期 var currDate string // 日期是否异常，异常的话，就跳过，进入下一行数据处理 var dateErr bool // 数据处理 for k, v := range cells { if k == 0 { tmpCurrDate, tmpErr := getExcelDate(v.Value) if tmpErr != nil { dateErr = true } currDate = tmpCurrDate continue } key := k - 1 // 日期异常，所以不处理该行了 if dateErr { var indexName, unit, classifyName, frequency string if key <= maxNameIndex { indexName = indexNameList[key] } if key <= maxUnitIndex { unit = unitList[key] } if key <= maxClassNameIndex { classifyName = varietyList[key] } if key <= maxFrequencyIndex { frequency = frequencyList[key] } failDataList = append(failDataList, &models.EdbdataImportFail{ //Id: 0, ClassifyName: classifyName, CreateDate: currDate, SecName: indexName, Close: v.Value, Remark: "日期格式异常", SysUserId: strconv.Itoa(sysUserId), Frequency: frequency, Unit: unit, }) continue } _, ok := indexDateValueMap[key] if !ok { indexDateValueMap[key] = make(map[string]string) } closeVal := v.Value //值 // 没有数据，说明是空串 if strings.Contains(closeVal, "#N/A") { indexDateValueMap[key][currDate] = "" continue } closeValFloat, tmpErr := v.Float() //值 if tmpErr != nil { indexDateValueMap[key][currDate] = "" continue } newDecimal := decimal.NewFromFloat(closeValFloat) newDecimal.Round(4) closeVal = newDecimal.String() if strings.Contains(closeVal, "#N/A") { indexDateValueMap[key][currDate] = "" continue } indexDateValueMap[key][currDate] = closeVal } } for i, indexName := range indexNameList { var unit, classifyName, frequency string if i <= maxUnitIndex { unit = unitList[i] } if i <= maxClassNameIndex { classifyName = varietyList[i] } if i <= maxFrequencyIndex { frequency = frequencyList[i] } indexData := ImportManualIndexData{ IndexName: indexName, Unit: unit, ClassName: classifyName, Frequency: frequency, DataMap: indexDateValueMap[i], } indexDataList = append(indexDataList, indexData) } return } // getExcelDate // @Description: 获取excel的日期 // @author: Roc // @datetime 2024-07-23 17:26:12 // @param createDate string // @return newCreateDate string // @return err error func getExcelDate(createDate string) (newCreateDate string, err error) { if strings.Contains(createDate, "-") { //如果是带有 - 的普通日期格式文本 _, err = time.Parse("2006-1-2", createDate) if err == nil { newCreateDate = createDate } } else if strings.Contains(createDate, "/") { //如果是带有 / 的普通日期格式文本 createDateTime, timeErr := time.Parse("2006/1/2", createDate) if timeErr != nil { err = timeErr } else { newCreateDate = createDateTime.Format("2006-01-02") } } else { //可能是excel的日期格式 _, tmpErr := strconv.Atoi(createDate) if tmpErr != nil { err = tmpErr } else { newCreateDate = utils.ConvertToFormatDay(createDate) //录入日期 } } return } // ModifyManualEdbMaxMinDate // @Description: 修改手动录入的edb数据的最大最小日期 // @author: Roc // @datetime 2024-08-01 15:34:31 // @param tradeCode string func ModifyManualEdbMaxMinDate(tradeCode string) { // 获取最大最小日期 item, err := models.GetEdbdataMaxMinDate(tradeCode) if err != nil { return } // 最新值 latestValue, err := models.GetEdbdataLatestValue(tradeCode) if err != nil { return } // 修改指标的最大最小日期和最新值 err = models.ModifyEdbinfoMaxMinDate(tradeCode, item.MinDate, item.MaxDate, latestValue) return } // GetUserManualClassifyIdList // @Description: 获取用户手动录入的分类id列表 // @author: Roc // @datetime 2024-08-02 15:09:11 // @param userId int // @return classifyIdList []int // @return err error func GetUserManualClassifyIdList(userId int) (classifyIdList []int, err error) { classifyIdList = make([]int, 0) // 获取有用权限的分类 classifyList, err := models.GetEdbdataClassify(int64(userId)) if err != nil { return } if len(classifyList) > 0 { for _, v := range classifyList { classifyIdList = append(classifyIdList, v.ClassifyId) if v.Child != nil && len(v.Child) > 0 { for _, v2 := range v.Child { classifyIdList = append(classifyIdList, v2.ClassifyId) } } } } return }