package controllers import ( "encoding/json" "eta/eta_api/models" "eta/eta_api/models/report" "eta/eta_api/models/report_approve" "eta/eta_api/models/smart_report" "eta/eta_api/models/system" "eta/eta_api/services" "eta/eta_api/services/alarm_msg" "eta/eta_api/services/data" "eta/eta_api/utils" "fmt" "github.com/rdlucklib/rdluck_tools/paging" "html" "io" "os" "os/exec" "strconv" "strings" "time" ) // ListReport // @Title 获取报告列表接口 // @Description 获取报告列表 // @Param PageSize query int true "每页数据条数" // @Param CurrentIndex query int true "当前页页码，从1开始" // @Param TimeType query string true "筛选的时间类别：publish_time(发布时间)，modify_time（更新时间）；approve_time（审批时间）" // @Param StartDate query string true "开始时间" // @Param EndDate query string true "结束时间" // @Param Frequency query string true "频度" // @Param ClassifyIdFirst query int true "一级分类id" // @Param ClassifyIdSecond query int true "二级分类id" // @Param ClassifyIdThird query int true "三级分类id" // @Param State query int true "状态" // @Param KeyWord query string true "搜索关键词" // @Param PublishSort query string true "desc:降序，asc 升序（预留）" // @Param FilterReportType query string true "筛选报告类型，1：公共研报，2：共享研报，3：我的研报" // @Success 200 {object} models.ReportListResp // @router /list [get] func (this *ReportController) ListReport() { br := new(models.BaseResponse).Init() defer func() { this.Data["json"] = br this.ServeJSON() }() pageSize, _ := this.GetInt("PageSize") currentIndex, _ := this.GetInt("CurrentIndex") timeType := this.GetString("TimeType") startDate := this.GetString("StartDate") endDate := this.GetString("EndDate") frequency := this.GetString("Frequency") classifyIdFirst, _ := this.GetInt("ClassifyIdFirst", 0) classifyIdSecond, _ := this.GetInt("ClassifyIdSecond", 0) classifyIdThird, _ := this.GetInt("ClassifyIdThird", 0) state, _ := this.GetInt("State") keyWord := this.GetString("KeyWord") msgIsSend, _ := this.GetInt("MsgIsSend") filterReportType, _ := this.GetInt("FilterReportType", 1) var startSize int if pageSize <= 0 { pageSize = utils.PageSize20 } if currentIndex <= 0 { currentIndex = 1 } startSize = utils.StartIndex(currentIndex, pageSize) if timeType == "" { timeType = "publish_time" } if timeType != "publish_time" && timeType != "modify_time" && timeType != "approve_time" { br.Msg = "请选择正确的时间" br.ErrMsg = "请选择正确的时间" return } var condition string var pars []interface{} if keyWord != "" { condition += ` AND (a.title LIKE ? OR a.admin_real_name LIKE ? ) ` pars = utils.GetLikeKeywordPars(pars, keyWord, 2) } if startDate != "" { condition += ` AND a.` + timeType + ` >= ? ` pars = append(pars, startDate) } if endDate != "" { condition += ` AND a.` + timeType + ` <= ? ` pars = append(pars, endDate) } if frequency != "" { condition += ` AND a.frequency = ? ` pars = append(pars, frequency) } if classifyIdFirst > 0 { condition += ` AND a.classify_id_first = ? ` pars = append(pars, classifyIdFirst) } if classifyIdSecond > 0 { condition += ` AND a.classify_id_second = ? ` pars = append(pars, classifyIdSecond) } if classifyIdThird > 0 { condition += ` AND a.classify_id_third = ? ` pars = append(pars, classifyIdThird) } if state > 0 { condition += ` AND a.state = ? ` pars = append(pars, state) } // 消息是否已推送 1-未推送; 2-已推送 if msgIsSend == 1 { condition += ` AND (a.msg_is_send = 0 OR a.ths_msg_is_send = 0) ` } if msgIsSend == 2 { condition += ` AND a.msg_is_send = 1 AND a.ths_msg_is_send = 1 ` } var err error var total int var list []*models.ReportList switch filterReportType { // 筛选报告类型，1：公共研报，2：共享研报，3：我的研报 case 1: condition += ` AND a.is_public_publish = ? ` pars = append(pars, 1) condition += ` AND a.state in (2,6) ` case 3: condition += ` AND a.admin_id = ? ` pars = append(pars, this.SysUser.AdminId) case 2: condition += ` AND (a.admin_id = ? or b.admin_id = ?) ` pars = append(pars, this.SysUser.AdminId, this.SysUser.AdminId) } // 共享报告需要连表查询，所以需要单独写 if filterReportType == 2 { total, err = models.GetReportListCountByGrant(condition, pars) if err != nil { br.Msg = "获取失败" br.ErrMsg = "获取失败，Err:" + err.Error() return } list, err = models.GetReportListByGrant(condition, pars, startSize, pageSize) if err != nil { br.Msg = "获取失败" br.ErrMsg = "获取失败，Err:" + err.Error() return } } else { total, err = models.GetReportListCountV1(condition, pars) if err != nil { br.Msg = "获取失败" br.ErrMsg = "获取失败，Err:" + err.Error() return } list, err = models.GetReportListV1(condition, pars, startSize, pageSize) if err != nil { br.Msg = "获取失败" br.ErrMsg = "获取失败，Err:" + err.Error() return } } listLen := len(list) if listLen > 0 { pvMap := make(map[int]int) uvMap := make(map[int]int) reportIdArr := make([]int, 0) syncReportIdArr := make([]string, 0) // 同步过来的报告IDs oldAndNewReportIdMap := make(map[int]int) // 旧报告和新报告的id对应关系 for i := 0; i < listLen; i++ { reportIdArr = append(reportIdArr, list[i].Id) if list[i].OldReportId > 0 && list[i].ReportLayout == 1 { syncReportIdArr = append(syncReportIdArr, strconv.Itoa(list[i].OldReportId)) oldAndNewReportIdMap[list[i].OldReportId] = list[i].Id } pvMap[list[i].Id] = list[i].Pv uvMap[list[i].Id] = list[i].Uv } // 当下报告的pv，uv if len(reportIdArr) > 0 { pvList, e := models.GetReportPvUvByReportIdList(reportIdArr) if e != nil { br.Msg = "获取失败" br.ErrMsg = "获取同步报告对应的PV、UV失败, Err: " + e.Error() return } for _, v := range pvList { pv := pvMap[v.ReportId] uv := uvMap[v.ReportId] pvMap[v.ReportId] = v.PvTotal + pv uvMap[v.ReportId] = v.UvTotal + uv } } //reportIds := strings.Join(reportIdArr, ",") //syncReportIds := strings.Join(syncReportIdArr, ",") // 查询同步过来的报告对应的老报告PV+UV if len(syncReportIdArr) > 0 { puvList, e := models.GetPUVByResearchReportIds(syncReportIdArr) if e != nil { br.Msg = "获取失败" br.ErrMsg = "获取同步报告对应的PV、UV失败, Err: " + e.Error() return } puvLen := len(puvList) for i := 0; i < puvLen; i++ { newReportId, ok := oldAndNewReportIdMap[puvList[i].ResearchReportId] if ok { pv := pvMap[newReportId] uv := uvMap[newReportId] pvMap[newReportId] = puvList[i].Pv + pv uvMap[newReportId] = puvList[i].Uv + uv } } } // 晨周报音频列表 videoList, err := models.GetReportChapterVideoListByReportIds(reportIdArr) if err != nil { br.Msg = "获取失败" br.ErrMsg = "获取报告音频文件失败，Err:" + err.Error() return } for i := 0; i < listLen; i++ { list[i].Content = html.UnescapeString(list[i].Content) list[i].ContentSub = html.UnescapeString(list[i].ContentSub) // 除周报外其余报告均可推送客群 list[i].NeedThsMsg = 1 //if list[i].HasChapter == 1 && list[i].ChapterType == utils.REPORT_TYPE_WEEK { // list[i].NeedThsMsg = 0 //} chapterList := make([]*models.ReportChapterVideoList, 0) for ii := 0; ii < len(videoList); ii++ { if list[i].Id == videoList[ii].ReportId { chapterList = append(chapterList, videoList[ii]) } } list[i].ChapterVideoList = chapterList list[i].Pv = pvMap[list[i].Id] list[i].Uv = uvMap[list[i].Id] } // 多人协作的协作报告，需要判断是否可编辑 { grantObj := report.ReportGrant{} grantList, err := grantObj.GetGrantListByIdList(reportIdArr) if err != nil { br.Msg = "获取失败" br.ErrMsg = "获取报告授权失败，Err:" + err.Error() return } //grantMap := make(map[报告id]map[用户id]bool) grantMap := make(map[int]map[int]bool) for _, v := range grantList { grantUserMap, ok := grantMap[v.ReportId] if !ok { grantUserMap = make(map[int]bool) } grantUserMap[v.AdminId] = true grantMap[v.ReportId] = grantUserMap } for i, item := range list { if item.AdminId == this.SysUser.AdminId { list[i].HasAuth = true continue } // 查找授权 var hasAuth bool grantUserMap, ok := grantMap[item.Id] // 如果报告根本没有授权用户，说明没有授权当前用户 if !ok { continue } _, ok = grantUserMap[this.SysUser.AdminId] list[i].HasAuth = hasAuth // 如果报告关联用户找到，说明有授权当前用户 if ok { list[i].HasAuth = true } } } } for _, item := range list { /*key := fmt.Sprint(`crm:report:edit:`, item.Id) opUserId, _ := utils.Rc.RedisInt(key) //如果当前没有人操作，获取当前操作人是本人，那么编辑按钮可用 if opUserId <= 0 || (opUserId == this.SysUser.AdminId) || item.ClassifyNameFirst == "周报" || item.ClassifyNameFirst == "晨报" { item.CanEdit = true } else { adminInfo, errAdmin := system.GetSysUserById(opUserId) if errAdmin != nil { br.Msg = "获取失败" br.ErrMsg = "获取失败，Err:" + errAdmin.Error() return } item.Editor = adminInfo.RealName }*/ if item.HasChapter == 1 { item.CanEdit = true continue } markStatus, err := services.UpdateReportEditMark(item.Id, 0, this.SysUser.AdminId, 2, this.SysUser.RealName, this.Lang) if err != nil { br.Msg = "查询标记状态失败" br.ErrMsg = "查询标记状态失败，Err:" + err.Error() return } if markStatus.Status == 0 { item.CanEdit = true } else { item.Editor = markStatus.Editor } } page := paging.GetPaging(currentIndex, pageSize, total) resp := new(models.ReportListResp) resp.Paging = page resp.List = list br.Ret = 200 br.Success = true br.Msg = "获取成功" br.Data = resp } // Add // @Title 新增报告接口 // @Description 新增报告（不区分报告类型） // @Param request body models.AddReq true "type json string" // @Success 200 {object} models.AddResp // @router /add [post] func (this *ReportController) Add() { 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 models.AddReq err := json.Unmarshal(this.Ctx.Input.RequestBody, &req) if err != nil { br.Msg = "参数解析异常！" br.ErrMsg = "参数解析失败，Err:" + err.Error() return } if req.Title == `` { br.Msg = "标题不能为空" br.ErrMsg = "标题不能为空" br.IsSendEmail = false return } if req.ClassifyIdFirst <= 0 { br.Msg = "分类必填" br.ErrMsg = "分类必填" br.IsSendEmail = false return } var contentSub string if req.Content != "" { e := utils.ContentXssCheck(req.Content) if e != nil { br.Msg = "存在非法标签" br.ErrMsg = "存在非法标签, Err: " + e.Error() return } content, e := services.FilterReportContentBr(req.Content) if e != nil { br.Msg = "内容去除前后空格失败" br.ErrMsg = "内容去除前后空格失败, Err: " + e.Error() return } req.Content = content contentSub, err = services.GetReportContentSub(req.Content) if err != nil { go alarm_msg.SendAlarmMsg("ContentSub 失败，Err:"+err.Error(), 3) //utils.SendEmail(utils.APPNAME+"失败提醒", "解析 ContentSub 失败，Err:"+err.Error(), utils.EmailSendToUsers) } } // 报告期数 maxStage, err := models.GetReportStage(req.ClassifyIdFirst, req.ClassifyIdSecond, req.ClassifyIdThird) if err != nil { br.Msg = "期数获取失败！" br.ErrMsg = "期数获取失败，Err:" + err.Error() return } // 根据审批开关及审批流判断当前报告状态 state, e := services.CheckReportCurrState(report_approve.FlowReportTypeChinese, req.ClassifyIdFirst, req.ClassifyIdSecond, req.ClassifyIdThird, models.ReportOperateAdd) if e != nil { br.Msg = "操作失败" br.ErrMsg = "校验报告当前状态失败, Err: " + e.Error() return } // 协作方式，1：个人，2：多人协作。默认：1 if req.CollaborateType == 0 { req.CollaborateType = 1 } // 报告布局，1：常规布局，2：智能布局。默认：1 if req.ReportLayout == 0 { req.ReportLayout = 1 } // 是否公开发布，1：是，2：否 if req.IsPublicPublish == 0 { req.IsPublicPublish = 1 } classifyItemList, err := models.GetClassifyListByIdList([]int{req.ClassifyIdFirst, req.ClassifyIdSecond, req.ClassifyIdThird}) classifyMap := make(map[int]string) for _, v := range classifyItemList { classifyMap[v.Id] = v.ClassifyName } item := new(models.Report) item.AddType = req.AddType item.ReportVersion = 2 item.ClassifyIdFirst = req.ClassifyIdFirst item.ClassifyNameFirst = classifyMap[req.ClassifyIdFirst] item.ClassifyIdSecond = req.ClassifyIdSecond item.ClassifyNameSecond = classifyMap[req.ClassifyIdSecond] item.Title = req.Title item.Abstract = req.Abstract item.Author = req.Author item.Frequency = req.Frequency item.State = state item.Content = html.EscapeString(req.Content) item.Stage = maxStage + 1 item.ContentSub = html.EscapeString(contentSub) item.CreateTime = req.CreateTime item.ModifyTime = time.Now() item.ReportVersion = req.ReportVersion item.AdminId = sysUser.AdminId item.AdminRealName = sysUser.RealName item.ClassifyIdThird = req.ClassifyIdThird item.ClassifyNameThird = classifyMap[req.ClassifyIdThird] // 产品要求，如果是多人协作，那么就是章节类型的报告 if req.CollaborateType == 2 { item.HasChapter = 1 item.ChapterType = "" } item.LastModifyAdminId = sysUser.AdminId item.LastModifyAdminName = sysUser.RealName item.ContentModifyTime = time.Now() item.NeedSplice = 1 item.ContentStruct = html.EscapeString(req.ContentStruct) item.HeadImg = req.HeadImg item.EndImg = req.EndImg item.CanvasColor = req.CanvasColor item.HeadResourceId = req.HeadResourceId item.EndResourceId = req.EndResourceId item.CollaborateType = req.CollaborateType item.ReportLayout = req.ReportLayout item.IsPublicPublish = req.IsPublicPublish item.ReportCreateTime = time.Now() err, errMsg := services.AddReportAndChapter(item, req.InheritReportId, req.GrantAdminIdList) if err != nil { br.Msg = "保存失败" if errMsg != "" { br.Msg = errMsg } br.ErrMsg = "保存失败,Err:" + err.Error() return } recordItem := &models.ReportStateRecord{ ReportId: item.Id, ReportType: 1, State: 1, AdminId: this.SysUser.AdminId, AdminName: this.SysUser.AdminName, CreateTime: time.Now(), } go func() { _, _ = models.AddReportStateRecord(recordItem) }() resp := new(models.AddResp) resp.ReportId = int64(item.Id) resp.ReportCode = item.ReportCode br.Ret = 200 br.Success = true br.Msg = "保存成功" br.Data = resp } // Edit // @Title 编辑报告基础信息接口 // @Description 编辑报告基础信息（不区分报告类型） // @Param request body models.EditReq true "type json string" // @Success 200 {object} models.EditResp // @router /edit [post] func (this *ReportController) Edit() { 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 models.EditReq err := json.Unmarshal(this.Ctx.Input.RequestBody, &req) if err != nil { br.Msg = "参数解析异常！" br.ErrMsg = "参数解析失败，Err:" + err.Error() return } //if req.Content == "" { // br.Msg = "报告内容不能为空" // return //} //更新标记key markStatus, err := services.UpdateReportEditMark(int(req.ReportId), 0, sysUser.AdminId, 1, sysUser.RealName, this.Lang) if err != nil { br.Msg = err.Error() return } if markStatus.Status == 1 { br.Msg = markStatus.Msg //br.Ret = 202 //202 服务器已接受请求，但尚未处理。 return } reportInfo, e := models.GetReportByReportId(int(req.ReportId)) if e != nil { if e.Error() == utils.ErrNoRow() { br.Msg = "报告已被删除, 请刷新页面" return } br.Msg = "操作失败" br.ErrMsg = "获取报告失败, Err: " + e.Error() return } if reportInfo.State == models.ReportStatePublished || reportInfo.State == models.ReportStatePass { br.Msg = "该报告已发布，不允许编辑" br.ErrMsg = "该报告已发布，不允许编辑" br.IsSendEmail = false return } // 编辑报告信息 err, errMsg := services.EditReport(reportInfo, req, sysUser) if err != nil { br.Msg = "保存失败" if errMsg != "" { br.Msg = errMsg } br.ErrMsg = "保存失败,Err:" + err.Error() return } reportCode := utils.MD5(strconv.Itoa(int(req.ReportId))) resp := new(models.EditResp) resp.ReportId = req.ReportId resp.ReportCode = reportCode br.Ret = 200 br.Success = true br.Msg = "保存成功" br.Data = resp } // Detail // @Title 获取报告详情接口 // @Description 获取报告详情 // @Param request body models.ReportDetailReq true "type json string" // @Success 200 {object} models.Report // @router /detail [get] func (this *ReportController) Detail() { br := new(models.BaseResponse).Init() defer func() { this.Data["json"] = br this.ServeJSON() }() /*var req models.ReportDetailReq err := json.Unmarshal(this.Ctx.Input.RequestBody, &req) if err != nil { br.Msg = "参数解析异常！" br.ErrMsg = "参数解析失败，Err:" + err.Error() return } if req.ReportId <= 0 { br.Msg = "参数错误" return }*/ reportId, err := this.GetInt("ReportId") if err != nil { br.Msg = "获取参数失败！" br.ErrMsg = "获取参数失败，Err:" + err.Error() return } if reportId <= 0 { br.Msg = "参数错误" return } item, err := models.GetReportById(reportId) if err != nil { if err.Error() == utils.ErrNoRow() { br.Msg = "报告已被删除" return } br.Msg = "获取失败" br.ErrMsg = "获取失败，Err:" + err.Error() return } chapterList := make([]*models.ReportChapter, 0) if item.HasChapter == 1 { // 获取章节内容 tmpChapterList, err := models.GetPublishedChapterListByReportId(item.Id) if err != nil { br.Msg = "获取失败" br.ErrMsg = "获取晨/周报章节列表失败, Err: " + err.Error() return } if len(tmpChapterList) > 0 { // 章节类型的字段赋值 for _, item := range tmpChapterList { item.Content = html.UnescapeString(item.Content) item.ContentSub = html.UnescapeString(item.ContentSub) chapterList = append(chapterList, item) } } //item.Abstract = item.Title } item.Content = html.UnescapeString(item.Content) item.ContentSub = html.UnescapeString(item.ContentSub) item.ContentStruct = html.UnescapeString(item.ContentStruct) if item.HeadResourceId > 0 { headResource, err := smart_report.GetResourceItemById(item.HeadResourceId) if err != nil { br.Msg = "操作失败" br.ErrMsg = "获取资源库版头失败, Err: " + err.Error() return } item.HeadImg = headResource.ImgUrl item.HeadStyle = headResource.Style } if item.EndResourceId > 0 { endResource, err := smart_report.GetResourceItemById(item.EndResourceId) if err != nil { br.Msg = "操作失败" br.ErrMsg = "获取资源库版头失败, Err: " + err.Error() return } item.EndImg = endResource.ImgUrl item.EndStyle = endResource.Style } resp := &models.ReportDetailView{ ReportDetail: item, ChapterList: chapterList, } br.Ret = 200 br.Success = true br.Msg = "获取成功" br.Data = resp } // SaveReportContent // @Title 保存草稿 // @Description 保存草稿 // @Param request body models.SaveReportContent true "type json string" // @Success 200 {object} models.ReportAuthorResp // @router /saveReportContent [post] func (this *ReportController) SaveReportContent() { br := new(models.BaseResponse).Init() br.IsSendEmail = false 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 models.SaveReportContent err := json.Unmarshal(this.Ctx.Input.RequestBody, &req) if err != nil { br.Msg = "参数解析异常！" br.ErrMsg = "参数解析失败，Err:" + err.Error() return } reportId := req.ReportId noChangeFlag := req.NoChange if reportId <= 0 { resp := new(models.SaveReportContentResp) resp.ReportId = reportId br.Ret = 200 br.Success = true br.Msg = "保存成功" br.Data = resp return } // 获取报告详情 reportInfo, _ := models.GetReportByReportId(req.ReportId) if reportInfo != nil && reportInfo.State == 2 { br.Msg = "该报告已发布，不允许编辑" br.ErrMsg = "该报告已发布，不允许编辑" br.IsSendEmail = false return } // 标记更新中 { markStatus, err := services.UpdateReportEditMark(req.ReportId, 0, sysUser.AdminId, 1, sysUser.RealName, this.Lang) if err != nil { br.Msg = err.Error() return } if markStatus.Status == 1 { br.Msg = markStatus.Msg return } } // 内容有过修改的话，那么逻辑处理 if noChangeFlag != 1 { content := req.Content if content == "" { content = this.GetString("Content") } if content != "" { e := utils.ContentXssCheck(req.Content) if e != nil { br.Msg = "存在非法标签" br.ErrMsg = "存在非法标签, Err: " + e.Error() return } contentClean, e := services.FilterReportContentBr(req.Content) if e != nil { br.Msg = "内容去除前后空格失败" br.ErrMsg = "内容去除前后空格失败, Err: " + e.Error() return } content = contentClean contentSub, err := services.GetReportContentSub(content) if err != nil { go alarm_msg.SendAlarmMsg("解析 ContentSub 失败，Err："+err.Error(), 3) //utils.SendEmail(utils.APPNAME+"失败提醒", "解析 ContentSub 失败，Err:"+err.Error(), utils.EmailSendToUsers) } reportInfo.Content = html.EscapeString(content) reportInfo.ContentSub = html.EscapeString(contentSub) reportInfo.ContentStruct = html.EscapeString(req.ContentStruct) reportInfo.HeadImg = req.HeadImg reportInfo.EndImg = req.EndImg reportInfo.CanvasColor = req.CanvasColor reportInfo.HeadResourceId = req.HeadResourceId reportInfo.EndResourceId = req.EndResourceId reportInfo.ModifyTime = time.Now() reportInfo.ContentModifyTime = time.Now() updateCols := []string{"Content", "ContentSub", "ContentStruct", "HeadImg", "EndImg", "CanvasColor", "HeadResourceId", "EndResourceId", "ModifyTime", "ContentModifyTime"} err = reportInfo.UpdateReport(updateCols) if err != nil { br.Msg = "保存失败" br.ErrMsg = "保存失败，Err:" + err.Error() return } go models.AddReportSaveLog(reportId, this.SysUser.AdminId, reportInfo.Content, reportInfo.ContentSub, reportInfo.ContentStruct, reportInfo.CanvasColor, this.SysUser.AdminName, reportInfo.HeadResourceId, reportInfo.EndResourceId) } } resp := new(models.SaveReportContentResp) resp.ReportId = reportId br.Ret = 200 br.Success = true br.Msg = "保存成功" br.Data = resp } // AuthorizedListReport // @Title 获取有权限的报告列表接口 // @Description 获取有权限的报告列表接口 // @Param PageSize query int true "每页数据条数" // @Param CurrentIndex query int true "当前页页码，从1开始" // @Param Keyword query string true "搜索关键词" // @Param ClassifyIdFirst query int true "一级分类id" // @Param ClassifyIdSecond query int true "二级分类id" // @Param ClassifyIdThird query int true "三级分类id" // @Success 200 {object} models.ReportListResp // @router /list/authorized [get] func (this *ReportController) AuthorizedListReport() { br := new(models.BaseResponse).Init() defer func() { this.Data["json"] = br this.ServeJSON() }() pageSize, _ := this.GetInt("PageSize") currentIndex, _ := this.GetInt("CurrentIndex") keyword := this.GetString("Keyword") classifyIdFirst, _ := this.GetInt("ClassifyIdFirst", 0) classifyIdSecond, _ := this.GetInt("ClassifyIdSecond", 0) classifyIdThird, _ := this.GetInt("ClassifyIdThird", 0) var startSize int if pageSize <= 0 { pageSize = utils.PageSize20 } if currentIndex <= 0 { currentIndex = 1 } startSize = utils.StartIndex(currentIndex, pageSize) var list []*models.ReportList // 没有输入信息，那就不展示 if keyword == `` && classifyIdFirst <= 0 { page := paging.GetPaging(currentIndex, pageSize, 0) resp := new(models.ReportListResp) resp.Paging = page resp.List = list br.Ret = 200 br.Success = true br.Msg = "获取成功" br.Data = resp return } // 当前用户有权限的报告id列表 grantReportIdList := make([]int, 0) { obj := report.ReportGrant{} grantList, err := obj.GetGrantListByAdminId(this.SysUser.AdminId) if err != nil { br.Msg = "获取失败" br.ErrMsg = "获取授权报告id失败，Err:" + err.Error() return } for _, v := range grantList { grantReportIdList = append(grantReportIdList, v.ReportId) } } var condition string var pars []interface{} if classifyIdFirst > 0 { condition += ` AND a.classify_id_first = ? ` pars = append(pars, classifyIdFirst) } if classifyIdSecond > 0 { condition += ` AND a.classify_id_second = ? ` pars = append(pars, classifyIdSecond) } if classifyIdThird > 0 { condition += ` AND a.classify_id_third = ? ` pars = append(pars, classifyIdThird) } if keyword != `` { condition += ` AND a.title LIKE ? ` pars = utils.GetLikeKeywordPars(pars, keyword, 1) } var err error var total int orCondition := `AND ( (a.is_public_publish = ? AND a.state in (2,6)) or a.admin_id = ? ` pars = append(pars, 1, this.SysUser.AdminId) // 当前用户有权限的报告id列表 num := len(grantReportIdList) if num > 0 { orCondition += ` OR a.id in (` + utils.GetOrmInReplace(num) + `)` pars = append(pars, grantReportIdList) } orCondition += ` ) ` condition += orCondition total, err = models.GetReportListCountByAuthorized(condition, pars) if err != nil { br.Msg = "获取失败" br.ErrMsg = "获取失败，Err:" + err.Error() return } list, err = models.GetReportListByAuthorized(condition, pars, startSize, pageSize) if err != nil { br.Msg = "获取失败" br.ErrMsg = "获取失败，Err:" + err.Error() return } //for _, item := range list { // if item.HasChapter == 1 { // item.CanEdit = true // } else { // markStatus, err := services.UpdateReportEditMark(item.Id, this.SysUser.AdminId, 2, this.SysUser.RealName, this.Lang) // if err != nil { // br.Msg = "查询标记状态失败" // br.ErrMsg = "查询标记状态失败，Err:" + err.Error() // return // } // if markStatus.Status == 0 { // item.CanEdit = true // } else { // item.Editor = markStatus.Editor // } // } //} page := paging.GetPaging(currentIndex, pageSize, total) resp := new(models.ReportListResp) resp.Paging = page resp.List = list br.Ret = 200 br.Success = true br.Msg = "获取成功" br.Data = resp } // BaseDetail // @Title 获取报告基础信息详情接口 // @Description 获取报告基础信息详情接口 // @Param request body models.ReportDetailReq true "type json string" // @Success 200 {object} models.Report // @router /detail/base [get] func (this *ReportController) BaseDetail() { br := new(models.BaseResponse).Init() defer func() { this.Data["json"] = br this.ServeJSON() }() /*var req models.ReportDetailReq err := json.Unmarshal(this.Ctx.Input.RequestBody, &req) if err != nil { br.Msg = "参数解析异常！" br.ErrMsg = "参数解析失败，Err:" + err.Error() return } if req.ReportId <= 0 { br.Msg = "参数错误" return }*/ reportId, err := this.GetInt("ReportId") if err != nil { br.Msg = "获取参数失败！" br.ErrMsg = "获取参数失败，Err:" + err.Error() return } if reportId <= 0 { br.Msg = "参数错误" return } reportInfo, err := models.GetReportById(reportId) if err != nil { if err.Error() == utils.ErrNoRow() { br.Msg = "报告已被删除" return } br.Msg = "获取失败" br.ErrMsg = "获取失败，Err:" + err.Error() return } // 基础信息就不获取章节信息了 chapterList := make([]*models.ReportChapter, 0) reportInfo.Content = html.UnescapeString(reportInfo.Content) reportInfo.ContentSub = html.UnescapeString(reportInfo.ContentSub) grandAdminList := make([]models.ReportDetailViewAdmin, 0) permissionList := make([]models.ReportDetailViewPermission, 0) // 处理报告授权用户列表 { obj := report.ReportGrant{} grantList, tmpErr := obj.GetGrantListById(reportId) if tmpErr != nil { br.Msg = "获取章节id授权用户列表失败" br.ErrMsg = "获取章节id授权用户列表失败, Err: " + tmpErr.Error() return } if len(grantList) > 0 { grandAdminIdList := make([]int, 0) for _, v := range grantList { grandAdminIdList = append(grandAdminIdList, v.AdminId) } adminList, tmpErr := system.GetAdminListByIdList(grandAdminIdList) if tmpErr != nil { br.Msg = "获取章节id授权用户列表失败" br.ErrMsg = "获取章节id授权用户列表失败, Err: " + tmpErr.Error() return } for _, v := range adminList { grandAdminList = append(grandAdminList, models.ReportDetailViewAdmin{ AdminId: v.AdminId, AdminName: v.RealName, }) } } } // 处理章节id关联品种id列表 { minClassifyId := reportInfo.ClassifyIdThird if minClassifyId <= 0 { minClassifyId = reportInfo.ClassifyIdSecond } if minClassifyId <= 0 { minClassifyId = reportInfo.ClassifyIdFirst } if minClassifyId <= 0 { br.Msg = "分类异常" br.ErrMsg = "分类异常" return } // 获取分类关联的品种id classifyPermissionList, tmpErr := models.GetPermission(minClassifyId) if tmpErr != nil { br.Msg = "获取分类信息失败" br.ErrMsg = "获取失败，Err:" + tmpErr.Error() return } if len(classifyPermissionList) > 0 { permissionIdList := make([]int, 0) for _, v := range classifyPermissionList { permissionIdList = append(permissionIdList, v.ChartPermissionId) } adminList, tmpErr := models.GetChartPermissionByIdList(permissionIdList) if tmpErr != nil { br.Msg = "获取章节id授权用户列表失败" br.ErrMsg = "获取章节id授权用户列表失败, Err: " + tmpErr.Error() return } for _, v := range adminList { permissionList = append(permissionList, models.ReportDetailViewPermission{ PermissionId: v.ChartPermissionId, PermissionName: v.PermissionName, }) } } } resp := &models.ReportDetailView{ ReportDetail: reportInfo, ChapterList: chapterList, GrandAdminList: grandAdminList, PermissionList: permissionList, } br.Ret = 200 br.Success = true br.Msg = "获取成功" br.Data = resp } // EditLayoutImg // @Title 版图设置接口 // @Description 版图设置接口 // @Param request body models.EditLayoutImgReq true "type json string" // @Success 200 {object} models.EditResp // @router /layout_img/edit [post] func (this *ReportController) EditLayoutImg() { 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 models.EditLayoutImgReq err := json.Unmarshal(this.Ctx.Input.RequestBody, &req) if err != nil { br.Msg = "参数解析异常！" br.ErrMsg = "参数解析失败，Err:" + err.Error() return } //if req.Content == "" { // br.Msg = "报告内容不能为空" // return //} //更新标记key markStatus, err := services.UpdateReportEditMark(int(req.ReportId), 0, sysUser.AdminId, 1, sysUser.RealName, this.Lang) if err != nil { br.Msg = err.Error() return } if markStatus.Status == 1 { br.Msg = markStatus.Msg //br.Ret = 202 //202 服务器已接受请求，但尚未处理。 return } reportInfo, e := models.GetReportByReportId(int(req.ReportId)) if e != nil { if e.Error() == utils.ErrNoRow() { br.Msg = "报告已被删除, 请刷新页面" return } br.Msg = "操作失败" br.ErrMsg = "获取报告失败, Err: " + e.Error() return } if reportInfo.State == models.ReportStatePublished || reportInfo.State == models.ReportStatePass { br.Msg = "该报告已发布，不允许编辑" br.ErrMsg = "该报告已发布，不允许编辑" br.IsSendEmail = false return } // 编辑报告信息 err, errMsg := services.EditReportLayoutImg(reportInfo, req, sysUser) if err != nil { br.Msg = "保存失败" if errMsg != "" { br.Msg = errMsg } br.ErrMsg = "保存失败,Err:" + err.Error() return } reportCode := utils.MD5(strconv.Itoa(int(req.ReportId))) resp := new(models.EditResp) resp.ReportId = req.ReportId resp.ReportCode = reportCode br.Ret = 200 br.Success = true br.Msg = "保存成功" br.Data = resp } // PublishReport // @Title 发布报告接口 // @Description 发布报告 // @Param request body models.PublishReq true "type json string" // @Success 200 Ret=200 发布成功 // @router /publish [post] func (this *ReportController) PublishReport() { br := new(models.BaseResponse).Init() defer func() { this.Data["json"] = br this.ServeJSON() }() var req models.PublishReq err := json.Unmarshal(this.Ctx.Input.RequestBody, &req) if err != nil { br.Msg = "参数解析异常！" br.ErrMsg = "参数解析失败，Err:" + err.Error() return } reportIds := req.ReportIds if reportIds == "" { br.Msg = "参数错误" br.ErrMsg = "参数错误，报告id不可为空" return } // 这里实际上不会批量发布了... reportArr := strings.Split(reportIds, ",") tips := "" for _, v := range reportArr { vint, err := strconv.Atoi(v) if err != nil { br.Msg = "参数错误" br.ErrMsg = "参数错误,Err:" + err.Error() return } // 报告的图表刷新状态校验 refreshResult := data.CheckBatchChartRefreshResult("report", vint, 0) if !refreshResult { br.Msg = "图表刷新未完成，请稍后操作" br.ErrMsg = "图表刷新未完成，请稍后操作" br.IsSendEmail = false return } // 报告发布 tmpTips, err, errMsg := services.PublishReport(vint, req.ReportUrl, this.SysUser) if err != nil { br.Msg = errMsg br.ErrMsg = "报告发布失败,Err:" + err.Error() return } tips = tmpTips } // 发布晨周报部分章节未发布的提示 if tips != "" { br.Data = tips } br.Ret = 200 br.Success = true br.Msg = "发布成功" } // PublishCancelReport // @Title 取消发布报告接口 // @Description 取消发布报告 // @Param request body models.PublishCancelReq true "type json string" // @Success 200 Ret=200 取消发布成功 // @router /publish/cancle [post] func (this *ReportController) PublishCancelReport() { br := new(models.BaseResponse).Init() defer func() { this.Data["json"] = br this.ServeJSON() }() var req models.PublishCancelReq err := json.Unmarshal(this.Ctx.Input.RequestBody, &req) if err != nil { br.Msg = "参数解析异常！" br.ErrMsg = "参数解析失败，Err:" + err.Error() return } if req.ReportIds <= 0 { br.Msg = "参数错误" br.ErrMsg = "参数错误，报告id不可为空" return } publishTimeNullFlag := true reportInfo, err := models.GetReportById(req.ReportIds) if err != nil { br.Msg = "获取报告信息失败" br.ErrMsg = "获取报告信息失败,Err:" + err.Error() return } if reportInfo.MsgIsSend == 1 { publishTimeNullFlag = false } // 根据审批开关及审批流判断当前报告状态 state, e := services.CheckReportCurrState(report_approve.FlowReportTypeChinese, reportInfo.ClassifyIdFirst, reportInfo.ClassifyIdSecond, reportInfo.ClassifyIdThird, models.ReportOperateCancelPublish) if e != nil { br.Msg = "操作失败" br.ErrMsg = "校验报告当前状态失败, Err: " + e.Error() return } err = models.PublishCancelReport(req.ReportIds, state, publishTimeNullFlag, this.SysUser.AdminId, this.SysUser.RealName) if err != nil { br.Msg = "取消发布失败" br.ErrMsg = "取消发布失败,Err:" + err.Error() return } // 更新ES禁用 { go services.UpdateReportEs(req.ReportIds, 1) } //// 获取审批流设置 //confKey := "approval_flow" //confTmp, e := company.GetConfigDetailByCode(confKey) //if e != nil { // br.Msg = "获取审批流配置失败" // br.ErrMsg = "获取审批流配置失败, Err: " + e.Error() // return //} //if confTmp.ConfigValue == "1" || confTmp.ConfigValue == "2" || confTmp.ConfigValue == "3" { // br.Msg = "撤销成功" //} else { // br.Msg = "取消发布成功" //} recordItem := &models.ReportStateRecord{ ReportId: req.ReportIds, ReportType: 1, State: state, AdminId: this.SysUser.AdminId, AdminName: this.SysUser.AdminName, CreateTime: time.Now(), } go func() { _, _ = models.AddReportStateRecord(recordItem) }() br.Ret = 200 br.Success = true } // PrePublishReport // @Title 设置定时发布接口 // @Description 设置定时发布接口 // @Param request body models.PrePublishReq true "type json string" // @Success 200 Ret=200 发布成功 // @router /pre_publish [post] func (this *ReportController) PrePublishReport() { br := new(models.BaseResponse).Init() defer func() { this.Data["json"] = br this.ServeJSON() }() var req models.PrePublishReq err := json.Unmarshal(this.Ctx.Input.RequestBody, &req) if err != nil { br.Msg = "参数解析异常！" br.ErrMsg = "参数解析失败，Err:" + err.Error() return } reportId := req.ReportId if reportId == 0 { br.Msg = "参数错误" br.ErrMsg = "参数错误，报告id不可为空" return } if req.PrePublishTime == "" { br.Msg = "发布时间不能为空" return } if req.PreMsgSend != 0 && req.PreMsgSend != 1 { br.Msg = "参数错误" br.ErrMsg = "是否发送模版消息标识错误" return } prePublishTime, err := time.ParseInLocation(utils.FormatDateTime, req.PrePublishTime, time.Local) if err != nil { br.Msg = "发布时间格式错误" br.ErrMsg = "发布时间格式错误，Err:" + err.Error() return } if prePublishTime.Before(time.Now()) { br.Msg = "发布时间不允许选择过去时间" return } if prePublishTime.Before(time.Now().Add(2 * time.Minute)) { br.Msg = "发布时间距离当前时间太近了" return } reportDetail, err := models.GetReportById(reportId) if err != nil { br.Msg = "获取报告信息失败" br.ErrMsg = "获取报告信息失败,Err:" + err.Error() return } if reportDetail == nil { br.Msg = "报告不存在" return } // 如果是章节类型的报告，那么需要确认所有章节已发布 if reportDetail.HasChapter == 1 { chapterList, err := models.GetChapterListByReportId(reportId) if err != nil { return } for _, chapter := range chapterList { if chapter.PublishState == 1 { br.Msg = "还存在未发布的章节" br.ErrMsg = "还存在未发布的章节" return } } } else { if reportDetail.Content == "" { br.Msg = "报告内容为空，不可设置定时发布" br.ErrMsg = "报告内容为空，不可设置定时发布,report_id:" + strconv.Itoa(reportDetail.Id) return } } if reportDetail.State == 2 { br.Msg = "报告已发布，不可设置定时发布" return } // 校验是否开启了审批流 opening, e := services.CheckReportOpenApprove(report_approve.FlowReportTypeChinese, reportDetail.ClassifyIdFirst, reportDetail.ClassifyIdSecond, reportDetail.ClassifyIdThird) if e != nil { br.Msg = "操作失败" br.ErrMsg = "校验报告是否开启审批流失败, Err: " + e.Error() return } if opening { br.Msg = "报告已开启审批流, 不可设置定时发布" return } var tmpErr error if tmpErr = models.SetPrePublishReportById(reportDetail.Id, req.PrePublishTime, req.PreMsgSend); tmpErr != nil { br.Msg = "设置定时发布失败" br.ErrMsg = "设置定时发布失败, Err:" + tmpErr.Error() + ", report_id:" + strconv.Itoa(reportDetail.Id) return } // 生成报告pdf和长图 { reportPdfUrl := services.GetGeneralPdfUrl(reportDetail.ReportCode, reportDetail.ClassifyNameFirst, reportDetail.ReportLayout) go services.Report2pdfAndJpeg(reportPdfUrl, reportDetail.Id, 1) } br.Ret = 200 br.Success = true br.Msg = "定时发布成功" } // SubmitApprove // @Title 提交审批 // @Description 提交审批接口 // @Param request body models.ReportSubmitApproveReq true "type json string" // @Success 200 Ret=200 操作成功 // @router /approve/submit [post] func (this *ReportController) SubmitApprove() { 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" return } var req models.ReportSubmitApproveReq if e := json.Unmarshal(this.Ctx.Input.RequestBody, &req); e != nil { br.Msg = "参数有误" br.ErrMsg = "参数解析失败, Err: " + e.Error() return } reportId := req.ReportId if reportId <= 0 { br.Msg = "参数有误" br.ErrMsg = fmt.Sprintf("参数有误, ReportId: %d", req.ReportId) return } reportOb := new(models.Report) reportItem, e := reportOb.GetItemById(reportId) if e != nil { if e.Error() == utils.ErrNoRow() { br.Msg = "报告已被删除, 请刷新页面" return } br.Msg = "操作失败" br.ErrMsg = "获取报告失败, Err: " + e.Error() return } // 如果是章节类型的报告，那么需要确认所有章节已发布 if reportItem.HasChapter == 1 { chapterList, err := models.GetChapterListByReportId(reportId) if err != nil { return } for _, chapter := range chapterList { if chapter.PublishState == 1 { br.Msg = "还存在未发布的章节" br.ErrMsg = "还存在未发布的章节" return } } } else { if reportItem.Content == "" { br.Msg = "报告内容为空，不可提交" br.ErrMsg = "报告内容为空，不可提交,report_id:" + strconv.Itoa(reportItem.Id) return } } // 校验当前审批配置, 返回下一个状态 state, e := services.CheckReportCurrState(report_approve.FlowReportTypeChinese, reportItem.ClassifyIdFirst, reportItem.ClassifyIdSecond, reportItem.ClassifyIdThird, models.ReportOperateSubmitApprove) if e != nil { br.Msg = "操作失败" br.ErrMsg = "校验报告状态失败, Err: " + e.Error() return } // 下一个状态不为待审批时, 仅更新状态 if state != models.ReportStateWaitApprove { reportItem.State = state e = reportItem.UpdateReport([]string{"State"}) if e != nil { br.Msg = "操作失败" br.ErrMsg = "更新报告状态失败, Err: " + e.Error() return } br.Ret = 200 br.Success = true br.Msg = "操作成功" return } // 提交审批 approveId, e := services.SubmitReportApprove(report_approve.FlowReportTypeChinese, reportItem.Id, reportItem.Title, reportItem.ClassifyIdFirst, reportItem.ClassifyIdSecond, reportItem.ClassifyIdThird, sysUser.AdminId, sysUser.RealName) if e != nil { br.Msg = "操作失败" br.ErrMsg = "提交审批失败, Err: " + e.Error() return } reportItem.ApproveId = approveId reportItem.State = models.ReportStateWaitApprove reportItem.ModifyTime = time.Now().Local() e = reportItem.UpdateReport([]string{"ApproveId", "State", "ModifyTime"}) if e != nil { br.Msg = "操作失败" br.ErrMsg = "更新报告状态失败, Err: " + e.Error() return } // 报告权限处理 { go services.HandleReportPermission(reportItem) } br.Ret = 200 br.Success = true br.Msg = "操作成功" } // CancelApprove // @Title 撤销审批 // @Description 撤销审批 // @Param request body models.ReportCancelApproveReq true "type json string" // @Success 200 Ret=200 操作成功 // @router /approve/cancel [post] func (this *ReportController) CancelApprove() { 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" return } var req models.ReportCancelApproveReq if e := json.Unmarshal(this.Ctx.Input.RequestBody, &req); e != nil { br.Msg = "参数有误" br.ErrMsg = "参数解析失败, Err: " + e.Error() return } reportId := req.ReportId if reportId <= 0 { br.Msg = "参数有误" br.ErrMsg = fmt.Sprintf("参数有误, ReportId: %d", req.ReportId) return } reportOb := new(models.Report) reportItem, e := reportOb.GetItemById(reportId) if e != nil { if e.Error() == utils.ErrNoRow() { br.Msg = "报告已被删除, 请刷新页面" return } br.Msg = "操作失败" br.ErrMsg = "获取报告失败, Err: " + e.Error() return } // 校验当前审批配置, 返回下一个状态 state, e := services.CheckReportCurrState(report_approve.FlowReportTypeChinese, reportItem.ClassifyIdFirst, reportItem.ClassifyIdSecond, reportItem.ClassifyIdThird, models.ReportOperateCancelApprove) if e != nil { br.Msg = "操作失败" br.ErrMsg = "校验报告状态失败, Err: " + e.Error() return } // 下一个状态不为待提交时, 仅更新状态 if state != models.ReportStateWaitSubmit { reportItem.State = state e = reportItem.UpdateReport([]string{"State"}) if e != nil { br.Msg = "操作失败" br.ErrMsg = "更新报告状态失败, Err: " + e.Error() return } br.Ret = 200 br.Success = true br.Msg = "操作成功" return } //if reportItem.ApproveId <= 0 { // br.Msg = "报告审批不存在" // br.ErrMsg = fmt.Sprintf("报告审批不存在, ApproveId: %d", reportItem.ApproveId) // return //} // 撤销审批 e = services.CancelReportApprove(report_approve.FlowReportTypeChinese, reportItem.Id, reportItem.ApproveId, sysUser.AdminId, sysUser.RealName) if e != nil { br.Msg = "操作失败" br.ErrMsg = "撤销审批失败, Err: " + e.Error() return } //reportItem.ApproveId = 0 //reportItem.State = models.ReportStateWaitSubmit //reportItem.ModifyTime = time.Now().Local() //e = reportItem.UpdateReport([]string{"ApproveId", "State", "ModifyTime"}) //if e != nil { // br.Msg = "操作失败" // br.ErrMsg = "更新报告状态失败, Err: " + e.Error() // return //} br.Ret = 200 br.Success = true br.Msg = "操作成功" } // init // @Description: 修复历史报告数据 // @author: Roc // @datetime 2024-06-21 09:19:05 func init() { //fixApproveRecord() //fixChapterPermission() //fixReportEs() //fixSmartReport() } // 修复研报审批数据 func fixApproveRecord() { recordOb := new(report_approve.ReportApproveRecord) recordCond := fmt.Sprintf(` AND %s = ? AND %s in (?,?)`, report_approve.ReportApproveRecordCols.NodeType, report_approve.ReportApproveRecordCols.State) recordPars := make([]interface{}, 0) recordPars = append(recordPars, 0, report_approve.ReportApproveStatePass, report_approve.ReportApproveStateRefuse) list, e := recordOb.GetItemsByCondition(recordCond, recordPars, []string{}, "") if e != nil { fmt.Println("查找审批记录失败，Err:", e.Error()) return } for _, recordItem := range list { //fmt.Println(recordItem) recordItem.NodeState = recordItem.State recordItem.NodeApproveUserId = recordItem.ApproveUserId recordItem.NodeApproveUserName = recordItem.ApproveUserName recordItem.NodeApproveTime = recordItem.ApproveTime // 如果不是或签，那么只需要修复自己就好了 if recordItem.ApproveType != report_approve.NodeApproveTypeAny { recordCols := []string{"State", "ApproveTime", "ModifyTime", "NodeState", "NodeApproveUserId", "NodeApproveUserName", "NodeApproveTime"} if e = recordItem.Update(recordCols); e != nil { fmt.Println("更新审批记录状态失败，Err：", e.Error()) } continue } // 或签 // 需要将该审批的同一个节点的记录标记为已审批 if e := recordItem.UpdateNodeState(recordItem.ReportApproveId, recordItem.NodeId, recordItem.NodeState, recordItem.NodeApproveUserId, recordItem.NodeApproveUserName, recordItem.NodeApproveTime); e != nil { fmt.Println("更新同一节点的其他审批记录状态失败，Err：", e.Error()) } } fmt.Println("审批数据修复完成") } // fixChapterPermission // @Description: 修复章节关联的品种权限 // @author: Roc // @datetime 2024-06-20 18:08:34 func fixChapterPermission() { allChapterTypePermissionList, err := models.GetAllChapterTypePermission() if err != nil { fmt.Println("获取所有章节类型ID获取章节类型权限列表失败，Err:", err.Error()) return } currChapterTypePermissionIdListMap := make(map[int][]int) hasPermissionMap := make(map[string]bool) for _, v := range allChapterTypePermissionList { tmpChapterTypePermissionList, ok := currChapterTypePermissionIdListMap[v.ReportChapterTypeId] if !ok { tmpChapterTypePermissionList = make([]int, 0) } key := fmt.Sprint(v.ReportChapterTypeId, "-", v.ChartPermissionId) if _, has := hasPermissionMap[key]; !has { hasPermissionMap[key] = true currChapterTypePermissionIdListMap[v.ReportChapterTypeId] = append(tmpChapterTypePermissionList, v.ChartPermissionId) } } //notIdList := []int{9675, 9675, 9740, 9749, 9768, 9773, 9791, 9792, 9793, 9850, 9851, 9852, 9852, 9852, 9853, 9854, 9856, 9857, 9857, 9858, 9859, 9860, 9861, 9862, 9862, 9863, 9866} notIdList := []int{} allReportChapterList, err := models.GetAllReportChapter() if err != nil { fmt.Println("获取所有章节失败，Err:", err.Error()) return } addList := make([]*report.ReportChapterPermissionMapping, 0) for _, v := range allReportChapterList { // 如果是上面的章节id，那么就过滤掉，因为已经入库了 if utils.InArrayByInt(notIdList, v.ReportChapterId) { continue } permissionIdList, ok := currChapterTypePermissionIdListMap[v.TypeId] if !ok { continue } for _, permissionId := range permissionIdList { addList = append(addList, &report.ReportChapterPermissionMapping{ ReportChapterPermissionMappingId: 0, ReportChapterId: v.ReportChapterId, ChartPermissionId: permissionId, CreateTime: v.ModifyTime, }) } } obj := report.ReportChapterPermissionMapping{} err = obj.MultiAdd(addList) if err != nil { fmt.Println("批量添加报章节的品种权限失败，Err:", err.Error()) } return } // fixReportEs // @Description: 修复报告es数据 // @author: Roc // @datetime 2024-06-20 18:08:34 func fixReportEs() { //reportInfo, err := models.GetReportByReportId(3941) //if err != nil { // fmt.Println("查询信息失败，", err) // return //} //content := utils.TrimHtml(html.UnescapeString(reportInfo.Content)) //fmt.Println(content) // //fmt.Println("=========================") // //chapterInfo, err := models.GetReportChapterInfoById(9637) //if err != nil { // fmt.Println("查询信息失败2，", err) // return //} // //content = utils.TrimHtml(html.UnescapeString(chapterInfo.Content)) //fmt.Println(content) // //services.UpdateReportChapterEs(9637) //return var condition string var pars []interface{} condition += " AND state in (2,6) " list, err := models.GetReportListV1(condition, pars, 0, 100000) if err != nil { fmt.Println("查询信息失败，", err) return } num := len(list) fmt.Println(num, "条待修复报告es数据") for k, v := range list { fmt.Println("剩余", num-k, "条") services.UpdateReportEs(v.Id, 2) } fmt.Println("报告ES数据修复完成") return } // fixSmartReport // @Description: 修复智能研报的数据 // @author: Roc // @datetime 2024-06-27 16:54:41 func fixSmartReport() { fmt.Println("修复智能研报开始") // 先判断是否已经修复过数据，如果修复过，那么就不用修复了 { condition := ` AND report_layout=2 and old_report_id>0 ` list, err := models.GetReportByCondition(condition, []interface{}{}, []string{}, " order by id asc ", false, 0, 0) if err != nil { fmt.Println("获取已修复的报告列表失败, Err:" + err.Error()) return } if len(list) > 0 { fmt.Println("智能研报已经修复过数据，不需要再次修复") return } } var condition string var pars []interface{} reportOB := new(smart_report.SmartReport) list, e := reportOB.GetItemsByCondition(condition, pars, []string{}, " smart_report_id asc ") if e != nil { fmt.Println("获取智能报告列表失败, Err:" + e.Error()) return } addList := make([]*models.Report, 0) for _, v := range list { fmt.Println(v) addList = append(addList, &models.Report{ //Id: 0, AddType: 1, ClassifyIdFirst: v.ClassifyIdFirst, ClassifyNameFirst: v.ClassifyNameFirst, ClassifyIdSecond: v.ClassifyIdSecond, ClassifyNameSecond: v.ClassifyNameSecond, Title: v.Title, Abstract: v.Abstract, Author: v.Author, Frequency: v.Frequency, CreateTime: v.CreateTime.Format(utils.FormatDateTime), ModifyTime: v.ModifyTime, State: v.State, PublishTime: v.PublishTime, Stage: v.Stage, MsgIsSend: v.MsgIsSend, //ThsMsgIsSend: v.Tha, Content: v.Content, VideoUrl: v.VideoUrl, VideoName: v.VideoName, VideoPlaySeconds: fmt.Sprint(v.VideoPlaySeconds), VideoSize: v.VideoSize, ContentSub: v.ContentSub, ReportCode: fmt.Sprint(v.SmartReportId), ReportVersion: 1, HasChapter: 0, ChapterType: "", OldReportId: v.SmartReportId, MsgSendTime: v.MsgSendTime, AdminId: v.AdminId, AdminRealName: v.AdminRealName, ApproveTime: v.ApproveTime, ApproveId: v.ApproveId, DetailImgUrl: v.DetailImgUrl, DetailPdfUrl: v.DetailPdfUrl, ContentStruct: v.ContentStruct, LastModifyAdminId: v.LastModifyAdminId, LastModifyAdminName: v.LastModifyAdminName, ContentModifyTime: v.ContentModifyTime, Pv: v.Pv, Uv: v.Uv, HeadImg: v.HeadImg, EndImg: v.EndImg, CanvasColor: v.CanvasColor, NeedSplice: v.NeedSplice, HeadResourceId: v.HeadResourceId, EndResourceId: v.EndResourceId, ClassifyIdThird: 0, ClassifyNameThird: "", CollaborateType: 1, ReportLayout: 2, IsPublicPublish: 1, ReportCreateTime: v.CreateTime, InheritReportId: 0, }) } if len(addList) > 0 { err := models.InsertMultiReport(addList) if err != nil { fmt.Println("新增智能研报失败") return } reportMap := make(map[int]*models.Report) // 找出已经修复过的新的研报 { reportList, tmpErr := models.GetReportByCondition(` AND report_layout=2 and old_report_id>0 `, []interface{}{}, []string{}, " order by id asc ", false, 0, 0) if tmpErr != nil { fmt.Println("获取已修复的报告列表失败, Err:" + tmpErr.Error()) return } for _, v := range reportList { v.ReportCode = utils.MD5(strconv.Itoa(v.Id)) v.Update([]string{"ReportCode"}) reportMap[v.OldReportId] = v } } // 找出智能研报的审批单并处理 { // 智能研报走审批单的 reportApproveObj := new(report_approve.ReportApprove) approveList, tmpErr := reportApproveObj.GetItemsByCondition(` AND report_type=3 `, []interface{}{}, []string{}, " report_approve_id asc ") if tmpErr != nil { fmt.Println("获取已修复的报告列表失败, Err:" + tmpErr.Error()) return } for _, v := range approveList { reportInfo, ok := reportMap[v.ReportId] if ok { v.ReportId = reportInfo.Id v.ReportType = 1 err = v.Update([]string{"ReportType", "ReportId"}) if err != nil { fmt.Println(v.ReportApproveId, "数据修复失败,其对应的报告ID是：", v.ReportId, "；新的报告ID是：", reportInfo.Id) } } else { fmt.Println(v.ReportApproveId, "找不到对应的报告,其对应的报告ID是：", v.ReportId) } } } } fmt.Println("修复智能研报完成") } func InitPdf() { inFile := "铝年报：25年成本回落后，开启长周期故事20241204.pdf" f2, err := services.GeneralWaterMarkPdf(inFile, "上周美国馏分油库存累库95万桶，馏分油表需环比下降（-25.6万桶/日）。本期馏分油产量继续抬升-18170239278") //f2, err := services.GeneralWaterMarkPdf(inFile, "上周美国馏分油库存累库95万桶，馏分油表需环比下降（-25.6万桶/日）。本期馏分油产量继续抬升，在供增需减的环比变动下库存持续累库。馏分油供应的增加我们认为可能和进口的油种有关，今年以来美国进口的中重质原油占比不断走高，尤其是5") if err != nil { fmt.Println("生成失败，ERR:", err) return } // 创建一个新的文件 newPdf, err := os.Create("new0555.pdf") if err != nil { fmt.Println("创建临时文件失败，Err:", err) return } defer func() { _ = newPdf.Close() }() _, _ = io.Copy(newPdf, f2) } func addWaterMarkToImage() { //python3 static/watermarker.py --file 铝年报：25年成本回落后，开启长周期故事20241204.jpg -m 你好我是超长水印我是中国人我爱我的家乡-1244555432 -o newMarked.jpg fileName := "铝年报：25年成本回落后，开启长周期故事20241204.jpg" markStr := "你好我是超长水印我是中国人我爱我的家乡我家住在大海边喜欢的美食是海鲜喜欢的玩具是乐高喜欢" outFileName := "static" font := "static/SimHei.ttf" cmd := exec.Command("python3", "static/watermarker.py", "-f", fileName, "--font-family", font, "-m", markStr, "-o", outFileName) err := cmd.Run() if err != nil { fmt.Println("生成失败，ERR:", err) } }