123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731 |
- package data_manage
- import (
- "encoding/json"
- "fmt"
- "hongze/hz_crm_api/controllers"
- "hongze/hz_crm_api/models"
- "hongze/hz_crm_api/models/data_manage"
- "hongze/hz_crm_api/services/data"
- "hongze/hz_crm_api/utils"
- "net/url"
- "strconv"
- "strings"
- "time"
- )
- // 计算指标
- type EdbInfoCalculateController struct {
- controllers.BaseAuthController
- }
- // @Title 生成计算指标
- // @Description 生成计算指标接口
- // @Param request body data_manage.EdbInfoCalculateSaveReq true "type json string"
- // @Success Ret=200 返回图表id
- // @router /edb_info/calculate/save [post]
- func (this *ChartInfoController) CalculateSave() {
- 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
- }
- var req data_manage.EdbInfoCalculateSaveReq
- 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, " ")
- 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
- }
- if len(req.EdbInfoIdArr) <= 0 {
- br.Msg = "请选择指标"
- return
- }
- if req.CalculateFormula == "" {
- br.Msg = "请填写指标"
- return
- }
- calculateFormula := req.CalculateFormula
- calculateFormula = strings.Replace(calculateFormula, "(", "(", -1)
- calculateFormula = strings.Replace(calculateFormula, ")", ")", -1)
- calculateFormula = strings.Replace(calculateFormula, ",", ",", -1)
- calculateFormula = strings.Replace(calculateFormula, "。", ".", -1)
- calculateFormula = strings.Replace(calculateFormula, "%", "*0.01", -1)
- req.CalculateFormula = calculateFormula
- //判断是否重复指标
- edbInfoMap := make(map[int]string)
- //移除研究员选择指标中的未使用的指标
- {
- //转大写的计算公式
- upperCalculateFormulaStr := strings.ToUpper(req.CalculateFormula)
- //用到的指标
- newEdbInfoIdArr := make([]data_manage.EdbInfoFromTag, 0)
- for _, tmpEdbInfo := range req.EdbInfoIdArr {
- _, ok := edbInfoMap[tmpEdbInfo.EdbInfoId]
- if ok {
- br.Msg = "选择指标失败,请勿选择重复指标!"
- return
- }
- edbInfoMap[tmpEdbInfo.EdbInfoId] = tmpEdbInfo.FromTag
- upperFromTag := strings.ToUpper(tmpEdbInfo.FromTag)
- if strings.Contains(upperCalculateFormulaStr, upperFromTag) {
- newEdbInfoIdArr = append(newEdbInfoIdArr, tmpEdbInfo)
- }
- }
- req.EdbInfoIdArr = newEdbInfoIdArr
- }
- var condition string
- var pars []interface{}
- condition += " AND edb_name=? "
- pars = append(pars, req.EdbName)
- count, err := data_manage.GetEdbInfoCountByCondition(condition, pars)
- if err != nil {
- br.Msg = "判断指标名称是否存在失败"
- br.ErrMsg = "判断指标名称是否存在失败,Err:" + err.Error()
- return
- }
- if count > 0 {
- br.Msg = "指标名称已存在,请重新填写"
- br.ErrMsg = "指标名称已存在,请重新填写"
- br.IsSendEmail = false
- return
- }
- //检验公式
- var formulaStr string
- var edbInfoIdBytes []string
- for _, v := range req.EdbInfoIdArr {
- formulaStr += v.FromTag + ","
- edbInfoIdBytes = append(edbInfoIdBytes, v.FromTag)
- }
- formulaMap := data.CheckFormula(req.CalculateFormula)
- for _, v := range formulaMap {
- if !strings.Contains(formulaStr, v) {
- br.Msg = "公式错误,请重新填写"
- return
- }
- }
- // 调用指标库去新增
- req2 := &data_manage.EdbInfoCalculateBatchSaveReqByEdbLib{
- AdminId: sysUser.AdminId,
- AdminName: sysUser.RealName,
- //EdbInfoId: req.EdbInfoId,
- EdbName: req.EdbName,
- Frequency: req.Frequency,
- Unit: req.Unit,
- ClassifyId: req.ClassifyId,
- CalculateFormula: req.CalculateFormula,
- EdbInfoIdArr: req.EdbInfoIdArr,
- }
- reqJson, err := json.Marshal(req2)
- if err != nil {
- br.Msg = "参数解析异常!"
- br.ErrMsg = "参数解析失败,Err:" + err.Error()
- return
- }
- respItem, err := data.AddEdbCalculateData(string(reqJson))
- if err != nil {
- br.Msg = "新增失败"
- br.ErrMsg = "新增失败,Err:" + err.Error()
- return
- }
- if respItem.Ret != 200 {
- br.Msg = respItem.Msg
- br.ErrMsg = respItem.ErrMsg
- return
- }
- resp := respItem.Data
- //添加es
- data.AddOrEditEdbInfoToEs(resp.EdbInfoId)
- br.Ret = 200
- br.Success = true
- br.Msg = "保存成功"
- br.Data = resp
- br.IsAddLog = true
- }
- // @Title 获取计算指标详情
- // @Description 获取计算指标详情接口
- // @Param EdbInfoId query int true "指标id"
- // @Success 200 {object} data_manage.CalculateDetailResp
- // @router /edb_info/calculate/detail [get]
- func (this *ChartInfoController) CalculateDetail() {
- 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
- }
- edbInfoId, _ := this.GetInt("EdbInfoId")
- edbInfo, err := data_manage.GetEdbInfoById(edbInfoId)
- if err != nil {
- br.Msg = "获取失败"
- br.ErrMsg = "获取失败,Err:" + err.Error()
- return
- }
- calculateList, err := data_manage.GetEdbInfoCalculateDetail(edbInfoId, edbInfo.Source)
- if err != nil {
- br.Msg = "获取失败"
- br.ErrMsg = "获取失败,Err:" + err.Error()
- return
- }
- resp := new(data_manage.CalculateDetailResp)
- resp.EdbInfoDetail = edbInfo
- resp.CalculateList = calculateList
- br.Ret = 200
- br.Success = true
- br.Msg = "保存成功"
- br.Data = resp
- }
- // @Title 修改计算指标
- // @Description 修改计算指标接口
- // @Param request body data_manage.EdbInfoCalculateEditReq true "type json string"
- // @Success Ret=200 返回图表id
- // @router /edb_info/calculate/edit [post]
- func (this *ChartInfoController) CalculateEdit() {
- 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
- }
- var req data_manage.EdbInfoCalculateEditReq
- err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
- if err != nil {
- br.Msg = "参数解析异常!"
- br.ErrMsg = "参数解析失败,Err:" + err.Error()
- return
- }
- if req.EdbInfoId <= 0 {
- br.Msg = "参数错误"
- br.ErrMsg = "指标ID:" + strconv.Itoa(req.EdbInfoId)
- return
- }
- req.EdbName = strings.Trim(req.EdbName, " ")
- 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
- }
- if len(req.EdbInfoIdArr) <= 0 {
- br.Msg = "请选择指标"
- return
- }
- if req.CalculateFormula == "" {
- br.Msg = "请填写指标"
- return
- }
- calculateFormula := req.CalculateFormula
- calculateFormula = strings.Replace(calculateFormula, "(", "(", -1)
- calculateFormula = strings.Replace(calculateFormula, ")", ")", -1)
- calculateFormula = strings.Replace(calculateFormula, ",", ",", -1)
- calculateFormula = strings.Replace(calculateFormula, "。", ".", -1)
- calculateFormula = strings.Replace(calculateFormula, "%", "*0.01", -1)
- req.CalculateFormula = calculateFormula
- fmt.Println("calculateFormula:" + calculateFormula)
- //判断是否重复指标
- edbInfoMap := make(map[int]string)
- //移除研究员选择指标中的未使用的指标
- {
- //转大写的计算公式
- upperCalculateFormulaStr := strings.ToUpper(req.CalculateFormula)
- //用到的指标
- newEdbInfoIdArr := make([]data_manage.EdbInfoFromTag, 0)
- for _, tmpEdbInfo := range req.EdbInfoIdArr {
- _, ok := edbInfoMap[tmpEdbInfo.EdbInfoId]
- if ok {
- br.Msg = "选择指标失败,请勿选择重复指标!"
- return
- }
- edbInfoMap[tmpEdbInfo.EdbInfoId] = tmpEdbInfo.FromTag
- upperFromTag := strings.ToUpper(tmpEdbInfo.FromTag)
- if strings.Contains(upperCalculateFormulaStr, upperFromTag) {
- newEdbInfoIdArr = append(newEdbInfoIdArr, tmpEdbInfo)
- }
- }
- req.EdbInfoIdArr = newEdbInfoIdArr
- }
- var condition string
- var pars []interface{}
- condition += " AND edb_name=? "
- pars = append(pars, req.EdbName)
- condition += " AND edb_info_id<>? "
- pars = append(pars, req.EdbInfoId)
- count, err := data_manage.GetEdbInfoCountByCondition(condition, pars)
- if err != nil {
- br.Msg = "判断指标名称是否存在失败"
- br.ErrMsg = "判断指标名称是否存在失败,Err:" + err.Error()
- return
- }
- if count > 0 {
- br.Msg = "指标名称已存在,请重新填写"
- br.ErrMsg = "指标名称已存在,请重新填写"
- br.IsSendEmail = false
- return
- }
- //判断公式,指标是否有改动
- edbInfoDetail, err := data_manage.GetEdbInfoById(req.EdbInfoId)
- if err != nil {
- br.Msg = "修改失败"
- br.Msg = "获取指标信息失败,GetEdbInfoById Err:" + err.Error()
- return
- }
- if edbInfoDetail == nil {
- br.Msg = "修改失败"
- br.Msg = "指标信息不存在,EdbInfoId:" + strconv.Itoa(req.EdbInfoId)
- return
- }
- // 调用指标库去更新
- req2 := &data_manage.EdbInfoCalculateBatchSaveReqByEdbLib{
- AdminId: sysUser.AdminId,
- AdminName: sysUser.RealName,
- EdbInfoId: req.EdbInfoId,
- EdbName: req.EdbName,
- Frequency: req.Frequency,
- Unit: req.Unit,
- ClassifyId: req.ClassifyId,
- CalculateFormula: req.CalculateFormula,
- EdbInfoIdArr: req.EdbInfoIdArr,
- }
- reqJson, err := json.Marshal(req2)
- if err != nil {
- br.Msg = "参数解析异常!"
- br.ErrMsg = "参数解析失败,Err:" + err.Error()
- return
- }
- respItem, err := data.EditEdbCalculateData(string(reqJson))
- if err != nil {
- br.Msg = "编辑失败"
- br.ErrMsg = "编辑失败,Err:" + err.Error()
- return
- }
- if respItem.Ret != 200 {
- br.Msg = respItem.Msg
- br.ErrMsg = respItem.ErrMsg
- return
- }
- resp := respItem.Data
- //添加es
- data.AddOrEditEdbInfoToEs(req.EdbInfoId)
- br.Ret = 200
- br.Success = true
- br.Msg = "保存成功"
- br.IsAddLog = true
- br.Data = resp
- }
- // @Title 累计值转月-同比值-同差等计算新增
- // @Description 累计值转月-同比值-同差等计算新增接口
- // @Param request body data_manage.EdbInfoCalculateBatchSaveReq true "type json string"
- // @Success Ret=200 返回指标id
- // @router /edb_info/calculate/batch/save [post]
- func (this *ChartInfoController) CalculateBatchSave() {
- 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
- }
- var req data_manage.EdbInfoCalculateBatchSaveReq
- 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, " ")
- 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
- }
- fromEdbInfoId := req.FromEdbInfoId
- var formulaInt int
- switch req.Source {
- case utils.DATA_SOURCE_CALCULATE_NSZYDPJJS, utils.DATA_SOURCE_CALCULATE_HBZ, utils.DATA_SOURCE_CALCULATE_HCZ, utils.DATA_SOURCE_CALCULATE_TIME_SHIFT, utils.DATA_SOURCE_CALCULATE_CJJX:
- if req.Formula == "" {
- br.Msg = "请填写N值"
- return
- }
- formulaInt, _ = strconv.Atoi(req.Formula)
- if formulaInt <= 0 {
- br.Msg = "N值输入错误,请重新输入"
- return
- }
- case utils.DATA_SOURCE_CALCULATE_ZJPJ:
- //直接拼接指标
- //校验时间格式
- _, err = time.ParseInLocation(utils.FormatDate, req.Formula, time.Local)
- if err != nil {
- br.Msg = "拼接日期有误,请重新输入"
- return
- }
- case utils.DATA_SOURCE_CALCULATE_NHCC: //拟合残差指标
- //指标校验
- if len(req.EdbInfoIdArr) != 2 {
- br.Msg = "选择的指标异常,请重新选择"
- return
- }
- fromEdbInfoId = req.EdbInfoIdArr[0].EdbInfoId
- //校验时间格式
- //数据格式:2022-11-01,2022-11-10
- timeList := strings.Split(req.Formula, ",")
- if len(timeList) != 2 {
- br.Msg = "选择时间有误,请重新输入"
- return
- }
- startDate, err := time.ParseInLocation(utils.FormatDate, timeList[0], time.Local)
- if err != nil {
- br.Msg = "开始日期有误,请重新输入"
- return
- }
- endDate, err := time.ParseInLocation(utils.FormatDate, timeList[1], time.Local)
- if err != nil {
- br.Msg = "结束日期有误,请重新输入"
- return
- }
- if utils.GetTimeSubDay(startDate, endDate) < 2 {
- br.Msg = "日期间隔不得少于两天"
- return
- }
- }
- notNeedFromEdbSourceList := []int{utils.DATA_SOURCE_CALCULATE_KSZS} // 不需要传入来源指标id的 指标类型
- if fromEdbInfoId <= 0 && !utils.InArrayByInt(notNeedFromEdbSourceList, req.Source) {
- br.Msg = "请选择指标"
- return
- }
- //加入缓存机制,避免创建同一个名称的指标 start
- redisKey := fmt.Sprint("edb_info:calculate:batch:save:", req.Source, ":", req.EdbName)
- isExist := utils.Rc.IsExist(redisKey)
- if isExist {
- br.Msg = "指标正在处理,请勿重复提交"
- return
- } else {
- //设置3分钟缓存
- utils.Rc.SetNX(redisKey, 1, time.Second*300)
- defer func() {
- utils.Rc.Delete(redisKey)
- }()
- }
- //加入缓存机制,避免创建同一个名称的指标 end
- req2 := &data_manage.EdbInfoCalculateBatchSaveReqByEdbLib{
- AdminId: sysUser.AdminId,
- AdminName: sysUser.RealName,
- EdbInfoId: req.EdbInfoId,
- EdbName: req.EdbName,
- Frequency: req.Frequency,
- Unit: req.Unit,
- ClassifyId: req.ClassifyId,
- Formula: req.Formula, //N数值移动平均计算、环比值、环差值
- FromEdbInfoId: req.FromEdbInfoId,
- CalculateFormula: req.CalculateFormula,
- Source: req.Source,
- MoveType: req.MoveType,
- MoveFrequency: req.MoveFrequency,
- //CalculateFormula: edbInfo.CalculateFormula,
- EdbInfoIdArr: req.EdbInfoIdArr,
- Calendar: req.Calendar,
- }
- // 调用指标库去更新
- reqJson, err := json.Marshal(req2)
- if err != nil {
- br.Msg = "参数解析异常!"
- br.ErrMsg = "参数解析失败,Err:" + err.Error()
- return
- }
- respItem, err := data.BatchSaveEdbCalculateData(string(reqJson))
- if err != nil {
- br.Msg = "新增失败"
- br.ErrMsg = "新增失败,Err:" + err.Error()
- return
- }
- if respItem.Ret != 200 {
- br.Msg = respItem.Msg
- br.ErrMsg = respItem.ErrMsg
- return
- }
- resp := respItem.Data
- //添加es
- data.AddOrEditEdbInfoToEs(resp.EdbInfoId)
- br.Ret = 200
- br.Success = true
- br.Msg = "保存成功"
- br.Data = resp
- br.IsAddLog = true
- }
- // @Title 累计值转月-同比值-同差等计算编辑
- // @Description 累计值转月-同比值-同差等计算编辑接口
- // @Param request body data_manage.EdbInfoCalculateBatchEditReq true "type json string"
- // @Success Ret=200 返回指标id
- // @router /edb_info/calculate/batch/edit [post]
- func (this *ChartInfoController) CalculateBatchEdit() {
- 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
- }
- var req data_manage.EdbInfoCalculateBatchEditReq
- err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
- if err != nil {
- br.Msg = "参数解析异常!"
- br.ErrMsg = "参数解析失败,Err:" + err.Error()
- 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
- }
- if req.EdbInfoId <= 0 {
- br.Msg = "请选择指标"
- return
- }
- var condition string
- var pars []interface{}
- condition += " AND edb_name=? "
- pars = append(pars, req.EdbName)
- condition += " AND edb_info_id<>? "
- pars = append(pars, req.EdbInfoId)
- count, err := data_manage.GetEdbInfoCountByCondition(condition, pars)
- if err != nil {
- br.Msg = "判断指标名称是否存在失败"
- br.ErrMsg = "判断指标名称是否存在失败,Err:" + err.Error()
- return
- }
- if count > 0 {
- br.Msg = "指标名称已存在,请重新填写"
- br.ErrMsg = "指标名称已存在,请重新填写"
- br.IsSendEmail = false
- return
- }
- edbInfo, err := data_manage.GetEdbInfoById(req.EdbInfoId)
- if err != nil {
- if err.Error() == utils.ErrNoRow() {
- br.Msg = "指标已被删除,请刷新页面"
- br.ErrMsg = "指标已被删除,请刷新页面:Err:" + err.Error()
- return
- }
- br.Msg = "获取指标信息失败"
- br.ErrMsg = "获取指标信息失败:Err:" + err.Error()
- return
- }
- if req.Source <= 0 {
- req.Source = edbInfo.Source
- }
- var formulaInt int
- switch req.Source {
- case utils.DATA_SOURCE_CALCULATE_NSZYDPJJS, utils.DATA_SOURCE_CALCULATE_HBZ, utils.DATA_SOURCE_CALCULATE_HCZ, utils.DATA_SOURCE_CALCULATE_TIME_SHIFT, utils.DATA_SOURCE_CALCULATE_CJJX:
- if req.Formula == "" {
- br.Msg = "请填写N值"
- return
- }
- formulaInt, _ = strconv.Atoi(req.Formula)
- if formulaInt <= 0 {
- br.Msg = "N值输入错误,请重新输入"
- return
- }
- case utils.DATA_SOURCE_CALCULATE_ZJPJ:
- //直接拼接指标
- //校验时间格式
- _, err = time.ParseInLocation(utils.FormatDate, req.Formula, time.Local)
- if err != nil {
- br.Msg = "拼接日期有误,请重新输入"
- return
- }
- case utils.DATA_SOURCE_CALCULATE_NHCC: //拟合残差指标
- //指标校验
- if len(req.EdbInfoIdArr) != 2 {
- br.Msg = "选择的指标异常,请重新选择"
- return
- }
- //校验时间格式
- //数据格式:2022-11-01,2022-11-10
- timeList := strings.Split(req.Formula, ",")
- if len(timeList) != 2 {
- br.Msg = "选择时间有误,请重新输入"
- return
- }
- startDate, err := time.ParseInLocation(utils.FormatDate, timeList[0], time.Local)
- if err != nil {
- br.Msg = "开始日期有误,请重新输入"
- return
- }
- endDate, err := time.ParseInLocation(utils.FormatDate, timeList[1], time.Local)
- if err != nil {
- br.Msg = "结束日期有误,请重新输入"
- return
- }
- if utils.GetTimeSubDay(startDate, endDate) < 2 {
- br.Msg = "日期间隔不得少于两天"
- return
- }
- }
- // 构造请求
- req2 := &data_manage.EdbInfoCalculateBatchEditReqByEdbLib{
- EdbInfoId: req.EdbInfoId,
- EdbName: req.EdbName,
- Frequency: req.Frequency,
- Unit: req.Unit,
- ClassifyId: req.ClassifyId,
- Formula: req.Formula, //N数值移动平均计算、环比值、环差值
- FromEdbInfoId: req.FromEdbInfoId,
- Source: req.Source,
- MoveType: req.MoveType,
- MoveFrequency: req.MoveFrequency,
- EdbInfoIdArr: req.EdbInfoIdArr,
- Calendar: req.Calendar,
- }
- // 调用指标库去更新
- reqJson, err := json.Marshal(req2)
- if err != nil {
- br.Msg = "参数解析异常!"
- br.ErrMsg = "参数解析失败,Err:" + err.Error()
- return
- }
- respItem, err := data.BatchEditEdbCalculateData(string(reqJson))
- if err != nil {
- br.Msg = "编辑失败"
- br.ErrMsg = "编辑失败,Err:" + err.Error()
- return
- }
- if respItem.Ret != 200 {
- br.Msg = respItem.Msg
- br.ErrMsg = respItem.ErrMsg
- return
- }
- resp := respItem.Data
- //添加es
- data.AddOrEditEdbInfoToEs(resp.EdbInfoId)
- br.Ret = 200
- br.Success = true
- br.Msg = "保存成功"
- br.Data = resp
- br.IsAddLog = true
- }
- // @Title 累计值转月-同比值-同差等重新计算
- // @Description 累计值转月-同比值-同差等重新计算接口
- // @Param request body data_manage.EdbInfoCalculateBatchSaveReq true "type json string"
- // @Success Ret=200 返回指标id
- // @router /edb_info/calculate/batch/reset [post]
- func (this *ChartInfoController) CalculateBatchReset() {
- 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
- }
- var req data_manage.EdbInfoCalculateBatchSaveReq
- err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
- if err != nil {
- br.Msg = "参数解析异常!"
- br.ErrMsg = "参数解析失败,Err:" + err.Error()
- return
- }
- if req.EdbInfoId <= 0 {
- br.Msg = "参数错误"
- br.ErrMsg = "参数错误:EdbInfoId:" + strconv.Itoa(req.EdbInfoId)
- return
- }
- edbInfo, err := data_manage.GetEdbInfoById(req.EdbInfoId)
- if err != nil {
- if err.Error() == utils.ErrNoRow() {
- br.Msg = "指标已被删除,请刷新页面"
- br.ErrMsg = "指标已删除,请刷新页面:Err:" + err.Error()
- return
- }
- br.Msg = "重新计算失败"
- br.ErrMsg = "获取指标信息失败:Err:" + err.Error()
- return
- }
- uniqueCode := edbInfo.UniqueCode
- sourName := edbInfo.SourceName
- edbInfoId := edbInfo.EdbInfoId
- //if source == utils.DATA_SOURCE_CALCULATE {
- // //检验公式
- // var formulaStr string
- // var edbInfoIdBytes []string
- // for _, v := range fromEdbInfoList {
- // formulaStr += v.FromTag + ","
- // edbInfoIdBytes = append(edbInfoIdBytes, v.FromTag)
- // }
- // formulaStr = strings.Trim(formulaStr, ",")
- // formulaMap := data.CheckFormula(req.CalculateFormula)
- // for _, v := range formulaMap {
- // if !strings.Contains(formulaStr, v) {
- // br.Msg = "公式错误,请重新填写"
- // return
- // }
- // }
- //
- // edbInfoList := make([]*data_manage.EdbInfo, 0)
- //
- // for _, v := range fromEdbInfoList {
- // edbInfo, err := data_manage.GetEdbInfoById(v.FromEdbInfoId)
- // if err != nil {
- // if err.Error() == utils.ErrNoRow() {
- // br.Msg = "重新计算失败"
- // br.Msg = "指标 " + strconv.Itoa(v.FromEdbInfoId) + " 不存在"
- // return
- // }
- // br.Msg = "重新计算失败"
- // br.Msg = "获取指标失败:Err:" + err.Error()
- // return
- // }
- // edbInfoList = append(edbInfoList, edbInfo)
- // }
- // //清除历史数据
- // err = data_manage.DeleteCalculateData(edbInfoId)
- // if err != nil {
- // br.Msg = "重新计算失败"
- // br.Msg = "获取指标失败:Err:" + err.Error()
- // return
- // }
- // err = data.Calculate(edbInfoList, int(edbInfoId), edbCode, edbInfo.CalculateFormula, edbInfoIdBytes)
- // if err != nil {
- // br.Msg = "重新计算失败"
- // br.Msg = "生成计算指标失败,Calculate Err:" + err.Error()
- // return
- // }
- //} else if source == utils.DATA_SOURCE_CALCULATE_LJZZY {
- // _, err = data_manage.AddCalculateLjzzy(&req, fromEdbInfo, edbCode, uniqueCode, sysUser.AdminId, sysUser.RealName)
- //} else if source == utils.DATA_SOURCE_CALCULATE_TBZ {
- // sourName = "同比值"
- // edbInfoId, err = data_manage.AddCalculateTbz(&req, fromEdbInfo, edbCode, uniqueCode, sysUser.AdminId, sysUser.RealName)
- //} else if source == utils.DATA_SOURCE_CALCULATE_TCZ {
- // sourName = "同差值"
- // edbInfoId, err = data_manage.AddCalculateTcz(&req, fromEdbInfo, edbCode, uniqueCode, sysUser.AdminId, sysUser.RealName)
- //} else if source == utils.DATA_SOURCE_CALCULATE_NSZYDPJJS {
- // sourName = "N数值移动平均计算"
- // formulaInt, _ := strconv.Atoi(edbInfo.CalculateFormula)
- // edbInfoId, err = data_manage.AddCalculateNszydpjjs(&req, fromEdbInfo, edbCode, uniqueCode, sysUser.AdminId, sysUser.RealName, formulaInt)
- //} else if source == utils.DATA_SOURCE_CALCULATE_HBZ {
- // sourName = "环比值"
- // formulaInt, _ := strconv.Atoi(edbInfo.CalculateFormula)
- // edbInfoId, err = data_manage.AddCalculateHbz(&req, fromEdbInfo, edbCode, uniqueCode, sysUser.AdminId, sysUser.RealName, formulaInt)
- //} else if source == utils.DATA_SOURCE_CALCULATE_HCZ {
- // sourName = "环差值"
- // formulaInt, _ := strconv.Atoi(edbInfo.CalculateFormula)
- // edbInfoId, err = data_manage.AddCalculateHcz(&req, fromEdbInfo, edbCode, uniqueCode, sysUser.AdminId, sysUser.RealName, formulaInt)
- //} else if source == utils.DATA_SOURCE_CALCULATE_BP {
- // sourName = "变频"
- // edbInfoId, err = data_manage.AddCalculateBp(&req, fromEdbInfo, edbCode, uniqueCode, sysUser.AdminId, sysUser.RealName)
- //} else if source == utils.DATA_SOURCE_CALCULATE_TIME_SHIFT {
- // sourName = "时间移位"
- // if req.MoveFrequency == "" {
- // req.MoveFrequency = edbInfo.MoveFrequency
- // }
- // if req.Formula == "" {
- // req.Formula = edbInfo.CalculateFormula
- // }
- // edbInfoId, err = data_manage.AddCalculateTimeShift(&req, fromEdbInfo, edbCode, uniqueCode, sysUser.AdminId, sysUser.RealName)
- //} else if source == utils.DATA_SOURCE_CALCULATE_ZJPJ {
- // sourName = "直接拼接"
- // err = data_manage.RefreshAllCalculateZjpj(edbInfo)
- //} else if source == utils.DATA_SOURCE_CALCULATE_LJZTBPJ {
- // sourName = "累计值同比拼接"
- // err = data_manage.RefreshAllCalculateLjztbpj(edbInfo)
- //} else {
- // br.Msg = "无效计算方式"
- // br.ErrMsg = "无效计算方式,source:" + strconv.Itoa(source)
- // return
- //}
- // 重新计算
- err = data.EdbInfoRefreshAllFromBaseV2(edbInfoId, true)
- if err != nil {
- fmt.Println(edbInfoId, "RefreshEdbCalculateData err", time.Now())
- br.Msg = "重新计算失败"
- br.ErrMsg = "生成" + sourName + "失败,RefreshEdbCalculateData Err:" + err.Error()
- return
- }
- resp := new(data_manage.AddEdbInfoResp)
- resp.EdbInfoId = edbInfoId
- resp.UniqueCode = uniqueCode
- br.Ret = 200
- br.Success = true
- br.Msg = "重新计算成功"
- br.Data = resp
- br.IsAddLog = true
- }
- //累计值转月值
- //func init() {
- // fmt.Println("start CalculateBatchSave")
- // req := new(data_manage.EdbInfoCalculateBatchSaveReq)
- // req.EdbInfoId = 100282
- //
- // randStr := utils.GetRandDigit(4)
- // edbCode := `C` + time.Now().Format("060102") + randStr
- // timestamp := strconv.FormatInt(time.Now().UnixNano(), 10)
- // uniqueCode := utils.MD5(utils.DATA_PREFIX + "_" + timestamp)
- // fromEdbInfo, _ := data_manage.GetEdbInfoById(req.EdbInfoId)
- // data_manage.AddCalculateLjzzy(req, fromEdbInfo, edbCode, uniqueCode, 1, "rdluck")
- // fmt.Println("end CalculateBatchSave")
- // return
- //}
- ////同比值
- //func init() {
- // fmt.Println("start AddCalculateTbz")
- // req := new(data_manage.EdbInfoCalculateBatchSaveReq)
- // req.EdbInfoId = 100092
- //
- // randStr := utils.GetRandDigit(4)
- // edbCode := `C` + time.Now().Format("060102") + randStr
- // timestamp := strconv.FormatInt(time.Now().UnixNano(), 10)
- // uniqueCode := utils.MD5(utils.DATA_PREFIX + "_" + timestamp)
- // fromEdbInfo, _ := data_manage.GetEdbInfoById(req.EdbInfoId)
- //
- // data_manage.AddCalculateTbz(req, fromEdbInfo, edbCode, uniqueCode, 1, "rdluck")
- // fmt.Println("end AddCalculateTbz")
- // return
- //}
- //同差值
- //func init() {
- // fmt.Println("start AddCalculateTbz")
- // req := new(data_manage.EdbInfoCalculateBatchSaveReq)
- // req.EdbInfoId = 100092
- //
- // randStr := utils.GetRandDigit(4)
- // edbCode := `C` + time.Now().Format("060102") + randStr
- // timestamp := strconv.FormatInt(time.Now().UnixNano(), 10)
- // uniqueCode := utils.MD5(utils.DATA_PREFIX + "_" + timestamp)
- // fromEdbInfo, _ := data_manage.GetEdbInfoById(req.EdbInfoId)
- //
- // data_manage.AddCalculateTcz(req, fromEdbInfo, edbCode, uniqueCode, 1, "rdluck")
- //
- // fmt.Println("end AddCalculateTbz")
- // return
- //}
- //N数值移动平均计算
- //func init() {
- // fmt.Println("start AddCalculateTbz")
- // req := new(data_manage.EdbInfoCalculateBatchSaveReq)
- // req.FromEdbInfoId = 100081
- //
- // randStr := utils.GetRandDigit(4)
- // edbCode := `C` + time.Now().Format("060102") + randStr
- // timestamp := strconv.FormatInt(time.Now().UnixNano(), 10)
- // uniqueCode := utils.MD5(utils.DATA_PREFIX + "_" + timestamp)
- // fromEdbInfo, _ := data_manage.GetEdbInfoById(req.FromEdbInfoId)
- // formulaInt := 3
- // fmt.Println("edbCode:", edbCode)
- // data_manage.AddCalculateNszydpjjs(req, fromEdbInfo, edbCode, uniqueCode, 1, "rdluck", formulaInt)
- //
- // fmt.Println("end AddCalculateTbz")
- // return
- //}
- // ExecPythonCode
- // @Title 执行python代码
- // @Description 执行python代码接口
- // @Param request body data_manage.EdbInfoCalculateSaveReq true "type json string"
- // @Success Ret=200 返回图表id
- // @router /edb_info/python/exec [post]
- func (this *ChartInfoController) ExecPythonCode() {
- 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
- }
- var req data_manage.ExecPythonEdbReq
- err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
- if err != nil {
- br.Msg = "参数解析异常!"
- br.ErrMsg = "参数解析失败,Err:" + err.Error()
- return
- }
- if req.PythonCode == "" {
- br.Msg = "python代码不能为空"
- return
- }
- resp, err := data.ExecPythonCode(req.PythonCode)
- if err != nil {
- br.Msg = "python代码执行失败,请检查代码"
- br.ErrMsg = "python代码执行失败,Err:" + err.Error()
- return
- }
- if resp.Ret != 200 {
- br.Msg = "python代码执行失败,请检查代码"
- br.ErrMsg = "python代码执行失败,Err:" + resp.ErrMsg
- return
- }
- br.Ret = 200
- br.Success = true
- br.Msg = "运行成功"
- br.Data = resp.Data
- }
- // AddPythonEdb
- // @Title 添加python指标
- // @Description 添加python指标接口
- // @Param request body data_manage.AddPythonEdbReq true "type json string"
- // @Success 200 {object} data_manage.AddEdbInfoResp
- // @router /edb_info/python/add [post]
- func (this *ChartInfoController) AddPythonEdb() {
- 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
- }
- var req data_manage.PostAddPythonEdbReq
- err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
- if err != nil {
- br.Msg = "参数解析异常!"
- br.ErrMsg = "参数解析失败,Err:" + err.Error()
- return
- }
- //传入操作人
- req.AdminId = sysUser.AdminId
- req.AdminName = sysUser.RealName
- reqJson, err := json.Marshal(req)
- if err != nil {
- br.Msg = "参数解析异常!"
- br.ErrMsg = "参数解析失败,Err:" + err.Error()
- return
- }
- // 判断指标名称是否重复
- {
- var condition string
- var pars []interface{}
- condition += " AND edb_name=? "
- pars = append(pars, req.EdbName)
- condition += " AND edb_info_id<>? "
- pars = append(pars, req.EdbInfoId)
- count, err := data_manage.GetEdbInfoCountByCondition(condition, pars)
- if err != nil {
- br.Msg = "判断指标名称是否存在失败"
- br.ErrMsg = "判断指标名称是否存在失败,Err:" + err.Error()
- return
- }
- if count > 0 {
- br.Msg = "指标名称已存在,请重新填写"
- br.ErrMsg = "指标名称已存在,请重新填写"
- br.IsSendEmail = false
- return
- }
- }
- // 添加python指标
- respItem, err := data.AddPythonEdbData(string(reqJson))
- if err != nil {
- br.Msg = "添加失败"
- br.ErrMsg = "获取失败,Err:" + err.Error()
- return
- }
- if respItem.Ret != 200 {
- br.Msg = "添加失败"
- br.ErrMsg = respItem.ErrMsg + ";EdbName:" + req.EdbName
- return
- }
- respData := respItem.Data.(map[string]interface{})
- tmpEdbInfoId, ok := respData["EdbInfoId"]
- if !ok {
- br.Msg = `数据转换异常`
- br.ErrMsg = fmt.Sprint("respData.EdbInfoId 数据转换异常;respData:", respData)
- return
- }
- //指标id
- edbInfoId := int64(tmpEdbInfoId.(float64))
- uniqueCode, ok := respData["UniqueCode"]
- if !ok {
- br.Msg = `数据转换异常`
- br.ErrMsg = fmt.Sprint("respData.UniqueCode 数据转换异常;respData:", respData)
- return
- }
- //添加es
- data.AddOrEditEdbInfoToEs(int(edbInfoId))
- //
- resp := data_manage.AddEdbInfoResp{
- EdbInfoId: int(edbInfoId),
- UniqueCode: uniqueCode.(string),
- }
- br.Ret = 200
- br.Success = true
- br.Msg = "保存成功"
- br.Data = resp
- br.IsAddLog = true
- }
- // EditPythonEdb
- // @Title 修改python运算指标
- // @Description 修改python运算指标接口
- // @Param request body data_manage.AddPythonEdbReq true "type json string"
- // @Success Ret=200 返回成功
- // @router /edb_info/python/edit [post]
- func (this *ChartInfoController) EditPythonEdb() {
- 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
- }
- var req data_manage.PostAddPythonEdbReq
- err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
- if err != nil {
- br.Msg = "参数解析异常!"
- br.ErrMsg = "参数解析失败,Err:" + err.Error()
- return
- }
- //传入操作人
- req.AdminId = sysUser.AdminId
- req.AdminName = sysUser.RealName
- reqJson, err := json.Marshal(req)
- if err != nil {
- br.Msg = "参数解析异常!"
- br.ErrMsg = "参数解析失败,Err:" + err.Error()
- return
- }
- // 判断指标名称是否重复
- {
- var condition string
- var pars []interface{}
- condition += " AND edb_name=? "
- pars = append(pars, req.EdbName)
- condition += " AND edb_info_id<>? "
- pars = append(pars, req.EdbInfoId)
- count, err := data_manage.GetEdbInfoCountByCondition(condition, pars)
- if err != nil {
- br.Msg = "判断指标名称是否存在失败"
- br.ErrMsg = "判断指标名称是否存在失败,Err:" + err.Error()
- return
- }
- if count > 0 {
- br.Msg = "指标名称已存在,请重新填写"
- br.ErrMsg = "指标名称已存在,请重新填写"
- br.IsSendEmail = false
- return
- }
- }
- //获取指标信息
- edbInfo, err := data_manage.GetEdbInfoById(req.EdbInfoId)
- if err != nil {
- if err.Error() == utils.ErrNoRow() {
- br.Msg = "指标已被删除,请刷新页面"
- br.ErrMsg = "指标已被删除,请刷新页面:Err:" + err.Error()
- return
- }
- br.Msg = "获取指标信息失败"
- br.ErrMsg = "获取指标信息失败:Err:" + err.Error()
- return
- }
- // 修改python运算指标
- respItem, err := data.EditPythonEdbData(string(reqJson))
- if err != nil {
- br.Msg = "保存失败"
- br.ErrMsg = "保存失败,Err:" + err.Error()
- return
- }
- if respItem.Ret != 200 {
- br.Msg = "保存失败"
- br.ErrMsg = respItem.ErrMsg + ";EdbName:" + req.EdbName
- return
- }
- //添加es
- data.AddOrEditEdbInfoToEs(req.EdbInfoId)
- resp := new(data_manage.AddEdbInfoResp)
- resp.EdbInfoId = req.EdbInfoId
- resp.UniqueCode = edbInfo.UniqueCode
- br.Ret = 200
- br.Success = true
- br.Data = resp
- br.Msg = "保存成功"
- br.IsAddLog = true
- }
- // GetPythonEdbCode
- // @Title 获取python代码
- // @Description 获取python代码接口
- // @Param EdbInfoId query int true "指标id"
- // @Success 200 {object} data_manage.PythonEdbDetailResp
- // @router /edb_info/python/detail [get]
- func (this *ChartInfoController) GetPythonEdbCode() {
- 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
- }
- edbInfoId, _ := this.GetInt("EdbInfoId")
- edbInfo, err := data_manage.GetEdbInfoById(edbInfoId)
- if err != nil {
- br.Msg = "获取失败"
- br.ErrMsg = "获取失败,Err:" + err.Error()
- return
- }
- if edbInfo.Source != utils.DATA_SOURCE_PYTHON {
- br.Msg = "该指标不是Python运算指标"
- br.ErrMsg = "该指标不是Python运算指标"
- return
- }
- pythonCode, err := data_manage.GetEdbPythonCodeById(edbInfoId)
- if err != nil {
- br.Msg = "获取失败"
- br.ErrMsg = "获取失败,Err:" + err.Error()
- return
- }
- resp := data_manage.PythonEdbDetailResp{
- EdbInfoDetail: edbInfo,
- PythonCode: pythonCode.PythonCode,
- }
- br.Ret = 200
- br.Success = true
- br.Msg = "保存成功"
- br.Data = resp
- }
- // QueryEdbDataTable
- // @Title 指标筛选接口
- // @Description 指标筛选接口
- // @Param EdbInfoId query int false "指标id"
- // @Success 200 {object} data_manage.EdbInfoList
- // @router /edb_data/table/detail [get]
- func (this *EdbInfoController) QueryEdbDataTable() {
- br := new(models.BaseResponse).Init()
- defer func() {
- this.Data["json"] = br
- this.ServeJSON()
- }()
- edbInfoId, _ := this.GetInt("EdbInfoId")
- if edbInfoId <= 0 {
- br.Msg = "请选择指标"
- br.ErrMsg = "请选择指标"
- return
- }
- edbInfo, err := data_manage.GetEdbInfoById(edbInfoId)
- if err != nil {
- br.Msg = "获取指标信息失败"
- br.ErrMsg = "获取指标信息失败,err:" + err.Error()
- return
- }
- tableName := data_manage.GetEdbDataTableName(edbInfo.Source)
- columnList := []map[string]string{
- {
- "edb_code": "指标code",
- },
- {
- "data_time": "指标日期",
- },
- {
- "value": "指标对应的值",
- },
- }
- templateStr := fmt.Sprintf("sql1 = f\"\"\"SELECT data_time,`value` FROM %s WHERE edb_code = '%s' ORDER BY data_time DESC;\"\"\"\nraw = pandas_fetch_all(sql1, db)", tableName, edbInfo.EdbCode)
- info := data_manage.TableInfoResp{
- ColumnList: columnList,
- TableName: tableName,
- TemplateStr: url.PathEscape(templateStr),
- }
- br.Ret = 200
- br.Success = true
- br.Msg = "获取成功"
- br.Data = info
- }
- // SaveAdjustEdbInfo
- // @Title 保存数据调整接口
- // @Description 保存数据调整接口
- // @Param request body data_manage.AddPythonEdbReq true "type json string"
- // @Success 200 {object} data_manage.AddEdbInfoResp
- // @router /edb_info/adjust/save [post]
- func (this *ChartInfoController) SaveAdjustEdbInfo() {
- 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
- }
- var req data_manage.SaveAdjustEdbInfoReq
- err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
- if err != nil {
- br.Msg = "参数解析异常!"
- br.ErrMsg = "参数解析失败,Err:" + err.Error()
- return
- }
- //传入操作人
- req.AdminId = sysUser.AdminId
- req.AdminName = sysUser.RealName
- reqJson, err := json.Marshal(req)
- if err != nil {
- br.Msg = "参数解析异常!"
- br.ErrMsg = "参数解析失败,Err:" + err.Error()
- return
- }
- // 判断指标名称是否重复
- {
- var condition string
- var pars []interface{}
- condition += " AND edb_name=? "
- pars = append(pars, req.EdbName)
- condition += " AND edb_info_id<>? "
- pars = append(pars, req.EdbInfoId)
- count, err := data_manage.GetEdbInfoCountByCondition(condition, pars)
- if err != nil {
- br.Msg = "判断指标名称是否存在失败"
- br.ErrMsg = "判断指标名称是否存在失败,Err:" + err.Error()
- return
- }
- if count > 0 {
- br.Msg = "指标名称已存在,请重新填写"
- br.ErrMsg = "指标名称已存在,请重新填写"
- br.IsSendEmail = false
- return
- }
- }
- // 添加数据调整指标
- respItem, err := data.SaveAdjustEdbInfo(string(reqJson))
- if err != nil {
- br.Msg = "添加失败"
- br.ErrMsg = "获取失败,Err:" + err.Error()
- return
- }
- if respItem.Ret != 200 {
- br.Msg = "添加失败"
- br.ErrMsg = respItem.ErrMsg + ";EdbName:" + req.EdbName
- return
- }
- respData := respItem.Data.(map[string]interface{})
- tmpEdbInfoId, ok := respData["EdbInfoId"]
- if !ok {
- br.Msg = `数据转换异常`
- br.ErrMsg = fmt.Sprint("respData.EdbInfoId 数据转换异常;respData:", respData)
- return
- }
- //指标id
- edbInfoId := int64(tmpEdbInfoId.(float64))
- uniqueCode, ok := respData["UniqueCode"]
- if !ok {
- br.Msg = `数据转换异常`
- br.ErrMsg = fmt.Sprint("respData.UniqueCode 数据转换异常;respData:", respData)
- return
- }
- //添加es
- data.AddOrEditEdbInfoToEs(int(edbInfoId))
- //
- resp := data_manage.AddEdbInfoResp{
- EdbInfoId: int(edbInfoId),
- UniqueCode: uniqueCode.(string),
- }
- br.Ret = 200
- br.Success = true
- br.Msg = "保存成功"
- br.Data = resp
- br.IsAddLog = true
- }
- // BatchCalculateBatchSave
- // @Title 批量添加 累计值转月-同比值-同差等计算新增
- // @Description 批量添加 累计值转月-同比值-同差等计算新增接口
- // @Param request body data_manage.EdbInfoCalculateBatchSaveReq true "type json string"
- // @Success Ret=200 返回指标id
- // @router /edb_info/calculate/batch/save/batch [post]
- func (this *ChartInfoController) BatchCalculateBatchSave() {
- 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
- }
- var reqList []data_manage.BatchEdbInfoCalculateBatchSaveReq
- err := json.Unmarshal(this.Ctx.Input.RequestBody, &reqList)
- if err != nil {
- br.Msg = "参数解析异常!"
- br.ErrMsg = "参数解析失败,Err:" + err.Error()
- return
- }
- resp := data_manage.BatchEdbInfoCalculateBatchSaveResp{
- Fail: make([]data_manage.BatchEdbInfoCalculateBatchSaveFailResp, 0),
- Success: make([]data_manage.BatchEdbInfoCalculateBatchSaveSuccessResp, 0),
- }
- redisKeyList := make([]string, 0) //需要清理的缓存key列表
- defer func() {
- for _, redisKey := range redisKeyList {
- utils.Rc.Delete(redisKey)
- }
- }()
- for _, v := range reqList {
- req := v.CalculateInfo
- req.EdbName = strings.Trim(req.EdbName, " ")
- if req.EdbName == "" {
- resp.Fail = append(resp.Fail, data_manage.BatchEdbInfoCalculateBatchSaveFailResp{
- CalculateId: v.CalculateId,
- Msg: "指标名称不能为空",
- })
- continue
- }
- if req.Frequency == "" {
- resp.Fail = append(resp.Fail, data_manage.BatchEdbInfoCalculateBatchSaveFailResp{
- CalculateId: v.CalculateId,
- Msg: "频率不能为空",
- })
- continue
- }
- if req.Unit == "" {
- resp.Fail = append(resp.Fail, data_manage.BatchEdbInfoCalculateBatchSaveFailResp{
- CalculateId: v.CalculateId,
- Msg: "单位不能为空",
- })
- continue
- }
- if req.ClassifyId <= 0 {
- resp.Fail = append(resp.Fail, data_manage.BatchEdbInfoCalculateBatchSaveFailResp{
- CalculateId: v.CalculateId,
- Msg: "请选择分类",
- })
- continue
- }
- fromEdbInfoId := req.FromEdbInfoId
- var formulaInt int
- switch req.Source {
- case utils.DATA_SOURCE_CALCULATE_NSZYDPJJS, utils.DATA_SOURCE_CALCULATE_HBZ, utils.DATA_SOURCE_CALCULATE_HCZ, utils.DATA_SOURCE_CALCULATE_TIME_SHIFT, utils.DATA_SOURCE_CALCULATE_CJJX:
- if req.Formula == "" {
- resp.Fail = append(resp.Fail, data_manage.BatchEdbInfoCalculateBatchSaveFailResp{
- CalculateId: v.CalculateId,
- Msg: "请选择分类",
- })
- continue
- }
- formulaInt, _ = strconv.Atoi(req.Formula)
- if formulaInt <= 0 {
- resp.Fail = append(resp.Fail, data_manage.BatchEdbInfoCalculateBatchSaveFailResp{
- CalculateId: v.CalculateId,
- Msg: "N值输入错误,请重新输入",
- })
- continue
- }
- case utils.DATA_SOURCE_CALCULATE_ZJPJ:
- //直接拼接指标
- //校验时间格式
- _, err = time.ParseInLocation(utils.FormatDate, req.Formula, time.Local)
- if err != nil {
- resp.Fail = append(resp.Fail, data_manage.BatchEdbInfoCalculateBatchSaveFailResp{
- CalculateId: v.CalculateId,
- Msg: "拼接日期有误,请重新输入",
- })
- continue
- }
- case utils.DATA_SOURCE_CALCULATE_NHCC: //拟合残差指标
- //指标校验
- if len(req.EdbInfoIdArr) != 2 {
- resp.Fail = append(resp.Fail, data_manage.BatchEdbInfoCalculateBatchSaveFailResp{
- CalculateId: v.CalculateId,
- Msg: "选择的指标异常,请重新选择",
- })
- continue
- }
- fromEdbInfoId = req.EdbInfoIdArr[0].EdbInfoId
- //校验时间格式
- //数据格式:2022-11-01,2022-11-10
- timeList := strings.Split(req.Formula, ",")
- if len(timeList) != 2 {
- resp.Fail = append(resp.Fail, data_manage.BatchEdbInfoCalculateBatchSaveFailResp{
- CalculateId: v.CalculateId,
- Msg: "选择时间有误,请重新输入",
- })
- continue
- }
- startDate, err := time.ParseInLocation(utils.FormatDate, timeList[0], time.Local)
- if err != nil {
- resp.Fail = append(resp.Fail, data_manage.BatchEdbInfoCalculateBatchSaveFailResp{
- CalculateId: v.CalculateId,
- Msg: "开始日期有误,请重新输入",
- })
- continue
- }
- endDate, err := time.ParseInLocation(utils.FormatDate, timeList[1], time.Local)
- if err != nil {
- resp.Fail = append(resp.Fail, data_manage.BatchEdbInfoCalculateBatchSaveFailResp{
- CalculateId: v.CalculateId,
- Msg: "结束日期有误,请重新输入",
- })
- continue
- }
- if utils.GetTimeSubDay(startDate, endDate) < 2 {
- resp.Fail = append(resp.Fail, data_manage.BatchEdbInfoCalculateBatchSaveFailResp{
- CalculateId: v.CalculateId,
- Msg: "日期间隔不得少于两天",
- })
- continue
- }
- }
- if fromEdbInfoId <= 0 {
- resp.Fail = append(resp.Fail, data_manage.BatchEdbInfoCalculateBatchSaveFailResp{
- CalculateId: v.CalculateId,
- Msg: "请选择指标",
- })
- continue
- }
- //加入缓存机制,避免创建同一个名称的指标 start
- redisKey := fmt.Sprint("edb_info:calculate:batch:save:", sysUser.AdminId, ":", req.Source, ":", v.CalculateId)
- isExist := utils.Rc.IsExist(redisKey)
- if isExist {
- resp.Fail = append(resp.Fail, data_manage.BatchEdbInfoCalculateBatchSaveFailResp{
- CalculateId: v.CalculateId,
- Msg: "指标正在处理,请勿重复提交",
- })
- continue
- } else {
- //设置3分钟缓存
- utils.Rc.SetNX(redisKey, 1, time.Second*300)
- redisKeyList = append(redisKeyList, redisKey)
- }
- //加入缓存机制,避免创建同一个名称的指标 end
- req2 := &data_manage.EdbInfoCalculateBatchSaveReqByEdbLib{
- AdminId: sysUser.AdminId,
- AdminName: sysUser.RealName,
- EdbInfoId: req.EdbInfoId,
- EdbName: req.EdbName,
- Frequency: req.Frequency,
- Unit: req.Unit,
- ClassifyId: req.ClassifyId,
- Formula: req.Formula, //N数值移动平均计算、环比值、环差值
- FromEdbInfoId: req.FromEdbInfoId,
- CalculateFormula: req.CalculateFormula,
- Source: req.Source,
- MoveType: req.MoveType,
- MoveFrequency: req.MoveFrequency,
- //CalculateFormula: edbInfo.CalculateFormula,
- EdbInfoIdArr: req.EdbInfoIdArr,
- Calendar: req.Calendar,
- }
- // 调用指标库去更新
- reqJson, err := json.Marshal(req2)
- if err != nil {
- resp.Fail = append(resp.Fail, data_manage.BatchEdbInfoCalculateBatchSaveFailResp{
- CalculateId: v.CalculateId,
- Msg: "参数解析异常!",
- ErrMsg: "参数解析失败,Err:" + err.Error(),
- })
- continue
- }
- respItem, err := data.BatchSaveEdbCalculateData(string(reqJson))
- if err != nil {
- resp.Fail = append(resp.Fail, data_manage.BatchEdbInfoCalculateBatchSaveFailResp{
- CalculateId: v.CalculateId,
- Msg: "新增失败!",
- ErrMsg: "新增失败,Err:" + err.Error(),
- })
- continue
- }
- if respItem.Ret != 200 {
- resp.Fail = append(resp.Fail, data_manage.BatchEdbInfoCalculateBatchSaveFailResp{
- CalculateId: v.CalculateId,
- Msg: respItem.Msg,
- ErrMsg: respItem.ErrMsg,
- })
- continue
- }
- result := respItem.Data
- //添加es
- data.AddOrEditEdbInfoToEs(result.EdbInfoId)
- resp.Success = append(resp.Success, data_manage.BatchEdbInfoCalculateBatchSaveSuccessResp{
- ClassifyId: req.ClassifyId,
- CalculateId: v.CalculateId,
- EdbInfoId: result.EdbInfoId,
- UniqueCode: result.UniqueCode,
- })
- }
- br.Ret = 200
- br.Success = true
- br.Msg = "批量保存成功"
- br.Data = resp
- br.IsAddLog = true
- }
- // CalculateMapping
- // @Title 获取计算指标与基础指标关联
- // @Description 获取计算指标与基础指标关联
- // @Param EdbInfoId query int true "指标id"
- // @Success 200 {object} data_manage.CalculateDetailResp
- // @router /edb_info/calculate/mapping [get]
- func (this *ChartInfoController) CalculateMapping() {
- 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
- }
- edbInfoId, _ := this.GetInt("EdbInfoId")
- if edbInfoId <= 0 {
- br.Msg = "参数有误"
- return
- }
- item, e := data_manage.GetEdbInfoCalculateMappingDetail(edbInfoId)
- if e != nil {
- br.Msg = "获取失败"
- br.ErrMsg = "获取失败,Err:" + e.Error()
- return
- }
- br.Ret = 200
- br.Success = true
- br.Msg = "保存成功"
- br.Data = item
- }
|