smart_report.go 18 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425
  1. package smart_report
  2. import (
  3. "eta/eta_api/global"
  4. "eta/eta_api/utils"
  5. "fmt"
  6. "github.com/rdlucklib/rdluck_tools/paging"
  7. "html"
  8. "strings"
  9. "time"
  10. )
  11. const (
  12. SmartReportStateWaitPublish = 1
  13. SmartReportStatePublished = 2
  14. )
  15. // SmartReport 智能研报
  16. type SmartReport struct {
  17. SmartReportId int `orm:"column(smart_report_id);pk" gorm:"primaryKey" description:"智能研报ID"`
  18. ReportCode string `description:"报告唯一编码"`
  19. ClassifyIdFirst int `description:"一级分类ID"`
  20. ClassifyNameFirst string `description:"一级分类名称"`
  21. ClassifyIdSecond int `description:"二级分类ID"`
  22. ClassifyNameSecond string `description:"二级分类名称"`
  23. AddType int `description:"新增方式:1-新增报告;2-继承报告"`
  24. Title string `description:"标题"`
  25. Abstract string `description:"摘要"`
  26. Author string `description:"作者"`
  27. Frequency string `description:"频度"`
  28. Stage int `description:"期数"`
  29. Content string `description:"内容"`
  30. ContentSub string `description:"内容前两个章节"`
  31. ContentStruct string `description:"内容组件"`
  32. VideoUrl string `description:"音频文件URL"`
  33. VideoName string `description:"音频文件名称"`
  34. VideoPlaySeconds float64 `description:"音频播放时长"`
  35. VideoSize string `description:"音频文件大小,单位M"`
  36. AdminId int `description:"创建者ID"`
  37. AdminRealName string `description:"创建者姓名"`
  38. State int `description:"1:未发布;2:已发布;3-待提交;4-待审批;5-已驳回;6-已通过"`
  39. LastModifyAdminId int `description:"最后更新人ID"`
  40. LastModifyAdminName string `description:"最后更新人姓名"`
  41. ContentModifyTime time.Time `description:"内容更新时间"`
  42. Pv int `description:"pv"`
  43. Uv int `description:"uv"`
  44. PublishTime time.Time `description:"发布时间"`
  45. PrePublishTime time.Time `description:"预发布时间"`
  46. PreMsgSend int `description:"定时发布后是否推送模版消息:0-否;1-是"`
  47. MsgIsSend int `description:"消息是否已发送:0-否;1-是"`
  48. MsgSendTime time.Time `description:"模版消息发送时间"`
  49. DetailImgUrl string `description:"报告详情长图地址"`
  50. DetailPdfUrl string `description:"报告详情PDF地址"`
  51. CreateTime time.Time `description:"创建时间"`
  52. ModifyTime time.Time `description:"修改时间"`
  53. HeadImg string `description:"报告头图地址"`
  54. EndImg string `description:"报告尾图地址"`
  55. CanvasColor string `description:"画布颜色"`
  56. ApproveTime time.Time `description:"审批时间"`
  57. ApproveId int `description:"审批ID"`
  58. NeedSplice int `description:"0-不需要 1-需要"`
  59. HeadResourceId int `description:"版头资源ID"`
  60. EndResourceId int `description:"版尾资源ID"`
  61. }
  62. func (m *SmartReport) TableName() string {
  63. return "smart_report"
  64. }
  65. func (m *SmartReport) PrimaryId() string {
  66. return "smart_report_id"
  67. }
  68. func (m *SmartReport) Create() (err error) {
  69. o := global.DbMap[utils.DbNameReport]
  70. err = o.Create(m).Error
  71. return
  72. }
  73. func (m *SmartReport) CreateMulti(items []*SmartReport) (err error) {
  74. if len(items) == 0 {
  75. return
  76. }
  77. o := global.DbMap[utils.DbNameReport]
  78. err = o.CreateInBatches(items, utils.MultiAddNum).Error
  79. return
  80. }
  81. func (m *SmartReport) Update(cols []string) (err error) {
  82. o := global.DbMap[utils.DbNameReport]
  83. err = o.Select(cols).Updates(m).Error
  84. return
  85. }
  86. func (m *SmartReport) Del() (err error) {
  87. o := global.DbMap[utils.DbNameReport]
  88. sql := fmt.Sprintf(`DELETE FROM %s WHERE %s = ? LIMIT 1`, m.TableName(), m.PrimaryId())
  89. err = o.Exec(sql, m.SmartReportId).Error
  90. return
  91. }
  92. func (m *SmartReport) MultiDel(menuIds []int) (err error) {
  93. if len(menuIds) == 0 {
  94. return
  95. }
  96. o := global.DbMap[utils.DbNameReport]
  97. sql := fmt.Sprintf(`DELETE FROM %s WHERE %s IN (%s)`, m.TableName(), m.PrimaryId(), utils.GetOrmInReplace(len(menuIds)))
  98. err = o.Exec(sql, menuIds).Error
  99. return
  100. }
  101. func (m *SmartReport) GetItemById(id int) (item *SmartReport, err error) {
  102. o := global.DbMap[utils.DbNameReport]
  103. sql := fmt.Sprintf(`SELECT * FROM %s WHERE %s = ? LIMIT 1`, m.TableName(), m.PrimaryId())
  104. err = o.Raw(sql, id).First(&item).Error
  105. return
  106. }
  107. func (m *SmartReport) GetItemByCondition(condition string, pars []interface{}, orderRule string) (item *SmartReport, err error) {
  108. o := global.DbMap[utils.DbNameReport]
  109. order := ``
  110. if orderRule != "" {
  111. order = ` ORDER BY ` + orderRule
  112. }
  113. sql := fmt.Sprintf(`SELECT * FROM %s WHERE 1=1 %s %s LIMIT 1`, m.TableName(), condition, order)
  114. err = o.Raw(sql, pars...).First(&item).Error
  115. return
  116. }
  117. func (m *SmartReport) GetCountByCondition(condition string, pars []interface{}) (count int, err error) {
  118. o := global.DbMap[utils.DbNameReport]
  119. sql := fmt.Sprintf(`SELECT COUNT(1) FROM %s WHERE 1=1 %s`, m.TableName(), condition)
  120. err = o.Raw(sql, pars...).Scan(&count).Error
  121. return
  122. }
  123. func (m *SmartReport) GetItemsByCondition(condition string, pars []interface{}, fieldArr []string, orderRule string) (items []*SmartReport, err error) {
  124. o := global.DbMap[utils.DbNameReport]
  125. fields := strings.Join(fieldArr, ",")
  126. if len(fieldArr) == 0 {
  127. fields = `*`
  128. }
  129. order := `ORDER BY create_time DESC`
  130. if orderRule != "" {
  131. order = ` ORDER BY ` + orderRule
  132. }
  133. sql := fmt.Sprintf(`SELECT %s FROM %s WHERE 1=1 %s %s`, fields, m.TableName(), condition, order)
  134. err = o.Raw(sql, pars...).Find(&items).Error
  135. return
  136. }
  137. func (m *SmartReport) GetPageItemsByCondition(condition string, pars []interface{}, fieldArr []string, orderRule string, startSize, pageSize int) (items []*SmartReport, err error) {
  138. o := global.DbMap[utils.DbNameReport]
  139. fields := strings.Join(fieldArr, ",")
  140. if len(fieldArr) == 0 {
  141. fields = `*`
  142. }
  143. // 排序:1:未发布;2:已发布;3-待提交;4-待审批;5-已驳回;6-已通过
  144. order := `ORDER BY FIELD(state,3,1,4,5,6,2), modify_time DESC`
  145. if orderRule != "" {
  146. order = ` ORDER BY ` + orderRule
  147. }
  148. sql := fmt.Sprintf(`SELECT %s FROM %s WHERE 1=1 %s %s LIMIT ?,?`, fields, m.TableName(), condition, order)
  149. pars = append(pars, startSize, pageSize)
  150. err = o.Raw(sql, pars...).Find(&items).Error
  151. return
  152. }
  153. func (m *SmartReport) GetMaxStageByClassifyId(classifyId int) (stage int, err error) {
  154. o := global.DbMap[utils.DbNameReport]
  155. sql := fmt.Sprintf(`SELECT MAX(stage) AS max_stage FROM %s WHERE classify_id_second = ?`, m.TableName())
  156. err = o.Raw(sql, classifyId).Scan(&stage).Error
  157. return
  158. }
  159. // SmartReportItem 智能研报信息
  160. type SmartReportItem struct {
  161. SmartReportId int `description:"智能研报ID"`
  162. ReportCode string `description:"报告唯一编码"`
  163. ClassifyIdFirst int `description:"一级分类ID"`
  164. ClassifyNameFirst string `description:"一级分类名称"`
  165. ClassifyIdSecond int `description:"二级分类ID"`
  166. ClassifyNameSecond string `description:"二级分类名称"`
  167. AddType int `description:"新增方式:1-新增报告;2-继承报告"`
  168. Title string `description:"标题"`
  169. Abstract string `description:"摘要"`
  170. Author string `description:"作者"`
  171. Frequency string `description:"频度"`
  172. Stage int `description:"期数"`
  173. Content string `description:"内容"`
  174. ContentSub string `description:"内容前两个章节"`
  175. ContentStruct string `description:"内容组件"`
  176. VideoUrl string `description:"音频文件URL"`
  177. VideoName string `description:"音频文件名称"`
  178. VideoPlaySeconds float64 `description:"音频播放时长"`
  179. VideoSize string `description:"音频文件大小,单位M"`
  180. AdminId int `description:"创建者姓名"`
  181. AdminRealName string `description:"创建者姓名"`
  182. LastModifyAdminId int `description:"最后更新人ID"`
  183. LastModifyAdminName string `description:"最后更新人姓名"`
  184. ContentModifyTime string `description:"内容更新时间"`
  185. Pv int `description:"pv"`
  186. Uv int `description:"uv"`
  187. State int `description:"1:未发布;2:已发布;3-待提交;4-待审批;5-已驳回;6-已通过"`
  188. PublishTime string `description:"发布时间"`
  189. PrePublishTime string `description:"预发布时间"`
  190. MsgIsSend int `description:"消息是否已发送:0-否;1-是"`
  191. MsgSendTime string `description:"模版消息发送时间"`
  192. DetailImgUrl string `description:"报告详情长图地址"`
  193. DetailPdfUrl string `description:"报告详情PDF地址"`
  194. CreateTime string `description:"创建时间"`
  195. ModifyTime string `description:"修改时间"`
  196. ApproveTime string `description:"审批时间"`
  197. CanEdit bool `description:"是否可编辑"`
  198. Editor string `description:"当前编辑人"`
  199. HeadImg string `description:"报告头图地址"`
  200. EndImg string `description:"报告尾图地址"`
  201. CanvasColor string `description:"画布颜色"`
  202. NeedSplice int `description:"0-不需要 1-需要"`
  203. HeadResourceId int `description:"版头资源ID"`
  204. EndResourceId int `description:"版尾资源ID"`
  205. HeadStyle string `description:"版头样式"`
  206. EndStyle string `description:"版尾样式"`
  207. }
  208. // FormatSmartReport2Item 格式化智能研报数据格式
  209. func FormatSmartReport2Item(origin *SmartReport) (item *SmartReportItem) {
  210. item = new(SmartReportItem)
  211. if origin == nil {
  212. return
  213. }
  214. item.SmartReportId = origin.SmartReportId
  215. item.ReportCode = origin.ReportCode
  216. item.ClassifyIdFirst = origin.ClassifyIdFirst
  217. item.ClassifyNameFirst = origin.ClassifyNameFirst
  218. item.ClassifyIdSecond = origin.ClassifyIdSecond
  219. item.ClassifyNameSecond = origin.ClassifyNameSecond
  220. item.AddType = origin.AddType
  221. item.Title = origin.Title
  222. item.Abstract = origin.Abstract
  223. item.Author = origin.Author
  224. item.Frequency = origin.Frequency
  225. item.Stage = origin.Stage
  226. item.Content = html.UnescapeString(origin.Content)
  227. item.ContentSub = html.UnescapeString(origin.ContentSub)
  228. item.ContentStruct = html.UnescapeString(origin.ContentStruct)
  229. item.VideoUrl = origin.VideoUrl
  230. item.VideoName = origin.VideoName
  231. item.VideoPlaySeconds = origin.VideoPlaySeconds
  232. item.VideoSize = origin.VideoSize
  233. item.AdminId = origin.AdminId
  234. item.AdminRealName = origin.AdminRealName
  235. item.LastModifyAdminId = origin.LastModifyAdminId
  236. item.LastModifyAdminName = origin.LastModifyAdminName
  237. item.ContentModifyTime = utils.TimeTransferString(utils.FormatDateTime, origin.ContentModifyTime)
  238. item.Pv = origin.Pv
  239. item.Uv = origin.Uv
  240. item.State = origin.State
  241. item.PublishTime = utils.TimeTransferString(utils.FormatDateTime, origin.PublishTime)
  242. item.PrePublishTime = utils.TimeTransferString(utils.FormatDateTime, origin.PrePublishTime)
  243. item.MsgIsSend = origin.MsgIsSend
  244. item.MsgSendTime = utils.TimeTransferString(utils.FormatDateTime, origin.MsgSendTime)
  245. item.DetailImgUrl = origin.DetailImgUrl
  246. item.DetailPdfUrl = origin.DetailPdfUrl
  247. item.CreateTime = utils.TimeTransferString(utils.FormatDateTime, origin.CreateTime)
  248. item.ModifyTime = utils.TimeTransferString(utils.FormatDateTime, origin.ModifyTime)
  249. item.HeadImg = origin.HeadImg
  250. item.EndImg = origin.EndImg
  251. item.CanvasColor = origin.CanvasColor
  252. item.ApproveTime = utils.TimeTransferString(utils.FormatDateTime, origin.ApproveTime)
  253. item.NeedSplice = origin.NeedSplice
  254. item.HeadResourceId = origin.HeadResourceId
  255. item.EndResourceId = origin.EndResourceId
  256. return
  257. }
  258. // SmartReportAddReq 新增智能研报请求体
  259. type SmartReportAddReq struct {
  260. AddType int `description:"新增方式:1:新增报告,2:继承报告"`
  261. ClassifyIdFirst int `description:"一级分类ID"`
  262. ClassifyNameFirst string `description:"一级分类名称"`
  263. ClassifyIdSecond int `description:"二级分类ID"`
  264. ClassifyNameSecond string `description:"二级分类名称"`
  265. Title string `description:"标题"`
  266. Abstract string `description:"摘要"`
  267. Author string `description:"作者"`
  268. Frequency string `description:"频度"`
  269. HeadImg string `description:"报告头图地址"`
  270. EndImg string `description:"报告尾图地址"`
  271. CanvasColor string `description:"画布颜色"`
  272. HeadResourceId int `description:"版头资源ID"`
  273. EndResourceId int `description:"版尾资源ID"`
  274. }
  275. // SmartReportEditReq 编辑智能研报请求体
  276. type SmartReportEditReq struct {
  277. SmartReportAddReq
  278. SmartReportId int `description:"智能研报ID"`
  279. Content string `description:"内容"`
  280. ContentStruct string `description:"内容结构"`
  281. HeadImg string `description:"报告头图地址"`
  282. EndImg string `description:"报告尾图地址"`
  283. CanvasColor string `description:"画布颜色"`
  284. HeadResourceId int `description:"版头资源ID"`
  285. EndResourceId int `description:"版尾资源ID"`
  286. }
  287. // SmartReportRemoveReq 删除智能研报请求体
  288. type SmartReportRemoveReq struct {
  289. SmartReportId int `description:"智能研报ID"`
  290. }
  291. // SmartReportPublishReq 发布智能研报请求体
  292. type SmartReportPublishReq struct {
  293. SmartReportId int `description:"智能研报ID"`
  294. PublishState int `description:"1-取消发布; 2-发布"`
  295. ReportUrl string `description:"报告Url"`
  296. }
  297. // SmartReportPrePublishReq 预发布智能研报请求体
  298. type SmartReportPrePublishReq struct {
  299. SmartReportId int `description:"智能研报ID"`
  300. PrePublishTime string `description:"预发布时间"`
  301. PreMsgSend int `description:"定时发布成功后是否立即推送模版消息:0否,1是"`
  302. ReportUrl string `description:"报告Url"`
  303. }
  304. // SmartReportSaveContentReq 保存草稿请求体
  305. type SmartReportSaveContentReq struct {
  306. SmartReportId int `description:"智能研报ID"`
  307. Content string `description:"内容"`
  308. ContentStruct string `description:"内容结构"`
  309. NoChange int `description:"内容是否未改变:1:内容未改变"`
  310. HeadImg string `description:"报告头图地址"`
  311. EndImg string `description:"报告尾图地址"`
  312. CanvasColor string `description:"画布颜色"`
  313. HeadResourceId int `description:"版头资源ID"`
  314. EndResourceId int `description:"版尾资源ID"`
  315. }
  316. // SmartReportSaveContentResp 保存草稿响应体
  317. type SmartReportSaveContentResp struct {
  318. SmartReportId int `description:"智能研报ID"`
  319. }
  320. // SmartReportSendMsgReq 消息推送请求体
  321. type SmartReportSendMsgReq struct {
  322. SmartReportId int `description:"智能研报ID"`
  323. }
  324. // SmartReportMarkEditReq 标记编辑英文研报的请求数据
  325. type SmartReportMarkEditReq struct {
  326. SmartReportId int `description:"智能研报ID"`
  327. Status int `description:"标记状态: 1-编辑中; 2-编辑完成"`
  328. }
  329. // SmartReportListResp 智能研报
  330. type SmartReportListResp struct {
  331. List []*SmartReportItem
  332. Paging *paging.PagingItem `description:"分页数据"`
  333. }
  334. // ElasticSmartReport 智能研报es
  335. type ElasticSmartReport struct {
  336. SmartReportId int `description:"智能研报ID"`
  337. Title string `description:"标题"`
  338. Abstract string `description:"摘要"`
  339. BodyContent string `description:"内容"`
  340. PublishTime string `description:"发布时间"`
  341. PublishState int `description:"发布状态 1-未发布 2-已发布"`
  342. Author string `description:"作者"`
  343. ClassifyIdFirst int `description:"一级分类ID"`
  344. ClassifyNameFirst string `description:"一级分类名称"`
  345. ClassifyIdSecond int `description:"二级分类ID"`
  346. ClassifyNameSecond string `description:"二级分类名称"`
  347. StageStr string `description:"报告期数"`
  348. Frequency string `description:"频度"`
  349. }
  350. // Report2ImgQueueReq 报告详情生成长图队列请求体
  351. type Report2ImgQueueReq struct {
  352. ReportType int `description:"报告类型: 1-研报; 2-智能研报"`
  353. ReportCode string `description:"报告唯一编码"`
  354. }
  355. // UpdateSmartReportsStateByCond 批量更新报告状态
  356. func UpdateSmartReportsStateByCond(classifyFirstId, classifySecondId, oldState, newState int) (err error) {
  357. o := global.DbMap[utils.DbNameReport]
  358. cond := ``
  359. if classifyFirstId > 0 {
  360. cond += fmt.Sprintf(` AND classify_id_first = %d`, classifyFirstId)
  361. }
  362. if classifySecondId > 0 {
  363. cond += fmt.Sprintf(` AND classify_id_second = %d`, classifySecondId)
  364. }
  365. sql := fmt.Sprintf(`UPDATE smart_report SET state = ?, pre_publish_time = NULL WHERE state = ? %s`, cond)
  366. err = o.Exec(sql, newState, oldState).Error
  367. return
  368. }
  369. // UpdateSmartReportsStateBySecondIds 批量更新二级分类报告状态
  370. func UpdateSmartReportsStateBySecondIds(oldState, newState int, secondIds []int) (err error) {
  371. if len(secondIds) <= 0 {
  372. return
  373. }
  374. o := global.DbMap[utils.DbNameReport]
  375. // (有审批流的)未发布->待提交
  376. sql := fmt.Sprintf(`UPDATE smart_report SET state = ?, pre_publish_time = NULL WHERE state = ? AND classify_id_second IN (%s)`, utils.GetOrmInReplace(len(secondIds)))
  377. err = o.Exec(sql, newState, oldState, secondIds).Error
  378. if err != nil {
  379. return
  380. }
  381. // (无审批流的)待提交->未发布
  382. sql = fmt.Sprintf(`UPDATE smart_report SET state = ?, pre_publish_time = NULL WHERE state = ? AND classify_id_second NOT IN (%s)`, utils.GetOrmInReplace(len(secondIds)))
  383. err = o.Exec(sql, oldState, newState, secondIds).Error
  384. return
  385. }
  386. // UpdatePdfUrlSmartReportById 清空pdf相关字段
  387. func UpdatePdfUrlSmartReportById(reportId int) (err error) {
  388. o := global.DbMap[utils.DbNameReport]
  389. sql := `UPDATE smart_report SET detail_img_url = '',detail_pdf_url='',modify_time=NOW() WHERE smart_report_id = ? `
  390. err = o.Exec(sql, reportId).Error
  391. return
  392. }