report.go 31 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155
  1. package english_report
  2. import (
  3. "encoding/json"
  4. "eta/eta_api/controllers"
  5. "eta/eta_api/models"
  6. "eta/eta_api/models/company"
  7. "eta/eta_api/models/system"
  8. "eta/eta_api/services"
  9. "eta/eta_api/services/alarm_msg"
  10. "eta/eta_api/utils"
  11. "fmt"
  12. "github.com/rdlucklib/rdluck_tools/paging"
  13. "html"
  14. "strconv"
  15. "strings"
  16. "sync"
  17. "time"
  18. )
  19. // EnglishReportController 研报活动模块
  20. type EnglishReportController struct {
  21. controllers.BaseAuthController
  22. }
  23. // @Title 新增报告接口
  24. // @Description 新增报告
  25. // @Param request body models.AddEnglishReportReq true "type json string"
  26. // @Success 200 {object} models.AddEnglishReportResp
  27. // @router /add [post]
  28. func (this *EnglishReportController) Add() {
  29. br := new(models.BaseResponse).Init()
  30. defer func() {
  31. this.Data["json"] = br
  32. this.ServeJSON()
  33. }()
  34. sysUser := this.SysUser
  35. if sysUser == nil {
  36. br.Msg = "请登录"
  37. br.ErrMsg = "请登录,SysUser Is Empty"
  38. br.Ret = 408
  39. return
  40. }
  41. var req models.AddEnglishReportReq
  42. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  43. if err != nil {
  44. br.Msg = "参数解析异常!"
  45. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  46. return
  47. }
  48. if req.Content == "" {
  49. br.Msg = "请输入内容"
  50. return
  51. }
  52. if req.Overview == "" {
  53. br.Msg = "请输入overview"
  54. return
  55. }
  56. if req.Abstract == "" {
  57. br.Msg = "请输入摘要"
  58. return
  59. }
  60. var contentSub string
  61. if req.Content != "" {
  62. content, e := services.FilterReportContentBr(req.Content)
  63. if e != nil {
  64. br.Msg = "内容去除前后空格失败"
  65. br.ErrMsg = "内容去除前后空格失败, Err: " + e.Error()
  66. return
  67. }
  68. req.Content = content
  69. contentSub, err = services.GetReportContentSub(req.Content)
  70. if err != nil {
  71. go alarm_msg.SendAlarmMsg("ContentSub 失败,Err:"+err.Error(), 3)
  72. //utils.SendEmail(utils.APPNAME+"失败提醒", "解析 ContentSub 失败,Err:"+err.Error(), utils.EmailSendToUsers)
  73. }
  74. }
  75. maxStage, err := models.GetEnglishReportStage(req.ClassifyIdFirst, req.ClassifyIdSecond)
  76. if err != nil {
  77. br.Msg = "期数获取失败!"
  78. br.ErrMsg = "期数获取失败,Err:" + err.Error()
  79. return
  80. }
  81. item := new(models.EnglishReport)
  82. item.AddType = req.AddType
  83. item.ClassifyIdFirst = req.ClassifyIdFirst
  84. item.ClassifyNameFirst = req.ClassifyNameFirst
  85. item.ClassifyIdSecond = req.ClassifyIdSecond
  86. item.ClassifyNameSecond = req.ClassifyNameSecond
  87. item.Title = req.Title
  88. item.Abstract = req.Abstract
  89. item.Author = req.Author
  90. item.Frequency = req.Frequency
  91. item.State = req.State
  92. item.Content = html.EscapeString(req.Content)
  93. item.Stage = maxStage + 1
  94. item.ContentSub = html.EscapeString(contentSub)
  95. item.CreateTime = req.CreateTime
  96. item.ModifyTime = time.Now()
  97. item.Overview = req.Overview
  98. item.AdminId = sysUser.AdminId
  99. item.AdminRealName = sysUser.RealName
  100. newReportId, err := models.AddEnglishReport(item)
  101. if err != nil {
  102. br.Msg = "保存失败"
  103. br.ErrMsg = "保存失败,Err:" + err.Error()
  104. return
  105. }
  106. reportCode := utils.MD5(strconv.Itoa(int(newReportId)))
  107. //修改唯一编码
  108. {
  109. go models.ModifyEnglishReportCode(newReportId, reportCode)
  110. }
  111. resp := new(models.AddEnglishReportResp)
  112. resp.ReportId = newReportId
  113. resp.ReportCode = reportCode
  114. br.Ret = 200
  115. br.Success = true
  116. br.Msg = "保存成功"
  117. br.Data = resp
  118. }
  119. // @Title 编辑报告接口
  120. // @Description 编辑报告
  121. // @Param request body models.EditEnglishReportReq true "type json string"
  122. // @Success 200 {object} models.EditEnglishReportResp
  123. // @router /edit [post]
  124. func (this *EnglishReportController) Edit() {
  125. br := new(models.BaseResponse).Init()
  126. defer func() {
  127. this.Data["json"] = br
  128. this.ServeJSON()
  129. }()
  130. sysUser := this.SysUser
  131. if sysUser == nil {
  132. br.Msg = "请登录"
  133. br.ErrMsg = "请登录,SysUser Is Empty"
  134. br.Ret = 408
  135. return
  136. }
  137. var req models.EditEnglishReportReq
  138. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  139. if err != nil {
  140. br.Msg = "参数解析异常!"
  141. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  142. return
  143. }
  144. if req.Content == "" {
  145. br.Msg = "请输入内容"
  146. return
  147. }
  148. if req.Overview == "" {
  149. br.Msg = "请输入overview"
  150. return
  151. }
  152. if req.Abstract == "" {
  153. br.Msg = "请输入摘要"
  154. return
  155. }
  156. markStatus, err := services.UpdateEnReportEditMark(int(req.ReportId), sysUser.AdminId, 1, sysUser.RealName)
  157. if err != nil {
  158. br.Msg = err.Error()
  159. return
  160. }
  161. if markStatus.Status == 1 {
  162. br.Msg = markStatus.Msg
  163. return
  164. }
  165. var contentSub string
  166. if req.Content != "" {
  167. content, e := services.FilterReportContentBr(req.Content)
  168. if e != nil {
  169. br.Msg = "内容去除前后空格失败"
  170. br.ErrMsg = "内容去除前后空格失败, Err: " + e.Error()
  171. return
  172. }
  173. req.Content = content
  174. contentSub, err = services.GetReportContentSub(req.Content)
  175. if err != nil {
  176. go alarm_msg.SendAlarmMsg("解析 ContentSub 失败,Err:"+err.Error(), 3)
  177. //utils.SendEmail(utils.APPNAME+"失败提醒", "解析 ContentSub 失败,Err:"+err.Error(), utils.EmailSendToUsers)
  178. }
  179. }
  180. var stage int
  181. report, _ := models.GetEnglishReportById(int(req.ReportId))
  182. if report != nil {
  183. if report.ClassifyNameFirst != req.ClassifyNameFirst || report.ClassifyNameSecond != req.ClassifyNameSecond {
  184. maxStage, _ := models.GetEnglishReportStageEdit(req.ClassifyIdFirst, req.ClassifyIdSecond, int(req.ReportId))
  185. maxStage = maxStage + 1
  186. stage = maxStage
  187. } else {
  188. stage = report.Stage
  189. }
  190. }
  191. item := new(models.EnglishReport)
  192. item.ClassifyIdFirst = req.ClassifyIdFirst
  193. item.ClassifyNameFirst = req.ClassifyNameFirst
  194. item.ClassifyIdSecond = req.ClassifyIdSecond
  195. item.ClassifyNameSecond = req.ClassifyNameSecond
  196. item.Title = req.Title
  197. item.Abstract = req.Abstract
  198. item.Author = req.Author
  199. item.Frequency = req.Frequency
  200. item.State = req.State
  201. item.Stage = stage
  202. item.Content = html.EscapeString(req.Content)
  203. item.ContentSub = html.EscapeString(contentSub)
  204. item.CreateTime = req.CreateTime
  205. item.Overview = req.Overview
  206. err = models.EditEnglishReport(item, req.ReportId)
  207. if err != nil {
  208. br.Msg = "保存失败"
  209. br.ErrMsg = "保存失败,Err:" + err.Error()
  210. return
  211. }
  212. reportCode := utils.MD5(strconv.Itoa(int(req.ReportId)))
  213. resp := new(models.EditEnglishReportResp)
  214. resp.ReportId = req.ReportId
  215. resp.ReportCode = reportCode
  216. br.Ret = 200
  217. br.Success = true
  218. br.Msg = "保存成功"
  219. br.Data = resp
  220. }
  221. // @Title 获取报告详情接口
  222. // @Description 获取报告详情
  223. // @Param request body models.ReportDetailReq true "type json string"
  224. // @Success 200 {object} models.EnglishReportDetailView
  225. // @router /detail [get]
  226. func (this *EnglishReportController) Detail() {
  227. br := new(models.BaseResponse).Init()
  228. defer func() {
  229. this.Data["json"] = br
  230. this.ServeJSON()
  231. }()
  232. /*var req models.ReportDetailReq
  233. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  234. if err != nil {
  235. br.Msg = "参数解析异常!"
  236. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  237. return
  238. }
  239. if req.ReportId <= 0 {
  240. br.Msg = "参数错误"
  241. return
  242. }*/
  243. reportId, err := this.GetInt("ReportId")
  244. if err != nil {
  245. br.Msg = "获取参数失败!"
  246. br.ErrMsg = "获取参数失败,Err:" + err.Error()
  247. return
  248. }
  249. if reportId <= 0 {
  250. br.Msg = "参数错误"
  251. return
  252. }
  253. item, err := models.GetEnglishReportById(reportId)
  254. if err != nil {
  255. if err.Error() == utils.ErrNoRow() {
  256. br.Msg = "报告已被删除"
  257. return
  258. }
  259. br.Msg = "获取失败"
  260. br.ErrMsg = "获取失败,Err:" + err.Error()
  261. return
  262. }
  263. item.Content = html.UnescapeString(item.Content)
  264. item.ContentSub = html.UnescapeString(item.ContentSub)
  265. br.Ret = 200
  266. br.Success = true
  267. br.Msg = "获取成功"
  268. br.Data = item
  269. }
  270. // ListReport
  271. // @Title 获取报告列表接口
  272. // @Description 获取报告列表
  273. // @Param PageSize query int true "每页数据条数"
  274. // @Param CurrentIndex query int true "当前页页码,从1开始"
  275. // @Param TimeType query string true "筛选的时间类别:publish_time(发布时间),modify_time(更新时间)"
  276. // @Param StartDate query string true "开始时间"
  277. // @Param EndDate query string true "结束时间"
  278. // @Param Frequency query string true "频度"
  279. // @Param ClassifyIdFirst query string true "一级分类id"
  280. // @Param ClassifyIdSecond query string true "二级分类id"
  281. // @Param State query int true "状态"
  282. // @Param KeyWord query string true "搜索关键词"
  283. // @Param PublishSort query string true "desc:降序,asc 升序(预留)"
  284. // @Param CompanyType query string false "产品类型,枚举值:'ficc','权益';不传默认返回全部"
  285. // @Success 200 {object} models.ReportListResp
  286. // @router /list [get]
  287. func (this *EnglishReportController) ListReport() {
  288. br := new(models.BaseResponse).Init()
  289. defer func() {
  290. this.Data["json"] = br
  291. this.ServeJSON()
  292. }()
  293. sysUser := this.SysUser
  294. if sysUser == nil {
  295. br.Msg = "请登录"
  296. br.ErrMsg = "请登录,SysUser Is Empty"
  297. br.Ret = 408
  298. return
  299. }
  300. pageSize, _ := this.GetInt("PageSize")
  301. currentIndex, _ := this.GetInt("CurrentIndex")
  302. timeType := this.GetString("TimeType")
  303. startDate := this.GetString("StartDate")
  304. endDate := this.GetString("EndDate")
  305. frequency := this.GetString("Frequency")
  306. classifyIdFirst, _ := this.GetInt("ClassifyIdFirst")
  307. classifyIdSecond, _ := this.GetInt("ClassifyIdSecond")
  308. state, _ := this.GetInt("State")
  309. keyWord := this.GetString("KeyWord")
  310. companyType := this.GetString("CompanyType")
  311. // 群发邮件状态筛选
  312. emailState, _ := this.GetInt("EmailState")
  313. var startSize int
  314. if pageSize <= 0 {
  315. pageSize = utils.PageSize20
  316. }
  317. if currentIndex <= 0 {
  318. currentIndex = 1
  319. }
  320. startSize = utils.StartIndex(currentIndex, pageSize)
  321. var condition string
  322. var pars []interface{}
  323. if keyWord != "" {
  324. condition += ` AND (title LIKE '%` + keyWord + `%' OR admin_real_name LIKE '%` + keyWord + `%' ) `
  325. }
  326. if timeType == "" {
  327. timeType = "publish_time"
  328. }
  329. if timeType != "publish_time" && timeType != "modify_time" {
  330. br.Msg = "请选择正确的时间"
  331. br.ErrMsg = "请选择正确的时间"
  332. return
  333. }
  334. if startDate != "" {
  335. condition += ` AND ` + timeType + ` >= ? `
  336. pars = append(pars, startDate)
  337. }
  338. if endDate != "" {
  339. condition += ` AND ` + timeType + ` <= ? `
  340. pars = append(pars, endDate)
  341. }
  342. if frequency != "" {
  343. condition += ` AND frequency = ? `
  344. pars = append(pars, frequency)
  345. }
  346. if classifyIdFirst != 0 {
  347. condition += ` AND classify_id_first = ? `
  348. pars = append(pars, classifyIdFirst)
  349. }
  350. if classifyIdSecond != 0 {
  351. condition += ` AND classify_id_second = ? `
  352. pars = append(pars, classifyIdSecond)
  353. }
  354. if state > 0 {
  355. condition += ` AND state = ? `
  356. pars = append(pars, state)
  357. }
  358. // 未群发邮件(包含推送邮件失败的)
  359. if emailState == 1 {
  360. failIds, e := models.GetHasFailEmailLogReportIds()
  361. if e != nil {
  362. br.Msg = "获取失败"
  363. br.ErrMsg = "获取存在邮件推送失败记录的英文报告IDs失败, Err:" + e.Error()
  364. return
  365. }
  366. condition += ` AND email_state = 0`
  367. if len(failIds) > 0 {
  368. condition += ` OR id IN (` + utils.GetOrmInReplace(len(failIds)) + `)`
  369. pars = append(pars, failIds)
  370. }
  371. }
  372. // 已群发邮件
  373. if emailState == 2 {
  374. successIds, e := models.GetSuccessEmailLogReportIds()
  375. if e != nil {
  376. br.Msg = "获取失败"
  377. br.ErrMsg = "获取邮件推送记录均为成功的英文报告IDs失败, Err:" + e.Error()
  378. return
  379. }
  380. condition += ` AND email_state = 1`
  381. if len(successIds) > 0 {
  382. condition += ` AND id IN (` + utils.GetOrmInReplace(len(successIds)) + `)`
  383. pars = append(pars, successIds)
  384. }
  385. }
  386. var total int
  387. var errCount, errList, errOther error
  388. var authOk bool
  389. list := make([]*models.EnglishReportList, 0)
  390. failMap := make(map[int]bool, 0) // 有群发失败记录的研报
  391. adminMap := make(map[int]string, 0) // 编辑中的研究员姓名
  392. wg := sync.WaitGroup{}
  393. wg.Add(3)
  394. // 列表总数
  395. go func() {
  396. defer func() {
  397. wg.Done()
  398. }()
  399. t, e := models.GetEnglishReportListCount(condition, pars, companyType)
  400. if e != nil {
  401. errCount = fmt.Errorf("获取英文研报Count失败, Err: %s", e.Error())
  402. return
  403. }
  404. total = t
  405. }()
  406. // 列表数据
  407. go func() {
  408. defer func() {
  409. wg.Done()
  410. }()
  411. // 限制一下富文本字段, 列表用不到
  412. fieldArr := []string{
  413. "id", "add_type", "classify_id_first", "classify_name_first", "classify_id_second", "classify_name_second", "title", "abstract", "author",
  414. "frequency", "create_time", "modify_time", "state", "publish_time", "stage", "msg_is_send", "report_code", "pv", "share_url",
  415. "pv_email", "email_state", "from_report_id", "key_takeaways", "admin_id", "admin_real_name",
  416. }
  417. items, e := models.GetEnglishReportList(condition, pars, companyType, startSize, pageSize, fieldArr)
  418. if e != nil {
  419. errList = fmt.Errorf("获取英文研报列表失败, Err: %s", e.Error())
  420. return
  421. }
  422. list = items
  423. }()
  424. // 群发权限/失败记录
  425. go func() {
  426. defer func() {
  427. wg.Done()
  428. }()
  429. // 获取邮件配置-是否有权限群发
  430. conf := new(models.EnglishReportEmailConf)
  431. authKey := "english_report_email_conf"
  432. confAuth, e := company.GetConfigDetailByCode(authKey)
  433. if e != nil {
  434. errOther = fmt.Errorf("获取群发邮件权限失败, Err: %s", e.Error())
  435. return
  436. }
  437. if confAuth.ConfigValue == "" {
  438. errOther = fmt.Errorf("群发邮件配置为空")
  439. return
  440. }
  441. if e := json.Unmarshal([]byte(confAuth.ConfigValue), &conf); e != nil {
  442. errOther = fmt.Errorf("群发邮件配置有误")
  443. return
  444. }
  445. authArr := strings.Split(conf.SendAuthGroup, ",")
  446. if utils.InArrayByStr(authArr, sysUser.RoleTypeCode) {
  447. authOk = true
  448. }
  449. // 是否有群发邮件失败的记录,标记红点
  450. failList, e := models.GetEnglishReportEmailLogFailList(0)
  451. if e != nil {
  452. errOther = fmt.Errorf("获取群发邮件记录失败, Err: %s", e.Error())
  453. return
  454. }
  455. for i := range failList {
  456. failMap[failList[i].ReportId] = true
  457. }
  458. // 获取admin, 用于匹配编辑中的研究员姓名
  459. admins, e := system.GetSysAdminList("", make([]interface{}, 0), []string{"admin_id", "real_name"}, "")
  460. if e != nil {
  461. errOther = fmt.Errorf("获取系统用户列表失败, Err: %s", e.Error())
  462. return
  463. }
  464. for _, a := range admins {
  465. adminMap[a.AdminId] = a.RealName
  466. }
  467. }()
  468. wg.Wait()
  469. if errCount != nil {
  470. br.Msg = "获取失败"
  471. br.ErrMsg = errCount.Error()
  472. return
  473. }
  474. if errList != nil {
  475. br.Msg = "获取失败"
  476. br.ErrMsg = errList.Error()
  477. return
  478. }
  479. if errOther != nil {
  480. br.Msg = "获取失败"
  481. br.ErrMsg = errOther.Error()
  482. return
  483. }
  484. // 查询分类信息
  485. var classifyIdSecondSlice []int
  486. for _, item := range list {
  487. classifyIdSecondSlice = append(classifyIdSecondSlice, item.ClassifyIdSecond)
  488. }
  489. nameList, err := models.GetEnglishClassifyFullNameByIds(classifyIdSecondSlice)
  490. if err != nil {
  491. br.Msg = "获取分类名称失败"
  492. br.ErrMsg = "获取分类名称失败, ERR:" + err.Error()
  493. return
  494. }
  495. classifyNameMap := make(map[int]*models.EnglishClassifyFullName)
  496. for _, v := range nameList {
  497. classifyNameMap[v.Id] = v
  498. }
  499. for _, item := range list {
  500. if item.State == 2 {
  501. item.ShareUrl = "https://share.hzinsights.com/reportEn?code=" + item.ReportCode
  502. }
  503. item.EmailAuth = authOk
  504. item.EmailHasFail = failMap[item.Id]
  505. // 邮箱PV大于0的时候, 不展示最初版本的PV
  506. if item.PvEmail > 0 {
  507. item.Pv = 0
  508. }
  509. // 报告是否正在编辑中
  510. var opUser models.MarkReportItem
  511. key := fmt.Sprint(`crm:enReport:edit:`, item.Id)
  512. opUserId, e := utils.Rc.RedisInt(key)
  513. if e != nil {
  514. str, te := utils.Rc.RedisString(key)
  515. if te == nil {
  516. te = json.Unmarshal([]byte(str), &opUser)
  517. if te == nil {
  518. opUserId = opUser.AdminId
  519. }
  520. }
  521. }
  522. var ret models.MarkReportResp
  523. if opUserId > 0 && opUserId != sysUser.AdminId {
  524. editor := opUser.Editor
  525. if editor == "" {
  526. editor = adminMap[opUserId]
  527. }
  528. ret.Status = 1
  529. ret.Msg = fmt.Sprintf("当前%s正在编辑报告", editor)
  530. ret.Editor = editor
  531. }
  532. if ret.Status == 0 {
  533. item.CanEdit = true
  534. } else {
  535. item.Editor = ret.Editor
  536. }
  537. //处理分类名
  538. if n, ok := classifyNameMap[item.ClassifyIdSecond]; ok {
  539. if n.RootId == 0 {
  540. item.FullClassifyName = strings.Join([]string{n.ParentName, n.ClassifyName}, "/")
  541. } else {
  542. item.FullClassifyName = strings.Join([]string{n.RootName, n.ParentName, n.ClassifyName}, "/")
  543. }
  544. item.ClassifyIdRoot = n.RootId
  545. item.ClassifyNameRoot = n.RootName
  546. }
  547. }
  548. page := paging.GetPaging(currentIndex, pageSize, total)
  549. resp := new(models.EnglishReportListResp)
  550. resp.Paging = page
  551. resp.List = list
  552. br.Ret = 200
  553. br.Success = true
  554. br.Msg = "获取成功"
  555. br.Data = resp
  556. }
  557. // @Title 发布报告接口
  558. // @Description 发布报告
  559. // @Param request body models.PublishReq true "type json string"
  560. // @Success 200 Ret=200 发布成功
  561. // @router /publish [post]
  562. func (this *EnglishReportController) PublishReport() {
  563. br := new(models.BaseResponse).Init()
  564. defer func() {
  565. this.Data["json"] = br
  566. this.ServeJSON()
  567. }()
  568. var req models.PublishReq
  569. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  570. if err != nil {
  571. br.Msg = "参数解析异常!"
  572. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  573. return
  574. }
  575. reportIds := req.ReportIds
  576. if reportIds == "" {
  577. br.Msg = "参数错误"
  578. br.ErrMsg = "参数错误,报告id不可为空"
  579. return
  580. }
  581. reportArr := strings.Split(reportIds, ",")
  582. for _, v := range reportArr {
  583. vint, err := strconv.Atoi(v)
  584. if err != nil {
  585. br.Msg = "参数错误"
  586. br.ErrMsg = "参数错误,Err:" + err.Error()
  587. return
  588. }
  589. report, err := models.GetEnglishReportById(vint)
  590. if err != nil {
  591. br.Msg = "获取报告信息失败"
  592. br.ErrMsg = "获取报告信息失败,Err:" + err.Error()
  593. return
  594. }
  595. if report == nil {
  596. br.Msg = "报告不存在"
  597. return
  598. }
  599. var tmpErr error
  600. if report.Content == "" {
  601. br.Msg = "报告内容为空,不可发布"
  602. br.ErrMsg = "报告内容为空,不需要生成,report_id:" + strconv.Itoa(report.Id)
  603. return
  604. }
  605. var publishTime string
  606. if report.PublishTime != "" {
  607. // 发布时间固定为首次发布时间
  608. publishTime = report.PublishTime
  609. } else {
  610. publishTime = time.Now().Format(utils.FormatDateTime)
  611. }
  612. if tmpErr = models.PublishEnglishReportById(report.Id, publishTime); tmpErr != nil {
  613. br.Msg = "报告发布失败"
  614. br.ErrMsg = "报告发布失败, Err:" + tmpErr.Error() + ", report_id:" + strconv.Itoa(report.Id)
  615. return
  616. }
  617. go func() {
  618. _ = services.UpdateEnglishReportEs(report.Id, 2)
  619. }()
  620. }
  621. br.Ret = 200
  622. br.Success = true
  623. br.Msg = "发布成功"
  624. }
  625. // PrePublishReport
  626. // @Title 设置定时发布接口
  627. // @Description 设置定时发布接口
  628. // @Param request body models.PrePublishReq true "type json string"
  629. // @Success 200 Ret=200 发布成功
  630. // @router /pre_publish [post]
  631. func (this *EnglishReportController) PrePublishReport() {
  632. br := new(models.BaseResponse).Init()
  633. defer func() {
  634. this.Data["json"] = br
  635. this.ServeJSON()
  636. }()
  637. var req models.PrePublishReq
  638. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  639. if err != nil {
  640. br.Msg = "参数解析异常!"
  641. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  642. return
  643. }
  644. reportId := req.ReportId
  645. if reportId == 0 {
  646. br.Msg = "参数错误"
  647. br.ErrMsg = "参数错误,报告id不可为空"
  648. return
  649. }
  650. if req.PrePublishTime == "" {
  651. br.Msg = "发布时间不能为空"
  652. return
  653. }
  654. prePublishTime, err := time.ParseInLocation(utils.FormatDateTime, req.PrePublishTime, time.Local)
  655. if err != nil {
  656. br.Msg = "发布时间格式错误"
  657. br.ErrMsg = "发布时间格式错误,Err:" + err.Error()
  658. return
  659. }
  660. if prePublishTime.Before(time.Now()) {
  661. br.Msg = "发布时间不允许选择过去时间"
  662. return
  663. }
  664. if prePublishTime.Before(time.Now().Add(2 * time.Minute)) {
  665. br.Msg = "发布时间距离当前时间太近了"
  666. return
  667. }
  668. report, err := models.GetEnglishReportById(reportId)
  669. if err != nil {
  670. br.Msg = "获取报告信息失败"
  671. br.ErrMsg = "获取报告信息失败,Err:" + err.Error()
  672. return
  673. }
  674. if report == nil {
  675. br.Msg = "报告不存在"
  676. return
  677. }
  678. if report.Content == "" {
  679. br.Msg = "报告内容为空,不可发布"
  680. br.ErrMsg = "报告内容为空,不需要生成,report_id:" + strconv.Itoa(report.Id)
  681. return
  682. }
  683. if report.State == 2 {
  684. br.Msg = "报告已发布,不可设置定时发布"
  685. return
  686. }
  687. var tmpErr error
  688. if tmpErr = models.SetPrePublishEnglishReportById(report.Id, req.PrePublishTime); tmpErr != nil {
  689. br.Msg = "设置定时发布失败"
  690. br.ErrMsg = "设置定时发布失败, Err:" + tmpErr.Error() + ", report_id:" + strconv.Itoa(report.Id)
  691. return
  692. }
  693. br.Ret = 200
  694. br.Success = true
  695. br.Msg = "定时发布成功"
  696. }
  697. // @Title 取消发布报告接口
  698. // @Description 取消发布报告
  699. // @Param request body models.PublishCancelReq true "type json string"
  700. // @Success 200 Ret=200 取消发布成功
  701. // @router /publish/cancel [post]
  702. func (this *EnglishReportController) PublishCancleReport() {
  703. br := new(models.BaseResponse).Init()
  704. defer func() {
  705. this.Data["json"] = br
  706. this.ServeJSON()
  707. }()
  708. var req models.PublishCancelReq
  709. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  710. if err != nil {
  711. br.Msg = "参数解析异常!"
  712. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  713. return
  714. }
  715. if req.ReportIds <= 0 {
  716. br.Msg = "参数错误"
  717. br.ErrMsg = "参数错误,报告id不可为空"
  718. return
  719. }
  720. err = models.PublishCancelEnglishReport(req.ReportIds)
  721. if err != nil {
  722. br.Msg = "取消发布失败"
  723. br.ErrMsg = "取消发布失败,Err:" + err.Error()
  724. return
  725. }
  726. // 更新es里的报告状态
  727. go func() {
  728. _ = services.UpdateEnglishReportEs(req.ReportIds, 1)
  729. }()
  730. br.Ret = 200
  731. br.Success = true
  732. br.Msg = "取消发布成功"
  733. }
  734. // @Title 删除报告接口
  735. // @Description 删除报告
  736. // @Param request body models.DeleteReq true "type json string"
  737. // @Success 200 Ret=200 删除成功
  738. // @router /delete [post]
  739. func (this *EnglishReportController) Delete() {
  740. br := new(models.BaseResponse).Init()
  741. defer func() {
  742. this.Data["json"] = br
  743. this.ServeJSON()
  744. }()
  745. var req models.DeleteReq
  746. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  747. if err != nil {
  748. br.Msg = "参数解析异常!"
  749. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  750. return
  751. }
  752. if req.ReportIds <= 0 {
  753. br.Msg = "参数错误"
  754. br.ErrMsg = "参数错误,报告id不可为空"
  755. return
  756. }
  757. reportInfo, err := models.GetEnglishReportById(req.ReportIds)
  758. if err != nil {
  759. if err.Error() == utils.ErrNoRow() {
  760. br.Msg = "报告不存在"
  761. return
  762. }
  763. br.Msg = "查询报告信息失败"
  764. br.ErrMsg = "查询报告信息失败, Err" + err.Error()
  765. return
  766. }
  767. // 更新es里的报告状态
  768. _ = services.UpdateEnglishReportEs(req.ReportIds, 1)
  769. if err = models.DeleteEnglishReportAndChapter(reportInfo); err != nil {
  770. br.Msg = "删除失败"
  771. br.ErrMsg = "删除报告失败, Err: " + err.Error()
  772. return
  773. }
  774. // 重置PPT关联报告
  775. go func() {
  776. _ = services.ResetPPTReport(req.ReportIds, true)
  777. }()
  778. go func() {
  779. _ = services.EnglishPolicyReportSyncCancel(reportInfo)
  780. }()
  781. br.Ret = 200
  782. br.Success = true
  783. br.Msg = "删除成功"
  784. }
  785. // @Title 保存草稿
  786. // @Description 保存草稿
  787. // @Param request body models.SaveReportContent true "type json string"
  788. // @Success 200 {object} models.ReportAuthorResp
  789. // @router /saveReportContent [post]
  790. func (this *EnglishReportController) SaveReportContent() {
  791. br := new(models.BaseResponse).Init()
  792. br.IsSendEmail = false
  793. defer func() {
  794. this.Data["json"] = br
  795. this.ServeJSON()
  796. }()
  797. var req models.SaveReportContent
  798. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  799. if err != nil {
  800. br.Msg = "参数解析异常!"
  801. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  802. return
  803. }
  804. sysUser := this.SysUser
  805. if sysUser == nil {
  806. br.Msg = "请登录"
  807. br.ErrMsg = "请登录,SysUser Is Empty"
  808. br.Ret = 408
  809. return
  810. }
  811. reportId := req.ReportId
  812. noChangeFlag := req.NoChange
  813. fmt.Println("reportId:", reportId)
  814. if reportId <= 0 {
  815. br.Msg = "请输入报告ID"
  816. return
  817. }
  818. markStatus, err := services.UpdateEnReportEditMark(int(req.ReportId), sysUser.AdminId, 1, sysUser.RealName)
  819. if err != nil {
  820. br.Msg = err.Error()
  821. return
  822. }
  823. if markStatus.Status == 1 {
  824. br.Msg = markStatus.Msg
  825. return
  826. }
  827. if noChangeFlag != 1 {
  828. content := req.Content
  829. if content == "" {
  830. content = this.GetString("Content")
  831. }
  832. if content != "" {
  833. contentClean, e := services.FilterReportContentBr(req.Content)
  834. if e != nil {
  835. br.Msg = "内容去除前后空格失败"
  836. br.ErrMsg = "内容去除前后空格失败, Err: " + e.Error()
  837. return
  838. }
  839. content = contentClean
  840. contentSub, err := services.GetReportContentSub(content)
  841. if err != nil {
  842. go alarm_msg.SendAlarmMsg("解析 ContentSub 失败,Err:"+err.Error(), 3)
  843. //utils.SendEmail(utils.APPNAME+"失败提醒", "解析 ContentSub 失败,Err:"+err.Error(), utils.EmailSendToUsers)
  844. }
  845. content = html.EscapeString(content)
  846. contentSub = html.EscapeString(contentSub)
  847. err = models.EditEnglishReportContent(reportId, content, contentSub)
  848. if err != nil {
  849. br.Msg = "保存失败"
  850. br.ErrMsg = "保存失败,Err:" + err.Error()
  851. return
  852. }
  853. go models.AddEnglishReportSaveLog(reportId, this.SysUser.AdminId, content, contentSub, this.SysUser.AdminName)
  854. }
  855. }
  856. resp := new(models.SaveReportContentResp)
  857. resp.ReportId = reportId
  858. br.Ret = 200
  859. br.Success = true
  860. br.Msg = "保存成功成功"
  861. br.Data = resp
  862. }
  863. // @Title 根据分类获取最近一次报告详情接口
  864. // @Description 根据分类获取最近一次报告详情
  865. // @Param request body models.ClassifyIdDetailReq true "type json string"
  866. // @Success 200 {object} models.Report
  867. // @router /classifyIdDetail [get]
  868. func (this *EnglishReportController) ClassifyIdDetail() {
  869. br := new(models.BaseResponse).Init()
  870. defer func() {
  871. this.Data["json"] = br
  872. this.ServeJSON()
  873. }()
  874. //classify_id_first=34&classify_id_second=36
  875. /*var req models.ClassifyIdDetailReq
  876. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  877. if err != nil {
  878. br.Msg = "参数解析异常!"
  879. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  880. return
  881. }*/
  882. classifyIdFirst, _ := this.GetInt("classify_id_first")
  883. if classifyIdFirst <= 0 {
  884. classifyIdFirst, _ = this.GetInt("ClassifyIdFirst")
  885. }
  886. classifyIdSecond, _ := this.GetInt("classify_id_second")
  887. if classifyIdSecond <= 0 {
  888. classifyIdSecond, _ = this.GetInt("ClassifyIdSecond")
  889. }
  890. item, err := models.GetEnglishReportDetailByClassifyId(classifyIdFirst, classifyIdSecond)
  891. if err != nil && err.Error() != utils.ErrNoRow() {
  892. br.Msg = "获取失败!"
  893. br.ErrMsg = "获取失败,Err:" + err.Error()
  894. return
  895. }
  896. if item != nil {
  897. item.Content = html.UnescapeString(item.Content)
  898. item.ContentSub = html.UnescapeString(item.ContentSub)
  899. }
  900. br.Ret = 200
  901. br.Success = true
  902. br.Msg = "获取成功"
  903. br.Data = item
  904. }
  905. // Author
  906. // @Title 获取报告作者接口
  907. // @Description 获取报告作者
  908. // @Param AuthorType query int true "来源类型,1:中文,2:英文"
  909. // @Param Keyword query string true "搜索关键词"
  910. // @Param CurrentIndex query int true "当前页页码,从1开始"
  911. // @Param StartDate query string true "开始时间"
  912. // @Success 200 {object} models.ReportAuthorResp
  913. // @router /author [get]
  914. func (this *EnglishReportController) Author() {
  915. br := new(models.BaseResponse).Init()
  916. defer func() {
  917. this.Data["json"] = br
  918. this.ServeJSON()
  919. }()
  920. keyword := this.GetString("Keyword")
  921. var condition string
  922. var pars []interface{}
  923. condition += ` AND author_type = 2 AND enable = 1`
  924. if keyword != `` {
  925. condition += ` AND report_author like ? `
  926. pars = append(pars, "%"+keyword+"%")
  927. }
  928. _, items, err := models.GetReportAuthorList(condition, pars, 0, 10000)
  929. if err != nil {
  930. br.Msg = "获取失败!"
  931. br.ErrMsg = "获取失败,Err:" + err.Error()
  932. return
  933. }
  934. resp := models.ReportAuthorResp{
  935. List: items,
  936. }
  937. br.Ret = 200
  938. br.Success = true
  939. br.Msg = "获取成功"
  940. br.Data = resp
  941. }
  942. // MarkEditStatus
  943. // @Title 标记报告编辑状态
  944. // @Description 标记报告编辑状态接口
  945. // @Param request body request.MarkEditEnReport true "type json string"
  946. // @Success 200 标记成功 ;202 标记成功
  947. // @router /mark [post]
  948. func (this *EnglishReportController) MarkEditStatus() {
  949. br := new(models.BaseResponse).Init()
  950. defer func() {
  951. this.Data["json"] = br
  952. this.ServeJSON()
  953. }()
  954. sysUser := this.SysUser
  955. if sysUser == nil {
  956. br.Msg = "请登录"
  957. br.ErrMsg = "请登录,SysUser Is Empty"
  958. br.Ret = 408
  959. return
  960. }
  961. var req models.MarkEditEnReport
  962. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  963. if err != nil {
  964. br.Msg = "参数解析异常!"
  965. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  966. return
  967. }
  968. if req.ReportId <= 0 {
  969. br.Msg = "缺少报告Id"
  970. return
  971. }
  972. if req.Status <= 0 {
  973. br.Msg = "标记状态异常"
  974. return
  975. }
  976. //更新标记key
  977. data, err := services.UpdateEnReportEditMark(req.ReportId, sysUser.AdminId, req.Status, sysUser.RealName)
  978. if err != nil {
  979. br.Msg = err.Error()
  980. return
  981. }
  982. msg := "标记成功"
  983. br.Ret = 200
  984. br.Success = true
  985. br.Msg = msg
  986. br.Data = data
  987. }
  988. // @Title 编辑来自策略报告的报告
  989. // @Description 编辑来自策略报告的报告
  990. // @Param request body models.EditEnglishReportFromPolicyReq true "type json string"
  991. // @Success 200 {object} models.EditEnglishReportResp
  992. // @router /edit_policy [post]
  993. func (this *EnglishReportController) EditPolicy() {
  994. br := new(models.BaseResponse).Init()
  995. defer func() {
  996. this.Data["json"] = br
  997. this.ServeJSON()
  998. }()
  999. sysUser := this.SysUser
  1000. if sysUser == nil {
  1001. br.Msg = "请登录"
  1002. br.ErrMsg = "请登录,SysUser Is Empty"
  1003. br.Ret = 408
  1004. return
  1005. }
  1006. var req models.EditEnglishReportFromPolicyReq
  1007. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  1008. if err != nil {
  1009. br.Msg = "参数解析异常!"
  1010. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  1011. return
  1012. }
  1013. if req.ReportId <= 0 {
  1014. br.Msg = "请选择报告ID"
  1015. return
  1016. }
  1017. if req.Title == "" {
  1018. br.Msg = "请输入标题"
  1019. return
  1020. }
  1021. if req.Abstract == "" {
  1022. br.Msg = "请输入摘要"
  1023. return
  1024. }
  1025. if req.Frequency == "" {
  1026. br.Msg = "请输入频度"
  1027. return
  1028. }
  1029. if req.Author == "" {
  1030. br.Msg = "请输入作者"
  1031. return
  1032. }
  1033. if req.CreateTime == "" {
  1034. br.Msg = "请输入创建时间"
  1035. return
  1036. }
  1037. //查询报告是否存在
  1038. reportInfo, err := models.GetEnglishReportItemById(int(req.ReportId))
  1039. if err != nil {
  1040. if err.Error() == utils.ErrNoRow() {
  1041. br.Msg = "报告不存在"
  1042. return
  1043. }
  1044. br.Msg = "查询报告信息失败"
  1045. br.ErrMsg = "查询报告信息失败, Err" + err.Error()
  1046. return
  1047. }
  1048. if reportInfo.State == 2 {
  1049. br.Msg = "报告已发布不允许编辑"
  1050. return
  1051. }
  1052. markStatus, err := services.UpdateEnReportEditMark(int(req.ReportId), sysUser.AdminId, 1, sysUser.RealName)
  1053. if err != nil {
  1054. br.Msg = err.Error()
  1055. return
  1056. }
  1057. if markStatus.Status == 1 {
  1058. br.Msg = markStatus.Msg
  1059. return
  1060. }
  1061. item := reportInfo
  1062. item.Title = req.Title
  1063. item.Abstract = req.Abstract
  1064. item.Author = req.Author
  1065. item.Frequency = req.Frequency
  1066. item.CreateTime = req.CreateTime
  1067. item.ModifyTime = time.Now()
  1068. err = item.Update([]string{"Title", "Abstract", "Author", "Frequency", "CreateTime", "ModifyTime"})
  1069. if err != nil {
  1070. br.Msg = "保存失败"
  1071. br.ErrMsg = "保存失败,Err:" + err.Error()
  1072. return
  1073. }
  1074. reportCode := reportInfo.ReportCode
  1075. resp := new(models.EditEnglishReportResp)
  1076. resp.ReportId = req.ReportId
  1077. resp.ReportCode = reportCode
  1078. br.Ret = 200
  1079. br.Success = true
  1080. br.Msg = "保存成功"
  1081. br.Data = resp
  1082. }