smart_report.go 33 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216
  1. package smart_report
  2. import (
  3. "encoding/json"
  4. "eta/eta_api/controllers"
  5. "eta/eta_api/models"
  6. "eta/eta_api/models/smart_report"
  7. "eta/eta_api/models/system"
  8. "eta/eta_api/services"
  9. smartReportService "eta/eta_api/services/smart_report"
  10. "eta/eta_api/utils"
  11. "fmt"
  12. "github.com/kgiannakakis/mp3duration/src/mp3duration"
  13. "github.com/rdlucklib/rdluck_tools/paging"
  14. "html"
  15. "io/ioutil"
  16. "os"
  17. "path"
  18. "strings"
  19. "time"
  20. )
  21. // SmartReportController 智能研报
  22. type SmartReportController struct {
  23. controllers.BaseAuthController
  24. }
  25. // Add
  26. // @Title 新增
  27. // @Description 新增
  28. // @Param request body smart_report.SmartReportAddReq true "type json string"
  29. // @Success 200 {object} smart_report.SmartReportItem
  30. // @router /add [post]
  31. func (this *SmartReportController) Add() {
  32. br := new(models.BaseResponse).Init()
  33. defer func() {
  34. if br.ErrMsg == "" {
  35. br.IsSendEmail = false
  36. }
  37. this.Data["json"] = br
  38. this.ServeJSON()
  39. }()
  40. sysUser := this.SysUser
  41. if sysUser == nil {
  42. br.Msg = "请登录"
  43. br.ErrMsg = "请登录,SysUser Is Empty"
  44. br.Ret = 408
  45. return
  46. }
  47. var req smart_report.SmartReportAddReq
  48. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  49. if err != nil {
  50. br.Msg = "参数解析异常!"
  51. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  52. return
  53. }
  54. if req.AddType != 1 && req.AddType != 2 {
  55. br.Msg = "请选择新增方式"
  56. return
  57. }
  58. if req.ClassifyIdFirst <= 0 || req.ClassifyIdSecond <= 0 {
  59. br.Msg = "请选择分类"
  60. return
  61. }
  62. req.Title = strings.TrimSpace(req.Title)
  63. if req.Title == "" {
  64. br.Msg = "请输入标题"
  65. return
  66. }
  67. reportOB := new(smart_report.SmartReport)
  68. stageMax, e := reportOB.GetMaxStageByClassifyId(req.ClassifyIdSecond)
  69. if e != nil {
  70. br.Msg = "操作失败"
  71. br.ErrMsg = "获取期数失败, Err: " + e.Error()
  72. return
  73. }
  74. stageMax += 1
  75. item := new(smart_report.SmartReport)
  76. item.AddType = req.AddType
  77. item.ClassifyIdFirst = req.ClassifyIdFirst
  78. item.ClassifyNameFirst = req.ClassifyNameFirst
  79. item.ClassifyIdSecond = req.ClassifyIdSecond
  80. item.ClassifyNameSecond = req.ClassifyNameSecond
  81. item.Title = req.Title
  82. item.Abstract = req.Abstract
  83. item.Author = req.Author
  84. item.Frequency = req.Frequency
  85. item.Stage = stageMax
  86. item.AdminId = sysUser.AdminId
  87. item.AdminRealName = sysUser.RealName
  88. item.LastModifyAdminId = sysUser.AdminId
  89. item.LastModifyAdminName = sysUser.RealName
  90. item.State = smart_report.SmartReportStateWaitPublish
  91. item.CreateTime = time.Now().Local()
  92. item.ModifyTime = time.Now().Local()
  93. // 继承报告
  94. if req.AddType == 2 {
  95. ob := new(smart_report.SmartReport)
  96. cond := ` AND classify_id_first = ? AND classify_id_second = ?`
  97. pars := make([]interface{}, 0)
  98. pars = append(pars, req.ClassifyIdFirst, req.ClassifyIdSecond)
  99. lastReport, e := ob.GetItemByCondition(cond, pars, "stage DESC")
  100. if e != nil && e.Error() != utils.ErrNoRow() {
  101. br.Msg = "获取失败"
  102. br.ErrMsg = "获取往期研报失败, Err: " + e.Error()
  103. return
  104. }
  105. if lastReport != nil {
  106. item.Content = lastReport.Content
  107. item.ContentSub = lastReport.ContentSub
  108. item.ContentStruct = lastReport.ContentStruct
  109. }
  110. }
  111. if e = item.Create(); e != nil {
  112. br.Msg = "操作失败"
  113. br.ErrMsg = "新增研报失败, Err: " + e.Error()
  114. return
  115. }
  116. uniqueCode := utils.MD5(fmt.Sprint("smart_", item.SmartReportId))
  117. item.ReportCode = uniqueCode
  118. if e = item.Update([]string{"ReportCode"}); e != nil {
  119. br.Msg = "操作失败"
  120. br.ErrMsg = "更新研报编码失败, Err: " + e.Error()
  121. return
  122. }
  123. resp := smart_report.FormatSmartReport2Item(item)
  124. br.Ret = 200
  125. br.Success = true
  126. br.Msg = "操作成功"
  127. br.Data = resp
  128. }
  129. // Edit
  130. // @Title 编辑
  131. // @Description 编辑
  132. // @Param request body smart_report.SmartReportEditReq true "type json string"
  133. // @Success 200 {object} smart_report.SmartReportItem
  134. // @router /edit [post]
  135. func (this *SmartReportController) Edit() {
  136. br := new(models.BaseResponse).Init()
  137. defer func() {
  138. if br.ErrMsg == "" {
  139. br.IsSendEmail = false
  140. }
  141. this.Data["json"] = br
  142. this.ServeJSON()
  143. }()
  144. sysUser := this.SysUser
  145. if sysUser == nil {
  146. br.Msg = "请登录"
  147. br.ErrMsg = "请登录,SysUser Is Empty"
  148. br.Ret = 408
  149. return
  150. }
  151. var req smart_report.SmartReportEditReq
  152. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  153. if err != nil {
  154. br.Msg = "参数解析异常!"
  155. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  156. return
  157. }
  158. if req.SmartReportId <= 0 {
  159. br.Msg = "参数有误"
  160. br.ErrMsg = "报告ID为空"
  161. return
  162. }
  163. if req.ClassifyIdFirst <= 0 || req.ClassifyIdSecond <= 0 {
  164. br.Msg = "请选择分类"
  165. return
  166. }
  167. req.Title = strings.TrimSpace(req.Title)
  168. if req.Title == "" {
  169. br.Msg = "请输入标题"
  170. return
  171. }
  172. var subContent string
  173. if req.Content != "" {
  174. req.Content = html.EscapeString(req.Content)
  175. // 前两个章节
  176. sub, e := services.GetReportContentSub(req.Content)
  177. if e != nil {
  178. br.Msg = "操作失败"
  179. br.ErrMsg = "读取报告前两个章节内容失败, Err: " + e.Error()
  180. return
  181. }
  182. subContent = html.EscapeString(sub)
  183. }
  184. req.ContentStruct = html.EscapeString(req.ContentStruct)
  185. ob := new(smart_report.SmartReport)
  186. item, e := ob.GetItemById(req.SmartReportId)
  187. if e != nil {
  188. if e.Error() == utils.ErrNoRow() {
  189. br.Msg = "报告不存在, 请刷新页面"
  190. return
  191. }
  192. br.Msg = "操作失败"
  193. br.ErrMsg = "获取研报失败, Err: " + e.Error()
  194. return
  195. }
  196. if item.State == 2 {
  197. br.Msg = "报告已发布, 请取消发布后编辑"
  198. return
  199. }
  200. // 内容是否变更, 只有内容产生变更时, 才更新最后更新人和内容更新时间字段
  201. contentModify := false
  202. if item.ClassifyIdFirst != req.ClassifyIdFirst ||
  203. item.ClassifyIdSecond != req.ClassifyIdSecond ||
  204. item.Title != req.Title ||
  205. item.Abstract != req.Abstract ||
  206. item.Frequency != req.Frequency ||
  207. utils.MD5(item.Content) != utils.MD5(req.Content) {
  208. contentModify = true
  209. }
  210. cols := []string{"ClassifyIdFirst", "ClassifyNameFirst", "ClassifyIdSecond", "ClassifyNameSecond", "Title", "Abstract", "Author",
  211. "Frequency", "Content", "ContentSub", "ContentStruct", "ModifyTime"}
  212. item.ClassifyIdFirst = req.ClassifyIdFirst
  213. item.ClassifyNameFirst = req.ClassifyNameFirst
  214. item.ClassifyIdSecond = req.ClassifyIdSecond
  215. item.ClassifyNameSecond = req.ClassifyNameSecond
  216. item.Title = req.Title
  217. item.Abstract = req.Abstract
  218. item.Author = req.Author
  219. item.Frequency = req.Frequency
  220. item.Content = req.Content
  221. item.ContentSub = subContent
  222. item.ContentStruct = req.ContentStruct
  223. item.ModifyTime = time.Now().Local()
  224. if contentModify {
  225. //fmt.Println(contentModify)
  226. item.LastModifyAdminId = sysUser.AdminId
  227. item.LastModifyAdminName = sysUser.RealName
  228. item.ContentModifyTime = time.Now().Local()
  229. cols = append(cols, "LastModifyAdminId", "LastModifyAdminName", "ContentModifyTime")
  230. }
  231. if e := item.Update(cols); e != nil {
  232. br.Msg = "操作失败"
  233. br.ErrMsg = "更新报告失败, Err: " + e.Error()
  234. return
  235. }
  236. resp := smart_report.FormatSmartReport2Item(item)
  237. br.Ret = 200
  238. br.Success = true
  239. br.Msg = "操作成功"
  240. br.Data = resp
  241. }
  242. // Remove
  243. // @Title 删除
  244. // @Description 删除
  245. // @Param request body smart_report.SmartReportRemoveReq true "type json string"
  246. // @Success 200 string "操作成功"
  247. // @router /remove [post]
  248. func (this *SmartReportController) Remove() {
  249. br := new(models.BaseResponse).Init()
  250. defer func() {
  251. if br.ErrMsg == "" {
  252. br.IsSendEmail = false
  253. }
  254. this.Data["json"] = br
  255. this.ServeJSON()
  256. }()
  257. sysUser := this.SysUser
  258. if sysUser == nil {
  259. br.Msg = "请登录"
  260. br.ErrMsg = "请登录,SysUser Is Empty"
  261. br.Ret = 408
  262. return
  263. }
  264. var req smart_report.SmartReportRemoveReq
  265. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  266. if err != nil {
  267. br.Msg = "参数解析异常!"
  268. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  269. return
  270. }
  271. if req.SmartReportId <= 0 {
  272. br.Msg = "参数有误"
  273. br.ErrMsg = "报告ID为空"
  274. return
  275. }
  276. ob := new(smart_report.SmartReport)
  277. item, e := ob.GetItemById(req.SmartReportId)
  278. if e != nil {
  279. if e.Error() == utils.ErrNoRow() {
  280. br.Ret = 200
  281. br.Success = true
  282. br.Msg = "操作成功"
  283. return
  284. }
  285. br.Msg = "操作失败"
  286. br.ErrMsg = "获取研报失败, Err: " + e.Error()
  287. return
  288. }
  289. if e = item.Del(); e != nil {
  290. br.Msg = "操作失败"
  291. br.ErrMsg = "删除报告失败, Err: " + e.Error()
  292. return
  293. }
  294. // ES更新报告为未发布
  295. go func() {
  296. _ = smartReportService.SmartReportElasticUpsert(item.SmartReportId, 1)
  297. }()
  298. br.Ret = 200
  299. br.Success = true
  300. br.Msg = "操作成功"
  301. }
  302. // Detail
  303. // @Title 详情
  304. // @Description 详情
  305. // @Param SmartReportId query int true "智能研报ID"
  306. // @Success 200 {object} smart_report.SmartReportItem
  307. // @router /detail [get]
  308. func (this *SmartReportController) Detail() {
  309. br := new(models.BaseResponse).Init()
  310. defer func() {
  311. if br.ErrMsg == "" {
  312. br.IsSendEmail = false
  313. }
  314. this.Data["json"] = br
  315. this.ServeJSON()
  316. }()
  317. sysUser := this.SysUser
  318. if sysUser == nil {
  319. br.Msg = "请登录"
  320. br.ErrMsg = "请登录,SysUser Is Empty"
  321. br.Ret = 408
  322. return
  323. }
  324. reportId, _ := this.GetInt("SmartReportId")
  325. if reportId <= 0 {
  326. br.Msg = "参数有误"
  327. br.ErrMsg = "报告ID有误"
  328. return
  329. }
  330. ob := new(smart_report.SmartReport)
  331. item, e := ob.GetItemById(reportId)
  332. if e != nil {
  333. if e.Error() == utils.ErrNoRow() {
  334. br.Msg = "报告不存在, 请刷新页面"
  335. return
  336. }
  337. br.Msg = "操作失败"
  338. br.ErrMsg = "获取研报失败, Err: " + e.Error()
  339. return
  340. }
  341. resp := smart_report.FormatSmartReport2Item(item)
  342. br.Ret = 200
  343. br.Success = true
  344. br.Msg = "获取成功"
  345. br.Data = resp
  346. }
  347. // Publish
  348. // @Title 发布/取消发布
  349. // @Description 发布/取消发布
  350. // @Param request body smart_report.SmartReportPublishReq true "type json string"
  351. // @Success 200 string "操作成功"
  352. // @router /publish [post]
  353. func (this *SmartReportController) Publish() {
  354. br := new(models.BaseResponse).Init()
  355. defer func() {
  356. if br.ErrMsg == "" {
  357. br.IsSendEmail = false
  358. }
  359. this.Data["json"] = br
  360. this.ServeJSON()
  361. }()
  362. sysUser := this.SysUser
  363. if sysUser == nil {
  364. br.Msg = "请登录"
  365. br.ErrMsg = "请登录,SysUser Is Empty"
  366. br.Ret = 408
  367. return
  368. }
  369. var req smart_report.SmartReportPublishReq
  370. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  371. if err != nil {
  372. br.Msg = "参数解析异常!"
  373. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  374. return
  375. }
  376. if req.SmartReportId <= 0 {
  377. br.Msg = "参数有误"
  378. br.ErrMsg = "报告ID为空"
  379. return
  380. }
  381. if req.PublishState != smart_report.SmartReportStateWaitPublish && req.PublishState != smart_report.SmartReportStatePublished {
  382. br.Msg = "参数有误"
  383. return
  384. }
  385. ob := new(smart_report.SmartReport)
  386. item, e := ob.GetItemById(req.SmartReportId)
  387. if e != nil {
  388. if e.Error() == utils.ErrNoRow() {
  389. br.Msg = "报告不存在, 请刷新页面"
  390. return
  391. }
  392. br.Msg = "操作失败"
  393. br.ErrMsg = "获取研报失败, Err: " + e.Error()
  394. return
  395. }
  396. cols := []string{"State", "ModifyTime"}
  397. item.State = req.PublishState
  398. item.ModifyTime = time.Now().Local()
  399. if req.PublishState == smart_report.SmartReportStatePublished {
  400. cols = append(cols, "PublishTime")
  401. item.PublishTime = time.Now().Local()
  402. // 写入队列
  403. var queue smart_report.Report2ImgQueueReq
  404. queue.ReportType = 2
  405. queue.ReportCode = item.ReportCode
  406. _ = utils.Rc.LPush(utils.CACHE_CREATE_REPORT_IMGPDF_QUEUE, queue)
  407. }
  408. // 取消发布时同时清除掉Img和PDF的文件地址
  409. if req.PublishState == smart_report.SmartReportStateWaitPublish {
  410. cols = append(cols, "DetailImgUrl", "DetailPdfUrl")
  411. item.DetailImgUrl = ""
  412. item.DetailPdfUrl = ""
  413. }
  414. if e = item.Update(cols); e != nil {
  415. br.Msg = "操作失败"
  416. br.ErrMsg = "更新研报失败, Err: " + e.Error()
  417. return
  418. }
  419. // 生成音频
  420. if req.PublishState == smart_report.SmartReportStatePublished && item.VideoUrl == "" {
  421. go smartReportService.SmartReportBuildVideoAndUpdate(item)
  422. }
  423. // ES更新报告
  424. go func() {
  425. _ = smartReportService.SmartReportElasticUpsert(item.SmartReportId, req.PublishState)
  426. }()
  427. br.Ret = 200
  428. br.Success = true
  429. br.Msg = "操作成功"
  430. }
  431. // PrePublish
  432. // @Title 设置定时发布
  433. // @Description 设置定时发布
  434. // @Param request body smart_report.SmartReportPrePublishReq true "type json string"
  435. // @Success 200 string "操作成功"
  436. // @router /pre_publish [post]
  437. func (this *SmartReportController) PrePublish() {
  438. br := new(models.BaseResponse).Init()
  439. defer func() {
  440. if br.ErrMsg == "" {
  441. br.IsSendEmail = false
  442. }
  443. this.Data["json"] = br
  444. this.ServeJSON()
  445. }()
  446. sysUser := this.SysUser
  447. if sysUser == nil {
  448. br.Msg = "请登录"
  449. br.ErrMsg = "请登录,SysUser Is Empty"
  450. br.Ret = 408
  451. return
  452. }
  453. var req smart_report.SmartReportPrePublishReq
  454. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  455. if err != nil {
  456. br.Msg = "参数解析异常!"
  457. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  458. return
  459. }
  460. reportId := req.SmartReportId
  461. if reportId == 0 {
  462. br.Msg = "参数错误"
  463. br.ErrMsg = "参数错误,报告id不可为空"
  464. return
  465. }
  466. if req.PrePublishTime == "" {
  467. br.Msg = "发布时间不能为空"
  468. return
  469. }
  470. if req.PreMsgSend != 0 && req.PreMsgSend != 1 {
  471. br.Msg = "参数错误"
  472. br.ErrMsg = "是否发送模版消息标识错误"
  473. return
  474. }
  475. preTime, e := time.ParseInLocation(utils.FormatDateTime, req.PrePublishTime, time.Local)
  476. if e != nil {
  477. br.Msg = "发布时间格式错误"
  478. br.ErrMsg = "发布时间格式错误,Err:" + e.Error()
  479. return
  480. }
  481. if preTime.Before(time.Now()) {
  482. br.Msg = "发布时间不允许选择过去时间"
  483. return
  484. }
  485. if preTime.Before(time.Now().Add(2 * time.Minute)) {
  486. br.Msg = "发布时间距离当前时间太近了"
  487. return
  488. }
  489. reportOB := new(smart_report.SmartReport)
  490. item, e := reportOB.GetItemById(reportId)
  491. if e != nil {
  492. if e.Error() == utils.ErrNoRow() {
  493. br.Msg = "报告不存在, 请刷新页面"
  494. return
  495. }
  496. br.Msg = "获取报告失败"
  497. br.ErrMsg = "获取报告失败, Err:" + e.Error()
  498. return
  499. }
  500. if item.Content == "" {
  501. br.Msg = "报告内容为空, 不可发布"
  502. return
  503. }
  504. if item.State == 2 {
  505. br.Msg = "报告已发布, 不可设置定时发布"
  506. return
  507. }
  508. item.PrePublishTime = preTime
  509. item.PreMsgSend = req.PreMsgSend
  510. cols := []string{"PrePublishTime", "PreMsgSend"}
  511. if e = item.Update(cols); e != nil {
  512. br.Msg = "操作失败"
  513. br.ErrMsg = "更新报告预发布失败, Err: " + e.Error()
  514. return
  515. }
  516. br.Ret = 200
  517. br.Success = true
  518. br.Msg = "操作成功"
  519. }
  520. // SendMsg
  521. // @Title 消息推送
  522. // @Description 消息推送
  523. // @Param request body models.SendTemplateMsgReq true "type json string"
  524. // @Success 200 Ret=200 推送成功
  525. // @router /send_msg [post]
  526. func (this *SmartReportController) SendMsg() {
  527. br := new(models.BaseResponse).Init()
  528. defer func() {
  529. if br.ErrMsg == "" {
  530. br.IsSendEmail = false
  531. }
  532. this.Data["json"] = br
  533. this.ServeJSON()
  534. }()
  535. sysUser := this.SysUser
  536. if sysUser == nil {
  537. br.Msg = "请登录"
  538. br.ErrMsg = "请登录,SysUser Is Empty"
  539. br.Ret = 408
  540. return
  541. }
  542. var req smart_report.SmartReportSendMsgReq
  543. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  544. if err != nil {
  545. br.Msg = "参数解析异常!"
  546. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  547. return
  548. }
  549. if req.SmartReportId <= 0 {
  550. br.Msg = "参数有误"
  551. br.ErrMsg = fmt.Sprintf("参数有误, SmartReportId: %d", req.SmartReportId)
  552. return
  553. }
  554. // 避免重复推送
  555. {
  556. redisKey := fmt.Sprint(utils.CACHE_SMART_REPORT_SEND_MSG, req.SmartReportId)
  557. ok := utils.Rc.SetNX(redisKey, 1, time.Second*300)
  558. if !ok {
  559. br.Msg = "报告已推送, 请勿重复推送"
  560. return
  561. }
  562. defer func() {
  563. _ = utils.Rc.Delete(redisKey)
  564. }()
  565. }
  566. reportOB := new(smart_report.SmartReport)
  567. item, e := reportOB.GetItemById(req.SmartReportId)
  568. if e != nil {
  569. if e.Error() == utils.ErrNoRow() {
  570. br.Msg = "报告不存在, 请刷新页面"
  571. return
  572. }
  573. br.Msg = "操作失败"
  574. br.ErrMsg = "获取报告失败, Err: " + e.Error()
  575. return
  576. }
  577. if item.MsgIsSend == 1 {
  578. br.Msg = "消息已推送,请勿重复操作"
  579. return
  580. }
  581. if item.State != 2 {
  582. br.Msg = "报告未发布, 不可推送"
  583. return
  584. }
  585. item.MsgIsSend = 1
  586. item.MsgSendTime = time.Now().Local()
  587. item.ModifyTime = time.Now().Local()
  588. if e = item.Update([]string{"MsgIsSend", "MsgSendTime", "ModifyTime"}); e != nil {
  589. br.Msg = "操作失败"
  590. br.ErrMsg = "更新报告推送状态失败, Err: " + e.Error()
  591. return
  592. }
  593. br.Ret = 200
  594. br.Success = true
  595. br.Msg = "操作成功"
  596. br.IsAddLog = true
  597. }
  598. // SaveContent
  599. // @Title 保存草稿
  600. // @Description 保存草稿
  601. // @Param request body smart_report.SmartReportSaveContentReq true "type json string"
  602. // @Success 200 {object} smart_report.SmartReportSaveContentResp
  603. // @router /save_content [post]
  604. func (this *SmartReportController) SaveContent() {
  605. br := new(models.BaseResponse).Init()
  606. defer func() {
  607. if br.ErrMsg == "" {
  608. br.IsSendEmail = false
  609. }
  610. this.Data["json"] = br
  611. this.ServeJSON()
  612. }()
  613. sysUser := this.SysUser
  614. if sysUser == nil {
  615. br.Msg = "请登录"
  616. br.ErrMsg = "请登录,SysUser Is Empty"
  617. br.Ret = 408
  618. return
  619. }
  620. var req smart_report.SmartReportSaveContentReq
  621. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  622. if err != nil {
  623. br.Msg = "参数解析异常!"
  624. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  625. return
  626. }
  627. resp := new(smart_report.SmartReportSaveContentResp)
  628. if req.SmartReportId <= 0 {
  629. br.Ret = 200
  630. br.Success = true
  631. br.Msg = "操作成功"
  632. br.Data = resp
  633. return
  634. }
  635. reportOB := new(smart_report.SmartReport)
  636. item, _ := reportOB.GetItemById(req.SmartReportId)
  637. if item == nil {
  638. br.Ret = 200
  639. br.Success = true
  640. br.Msg = "操作成功"
  641. br.Data = resp
  642. return
  643. }
  644. if item.State == smart_report.SmartReportStatePublished {
  645. br.Msg = "报告已发布, 不允许编辑"
  646. return
  647. }
  648. // 更新编辑状态
  649. adminIdName := make(map[int]string)
  650. admins, e := system.GetSysAdminList(``, make([]interface{}, 0), []string{}, "")
  651. if e != nil {
  652. br.Msg = "获取失败"
  653. br.ErrMsg = "获取系统用户列表失败, Err: " + e.Error()
  654. return
  655. }
  656. for _, ad := range admins {
  657. adminIdName[ad.AdminId] = ad.RealName
  658. }
  659. editing, e := smartReportService.UpdateSmartReportEditing(req.SmartReportId, 1, sysUser.AdminId, sysUser.RealName, adminIdName)
  660. if e != nil {
  661. br.Msg = e.Error()
  662. return
  663. }
  664. if editing.Status == 1 {
  665. br.Msg = editing.Msg
  666. return
  667. }
  668. // 内容有改动
  669. if req.NoChange != 1 && req.Content != "" && req.ContentStruct != "" {
  670. req.Content = html.EscapeString(req.Content)
  671. req.ContentStruct = html.EscapeString(req.ContentStruct)
  672. sub, e := services.GetReportContentSub(req.Content)
  673. if e != nil {
  674. br.Msg = "操作失败"
  675. br.ErrMsg = "读取报告前两个章节内容失败, Err: " + e.Error()
  676. return
  677. }
  678. subContent := html.EscapeString(sub)
  679. item.Content = req.Content
  680. item.ContentSub = subContent
  681. item.ContentStruct = req.ContentStruct
  682. item.ContentModifyTime = time.Now().Local()
  683. item.LastModifyAdminId = sysUser.AdminId
  684. item.LastModifyAdminName = sysUser.RealName
  685. item.ModifyTime = time.Now().Local()
  686. cols := []string{"Content", "ContentSub", "ContentStruct", "ContentModifyTime", "LastModifyAdminId", "LastModifyAdminName", "ModifyTime"}
  687. if e = item.Update(cols); e != nil {
  688. br.Msg = "操作失败"
  689. br.ErrMsg = "更新报告内容失败"
  690. return
  691. }
  692. go func() {
  693. saveLog := new(smart_report.SmartReportSaveLog)
  694. saveLog.SmartReportId = item.SmartReportId
  695. saveLog.Content = item.Content
  696. saveLog.ContentSub = item.ContentSub
  697. saveLog.ContentStruct = item.ContentStruct
  698. saveLog.AdminId = sysUser.AdminId
  699. saveLog.AdminName = sysUser.RealName
  700. saveLog.CreateTime = time.Now().Local()
  701. _ = saveLog.Create()
  702. }()
  703. }
  704. resp.SmartReportId = item.SmartReportId
  705. br.Ret = 200
  706. br.Success = true
  707. br.Msg = "保存成功"
  708. br.Data = resp
  709. }
  710. // MarkEditStatus
  711. // @Title 标记报告编辑状态
  712. // @Description 标记报告编辑状态接口
  713. // @Param request body request.MarkEditEnReport true "type json string"
  714. // @Success 200 标记成功 ;202 标记成功
  715. // @router /mark_edit [post]
  716. func (this *SmartReportController) MarkEditStatus() {
  717. br := new(models.BaseResponse).Init()
  718. defer func() {
  719. if br.ErrMsg == "" {
  720. br.IsSendEmail = false
  721. }
  722. this.Data["json"] = br
  723. this.ServeJSON()
  724. }()
  725. sysUser := this.SysUser
  726. if sysUser == nil {
  727. br.Msg = "请登录"
  728. br.ErrMsg = "请登录,SysUser Is Empty"
  729. br.Ret = 408
  730. return
  731. }
  732. var req smart_report.SmartReportMarkEditReq
  733. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  734. if err != nil {
  735. br.Msg = "参数解析异常!"
  736. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  737. return
  738. }
  739. if req.SmartReportId <= 0 {
  740. br.Msg = "缺少报告Id"
  741. return
  742. }
  743. if req.Status <= 0 {
  744. br.Msg = "标记状态异常"
  745. return
  746. }
  747. // 获取系统用户列表-用于匹配编辑中的用户
  748. adminIdName := make(map[int]string)
  749. admins, e := system.GetSysAdminList(``, make([]interface{}, 0), []string{}, "")
  750. if e != nil {
  751. br.Msg = "获取失败"
  752. br.ErrMsg = "获取系统用户列表失败, Err: " + e.Error()
  753. return
  754. }
  755. for _, ad := range admins {
  756. adminIdName[ad.AdminId] = ad.RealName
  757. }
  758. data, e := smartReportService.UpdateSmartReportEditing(req.SmartReportId, req.Status, sysUser.AdminId, sysUser.RealName, adminIdName)
  759. if e != nil {
  760. br.Msg = e.Error()
  761. return
  762. }
  763. br.Ret = 200
  764. br.Success = true
  765. br.Msg = "标记成功"
  766. br.Data = data
  767. }
  768. // List
  769. // @Title 报告列表
  770. // @Description 报告列表
  771. // @Param PageSize query int true "每页数据条数"
  772. // @Param CurrentIndex query int true "当前页页码"
  773. // @Param TimeType query string false "筛选的时间类别: publish_time-发布时间, modify_time-更新时间"
  774. // @Param StartDate query string false "开始时间"
  775. // @Param EndDate query string false "结束时间"
  776. // @Param Frequency query string false "频度"
  777. // @Param ClassifyIdFirst query int false "一级分类ID"
  778. // @Param ClassifyIdSecond query int false "二级分类ID"
  779. // @Param State query int false "发布状态: 1-待发布; 2-已发布"
  780. // @Param Keyword query string false "搜索关键词"
  781. // @Success 200 {object} smart_report.SmartReportListResp
  782. // @router /list [get]
  783. func (this *SmartReportController) List() {
  784. br := new(models.BaseResponse).Init()
  785. defer func() {
  786. if br.ErrMsg == "" {
  787. br.IsSendEmail = false
  788. }
  789. this.Data["json"] = br
  790. this.ServeJSON()
  791. }()
  792. sysUser := this.SysUser
  793. if sysUser == nil {
  794. br.Msg = "请登录"
  795. br.ErrMsg = "请登录,SysUser Is Empty"
  796. br.Ret = 408
  797. return
  798. }
  799. type SmartReportListReq struct {
  800. PageSize int `form:"PageSize"`
  801. CurrentIndex int `form:"CurrentIndex"`
  802. TimeType string `form:"TimeType"`
  803. StartDate string `form:"StartDate"`
  804. EndDate string `form:"EndDate"`
  805. Frequency string `form:"Frequency"`
  806. ClassifyIdFirst int `form:"ClassifyIdFirst"`
  807. ClassifyIdSecond int `form:"ClassifyIdSecond"`
  808. State int `form:"State"`
  809. Keyword string `form:"Keyword"`
  810. }
  811. params := new(SmartReportListReq)
  812. if e := this.ParseForm(params); e != nil {
  813. br.Msg = "获取失败"
  814. br.ErrMsg = "入参解析失败, Err: " + e.Error()
  815. return
  816. }
  817. if params.TimeType == "" {
  818. params.TimeType = "publish_time"
  819. }
  820. if params.TimeType != "publish_time" && params.TimeType != "modify_time" {
  821. br.Msg = "请选择正确的时间类型"
  822. return
  823. }
  824. // 更新时间指的是内容更新时间
  825. if params.TimeType == "modify_time" {
  826. params.TimeType = "content_modify_time"
  827. }
  828. var condition string
  829. var pars []interface{}
  830. // 筛选项
  831. {
  832. keyword := strings.TrimSpace(params.Keyword)
  833. if keyword != "" {
  834. kw := fmt.Sprint("%", keyword, "%")
  835. condition += fmt.Sprintf(` AND (title LIKE ? OR admin_real_name LIKE ? OR last_modify_admin_name LIKE ?)`)
  836. pars = append(pars, kw, kw, kw)
  837. }
  838. if params.StartDate != "" && params.EndDate != "" {
  839. st := fmt.Sprintf("%s 00:00:00", params.StartDate)
  840. ed := fmt.Sprintf("%s 23:59:59", params.EndDate)
  841. condition += fmt.Sprintf(` AND %s >= ? AND %s <= ?`, params.TimeType, params.TimeType)
  842. pars = append(pars, st, ed)
  843. }
  844. if params.Frequency != "" {
  845. condition += ` AND frequency = ?`
  846. pars = append(pars, params.Frequency)
  847. }
  848. if params.ClassifyIdFirst > 0 {
  849. condition += ` AND classify_id_first = ?`
  850. pars = append(pars, params.ClassifyIdFirst)
  851. }
  852. if params.ClassifyIdSecond > 0 {
  853. condition += ` AND classify_id_second = ?`
  854. pars = append(pars, params.ClassifyIdSecond)
  855. }
  856. if params.State > 0 {
  857. condition += ` AND state = ?`
  858. pars = append(pars, params.State)
  859. }
  860. }
  861. resp := new(smart_report.SmartReportListResp)
  862. reportOB := new(smart_report.SmartReport)
  863. total, e := reportOB.GetCountByCondition(condition, pars)
  864. if e != nil {
  865. br.Msg = "获取失败"
  866. br.ErrMsg = "获取报告总数失败, Err:" + e.Error()
  867. return
  868. }
  869. if total <= 0 {
  870. page := paging.GetPaging(params.CurrentIndex, params.PageSize, total)
  871. resp.Paging = page
  872. br.Ret = 200
  873. br.Success = true
  874. br.Msg = "获取成功"
  875. br.Data = resp
  876. return
  877. }
  878. // 分页列表
  879. var startSize int
  880. if params.PageSize <= 0 {
  881. params.PageSize = utils.PageSize20
  882. }
  883. if params.CurrentIndex <= 0 {
  884. params.CurrentIndex = 1
  885. }
  886. startSize = utils.StartIndex(params.CurrentIndex, params.PageSize)
  887. // 列表查询过滤掉富文本内容
  888. fields := []string{
  889. "smart_report_id", "report_code", "classify_id_first", "classify_name_first", "classify_id_second", "classify_name_second", "add_type",
  890. "title", "abstract", "author", "frequency", "stage", "video_url", "video_name", "video_play_seconds", "video_size", "detail_img_url", "detail_pdf_url",
  891. "admin_id", "admin_real_name", "state", "publish_time", "pre_publish_time", "pre_msg_send", "msg_is_send", "msg_send_time", "create_time", "modify_time",
  892. "last_modify_admin_id", "last_modify_admin_name", "content_modify_time", "pv", "uv",
  893. }
  894. list, e := reportOB.GetPageItemsByCondition(condition, pars, fields, "", startSize, params.PageSize)
  895. if e != nil {
  896. br.Msg = "获取失败"
  897. br.ErrMsg = "获取报告分页列表失败, Err:" + e.Error()
  898. return
  899. }
  900. // 获取系统用户列表-用于匹配编辑中的用户
  901. adminIdName := make(map[int]string)
  902. admins, e := system.GetSysAdminList(``, make([]interface{}, 0), []string{}, "")
  903. if e != nil {
  904. br.Msg = "获取失败"
  905. br.ErrMsg = "获取系统用户列表失败, Err: " + e.Error()
  906. return
  907. }
  908. for _, ad := range admins {
  909. adminIdName[ad.AdminId] = ad.RealName
  910. }
  911. for _, v := range list {
  912. item := smart_report.FormatSmartReport2Item(v)
  913. mark, e := smartReportService.UpdateSmartReportEditing(v.SmartReportId, 2, sysUser.AdminId, sysUser.RealName, adminIdName)
  914. if e != nil {
  915. br.Msg = "获取失败"
  916. br.ErrMsg = "查询编辑中标记失败, Err:" + e.Error()
  917. return
  918. }
  919. if mark.Status == 0 {
  920. item.CanEdit = true
  921. } else {
  922. item.Editor = mark.Editor
  923. }
  924. resp.List = append(resp.List, item)
  925. }
  926. page := paging.GetPaging(params.CurrentIndex, params.PageSize, total)
  927. resp.Paging = page
  928. br.Ret = 200
  929. br.Success = true
  930. br.Msg = "获取成功"
  931. br.Data = resp
  932. }
  933. // DetailImg
  934. // @Title 生成长图
  935. // @Description 生成长图
  936. // @Param SmartReportId query int true "智能研报ID"
  937. // @Success 200 {object} smart_report.SmartReportSaveContentResp
  938. // @router /detail_img [get]
  939. func (this *SmartReportController) DetailImg() {
  940. br := new(models.BaseResponse).Init()
  941. defer func() {
  942. if br.ErrMsg == "" {
  943. br.IsSendEmail = false
  944. }
  945. this.Data["json"] = br
  946. this.ServeJSON()
  947. }()
  948. sysUser := this.SysUser
  949. if sysUser == nil {
  950. br.Msg = "请登录"
  951. br.ErrMsg = "请登录,SysUser Is Empty"
  952. br.Ret = 408
  953. return
  954. }
  955. reportId, _ := this.GetInt("SmartReportId", 0)
  956. if reportId <= 0 {
  957. br.Msg = "参数有误"
  958. br.ErrMsg = fmt.Sprintf("参数有误, SmartReportId: %d", reportId)
  959. return
  960. }
  961. reportOB := new(smart_report.SmartReport)
  962. item, e := reportOB.GetItemById(reportId)
  963. if e != nil {
  964. if e.Error() == utils.ErrNoRow() {
  965. br.Msg = "报告不存在, 请刷新页面"
  966. return
  967. }
  968. br.Msg = "操作失败"
  969. br.ErrMsg = "获取报告失败, Err: " + e.Error()
  970. return
  971. }
  972. if item.DetailImgUrl != "" {
  973. br.Data = item.DetailImgUrl
  974. br.Ret = 200
  975. br.Success = true
  976. br.Msg = "获取成功"
  977. return
  978. }
  979. // 写入队列
  980. var queue smart_report.Report2ImgQueueReq
  981. queue.ReportType = 2
  982. queue.ReportCode = item.ReportCode
  983. _ = utils.Rc.LPush(utils.CACHE_CREATE_REPORT_IMGPDF_QUEUE, queue)
  984. br.Msg = "图片正在生成中, 请稍后下载"
  985. return
  986. }
  987. // LastPublishedReport
  988. // @Title 上期已发布的报告
  989. // @Description 上期已发布的报告
  990. // @Param ClassifyIdFirst query int false "一级分类ID"
  991. // @Param ClassifyIdSecond query int false "二级分类ID"
  992. // @Success 200 {object} smart_report.SmartReportItem
  993. // @router /last_published_report [get]
  994. func (this *SmartReportController) LastPublishedReport() {
  995. br := new(models.BaseResponse).Init()
  996. defer func() {
  997. if br.ErrMsg == "" {
  998. br.IsSendEmail = false
  999. }
  1000. this.Data["json"] = br
  1001. this.ServeJSON()
  1002. }()
  1003. sysUser := this.SysUser
  1004. if sysUser == nil {
  1005. br.Msg = "请登录"
  1006. br.ErrMsg = "请登录,SysUser Is Empty"
  1007. br.Ret = 408
  1008. return
  1009. }
  1010. firstId, _ := this.GetInt("ClassifyIdFirst")
  1011. secondId, _ := this.GetInt("ClassifyIdSecond")
  1012. ob := new(smart_report.SmartReport)
  1013. cond := ` AND classify_id_first = ? AND classify_id_second = ?`
  1014. pars := make([]interface{}, 0)
  1015. pars = append(pars, firstId, secondId)
  1016. item, e := ob.GetItemByCondition(cond, pars, "stage DESC")
  1017. if e != nil && e.Error() != utils.ErrNoRow() {
  1018. br.Msg = "获取失败"
  1019. br.ErrMsg = "获取研报失败, Err: " + e.Error()
  1020. return
  1021. }
  1022. resp := smart_report.FormatSmartReport2Item(item)
  1023. br.Ret = 200
  1024. br.Success = true
  1025. br.Msg = "获取成功"
  1026. br.Data = resp
  1027. }
  1028. // VoiceUpload
  1029. // @Title 音频上传
  1030. // @Description 音频上传接口
  1031. // @Param file query file true "文件"
  1032. // @Param SmartReportId query int true "报告ID"
  1033. // @Success Ret=200 上传成功
  1034. // @router /voice_upload [post]
  1035. func (this *SmartReportController) VoiceUpload() {
  1036. br := new(models.BaseResponse).Init()
  1037. defer func() {
  1038. if br.ErrMsg == "" {
  1039. br.IsSendEmail = false
  1040. }
  1041. this.Data["json"] = br
  1042. this.ServeJSON()
  1043. }()
  1044. reportId, _ := this.GetInt("SmartReportId")
  1045. if reportId <= 0 {
  1046. br.Msg = "参数有误"
  1047. br.ErrMsg = fmt.Sprintf("参数有误, SmartReportId: %d", reportId)
  1048. return
  1049. }
  1050. f, h, err := this.GetFile("file")
  1051. if err != nil {
  1052. br.Msg = "获取资源信息失败"
  1053. br.ErrMsg = "获取资源信息失败,Err:" + err.Error()
  1054. return
  1055. }
  1056. defer func() {
  1057. _ = f.Close()
  1058. }()
  1059. reportOb := new(smart_report.SmartReport)
  1060. item, e := reportOb.GetItemById(reportId)
  1061. if e != nil {
  1062. if e.Error() == utils.ErrNoRow() {
  1063. br.Msg = "报告不存在, 请刷新页面"
  1064. return
  1065. }
  1066. br.Msg = "操作失败"
  1067. br.ErrMsg = "获取报告信息失败, Err:" + err.Error()
  1068. return
  1069. }
  1070. ext := path.Ext(h.Filename)
  1071. uploadDir := utils.STATIC_DIR + "hongze/" + time.Now().Format("20060102")
  1072. if e = os.MkdirAll(uploadDir, 0766); e != nil {
  1073. br.Msg = "存储目录创建失败"
  1074. br.ErrMsg = "存储目录创建失败, Err:" + e.Error()
  1075. return
  1076. }
  1077. ossFileName := utils.GetRandStringNoSpecialChar(28) + ext
  1078. filePath := uploadDir + "/" + ossFileName
  1079. if e = this.SaveToFile("file", filePath); e != nil {
  1080. br.Msg = "文件保存失败"
  1081. br.ErrMsg = "文件保存失败, Err:" + e.Error()
  1082. return
  1083. }
  1084. defer func() {
  1085. _ = os.Remove(filePath)
  1086. }()
  1087. ossDir := "static/audio/"
  1088. resourceUrl := ``
  1089. //上传到阿里云 和 minio
  1090. if utils.ObjectStorageClient == "minio" {
  1091. resourceUrl, e = services.UploadMinIoToDir(ossFileName, filePath, ossDir, "")
  1092. if e != nil {
  1093. br.Msg = "文件上传失败"
  1094. br.ErrMsg = "文件上传失败, Err:" + e.Error()
  1095. return
  1096. }
  1097. } else {
  1098. resourceUrl, e = services.UploadAliyunToDir(ossFileName, filePath, ossDir, "")
  1099. if e != nil {
  1100. br.Msg = "文件上传失败"
  1101. br.ErrMsg = "文件上传失败, Err:" + e.Error()
  1102. return
  1103. }
  1104. }
  1105. resource := new(models.Resource)
  1106. resource.ResourceUrl = resourceUrl
  1107. resource.ResourceType = 2
  1108. resource.CreateTime = time.Now()
  1109. newId, err := models.AddResource(resource)
  1110. if err != nil {
  1111. br.Msg = "资源上传失败"
  1112. br.ErrMsg = "资源上传失败,Err:" + err.Error()
  1113. return
  1114. }
  1115. //fmt.Println(filePath)
  1116. playSeconds, err := mp3duration.Calculate(filePath)
  1117. if playSeconds <= 0 {
  1118. playSeconds, err = utils.GetVideoPlaySeconds(filePath)
  1119. if err != nil {
  1120. br.Msg = "获取音频时间失败"
  1121. br.ErrMsg = "获取音频时间失败,Err:" + err.Error()
  1122. return
  1123. }
  1124. }
  1125. createTime := item.CreateTime.Format("0102")
  1126. videoName := item.Title + "(" + createTime + ")"
  1127. fileBody, err := ioutil.ReadFile(filePath)
  1128. videoSize := len(fileBody)
  1129. sizeFloat := (float64(videoSize) / float64(1024)) / float64(1024)
  1130. sizeStr := utils.SubFloatToFloatStr(sizeFloat, 2)
  1131. item.VideoUrl = resourceUrl
  1132. item.VideoName = videoName
  1133. item.VideoPlaySeconds = playSeconds
  1134. item.VideoSize = sizeStr
  1135. item.ModifyTime = time.Now().Local()
  1136. updateCols := []string{"VideoUrl", "VideoName", "VideoPlaySeconds", "VideoSize", "ModifyTime"}
  1137. if e = item.Update(updateCols); e != nil {
  1138. br.Msg = "上传失败"
  1139. br.ErrMsg = "上传失败,Err:" + e.Error()
  1140. return
  1141. }
  1142. resp := new(models.ResourceResp)
  1143. resp.Id = newId
  1144. resp.ResourceUrl = resourceUrl
  1145. br.Msg = "上传成功"
  1146. br.Ret = 200
  1147. br.Success = true
  1148. br.Data = resp
  1149. return
  1150. }