package line_equation import ( "encoding/json" "eta/eta_api/controllers" "eta/eta_api/models" "eta/eta_api/models/data_manage" "eta/eta_api/models/system" "eta/eta_api/services/data" "eta/eta_api/utils" "strconv" "time" ) // LineEquationChartClassifyController 拟合方程图表分类 type LineEquationChartClassifyController struct { controllers.BaseAuthController } // ChartClassifyList // @Title 拟合方程图表分类列表 // @Description 拟合方程图表分类列表接口 // @Param IsShowMe query bool true "是否只看我的,true、false" // @Param Source query int true "图表类型,3:拟合方程,4:滚动拟合方程" // @Success 200 {object} data_manage.ChartClassifyListResp // @router /chart_classify/list [get] func (this *LineEquationChartClassifyController) ChartClassifyList() { br := new(models.BaseResponse).Init() defer func() { this.Data["json"] = br this.ServeJSON() }() resp := new(data_manage.ChartClassifyListResp) // 获取当前账号的不可见指标 noPermissionChartIdMap := make(map[int]bool) { obj := data_manage.EdbInfoNoPermissionAdmin{} confList, err := obj.GetAllChartListByAdminId(this.SysUser.AdminId) if err != nil && err.Error() != utils.ErrNoRow() { br.Msg = "获取失败" br.ErrMsg = "获取不可见指标配置数据失败,Err:" + err.Error() return } for _, v := range confList { noPermissionChartIdMap[v.ChartInfoId] = true } } isShowMe, _ := this.GetBool("IsShowMe") if isShowMe { errMsg, err := getChartClassifyListForMe(*this.SysUser, resp) if err != nil { br.Msg = errMsg br.ErrMsg = err.Error() return } br.Ret = 200 br.Success = true br.Msg = "获取成功" br.Data = resp return } source := utils.CHART_SOURCE_LINE_EQUATION rootList, err := data_manage.GetChartClassifyByParentId(0, source) if err != nil && err.Error() != utils.ErrNoRow() { br.Msg = "获取失败" br.ErrMsg = "获取数据失败,Err:" + err.Error() return } allChartInfo, err := data_manage.GetChartInfoAll([]int{source}) if err != nil && err.Error() != utils.ErrNoRow() { br.Msg = "获取失败" br.ErrMsg = "获取图表信息失败,Err:" + err.Error() return } chartInfoMap := make(map[int][]*data_manage.ChartClassifyItems) for _, v := range allChartInfo { if !isShowMe { chartInfoMap[v.ChartClassifyId] = append(chartInfoMap[v.ChartClassifyId], v) continue } if v.SysUserId != this.SysUser.AdminId { continue } chartInfoMap[v.ChartClassifyId] = append(chartInfoMap[v.ChartClassifyId], v) } rootChildMap := make(map[int][]*data_manage.ChartClassifyItems) for _, v := range rootList { rootChildMap[v.ParentId] = append(rootChildMap[v.ParentId], v) if existItems, ok := chartInfoMap[v.ChartClassifyId]; ok { v.Children = existItems } else { items := make([]*data_manage.ChartClassifyItems, 0) v.Children = items } } // 移除没有权限的图表 allNodes := handleNoPermissionChart(rootList, noPermissionChartIdMap) resp.AllNodes = allNodes br.Ret = 200 br.Success = true br.Msg = "获取成功" br.Data = resp } // getChartClassifyListForMe 获取我创建的图表 func getChartClassifyListForMe(adminInfo system.Admin, resp *data_manage.ChartClassifyListResp) (errMsg string, err error) { // 获取所有的分类 rootList, err := data_manage.GetChartClassifyByParentId(0, utils.CHART_SOURCE_LINE_EQUATION) if err != nil && err.Error() != utils.ErrNoRow() { errMsg = "获取失败" return } // /获取所有的图表 allChartInfo, err := data_manage.GetChartInfoByAdminId([]int{utils.CHART_SOURCE_LINE_EQUATION}, adminInfo.AdminId) if err != nil && err.Error() != utils.ErrNoRow() { errMsg = "获取失败" return } chartInfoMap := make(map[int][]*data_manage.ChartClassifyItems) for _, v := range allChartInfo { chartInfoMap[v.ChartClassifyId] = append(chartInfoMap[v.ChartClassifyId], v) } for k, v := range rootList { if existItems, ok := chartInfoMap[v.ChartClassifyId]; ok { v.Children = existItems } else { items := make([]*data_manage.ChartClassifyItems, 0) v.Children = items } rootList[k] = v } resp.AllNodes = rootList return } // handleNoPermissionChart 图表列表返回,将没有权限的图表移除 func handleNoPermissionChart(allNodes []*data_manage.ChartClassifyItems, noPermissionChartIdMap map[int]bool) (newAllNodes []*data_manage.ChartClassifyItems) { // 移除没有权限的图表 newAllNodes = make([]*data_manage.ChartClassifyItems, 0) for _, node := range allNodes { // 二级分类 tmpNodeInfo := *node tmpNodeList := make([]*data_manage.ChartClassifyItems, 0) if node.Children != nil { for _, chartInfo := range node.Children { // 如果指标不可见,那么就不返回该指标 if _, ok := noPermissionChartIdMap[chartInfo.ChartInfoId]; ok { continue } tmpNodeList = append(tmpNodeList, chartInfo) } } tmpNodeInfo.Children = tmpNodeList newAllNodes = append(newAllNodes, &tmpNodeInfo) } return } // ChartClassifyItems // @Title 获取所有拟合方程图表分类接口-不包含图表 // @Description 获取所有拟合方程图表分类接口-不包含图表 // @Success 200 {object} data_manage.ChartClassifyListResp // @router /chart_classify/items [get] func (this *LineEquationChartClassifyController) ChartClassifyItems() { br := new(models.BaseResponse).Init() defer func() { this.Data["json"] = br this.ServeJSON() }() rootList, err := data_manage.GetChartClassifyByParentId(0, utils.CHART_SOURCE_LINE_EQUATION) if err != nil { br.Msg = "获取失败" br.ErrMsg = "获取数据失败,Err:" + err.Error() return } nodeAll := make([]*data_manage.ChartClassifyItems, 0) for k := range rootList { rootNode := rootList[k] nodeAll = append(nodeAll, rootNode) } resp := new(data_manage.ChartClassifyListResp) resp.AllNodes = nodeAll br.Ret = 200 br.Success = true br.Msg = "获取成功" br.Data = resp } // AddChartClassify // @Title 新增拟合方程图表分类 // @Description 新增拟合方程图表分类接口 // @Param request body data_manage.AddChartClassifyReq true "type json string" // @Success 200 Ret=200 保存成功 // @router /chart_classify/add [post] func (this *LineEquationChartClassifyController) AddChartClassify() { br := new(models.BaseResponse).Init() defer func() { this.Data["json"] = br this.ServeJSON() }() var req data_manage.AddChartClassifyReq err := json.Unmarshal(this.Ctx.Input.RequestBody, &req) if err != nil { br.Msg = "参数解析异常!" br.ErrMsg = "参数解析失败,Err:" + err.Error() return } if req.ChartClassifyName == "" { br.Msg = "请输入分类名称" br.IsSendEmail = false return } if req.ParentId < 0 { br.Msg = "参数错误" br.IsSendEmail = false return } source := utils.CHART_SOURCE_LINE_EQUATION count, err := data_manage.GetChartClassifyCount(req.ChartClassifyName, req.ParentId, source) if err != nil { br.Msg = "判断名称是否已存在失败" br.ErrMsg = "判断名称是否已存在失败,Err:" + err.Error() return } if count > 0 { br.Msg = "分类名称已存在,请重新输入" br.IsSendEmail = false return } //获取该层级下最大的排序数 maxSort, err := data_manage.GetChartClassifyMaxSort(req.ParentId, source) classify := new(data_manage.ChartClassify) //classify.ParentId = req.ParentId classify.ParentId = 0 classify.ChartClassifyName = req.ChartClassifyName classify.HasData = 0 classify.CreateTime = time.Now() classify.ModifyTime = time.Now() classify.SysUserId = this.SysUser.AdminId classify.SysUserRealName = this.SysUser.RealName classify.Level = req.Level + 1 timestamp := strconv.FormatInt(time.Now().UnixNano(), 10) classify.UniqueCode = utils.MD5(utils.DATA_PREFIX + "_" + timestamp) classify.Sort = maxSort + 1 classify.Source = source _, err = data_manage.AddChartClassify(classify) if err != nil { br.Msg = "保存分类失败" br.ErrMsg = "保存分类失败,Err:" + err.Error() return } br.Ret = 200 br.Msg = "添加成功" br.Success = true } // EditChartClassify // @Title 修改拟合方程图表分类 // @Description 修改拟合方程图表分类接口 // @Param request body data_manage.EditChartClassifyReq true "type json string" // @Success 200 Ret=200 修改成功 // @router /chart_classify/edit [post] func (this *LineEquationChartClassifyController) EditChartClassify() { br := new(models.BaseResponse).Init() defer func() { this.Data["json"] = br this.ServeJSON() }() var req data_manage.EditChartClassifyReq err := json.Unmarshal(this.Ctx.Input.RequestBody, &req) if err != nil { br.Msg = "参数解析异常!" br.ErrMsg = "参数解析失败,Err:" + err.Error() return } if req.ChartClassifyName == "" { br.Msg = "请输入分类名称" br.IsSendEmail = false return } if req.ChartClassifyId <= 0 { br.Msg = "参数错误" br.IsSendEmail = false return } item, err := data_manage.GetChartClassifyById(req.ChartClassifyId) if err != nil { br.Msg = "保存失败" br.Msg = "获取分类信息失败,Err:" + err.Error() return } source := utils.CHART_SOURCE_LINE_EQUATION if item.ChartClassifyName != req.ChartClassifyName { count, err := data_manage.GetChartClassifyCount(req.ChartClassifyName, item.ParentId, source) if err != nil { br.Msg = "判断名称是否已存在失败" br.ErrMsg = "判断名称是否已存在失败,Err:" + err.Error() return } if count > 0 { br.Msg = "分类名称已存在,请重新输入" br.IsSendEmail = false return } err = data_manage.EditChartClassify(req.ChartClassifyId, source, req.ChartClassifyName) if err != nil { br.Msg = "保存失败" br.ErrMsg = "保存失败,Err:" + err.Error() return } } br.Ret = 200 br.Msg = "修改成功" br.Success = true br.IsAddLog = true } // DeleteChartClassifyCheck // @Title 删除图表检测接口 // @Description 删除图表检测接口 // @Param request body data_manage.ChartClassifyDeleteCheckResp true "type json string" // @Success 200 Ret=200 检测成功 // @router /chart_classify/delete/check [post] func (this *LineEquationChartClassifyController) DeleteChartClassifyCheck() { br := new(models.BaseResponse).Init() defer func() { this.Data["json"] = br this.ServeJSON() }() var req data_manage.ChartClassifyDeleteCheckReq err := json.Unmarshal(this.Ctx.Input.RequestBody, &req) if err != nil { br.Msg = "参数解析异常!" br.ErrMsg = "参数解析失败,Err:" + err.Error() return } if req.ChartClassifyId < 0 && req.ChartInfoId <= 0 { br.Msg = "参数错误" br.IsSendEmail = false return } var deleteStatus int var tipsMsg string //删除分类 if req.ChartClassifyId > 0 && req.ChartInfoId == 0 { //判断拟合方程图表分类下,是否含有图表 count, err := data_manage.GetChartInfoCountByClassifyId(req.ChartClassifyId) if err != nil { br.Msg = "删除失败" br.ErrMsg = "分类下是否含有图表失败,Err:" + err.Error() return } if count > 0 { deleteStatus = 1 tipsMsg = "该分类下关联图表不可删除" } } if deleteStatus != 1 && req.ChartInfoId == 0 { classifyCount, err := data_manage.GetChartClassifyCountByClassifyId(req.ChartClassifyId) if err != nil && err.Error() != utils.ErrNoRow() { br.Msg = "删除失败" br.ErrMsg = "分类下是否含有图表失败,Err:" + err.Error() return } if classifyCount > 0 { deleteStatus = 2 tipsMsg = "确认删除当前目录及包含的子目录吗" } } if deleteStatus == 0 { tipsMsg = "可删除,进行删除操作" } resp := new(data_manage.ChartClassifyDeleteCheckResp) resp.DeleteStatus = deleteStatus resp.TipsMsg = tipsMsg br.Ret = 200 br.Msg = "检测成功" br.Success = true br.Data = resp } // DeleteChartClassify // @Title 删除拟合方程图表分类/图表 // @Description 删除拟合方程图表分类/图表接口 // @Param request body data_manage.DeleteChartClassifyReq true "type json string" // @Success 200 Ret=200 删除成功 // @router /chart_classify/delete [post] func (this *LineEquationChartClassifyController) DeleteChartClassify() { 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.DeleteChartClassifyReq err := json.Unmarshal(this.Ctx.Input.RequestBody, &req) if err != nil { br.Msg = "参数解析异常!" br.ErrMsg = "参数解析失败,Err:" + err.Error() return } if req.ChartClassifyId < 0 && req.ChartInfoId <= 0 { br.Msg = "参数错误" br.IsSendEmail = false return } //删除分类 if req.ChartClassifyId > 0 && req.ChartInfoId == 0 { //判断是否含有指标 count, err := data_manage.GetChartInfoCountByClassifyId(req.ChartClassifyId) if err != nil && err.Error() != utils.ErrNoRow() { br.Msg = "删除失败" br.ErrMsg = "判断名称是否已存在失败,Err:" + err.Error() return } if count > 0 { br.Msg = "该目录下存在关联指标,不可删除" br.IsSendEmail = false return } err = data_manage.DeleteChartClassify(req.ChartClassifyId) if err != nil { br.Msg = "删除失败" br.ErrMsg = "删除失败,Err:" + err.Error() return } } resp := new(data_manage.AddChartInfoResp) //删除图表 if req.ChartInfoId > 0 { chartInfo, err := data_manage.GetChartInfoById(req.ChartInfoId) if err != nil { if err.Error() == utils.ErrNoRow() { br.Msg = "图表已删除,请刷新页面" br.ErrMsg = "指标不存在,Err:" + err.Error() return } else { br.Msg = "删除失败" br.ErrMsg = "删除失败,获取指标信息失败,Err:" + err.Error() return } } if chartInfo == nil { br.Msg = "图表已删除,请刷新页面" return } //图表操作权限 ok := data.CheckOpChartPermission(sysUser, chartInfo.SysUserId) if !ok { br.Msg = "没有该图表的操作权限" br.ErrMsg = "没有该图表的操作权限" return } myIds := make([]int, 0) { // 获取引用该图表的MyCharts, 用于ES删除 var myCond string var myPars []interface{} myCond += ` AND a.chart_info_id = ? ` myPars = append(myPars, req.ChartInfoId) myCharts, e := data_manage.GetMyChartListGroupByCharyInfoIdAndAdminIdByCondition(myCond, myPars) if e != nil { br.Msg = "删除失败" br.ErrMsg = "获取引用图表的MyChats失败, Err: " + e.Error() return } for _, m := range myCharts { myIds = append(myIds, m.MyChartId) } } //删除图表及关联指标 err = data_manage.DeleteChartInfoAndData(chartInfo.ChartInfoId) if err != nil { br.Msg = "删除失败" br.ErrMsg = "删除失败,Err:" + err.Error() return } //删除ES { go data.EsDeleteChartInfo(chartInfo.ChartInfoId) // 删除MY ETA 图表 es数据 go data.EsDeleteMyChartInfoByMyChartIds(myIds) } source := chartInfo.Source // 拟合方程图表 var condition string var pars []interface{} condition += " AND chart_classify_id=? AND source = ? " pars = append(pars, chartInfo.ChartClassifyId, source) condition += " AND chart_info_id>? ORDER BY create_time ASC LIMIT 1 " pars = append(pars, req.ChartInfoId) nextItem, err := data_manage.GetChartInfoByCondition(condition, pars) if err != nil && err.Error() != utils.ErrNoRow() { br.Msg = "删除失败" br.ErrMsg = "获取下一级图库信息失败,Err:" + err.Error() return } if nextItem != nil { resp.UniqueCode = nextItem.UniqueCode resp.ChartInfoId = nextItem.ChartInfoId } else { var condition string var pars []interface{} condition += " AND level=1 " //pars = append(pars, chartInfo.ChartClassifyId) condition += " AND chart_classify_id>? ORDER BY chart_classify_id ASC LIMIT 1 " pars = append(pars, chartInfo.ChartClassifyId) classifyItem, err := data_manage.GetChartClassifyByCondition(condition, pars) if err != nil && err.Error() != utils.ErrNoRow() { br.Msg = "删除失败" br.ErrMsg = "获取下一级图库分类信息失败,Err:" + err.Error() return } if classifyItem != nil { nextItem, err = data_manage.GetNextChartInfo(chartInfo.ChartClassifyId) if err != nil && err.Error() != utils.ErrNoRow() { br.Msg = "删除失败" br.ErrMsg = "获取下一级图库信息失败,Err:" + err.Error() return } if nextItem != nil { resp.UniqueCode = nextItem.UniqueCode resp.ChartInfoId = nextItem.ChartInfoId } } } //新增操作日志 { chartLog := new(data_manage.ChartInfoLog) chartLog.ChartName = chartInfo.ChartName chartLog.ChartInfoId = req.ChartInfoId chartLog.ChartClassifyId = chartInfo.ChartClassifyId chartLog.SysUserId = sysUser.AdminId chartLog.SysUserRealName = sysUser.RealName chartLog.UniqueCode = chartInfo.UniqueCode chartLog.CreateTime = time.Now() chartLog.Content = string(this.Ctx.Input.RequestBody) chartLog.Status = "删除图表" chartLog.Method = this.Ctx.Input.URI() go data_manage.AddChartInfoLog(chartLog) } } br.Ret = 200 br.Msg = "删除成功" br.Success = true br.Data = resp br.IsAddLog = true } // ChartClassifyMove // @Title 拟合方程图表分类移动接口 // @Description 拟合方程图表分类移动接口 // @Success 200 {object} data_manage.MoveChartClassifyReq // @router /chart_classify/move [post] func (this *LineEquationChartClassifyController) ChartClassifyMove() { 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.MoveChartClassifyReq err := json.Unmarshal(this.Ctx.Input.RequestBody, &req) if err != nil { br.Msg = "参数解析异常!" br.ErrMsg = "参数解析失败,Err:" + err.Error() return } if req.ClassifyId <= 0 { br.Msg = "参数错误" br.ErrMsg = "分类id小于等于0" return } //判断分类是否存在 chartClassifyInfo, err := data_manage.GetChartClassifyById(req.ClassifyId) if err != nil { br.Msg = "移动失败" br.ErrMsg = "获取分类信息失败,Err:" + err.Error() return } if chartClassifyInfo.Source != utils.CHART_SOURCE_LINE_EQUATION { br.Msg = "分类异常" br.ErrMsg = "分类异常,不是拟合方程图表的分类" return } updateCol := make([]string, 0) //判断上级id是否一致,如果不一致的话,那么需要移动该分类层级 if chartClassifyInfo.ParentId != req.ParentClassifyId && req.ParentClassifyId != 0 { parentChartClassifyInfo, err := data_manage.GetChartClassifyById(req.ParentClassifyId) if err != nil { br.Msg = "移动失败" br.ErrMsg = "获取上级分类信息失败,Err:" + err.Error() return } chartClassifyInfo.ParentId = parentChartClassifyInfo.ChartClassifyId chartClassifyInfo.Level = parentChartClassifyInfo.Level + 1 chartClassifyInfo.ModifyTime = time.Now() updateCol = append(updateCol, "ParentId", "Level", "ModifyTime") } //如果有传入 上一个兄弟节点分类id if req.PrevClassifyId > 0 { //上一个兄弟节点 prevClassify, err := data_manage.GetChartClassifyById(req.PrevClassifyId) if err != nil { br.Msg = "移动失败" br.ErrMsg = "获取上一个兄弟节点分类信息失败,Err:" + err.Error() return } //如果是移动在两个兄弟节点之间 if req.NextClassifyId > 0 { //下一个兄弟节点 nextClassify, err := data_manage.GetChartClassifyById(req.NextClassifyId) if err != nil { br.Msg = "移动失败" br.ErrMsg = "获取下一个兄弟节点分类信息失败,Err:" + err.Error() return } //如果上一个兄弟与下一个兄弟的排序权重是一致的,那么需要将下一个兄弟(以及下个兄弟的同样排序权重)的排序权重+2,自己变成上一个兄弟的排序权重+1 if prevClassify.Sort == nextClassify.Sort || prevClassify.Sort == chartClassifyInfo.Sort { //变更兄弟节点的排序 updateSortStr := `sort + 2` _ = data_manage.UpdateChartClassifySortByParentId(prevClassify.ParentId, prevClassify.ChartClassifyId, prevClassify.Sort, updateSortStr) } else { //如果下一个兄弟的排序权重正好是上个兄弟节点的下一层,那么需要再加一层了 if nextClassify.Sort-prevClassify.Sort == 1 { //变更兄弟节点的排序 updateSortStr := `sort + 1` _ = data_manage.UpdateChartClassifySortByParentId(prevClassify.ParentId, 0, prevClassify.Sort, updateSortStr) } } } chartClassifyInfo.Sort = prevClassify.Sort + 1 chartClassifyInfo.ModifyTime = time.Now() updateCol = append(updateCol, "Sort", "ModifyTime") } else { firstClassify, err := data_manage.GetFirstChartClassifyByParentId(chartClassifyInfo.ParentId) if err != nil && err.Error() != utils.ErrNoRow() { br.Msg = "移动失败" br.ErrMsg = "获取获取当前父级分类下的排序第一条的分类信息失败,Err:" + err.Error() return } //如果该分类下存在其他分类,且第一个其他分类的排序等于0,那么需要调整排序 if firstClassify != nil && firstClassify.Sort == 0 { updateSortStr := ` sort + 1 ` _ = data_manage.UpdateChartClassifySortByParentId(firstClassify.ParentId, firstClassify.ChartClassifyId-1, 0, updateSortStr) } chartClassifyInfo.Sort = 0 //那就是排在第一位 chartClassifyInfo.ModifyTime = time.Now() updateCol = append(updateCol, "Sort", "ModifyTime") } //更新 if len(updateCol) > 0 { err = chartClassifyInfo.Update(updateCol) if err != nil { br.Msg = "移动失败" br.ErrMsg = "修改失败,Err:" + err.Error() return } } br.Ret = 200 br.Success = true br.Msg = "移动成功" }