|
@@ -0,0 +1,1171 @@
|
|
|
+package data_manage
|
|
|
+
|
|
|
+import (
|
|
|
+ "encoding/json"
|
|
|
+ "eta/eta_api/models"
|
|
|
+ "eta/eta_api/models/data_manage"
|
|
|
+ "eta/eta_api/models/system"
|
|
|
+ "eta/eta_api/services/data"
|
|
|
+ etaTrialService "eta/eta_api/services/eta_trial"
|
|
|
+ "eta/eta_api/utils"
|
|
|
+ "fmt"
|
|
|
+ "os"
|
|
|
+ "path/filepath"
|
|
|
+ "strconv"
|
|
|
+ "strings"
|
|
|
+ "time"
|
|
|
+
|
|
|
+ "github.com/rdlucklib/rdluck_tools/paging"
|
|
|
+ "github.com/tealeg/xlsx"
|
|
|
+)
|
|
|
+
|
|
|
+// UsdaFasClassify
|
|
|
+// @Title 美国农业部数据分类
|
|
|
+// @Description 美国农业部数据分类接口
|
|
|
+// @Success 200 {object} data_manage.BaseFromUsdaFasClassify
|
|
|
+// @router /usda_fas/classify [get]
|
|
|
+func (this *EdbInfoController) UsdaFasClassify() {
|
|
|
+ br := new(models.BaseResponse).Init()
|
|
|
+ defer func() {
|
|
|
+ this.Data["json"] = br
|
|
|
+ this.ServeJSON()
|
|
|
+ }()
|
|
|
+ sysUser := this.SysUser
|
|
|
+ if sysUser == nil {
|
|
|
+ br.Msg = "请登录"
|
|
|
+ br.ErrMsg = "请登录,SysUser Is Empty"
|
|
|
+ br.Ret = 408
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+ classifyAll, err := data_manage.GetAllBaseFromUsdaFasClassify()
|
|
|
+ if err != nil && err.Error() != utils.ErrNoRow() {
|
|
|
+ br.Msg = "获取失败"
|
|
|
+ br.ErrMsg = "获取数据失败,Err:" + err.Error()
|
|
|
+ return
|
|
|
+ }
|
|
|
+ //组装一级分类
|
|
|
+ rootMap := make(map[int][]*data_manage.BaseFromUsdaFasClassifyItems)
|
|
|
+ list := make([]*data_manage.BaseFromUsdaFasClassifyItems, 0)
|
|
|
+ for _, classify := range classifyAll {
|
|
|
+ classify.UniqueCode = strconv.Itoa(classify.ClassifyId)
|
|
|
+ if classify.ParentId == 0 {
|
|
|
+ if _, ok := rootMap[classify.ClassifyId]; !ok {
|
|
|
+ rootMap[classify.ClassifyId] = make([]*data_manage.BaseFromUsdaFasClassifyItems, 0)
|
|
|
+ list = append(list, classify)
|
|
|
+
|
|
|
+ }
|
|
|
+ } else {
|
|
|
+ child, ok := rootMap[classify.ParentId]
|
|
|
+ if ok {
|
|
|
+ child = append(child, classify)
|
|
|
+ rootMap[classify.ParentId] = child
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ for k, v := range list {
|
|
|
+ child, ok := rootMap[v.ClassifyId]
|
|
|
+ if ok {
|
|
|
+ list[k].Children = child
|
|
|
+ }
|
|
|
+ }
|
|
|
+ //组装二级分类
|
|
|
+ var ret data_manage.BaseFromUsdaFasClassifyResp
|
|
|
+ ret.List = list
|
|
|
+ br.Ret = 200
|
|
|
+ br.Success = true
|
|
|
+ br.Msg = "获取成功"
|
|
|
+ br.Data = ret
|
|
|
+}
|
|
|
+
|
|
|
+// UsdaFasIndexData
|
|
|
+// @Title 获取美国农业部数据
|
|
|
+// @Description 获取美国农业部数据接口
|
|
|
+// @Param PageSize query int true "每页数据条数"
|
|
|
+// @Param CurrentIndex query int true "当前页页码,从1开始"
|
|
|
+// @Param ClassifyId query string true "分类id"
|
|
|
+// @Success 200 {object} data_manage.LzFrequency
|
|
|
+// @router /usda_fas/index/data [get]
|
|
|
+func (this *EdbInfoController) UsdaFasIndexData() {
|
|
|
+ br := new(models.BaseResponse).Init()
|
|
|
+ defer func() {
|
|
|
+ this.Data["json"] = br
|
|
|
+ this.ServeJSON()
|
|
|
+ }()
|
|
|
+ sysUser := this.SysUser
|
|
|
+ if sysUser == nil {
|
|
|
+ br.Msg = "请登录"
|
|
|
+ br.ErrMsg = "请登录,SysUser Is Empty"
|
|
|
+ br.Ret = 408
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+ pageSize, _ := this.GetInt("PageSize")
|
|
|
+ currentIndex, _ := this.GetInt("CurrentIndex")
|
|
|
+ var startSize int
|
|
|
+
|
|
|
+ if pageSize <= 0 {
|
|
|
+ pageSize = utils.PageSize20
|
|
|
+ }
|
|
|
+ if currentIndex <= 0 {
|
|
|
+ currentIndex = 1
|
|
|
+ }
|
|
|
+ startSize = utils.StartIndex(currentIndex, pageSize)
|
|
|
+
|
|
|
+ classifyId, _ := this.GetInt("ClassifyId")
|
|
|
+ if classifyId < 0 {
|
|
|
+ br.Msg = "请选择分类"
|
|
|
+ br.ErrMsg = "请选择分类"
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+ //获取指标
|
|
|
+ var condition string
|
|
|
+ var pars []interface{}
|
|
|
+
|
|
|
+ if classifyId >= 0 {
|
|
|
+ classifyInfo, err := data_manage.GetBaseFromUsdaFasClassifyById(classifyId)
|
|
|
+ if err != nil {
|
|
|
+ if err.Error() == utils.ErrNoRow() {
|
|
|
+ br.Msg = "分类不存在"
|
|
|
+ return
|
|
|
+ }
|
|
|
+ br.Msg = "获取分类信息失败"
|
|
|
+ br.ErrMsg = "获取分类信息失败,Err:" + err.Error()
|
|
|
+ return
|
|
|
+ }
|
|
|
+ if classifyInfo.Level == 2 || classifyInfo.ParentId > 0 {
|
|
|
+ condition += ` AND classify_id=? `
|
|
|
+ pars = append(pars, classifyId)
|
|
|
+ } else if classifyInfo.Level == 1 {
|
|
|
+ childClassify, err := data_manage.GetBaseFromUsdaFasClassifyByParentId(classifyId)
|
|
|
+ if err != nil {
|
|
|
+ br.Msg = "获取分类信息失败"
|
|
|
+ br.ErrMsg = "获取子分类信息失败,Err:" + err.Error()
|
|
|
+ return
|
|
|
+ }
|
|
|
+ var classifyList []int
|
|
|
+ for _, v := range childClassify {
|
|
|
+ classifyList = append(classifyList, v.ClassifyId)
|
|
|
+ }
|
|
|
+ condition += ` AND classify_id IN (` + utils.GetOrmInReplace(len(classifyList)) + `) `
|
|
|
+ pars = append(pars, classifyList)
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ UsdaFasList, err := data_manage.GetUsdaFasIndex(condition, pars)
|
|
|
+ if err != nil {
|
|
|
+ br.Msg = "获取数据失败"
|
|
|
+ br.ErrMsg = "获取数据失败,Err:" + err.Error()
|
|
|
+ return
|
|
|
+ }
|
|
|
+ edbCodeList := make([]string, 0)
|
|
|
+ for _, v := range UsdaFasList {
|
|
|
+ edbCodeList = append(edbCodeList, v.IndexCode)
|
|
|
+ }
|
|
|
+ edbInfoList, err := data_manage.GetEdbInfoByEdbCodeList(utils.DATA_SOURCE_USDA_FAS, edbCodeList)
|
|
|
+ if err != nil {
|
|
|
+ br.Msg = "获取数据源失败"
|
|
|
+ br.ErrMsg = "获取指标数据失败,Err:" + err.Error()
|
|
|
+ return
|
|
|
+ }
|
|
|
+ edbInfoMap := make(map[string]*data_manage.EdbInfo)
|
|
|
+ for _, v := range edbInfoList {
|
|
|
+ edbInfoMap[v.EdbCode] = v
|
|
|
+ }
|
|
|
+
|
|
|
+ resultList := make([]*data_manage.BaseFromUsdaFasIndexList, 0)
|
|
|
+ for _, v := range UsdaFasList {
|
|
|
+ product := new(data_manage.BaseFromUsdaFasIndexList)
|
|
|
+ product.BaseFromUsdaFasIndexId = v.BaseFromUsdaFasIndexId
|
|
|
+ product.Unit = v.Unit
|
|
|
+ product.IndexCode = v.IndexCode
|
|
|
+ product.IndexName = v.IndexName
|
|
|
+ product.Frequency = v.Frequency
|
|
|
+ product.ModifyTime = v.ModifyTime
|
|
|
+ if edb, ok := edbInfoMap[v.IndexCode]; ok {
|
|
|
+ product.EdbInfoId = edb.EdbInfoId
|
|
|
+ product.EdbExist = 1
|
|
|
+ }
|
|
|
+
|
|
|
+ total, err := data_manage.GetUsdaFasIndexDataCount(v.IndexCode)
|
|
|
+ if err != nil {
|
|
|
+ br.Msg = "获取数据失败"
|
|
|
+ br.ErrMsg = "获取指标数据失败,Err:" + err.Error()
|
|
|
+ }
|
|
|
+ page := paging.GetPaging(currentIndex, pageSize, total)
|
|
|
+ dataList, err := data_manage.GetUsdaFasIndexData(v.IndexCode, startSize, pageSize)
|
|
|
+ if err != nil {
|
|
|
+ br.Msg = "获取数据失败"
|
|
|
+ br.ErrMsg = "获取指标数据失败,Err:" + err.Error()
|
|
|
+ return
|
|
|
+ }
|
|
|
+ if dataList == nil {
|
|
|
+ dataList = make([]*data_manage.BaseFromUsdaFasData, 0)
|
|
|
+ }
|
|
|
+ product.DataList = dataList
|
|
|
+ product.Paging = page
|
|
|
+ resultList = append(resultList, product)
|
|
|
+ }
|
|
|
+ br.Ret = 200
|
|
|
+ br.Success = true
|
|
|
+ br.Msg = "获取成功"
|
|
|
+ br.Data = resultList
|
|
|
+}
|
|
|
+
|
|
|
+// UsdaFasSearchList
|
|
|
+// @Title UsdaFas模糊搜索
|
|
|
+// @Description UsdaFas模糊搜索
|
|
|
+// @Param Keyword query string ture "关键字搜索"
|
|
|
+// @Success 200 {object} models.BaseResponse
|
|
|
+// @router /usda_fas/search_list [get]
|
|
|
+func (this *EdbInfoController) UsdaFasSearchList() {
|
|
|
+ br := new(models.BaseResponse).Init()
|
|
|
+ defer func() {
|
|
|
+ this.Data["json"] = br
|
|
|
+ this.ServeJSON()
|
|
|
+ }()
|
|
|
+
|
|
|
+ sysUser := this.SysUser
|
|
|
+ if sysUser == nil {
|
|
|
+ br.Msg = "请重新登录"
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+ list := make([]*data_manage.BaseFromUsdaFasIndexSearchItem, 0)
|
|
|
+ var err error
|
|
|
+ //关键字
|
|
|
+ keyword := this.GetString("Keyword")
|
|
|
+ if keyword != "" {
|
|
|
+ keyWordArr := strings.Split(keyword, " ")
|
|
|
+
|
|
|
+ if len(keyWordArr) > 0 {
|
|
|
+ condition := ""
|
|
|
+ for _, v := range keyWordArr {
|
|
|
+ condition += ` AND CONCAT(index_name,index_code) LIKE '%` + v + `%'`
|
|
|
+ }
|
|
|
+ list, err = data_manage.GetUsdaFasItemList(condition)
|
|
|
+ if err != nil {
|
|
|
+ br.ErrMsg = "获取失败,Err:" + err.Error()
|
|
|
+ br.Msg = "获取失败"
|
|
|
+ return
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ } else {
|
|
|
+ // todo es 模糊搜索
|
|
|
+ list, err = data_manage.GetUsdaFasItemList("")
|
|
|
+ if err != nil {
|
|
|
+ br.ErrMsg = "获取失败,Err:" + err.Error()
|
|
|
+ br.Msg = "获取失败"
|
|
|
+ return
|
|
|
+ }
|
|
|
+ }
|
|
|
+ classifyIds := make([]int, 0)
|
|
|
+ for _, v := range list {
|
|
|
+ classifyIds = append(classifyIds, v.ClassifyId)
|
|
|
+ }
|
|
|
+ classifyList, err := data_manage.GetBaseFromUsdaFasClassifyByIds(classifyIds)
|
|
|
+ if err != nil {
|
|
|
+ br.Msg = "搜索失败"
|
|
|
+ br.ErrMsg = "获取分类信息失败,Err:" + err.Error()
|
|
|
+ return
|
|
|
+ }
|
|
|
+ classifyMap := make(map[int]int)
|
|
|
+ for _, v := range classifyList {
|
|
|
+ classifyMap[v.ClassifyId] = v.ParentId
|
|
|
+ }
|
|
|
+ for _, v := range list {
|
|
|
+ v.ParentClassifyId = classifyMap[v.ClassifyId]
|
|
|
+ }
|
|
|
+
|
|
|
+ br.Ret = 200
|
|
|
+ br.Success = true
|
|
|
+ br.Msg = "获取成功"
|
|
|
+ br.Data = list
|
|
|
+}
|
|
|
+
|
|
|
+// UsdaFasSingleData
|
|
|
+// @Title 获取UsdaFas数据
|
|
|
+// @Description 获取UsdaFas单条数据接口
|
|
|
+// @Param IndexCode query string true "指标唯一编码"
|
|
|
+// @Success 200 {object} models.BaseResponse
|
|
|
+// @router /usda_fas/single_data [get]
|
|
|
+func (this *EdbInfoController) UsdaFasSingleData() {
|
|
|
+ br := new(models.BaseResponse).Init()
|
|
|
+ defer func() {
|
|
|
+ this.Data["json"] = br
|
|
|
+ this.ServeJSON()
|
|
|
+ }()
|
|
|
+ sysUser := this.SysUser
|
|
|
+ if sysUser == nil {
|
|
|
+ br.Msg = "请登录"
|
|
|
+ br.ErrMsg = "请登录,SysUser Is Empty"
|
|
|
+ br.Ret = 408
|
|
|
+ return
|
|
|
+ }
|
|
|
+ indexCode := this.GetString("IndexCode")
|
|
|
+ indexInfo, err := data_manage.GetBaseFromUsdaFasIndexByIndexCode(indexCode)
|
|
|
+ if err != nil {
|
|
|
+ br.Msg = "获取指标信息失败"
|
|
|
+ br.ErrMsg = "获取指标信息失败,Err:" + err.Error()
|
|
|
+ return
|
|
|
+ }
|
|
|
+ dataTmpList, err := data_manage.GetUsdaFasIndexDataByCode(indexCode)
|
|
|
+ if err != nil {
|
|
|
+ br.Msg = "获取数据失败"
|
|
|
+ br.ErrMsg = "获取数据失败,Err:" + err.Error()
|
|
|
+ return
|
|
|
+ }
|
|
|
+ edbInfo, err := data_manage.GetEdbInfoByEdbCode(utils.DATA_SOURCE_USDA_FAS, indexCode)
|
|
|
+ if err != nil && err.Error() != utils.ErrNoRow() {
|
|
|
+ br.Msg = "获取数据源失败"
|
|
|
+ br.ErrMsg = "获取数据源失败,Err:" + err.Error()
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+ var ret data_manage.UsdaFasSingleDataResp
|
|
|
+ var dataList []*data_manage.UsdaFasSingleData
|
|
|
+
|
|
|
+ if edbInfo != nil {
|
|
|
+ ret.EdbInfoId = edbInfo.EdbInfoId
|
|
|
+ ret.EdbExist = 1
|
|
|
+ }
|
|
|
+ ret.ClassifyId = indexInfo.ClassifyId
|
|
|
+ ret.BaseFromUsdaFasIndexId = indexInfo.BaseFromUsdaFasIndexId
|
|
|
+ ret.IndexCode = indexInfo.IndexCode
|
|
|
+ ret.IndexName = indexInfo.IndexName
|
|
|
+ ret.Frequency = indexInfo.Frequency
|
|
|
+ ret.CreateTime = indexInfo.CreateTime.Format(utils.FormatDateTime)
|
|
|
+ ret.ModifyTime = indexInfo.ModifyTime.Format(utils.FormatDateTime)
|
|
|
+ ret.Unit = indexInfo.Unit
|
|
|
+ for _, v := range dataTmpList {
|
|
|
+ tmp := &data_manage.UsdaFasSingleData{
|
|
|
+ Value: v.Value,
|
|
|
+ DataTime: v.DataTime,
|
|
|
+ }
|
|
|
+ dataList = append(dataList, tmp)
|
|
|
+ }
|
|
|
+ ret.Data = dataList
|
|
|
+
|
|
|
+ br.Ret = 200
|
|
|
+ br.Success = true
|
|
|
+ br.Msg = "获取成功"
|
|
|
+ br.Data = ret
|
|
|
+}
|
|
|
+
|
|
|
+// UsdaFasIndexList
|
|
|
+// @Title 美国农业部指标列表
|
|
|
+// @Description 美国农业部指标列表
|
|
|
+// @Param ClassifyId query int true "分类id"
|
|
|
+// @Success 200 {object} data_manage.BaseFromMysteelChemicalIndexResp
|
|
|
+// @router /usda_fas/classify/index/list [get]
|
|
|
+func (this *EdbInfoController) UsdaFasIndexList() {
|
|
|
+ br := new(models.BaseResponse).Init()
|
|
|
+ defer func() {
|
|
|
+ this.Data["json"] = br
|
|
|
+ this.ServeJSON()
|
|
|
+ }()
|
|
|
+ classifyId, _ := this.GetInt("ClassifyId", 0)
|
|
|
+ indexList, err := data_manage.GetUsdaFasIndexByClassifyId(classifyId)
|
|
|
+ if err != nil && err.Error() != utils.ErrNoRow() {
|
|
|
+ br.Msg = "获取失败"
|
|
|
+ br.ErrMsg = "获取指标信息失败,Err:" + err.Error()
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+ var ret data_manage.BaseFromUsdaFasClassifyResp
|
|
|
+ list := make([]*data_manage.BaseFromUsdaFasClassifyItems, 0)
|
|
|
+ for _, v := range indexList {
|
|
|
+ classify := new(data_manage.BaseFromUsdaFasClassifyItems)
|
|
|
+ classify.ClassifyId = classifyId
|
|
|
+ classify.BaseFromUsdaFasIndexId = v.BaseFromUsdaFasIndexId
|
|
|
+ classify.IndexCode = v.IndexCode
|
|
|
+ classify.ClassifyName = v.IndexName
|
|
|
+ classify.UniqueCode = fmt.Sprintf("%d_%d", classifyId, v.BaseFromUsdaFasIndexId)
|
|
|
+ list = append(list, classify)
|
|
|
+ }
|
|
|
+ ret.List = list
|
|
|
+
|
|
|
+ br.Ret = 200
|
|
|
+ br.Success = true
|
|
|
+ br.Msg = "获取成功"
|
|
|
+ br.Data = ret
|
|
|
+}
|
|
|
+
|
|
|
+// UsdaFasBatchSearch
|
|
|
+// @Title 美国农业部指标查询
|
|
|
+// @Description 美国农业部指标查询
|
|
|
+// @Param ClassifyIds query string true "分类id, 多个分类用英文"
|
|
|
+// @Param Keyword query string true "关键词, 指标ID/指标名称"
|
|
|
+// @Success 200 {object} data_manage.LzFrequency
|
|
|
+// @router /usda_fas/batch_search [get]
|
|
|
+func (this *EdbInfoController) UsdaFasBatchSearch() {
|
|
|
+ br := new(models.BaseResponse).Init()
|
|
|
+ defer func() {
|
|
|
+ this.Data["json"] = br
|
|
|
+ this.ServeJSON()
|
|
|
+ }()
|
|
|
+ sysUser := this.SysUser
|
|
|
+ if sysUser == nil {
|
|
|
+ br.Msg = "请登录"
|
|
|
+ br.ErrMsg = "请登录,SysUser Is Empty"
|
|
|
+ br.Ret = 408
|
|
|
+ return
|
|
|
+ }
|
|
|
+ classifyIdStr := this.GetString("ClassifyIds")
|
|
|
+
|
|
|
+ var condition string
|
|
|
+ var pars []interface{}
|
|
|
+ classifyIds := strings.Split(classifyIdStr, ",")
|
|
|
+ if len(classifyIds) > 0 && classifyIds[0] != `` {
|
|
|
+ condition += " AND classify_id IN (" + utils.GetOrmInReplace(len(classifyIds)) + " ) "
|
|
|
+ pars = append(pars, classifyIds)
|
|
|
+ }
|
|
|
+ keyword := this.GetString("Keyword")
|
|
|
+ if keyword != `` {
|
|
|
+ condition += " AND (index_name like ? OR index_code like ?) "
|
|
|
+ pars = utils.GetLikeKeywordPars(pars, keyword, 2)
|
|
|
+ }
|
|
|
+
|
|
|
+ if classifyIdStr == `` && keyword == `` {
|
|
|
+ var list = make([]*data_manage.BaseFromUsdaFasIndexList, 0)
|
|
|
+ br.Ret = 200
|
|
|
+ br.Success = true
|
|
|
+ br.Msg = "获取成功"
|
|
|
+ br.Data = list
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+ list, err := data_manage.GetUsdaFasIndex(condition, pars)
|
|
|
+ if err != nil {
|
|
|
+ br.Msg = "获取失败"
|
|
|
+ br.ErrMsg = "获取失败,Err:" + err.Error()
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+ br.Ret = 200
|
|
|
+ br.Success = true
|
|
|
+ br.Msg = "获取成功"
|
|
|
+ br.Data = list
|
|
|
+}
|
|
|
+
|
|
|
+// UsdaFasBatchAdd
|
|
|
+// @Title 美国农业部批量新增
|
|
|
+// @Description 美国农业部批量新增
|
|
|
+// @Param request body data_manage.AddEdbInfoReq true "type json string"
|
|
|
+// @Success 200 string "操作成功"
|
|
|
+// @router /usda_fas/batch_add [post]
|
|
|
+func (this *EdbInfoController) UsdaFasBatchAdd() {
|
|
|
+ /*br := new(models.BaseResponse).Init()
|
|
|
+ defer func() {
|
|
|
+ if br.ErrMsg == "" {
|
|
|
+ br.IsSendEmail = false
|
|
|
+ }
|
|
|
+ this.Data["json"] = br
|
|
|
+ this.ServeJSON()
|
|
|
+ }()
|
|
|
+ sysUser := this.SysUser
|
|
|
+ if sysUser == nil {
|
|
|
+ br.Msg = "请登录"
|
|
|
+ br.ErrMsg = "请登录,SysUser Is Empty"
|
|
|
+ br.Ret = 408
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+ deleteCache := true
|
|
|
+ cacheKey := "CACHE_EDB_INFO_BATCH_ADD_UsdaFas_" + strconv.Itoa(sysUser.AdminId)
|
|
|
+ defer func() {
|
|
|
+ if deleteCache {
|
|
|
+ _ = utils.Rc.Delete(cacheKey)
|
|
|
+ }
|
|
|
+ }()
|
|
|
+ if !utils.Rc.SetNX(cacheKey, 1, 30*time.Second) {
|
|
|
+ deleteCache = false
|
|
|
+ br.Msg = "系统处理中,请稍后重试!"
|
|
|
+ br.ErrMsg = "系统处理中,请稍后重试!" + sysUser.RealName + ";data:" + string(this.Ctx.Input.RequestBody)
|
|
|
+ return
|
|
|
+ }
|
|
|
+ var req []*data_manage.AddEdbInfoReq
|
|
|
+ if e := json.Unmarshal(this.Ctx.Input.RequestBody, &req); e != nil {
|
|
|
+ br.Msg = "参数解析异常!"
|
|
|
+ br.ErrMsg = "参数解析失败,Err:" + e.Error()
|
|
|
+ return
|
|
|
+ }
|
|
|
+ if len(req) == 0 {
|
|
|
+ br.Msg = "请选择指标"
|
|
|
+ return
|
|
|
+ }
|
|
|
+ if len(req) > 30 {
|
|
|
+ br.Msg = "批量添加指标数量不得超过30个"
|
|
|
+ return
|
|
|
+ }
|
|
|
+ for _, v := range req {
|
|
|
+ v.EdbCode = strings.TrimSpace(v.EdbCode)
|
|
|
+ if v.EdbCode == "" {
|
|
|
+ br.Msg = "指标ID不可为空"
|
|
|
+ return
|
|
|
+ }
|
|
|
+ v.EdbName = strings.TrimSpace(v.EdbName)
|
|
|
+ if v.EdbName == "" {
|
|
|
+ br.Msg = "请输入指标名称"
|
|
|
+ return
|
|
|
+ }
|
|
|
+ v.Frequency = strings.TrimSpace(v.Frequency)
|
|
|
+ if v.Frequency == "" {
|
|
|
+ br.Msg = "请选择频度"
|
|
|
+ return
|
|
|
+ }
|
|
|
+ v.Unit = strings.TrimSpace(v.Unit)
|
|
|
+ if v.Unit == "" {
|
|
|
+ br.Msg = "请输入单位"
|
|
|
+ return
|
|
|
+ }
|
|
|
+ if v.ClassifyId <= 0 {
|
|
|
+ br.Msg = "请选择分类"
|
|
|
+ return
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ // 限定同一时间最多批量新增30个指标
|
|
|
+ for _, v := range req {
|
|
|
+ var r data_manage.UsdaFasIndexSource2EdbReq
|
|
|
+ r.EdbCode = v.EdbCode
|
|
|
+ r.EdbName = v.EdbName
|
|
|
+ r.Frequency = v.Frequency
|
|
|
+ r.Unit = v.Unit
|
|
|
+ r.ClassifyId = v.ClassifyId
|
|
|
+ r.AdminId = sysUser.AdminId
|
|
|
+ r.AdminRealName = sysUser.RealName
|
|
|
+
|
|
|
+ edbInfo, e, errMsg, skip := data.UsdaFasIndexSource2Edb(r, this.Lang)
|
|
|
+ if e != nil {
|
|
|
+ br.Msg = "操作失败"
|
|
|
+ if errMsg != "" {
|
|
|
+ br.Msg = errMsg
|
|
|
+ }
|
|
|
+ br.ErrMsg = e.Error()
|
|
|
+ return
|
|
|
+ }
|
|
|
+ if skip {
|
|
|
+ continue
|
|
|
+ }
|
|
|
+
|
|
|
+ // 试用平台更新用户累计新增指标数
|
|
|
+ if utils.BusinessCode == utils.BusinessCodeSandbox {
|
|
|
+ go func() {
|
|
|
+ adminItem, e := system.GetSysAdminById(sysUser.AdminId)
|
|
|
+ if e != nil {
|
|
|
+ tips := fmt.Sprintf("试用平台更新用户累计新增指标数-获取用户失败, Err: " + e.Error())
|
|
|
+ utils.FileLog.Info(tips)
|
|
|
+ return
|
|
|
+ }
|
|
|
+ if adminItem.DepartmentName != "ETA试用客户" {
|
|
|
+ return
|
|
|
+ }
|
|
|
+ var ur etaTrialService.EtaTrialUserReq
|
|
|
+ ur.Mobile = adminItem.Mobile
|
|
|
+ _, _ = etaTrialService.UpdateUserIndexNum(ur)
|
|
|
+ }()
|
|
|
+ }
|
|
|
+
|
|
|
+ // 新增操作日志
|
|
|
+ {
|
|
|
+ edbLog := new(data_manage.EdbInfoLog)
|
|
|
+ edbLog.EdbInfoId = edbInfo.EdbInfoId
|
|
|
+ edbLog.SourceName = edbInfo.SourceName
|
|
|
+ edbLog.Source = edbInfo.Source
|
|
|
+ edbLog.EdbCode = edbInfo.EdbCode
|
|
|
+ edbLog.EdbName = edbInfo.EdbName
|
|
|
+ edbLog.ClassifyId = edbInfo.ClassifyId
|
|
|
+ edbLog.SysUserId = sysUser.AdminId
|
|
|
+ edbLog.SysUserRealName = sysUser.RealName
|
|
|
+ edbLog.CreateTime = time.Now()
|
|
|
+ edbLog.Content = string(this.Ctx.Input.RequestBody)
|
|
|
+ edbLog.Status = "新增指标"
|
|
|
+ edbLog.Method = this.Ctx.Input.URI()
|
|
|
+ go data_manage.AddEdbInfoLog(edbLog)
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ br.Msg = "操作成功"
|
|
|
+ br.Ret = 200
|
|
|
+ br.Success = true
|
|
|
+ br.IsAddLog = true*/
|
|
|
+}
|
|
|
+
|
|
|
+// UsdaFasNameCheck
|
|
|
+// @Title 加入指标库的重名检测
|
|
|
+// @Description 加入指标库的重名检测
|
|
|
+// @Param ClassifyIds query string true "分类id, 多个分类用英文"
|
|
|
+// @Param Keyword query string true "关键词, 指标ID/指标名称"
|
|
|
+// @Success 200 {object} NameCheckResult
|
|
|
+// @router /usda_fas/edb_info/name_check [post]
|
|
|
+func (this *EdbInfoController) UsdaFasNameCheck() {
|
|
|
+ br := new(models.BaseResponse).Init()
|
|
|
+ defer func() {
|
|
|
+ if br.ErrMsg == "" {
|
|
|
+ br.IsSendEmail = false
|
|
|
+ }
|
|
|
+ this.Data["json"] = br
|
|
|
+ this.ServeJSON()
|
|
|
+ }()
|
|
|
+ sysUser := this.SysUser
|
|
|
+ if sysUser == nil {
|
|
|
+ br.Msg = "请登录"
|
|
|
+ br.ErrMsg = "请登录,SysUser Is Empty"
|
|
|
+ br.Ret = 408
|
|
|
+ return
|
|
|
+ }
|
|
|
+ var req []*data_manage.NameCheckEdbInfoReq
|
|
|
+ if e := json.Unmarshal(this.Ctx.Input.RequestBody, &req); e != nil {
|
|
|
+ br.Msg = "参数解析异常!"
|
|
|
+ br.ErrMsg = "参数解析失败,Err:" + e.Error()
|
|
|
+ return
|
|
|
+ }
|
|
|
+ if len(req) == 0 {
|
|
|
+ br.Msg = "请选择指标"
|
|
|
+ return
|
|
|
+ }
|
|
|
+ codeMax := 30
|
|
|
+ codeLen := len(req)
|
|
|
+ if codeLen > codeMax {
|
|
|
+ br.Msg = fmt.Sprintf("最多只能选择%d个指标", codeMax)
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+ type NameCheckResult struct {
|
|
|
+ EdbCode string
|
|
|
+ EdbName string
|
|
|
+ Exist bool
|
|
|
+ }
|
|
|
+ indexNames := make([]string, 0)
|
|
|
+ resp := make([]*NameCheckResult, 0)
|
|
|
+ for _, v := range req {
|
|
|
+ v.EdbCode = strings.TrimSpace(v.EdbCode)
|
|
|
+ if v.EdbCode == "" {
|
|
|
+ br.Msg = "指标ID不可为空"
|
|
|
+ return
|
|
|
+ }
|
|
|
+ v.EdbName = strings.TrimSpace(v.EdbName)
|
|
|
+ if v.EdbName == "" {
|
|
|
+ br.Msg = "请输入指标名称"
|
|
|
+ return
|
|
|
+ }
|
|
|
+ indexNames = append(indexNames, v.EdbName)
|
|
|
+ resp = append(resp, &NameCheckResult{
|
|
|
+ EdbCode: v.EdbCode,
|
|
|
+ EdbName: v.EdbName,
|
|
|
+ })
|
|
|
+ dataItems, err := data_manage.GetEdbDataAllByEdbCode(v.EdbCode, utils.DATA_SOURCE_USDA_FAS, 0, utils.EDB_DATA_LIMIT)
|
|
|
+ if err != nil && err.Error() != utils.ErrNoRow() {
|
|
|
+ br.Msg = "获取失败"
|
|
|
+ br.ErrMsg = "获取钢联已存在信息失败,Err:" + err.Error()
|
|
|
+ return
|
|
|
+ }
|
|
|
+ if len(dataItems) <= 0 {
|
|
|
+ respItem, err := data.AddEdbData(utils.DATA_SOURCE_USDA_FAS, v.EdbCode, v.Frequency)
|
|
|
+ if err != nil {
|
|
|
+ br.Msg = "获取失败"
|
|
|
+ br.ErrMsg = "获取失败,Err:" + err.Error()
|
|
|
+ return
|
|
|
+ }
|
|
|
+ if respItem.Ret != 200 {
|
|
|
+ br.Msg = "未搜索到该指标"
|
|
|
+ br.ErrMsg = respItem.ErrMsg + ";EdbCode:" + v.EdbCode
|
|
|
+ return
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ // 重名校验
|
|
|
+ edbList, e := data_manage.GetEdbInfoByNameArr(indexNames, utils.EDB_INFO_TYPE)
|
|
|
+ if e != nil {
|
|
|
+ br.Msg = "操作失败"
|
|
|
+ br.ErrMsg = "获取重名指标失败, Err: " + e.Error()
|
|
|
+ return
|
|
|
+ }
|
|
|
+ nameExists := make(map[string]bool)
|
|
|
+ for _, v := range edbList {
|
|
|
+ nameExists[v.EdbName] = true
|
|
|
+ }
|
|
|
+ if len(nameExists) > 0 {
|
|
|
+ for _, v := range resp {
|
|
|
+ v.Exist = nameExists[v.EdbName]
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ br.Data = resp
|
|
|
+ br.Msg = "校验成功"
|
|
|
+ br.Ret = 200
|
|
|
+ br.Success = true
|
|
|
+}
|
|
|
+
|
|
|
+// UsdaFasAddCheck
|
|
|
+// @Title 加入指标库指标Id检测
|
|
|
+// @Description 加入指标库指标Id检测
|
|
|
+// @Param request body request.BatchAddCheckReq true "type json string"
|
|
|
+// @Success 200 string "操作成功"
|
|
|
+// @router /usda_fas/edb_info/add_check [post]
|
|
|
+func (c *EdbInfoController) UsdaFasAddCheck() {
|
|
|
+ /*br := new(models.BaseResponse).Init()
|
|
|
+ defer func() {
|
|
|
+ if br.ErrMsg == "" {
|
|
|
+ br.IsSendEmail = false
|
|
|
+ }
|
|
|
+ c.Data["json"] = br
|
|
|
+ c.ServeJSON()
|
|
|
+ }()
|
|
|
+ sysUser := c.SysUser
|
|
|
+ if sysUser == nil {
|
|
|
+ br.Msg = "请登录"
|
|
|
+ br.ErrMsg = "请登录,SysUser Is Empty"
|
|
|
+ br.Ret = 408
|
|
|
+ return
|
|
|
+ }
|
|
|
+ var req data_manage.BatchAddCheckReq
|
|
|
+ if e := json.Unmarshal(c.Ctx.Input.RequestBody, &req); e != nil {
|
|
|
+ br.Msg = "参数解析异常!"
|
|
|
+ br.ErrMsg = "参数解析失败,Err:" + e.Error()
|
|
|
+ return
|
|
|
+ }
|
|
|
+ codeMax := 30
|
|
|
+ codeLen := len(req.IndexCodes)
|
|
|
+
|
|
|
+ // 获取指标库已有指标
|
|
|
+ existsEdb, e := data_manage.GetEdbCodesBySource(utils.DATA_SOURCE_USDA_FAS)
|
|
|
+ if e != nil {
|
|
|
+ br.Msg = "获取失败"
|
|
|
+ br.ErrMsg = "获取自有数据已添加的指标失败, Err: " + e.Error()
|
|
|
+ return
|
|
|
+ }
|
|
|
+ existMap := make(map[string]*data_manage.EdbInfo)
|
|
|
+ for _, v := range existsEdb {
|
|
|
+ existMap[v.EdbCode] = v
|
|
|
+ }
|
|
|
+
|
|
|
+ if codeLen == 0 {
|
|
|
+ br.Msg = "请选择指标"
|
|
|
+ return
|
|
|
+ }
|
|
|
+ if codeLen > codeMax {
|
|
|
+ br.Msg = fmt.Sprintf("最多只能选择%d个指标", codeMax)
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+ // 查询选中的指标
|
|
|
+ cond := fmt.Sprintf(` AND index_code IN (%s)`, utils.GetOrmInReplace(codeLen))
|
|
|
+ pars := make([]interface{}, 0)
|
|
|
+ pars = append(pars, req.IndexCodes)
|
|
|
+ list, err := data_manage.GetUsdaFasIndex(cond, pars)
|
|
|
+ if err != nil {
|
|
|
+ br.Msg = "获取失败"
|
|
|
+ br.ErrMsg = "获取钢联已存在信息失败,Err:" + err.Error()
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+ if len(list) > codeMax {
|
|
|
+ br.Msg = fmt.Sprintf("最多只能选择%d个指标", codeMax)
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+ resp := make([]*data_manage.BaseFromUsdaFasIndexList, 0)
|
|
|
+ for _, v := range list {
|
|
|
+ if edb, ok := existMap[v.IndexCode]; ok {
|
|
|
+ v.EdbInfoId = edb.EdbInfoId
|
|
|
+ v.EdbClassifyId = edb.ClassifyId
|
|
|
+ v.EdbUniqueCode = edb.UniqueCode
|
|
|
+ v.EdbExist = 1
|
|
|
+ }
|
|
|
+ resp = append(resp, v)
|
|
|
+ }
|
|
|
+
|
|
|
+ br.Data = resp
|
|
|
+ br.Msg = "校验成功"
|
|
|
+ br.Ret = 200
|
|
|
+ br.Success = true*/
|
|
|
+}
|
|
|
+
|
|
|
+// UsdaFasEdbInfoAdd
|
|
|
+// @Title 新增指标接口
|
|
|
+// @Description 新增指标接口
|
|
|
+// @Param request body data_manage.AddEdbInfoReq true "type json string"
|
|
|
+// @Success Ret=200 保存成功
|
|
|
+// @router /usda_fas/edb_info/add [post]
|
|
|
+func (this *EdbInfoController) UsdaFasEdbInfoAdd() {
|
|
|
+ br := new(models.BaseResponse).Init()
|
|
|
+ defer func() {
|
|
|
+ this.Data["json"] = br
|
|
|
+ this.ServeJSON()
|
|
|
+ }()
|
|
|
+
|
|
|
+ sysUser := this.SysUser
|
|
|
+ if sysUser == nil {
|
|
|
+ br.Msg = "请登录"
|
|
|
+ br.ErrMsg = "请登录,SysUser Is Empty"
|
|
|
+ br.Ret = 408
|
|
|
+ return
|
|
|
+ }
|
|
|
+ deleteCache := true
|
|
|
+ cacheKey := "CACHE_EDB_INFO_ADD_" + strconv.Itoa(sysUser.AdminId)
|
|
|
+ defer func() {
|
|
|
+ if deleteCache {
|
|
|
+ utils.Rc.Delete(cacheKey)
|
|
|
+ }
|
|
|
+ }()
|
|
|
+ if !utils.Rc.SetNX(cacheKey, 1, 30*time.Second) {
|
|
|
+ deleteCache = false
|
|
|
+ br.Msg = "系统处理中,请稍后重试!"
|
|
|
+ br.ErrMsg = "系统处理中,请稍后重试!" + sysUser.RealName + ";data:" + string(this.Ctx.Input.RequestBody)
|
|
|
+ return
|
|
|
+ }
|
|
|
+ var req data_manage.AddEdbInfoReq
|
|
|
+ err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
|
|
|
+ if err != nil {
|
|
|
+ br.Msg = "参数解析异常!"
|
|
|
+ br.ErrMsg = "参数解析失败,Err:" + err.Error()
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+ req.EdbName = strings.Trim(req.EdbName, " ")
|
|
|
+ req.EdbCode = strings.Trim(req.EdbCode, " ")
|
|
|
+
|
|
|
+ if req.EdbCode == "" {
|
|
|
+ br.Msg = "指标ID不能为空"
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+ if req.EdbName == "" {
|
|
|
+ br.Msg = "指标名称不能为空"
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+ if req.Frequency == "" {
|
|
|
+ br.Msg = "频率不能为空"
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+ if req.Unit == "" {
|
|
|
+ br.Msg = "单位不能为空"
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+ if req.ClassifyId <= 0 {
|
|
|
+ br.Msg = "请选择分类"
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+ count, err := data_manage.GetUsdaFasIndexDataCount(req.EdbCode)
|
|
|
+ if err != nil {
|
|
|
+ br.Msg = "获取失败"
|
|
|
+ br.ErrMsg = "获取失败,Err:" + err.Error()
|
|
|
+ return
|
|
|
+ }
|
|
|
+ if count == 0 {
|
|
|
+ br.Msg = "指标不存在"
|
|
|
+ }
|
|
|
+
|
|
|
+ // 指标入库
|
|
|
+ edbInfo, err, errMsg, isSendEmail := data.EdbInfoAdd(utils.DATA_SOURCE_USDA_FAS, utils.DATA_SUB_SOURCE_EDB, req.ClassifyId, req.EdbCode, req.EdbName, req.Frequency, req.Unit, req.StartDate, req.EndDate, sysUser.AdminId, sysUser.RealName, this.Lang)
|
|
|
+ if err != nil {
|
|
|
+ br.Msg = "保存失败"
|
|
|
+ if errMsg != `` {
|
|
|
+ br.Msg = errMsg
|
|
|
+ }
|
|
|
+ br.ErrMsg = err.Error()
|
|
|
+ br.IsSendEmail = isSendEmail
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+ // 试用平台更新用户累计新增指标数
|
|
|
+ adminItem, e := system.GetSysAdminById(sysUser.AdminId)
|
|
|
+ if e != nil {
|
|
|
+ br.Msg = "操作失败"
|
|
|
+ br.ErrMsg = "获取系统用户数据失败,Err:" + e.Error()
|
|
|
+ return
|
|
|
+ }
|
|
|
+ if utils.BusinessCode == utils.BusinessCodeSandbox && adminItem.DepartmentName == "ETA试用客户" {
|
|
|
+ go func() {
|
|
|
+ var r etaTrialService.EtaTrialUserReq
|
|
|
+ r.Mobile = adminItem.Mobile
|
|
|
+ _, _ = etaTrialService.UpdateUserIndexNum(r)
|
|
|
+ }()
|
|
|
+ }
|
|
|
+
|
|
|
+ //新增操作日志
|
|
|
+ {
|
|
|
+ edbLog := new(data_manage.EdbInfoLog)
|
|
|
+ edbLog.EdbInfoId = edbInfo.EdbInfoId
|
|
|
+ edbLog.SourceName = edbInfo.SourceName
|
|
|
+ edbLog.Source = edbInfo.Source
|
|
|
+ edbLog.EdbCode = edbInfo.EdbCode
|
|
|
+ edbLog.EdbName = edbInfo.EdbName
|
|
|
+ edbLog.ClassifyId = edbInfo.ClassifyId
|
|
|
+ edbLog.SysUserId = sysUser.AdminId
|
|
|
+ edbLog.SysUserRealName = sysUser.RealName
|
|
|
+ edbLog.CreateTime = time.Now()
|
|
|
+ edbLog.Content = string(this.Ctx.Input.RequestBody)
|
|
|
+ edbLog.Status = "新增指标"
|
|
|
+ edbLog.Method = this.Ctx.Input.URI()
|
|
|
+ go data_manage.AddEdbInfoLog(edbLog)
|
|
|
+ }
|
|
|
+
|
|
|
+ // 更新es
|
|
|
+ go data.AddOrEditEdbInfoToEs(edbInfo.EdbInfoId)
|
|
|
+
|
|
|
+ resp := new(data_manage.AddEdbInfoResp)
|
|
|
+ resp.EdbInfoId = edbInfo.EdbInfoId
|
|
|
+ resp.UniqueCode = edbInfo.UniqueCode
|
|
|
+ br.Ret = 200
|
|
|
+ br.Success = true
|
|
|
+ br.Msg = "保存成功"
|
|
|
+ br.Data = resp
|
|
|
+ br.IsAddLog = true
|
|
|
+}
|
|
|
+
|
|
|
+// ExportUsdaFasList
|
|
|
+// @Title 导出美国农业部数据
|
|
|
+// @Description 导出美国农业部数据
|
|
|
+// @Param ClassifyId query int true "关键字搜索"
|
|
|
+// @Param IndexCode query string true "指标编码"
|
|
|
+// @Success 200 导出成功
|
|
|
+// @router /usda_fas/export [get]
|
|
|
+func (this *EdbInfoController) ExportUsdaFasList() {
|
|
|
+ br := new(models.BaseResponse).Init()
|
|
|
+ defer func() {
|
|
|
+ this.Data["json"] = br
|
|
|
+ this.ServeJSON()
|
|
|
+ }()
|
|
|
+
|
|
|
+ sysUser := this.SysUser
|
|
|
+ if sysUser == nil {
|
|
|
+ br.Msg = "请重新登录"
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+ classifyId, _ := this.GetInt("ClassifyId")
|
|
|
+ indexCode := this.GetString("IndexCode")
|
|
|
+ if classifyId < 0 {
|
|
|
+ br.Msg = "请选择分类"
|
|
|
+ return
|
|
|
+ }
|
|
|
+ dir, _ := os.Executable()
|
|
|
+ exPath := filepath.Dir(dir)
|
|
|
+ downLoadnFilePath := exPath + "/" + time.Now().Format(utils.FormatDateTimeUnSpace) + ".xlsx"
|
|
|
+ xlsxFile := xlsx.NewFile()
|
|
|
+
|
|
|
+ var condition string
|
|
|
+ var pars []interface{}
|
|
|
+ var classifyName string
|
|
|
+ if classifyId > 0 {
|
|
|
+ classifyInfo, err := data_manage.GetBaseFromUsdaFasClassifyById(classifyId)
|
|
|
+ if err != nil {
|
|
|
+ if err.Error() == utils.ErrNoRow() {
|
|
|
+ br.Msg = "分类不存在"
|
|
|
+ return
|
|
|
+ }
|
|
|
+ br.Msg = "下载失败"
|
|
|
+ br.ErrMsg = "获取分类失败,Err:" + err.Error()
|
|
|
+ return
|
|
|
+ }
|
|
|
+ classifyName = classifyInfo.ClassifyName
|
|
|
+ childClassify, err := data_manage.GetBaseFromUsdaFasClassifyByParentId(classifyId)
|
|
|
+ if err != nil {
|
|
|
+ br.Msg = "下载失败"
|
|
|
+ br.ErrMsg = "获取分类失败,Err:" + err.Error()
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+ if len(childClassify) > 0 {
|
|
|
+ condition += `AND classify_id IN (` + utils.GetOrmInReplace(len(childClassify)) + `)`
|
|
|
+ for _, child := range childClassify {
|
|
|
+ pars = append(pars, child.ClassifyId)
|
|
|
+ }
|
|
|
+ } else {
|
|
|
+ condition += ` AND classify_id=?`
|
|
|
+ pars = append(pars, classifyId)
|
|
|
+ }
|
|
|
+ } else {
|
|
|
+ condition += ` AND classify_id=?`
|
|
|
+ pars = append(pars, classifyId)
|
|
|
+ }
|
|
|
+ if indexCode != "" {
|
|
|
+ condition += ` AND index_code=? `
|
|
|
+ pars = append(pars, indexCode)
|
|
|
+ }
|
|
|
+
|
|
|
+ indexList, err := data_manage.GetUsdaFasIndex(condition, pars)
|
|
|
+ if err != nil {
|
|
|
+ fmt.Println("获取数据失败,Err:" + err.Error())
|
|
|
+ return
|
|
|
+ }
|
|
|
+ if len(indexList) <= 0 {
|
|
|
+ fmt.Println("indexList 为空")
|
|
|
+ br.Ret = 200
|
|
|
+ br.Success = true
|
|
|
+ br.Msg = "success"
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+ indexCodeList := make([]string, 0)
|
|
|
+ frequenciesMap := make(map[string][]*data_manage.BaseFromUsdaFasIndexList)
|
|
|
+ for _, v := range indexList {
|
|
|
+ indexCodeList = append(indexCodeList, v.IndexCode)
|
|
|
+ frequenciesMap[v.Frequency] = append(frequenciesMap[v.Frequency], v)
|
|
|
+ }
|
|
|
+ dataListMap := make(map[string][]*data_manage.BaseFromUsdaFasData)
|
|
|
+ if len(indexList) > 0 {
|
|
|
+ allDataList, e := data_manage.GetUsdaFasIndexDataByCodes(indexCodeList)
|
|
|
+ if e != nil {
|
|
|
+ br.Msg = "获取数据失败"
|
|
|
+ br.ErrMsg = "获取数据失败,Err:" + e.Error()
|
|
|
+ return
|
|
|
+ }
|
|
|
+ for _, v := range allDataList {
|
|
|
+ dataListMap[v.IndexCode] = append(dataListMap[v.IndexCode], v)
|
|
|
+ }
|
|
|
+ }
|
|
|
+ // 按照频率分组排序
|
|
|
+ frequencies := []string{
|
|
|
+ "日度", "周度", "旬度", "月度", "季度", "半年度", "年度",
|
|
|
+ }
|
|
|
+ for _, frequency := range frequencies {
|
|
|
+ //获取指标
|
|
|
+ indexCodeList, ok := frequenciesMap[frequency]
|
|
|
+ if !ok {
|
|
|
+ continue
|
|
|
+ }
|
|
|
+ if len(indexCodeList) <= 0 {
|
|
|
+ fmt.Printf("sheet:%s, 不存在指标", frequency)
|
|
|
+ return
|
|
|
+ }
|
|
|
+ var sheetName string
|
|
|
+ switch frequency {
|
|
|
+ case "日度":
|
|
|
+ sheetName = "日度(Daily)"
|
|
|
+ case "周度":
|
|
|
+ sheetName = "周度(Weekly)"
|
|
|
+ case "旬度":
|
|
|
+ sheetName = "旬度(ten-day)"
|
|
|
+ case "月度":
|
|
|
+ sheetName = "月度(Monthly)"
|
|
|
+ case "季度":
|
|
|
+ sheetName = "季度(Quarterly)"
|
|
|
+ case "半年度":
|
|
|
+ sheetName = "半年度(Semi-annual)"
|
|
|
+ case "年度":
|
|
|
+ sheetName = "年度(Annual)"
|
|
|
+ default:
|
|
|
+ sheetName = "其他数据"
|
|
|
+ }
|
|
|
+ sheetNew, err := xlsxFile.AddSheet(sheetName)
|
|
|
+ if err != nil {
|
|
|
+ fmt.Println("新增Sheet失败", err.Error())
|
|
|
+ return
|
|
|
+ }
|
|
|
+ secNameRow := sheetNew.AddRow()
|
|
|
+ frequencyRow := sheetNew.AddRow()
|
|
|
+ unitRow := sheetNew.AddRow()
|
|
|
+ lastModifyDateRow := sheetNew.AddRow()
|
|
|
+
|
|
|
+ var indexIdList []int
|
|
|
+ for _, idx := range frequenciesMap[frequency] {
|
|
|
+ indexIdList = append(indexIdList, idx.BaseFromUsdaFasIndexId)
|
|
|
+ }
|
|
|
+ dataTimeList, err := data_manage.GetUsdaFasDataDataTimeByIndexId(indexIdList)
|
|
|
+ if err != nil {
|
|
|
+ fmt.Println("获取数据时间失败", err.Error())
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+ // 添加excel左侧指标日期
|
|
|
+ setRowIndex := 4
|
|
|
+ for rk, dv := range dataTimeList {
|
|
|
+ rowIndex := setRowIndex + rk
|
|
|
+ row := sheetNew.Row(rowIndex)
|
|
|
+ displayDate, _ := time.Parse(utils.FormatDate, dv)
|
|
|
+ displayDateCell := row.AddCell()
|
|
|
+ style := new(xlsx.Style)
|
|
|
+ style.ApplyAlignment = true
|
|
|
+ style.Alignment.WrapText = true
|
|
|
+ displayDateCell.SetStyle(style)
|
|
|
+ displayDateCell.SetDate(displayDate)
|
|
|
+
|
|
|
+ }
|
|
|
+ for k, icl := range indexCodeList {
|
|
|
+ // 获取数据
|
|
|
+ dataList, ok := dataListMap[icl.IndexCode]
|
|
|
+ if !ok {
|
|
|
+ continue
|
|
|
+ }
|
|
|
+ if k == 0 {
|
|
|
+ secNameRow.AddCell().SetValue("指标名称/Metric Name")
|
|
|
+ frequencyRow.AddCell().SetValue("频度/Frequency")
|
|
|
+ unitRow.AddCell().SetValue("单位/Unit")
|
|
|
+ lastModifyDateRow.AddCell().SetValue("更新时间/Update Time")
|
|
|
+ min := k * 3
|
|
|
+ sheetNew.SetColWidth(min, min, 15)
|
|
|
+ }
|
|
|
+ if len(dataList) == 0 {
|
|
|
+ continue
|
|
|
+ }
|
|
|
+ secNameRow.AddCell().SetValue(icl.IndexName)
|
|
|
+ frequencyRow.AddCell().SetValue(icl.Frequency)
|
|
|
+ unitRow.AddCell().SetValue(icl.Unit)
|
|
|
+
|
|
|
+ timeDate, err := time.Parse(utils.FormatDateTime, dataList[0].ModifyTime)
|
|
|
+ if err != nil {
|
|
|
+ continue
|
|
|
+ }
|
|
|
+ lastModifyDateRow.AddCell().SetValue(timeDate.Format(utils.FormatDate))
|
|
|
+ dataInfoMap := make(map[string]*data_manage.BaseFromUsdaFasData)
|
|
|
+ for _, v := range dataList {
|
|
|
+ dataInfoMap[v.DataTime] = v
|
|
|
+ }
|
|
|
+
|
|
|
+ for rk, dtv := range dataTimeList {
|
|
|
+ rowIndex := setRowIndex + rk
|
|
|
+ row := sheetNew.Row(rowIndex)
|
|
|
+ displayDateCell := row.AddCell()
|
|
|
+ tmpData, ok := dataInfoMap[dtv]
|
|
|
+ if ok {
|
|
|
+ displayDateCell.SetValue(tmpData.Value)
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ err = xlsxFile.Save(downLoadnFilePath)
|
|
|
+ if err != nil {
|
|
|
+ //有指标无数据时先导出一遍空表
|
|
|
+ sheet, err := xlsxFile.AddSheet("无数据")
|
|
|
+ if err != nil {
|
|
|
+ br.Msg = "新增Sheet失败"
|
|
|
+ br.ErrMsg = "新增Sheet失败,Err:" + err.Error()
|
|
|
+ return
|
|
|
+ }
|
|
|
+ rowSecName := sheet.AddRow()
|
|
|
+ celSecName := rowSecName.AddCell()
|
|
|
+ celSecName.SetValue("")
|
|
|
+ err = xlsxFile.Save(downLoadnFilePath)
|
|
|
+ if err != nil {
|
|
|
+ br.Msg = "保存文件失败"
|
|
|
+ br.ErrMsg = "保存文件失败"
|
|
|
+ return
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ fileName := classifyName
|
|
|
+ if indexCode != "" && len(indexList) == 1 {
|
|
|
+ fileName = indexList[0].IndexName
|
|
|
+ }
|
|
|
+ fileName += time.Now().Format("06.01.02") + `.xlsx` //文件名称
|
|
|
+ this.Ctx.Output.Download(downLoadnFilePath, fileName)
|
|
|
+ defer func() {
|
|
|
+ os.Remove(downLoadnFilePath)
|
|
|
+ }()
|
|
|
+ br.Ret = 200
|
|
|
+ br.Success = true
|
|
|
+ br.Msg = "success"
|
|
|
+}
|