media_service.go 9.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244
  1. package media
  2. import (
  3. "encoding/json"
  4. "eta/eta_mini_ht_api/common/component/config"
  5. "eta/eta_mini_ht_api/common/component/es"
  6. logger "eta/eta_mini_ht_api/common/component/log"
  7. "eta/eta_mini_ht_api/common/contants"
  8. "eta/eta_mini_ht_api/common/utils/page"
  9. reportService "eta/eta_mini_ht_api/domian/report"
  10. "eta/eta_mini_ht_api/models"
  11. mediaDao "eta/eta_mini_ht_api/models/media"
  12. "sort"
  13. "strconv"
  14. "time"
  15. )
  16. var (
  17. htConfig = config.GetConfig(contants.HT).(*config.HTBizConfig)
  18. )
  19. const (
  20. DESC models.Order = "desc"
  21. ASC models.Order = "asc"
  22. ESColumn = "mediaName"
  23. ESRangeColumn = "mediaId"
  24. ConditionColumn = "mediaType"
  25. )
  26. var (
  27. sortField = []string{"_score:desc"}
  28. )
  29. func elastic() *es.ESClient {
  30. return es.GetInstance()
  31. }
  32. type MediaDTO struct {
  33. MediaId int `json:"mediaId"`
  34. AuthorId int `json:"authorId"`
  35. AuthorName string `json:"authorName,omitempty"`
  36. MediaType string `json:"mediaType"`
  37. Src string `json:"src"`
  38. MediaName string `json:"mediaName"`
  39. MediaTitle string `json:"mediaTitle,omitempty"`
  40. CoverSrc string `json:"coverSrc"`
  41. SourceType string `json:"sourceType"`
  42. MediaPlayMilliseconds int `json:"mediaPlayMilliseconds"`
  43. PublishedTime string `json:"publishedTime"`
  44. PermissionIDs string `json:"permissionIds"`
  45. PermissionNames []string `json:"permissionNames,omitempty"`
  46. Highlight []string `json:"highlight,omitempty"`
  47. Login bool `json:"login"`
  48. RiskLevelStatus string `json:"riskLevelStatus"`
  49. }
  50. func SearchMediaList(_ string, key string, mediaIds []int, from int, size int, max int64) (reports []MediaDTO, err error) {
  51. //同步es
  52. var docIds []string
  53. for _, id := range mediaIds {
  54. docIds = append(docIds, strconv.Itoa(id))
  55. }
  56. sorts := append(sortField, "publishedTime:desc")
  57. request := matchRangeWithDocIds(key, from, size, max, sorts, docIds)
  58. re, err := elastic().Search(request)
  59. if err != nil {
  60. logger.Error("es搜索失败:%v", err)
  61. }
  62. hits := elastic().GetSource(re.Hits)
  63. for _, hit := range hits {
  64. var content map[string][]string
  65. err = json.Unmarshal(hit.Highlight, &content)
  66. media := MediaDTO{}
  67. err = json.Unmarshal(hit.Source, &media)
  68. if err != nil {
  69. logger.Error("解析媒体数据失败:%v", err)
  70. continue
  71. }
  72. media.Highlight = content[ESColumn]
  73. media.PublishedTime = media.PublishedTime[:10]
  74. media.MediaTitle = media.Highlight[0]
  75. reports = append(reports, media)
  76. }
  77. return
  78. }
  79. func SearchMaxMediaId(mediaType string, key string) (total int64, latestId int64) {
  80. mediaSort := []string{"mediaId:desc"}
  81. request := matchAllByCondition(mediaSort, key, ConditionColumn, mediaType)
  82. //同步es
  83. //re, err := elastic().Search(request)
  84. re, err := elastic().Count(request)
  85. if err != nil {
  86. logger.Error("es搜索失败:%v", err)
  87. }
  88. count := re.Count
  89. total = int64(count)
  90. if total > 0 {
  91. request = matchByCondition(mediaSort, key, ConditionColumn, mediaType, 0, count)
  92. re, err = elastic().Search(request)
  93. if err != nil {
  94. logger.Error("es搜索异常:%v", err)
  95. }
  96. hits := elastic().GetSource(re.Hits)
  97. data := hits[0].Source
  98. media := MediaDTO{}
  99. err = json.Unmarshal(data, &media)
  100. if err != nil {
  101. logger.Error("获取当前最大媒体id失败:%v", err)
  102. return
  103. }
  104. total = int64(len(hits))
  105. latestId = int64(media.MediaId)
  106. return
  107. }
  108. return
  109. }
  110. func GetMediaPermissionMappingByPermissionIds(mediaType string, permissionIds []int) (total int64, latestId int64, ids []int) {
  111. ids, err := mediaDao.GetMediaPermissionMappingByPermissionId(mediaType, permissionIds)
  112. if err != nil {
  113. logger.Error("获取当前最大媒体id失败:%v", err)
  114. return 0, 0, ids
  115. }
  116. sort.Slice(ids, func(i, j int) bool {
  117. return ids[i] > ids[j]
  118. })
  119. return int64(len(ids)), int64(ids[0]), ids
  120. }
  121. func GetAnalystMediaPermissionMappingByPermissionIds(mediaType string, permissionIds []int, analystId int) (total int64, latestId int64, ids []int) {
  122. ids, err := mediaDao.GetMediaPermissionMappingByPermissionId(mediaType, permissionIds)
  123. if err != nil {
  124. logger.Error("获取当前最大媒体id失败:%v", err)
  125. return 0, 0, ids
  126. }
  127. ids, err = mediaDao.GetAnalystMediaRangeReportIds(mediaType, ids, analystId)
  128. if err != nil {
  129. logger.Error("根据研究员过滤媒体列表id失败:%v", err)
  130. return 0, 0, ids
  131. }
  132. if len(ids) == 0 {
  133. logger.Info("根据研究员过滤媒体列表id为空")
  134. return 0, 0, ids
  135. }
  136. sort.Slice(ids, func(i, j int) bool {
  137. return ids[i] > ids[j]
  138. })
  139. return int64(len(ids)), int64(ids[0]), ids
  140. }
  141. func GetTotalPageCount(mediaType string) (count int64, latestId int64) {
  142. return mediaDao.GetCountByMediaType(mediaType)
  143. }
  144. func GetTotalPageCountByAnalystId(mediaType string, analystId int) (count int64, latestId int64) {
  145. return mediaDao.GetCountByMediaTypeByAnalystId(mediaType, analystId)
  146. }
  147. func GetMediaPageByIds(mediaType string, pageInfo page.PageInfo, mediaIds []int) (list []MediaDTO, err error) {
  148. offset := page.StartIndex(pageInfo.Current, pageInfo.PageSize)
  149. medias, err := mediaDao.GetMediaPageByIds(pageInfo.LatestId, pageInfo.PageSize, offset, mediaType, mediaIds)
  150. if err == nil && medias != nil {
  151. for _, media := range medias {
  152. dto := convertMediaDTO(media, false)
  153. list = append(list, dto)
  154. }
  155. }
  156. return
  157. }
  158. func GetMediaPageByAnalystId(mediaType string, pageInfo page.PageInfo, analystId int, mediaIds []int) (list []MediaDTO, err error) {
  159. offset := page.StartIndex(pageInfo.Current, pageInfo.PageSize)
  160. medias, err := mediaDao.GetMediaPageByAnalystId(pageInfo.LatestId, pageInfo.PageSize, offset, mediaType, analystId, mediaIds)
  161. if err == nil && medias != nil {
  162. for _, media := range medias {
  163. dto := convertMediaDTO(media, false)
  164. list = append(list, dto)
  165. }
  166. }
  167. return
  168. }
  169. func convertMediaDTO(media mediaDao.Media, fullTime bool) MediaDTO {
  170. if fullTime {
  171. return MediaDTO{
  172. MediaId: media.Id,
  173. AuthorId: media.AuthorID,
  174. AuthorName: media.AuthorName,
  175. MediaType: string(media.MediaType),
  176. Src: media.Src,
  177. MediaName: media.MediaName,
  178. SourceType: media.SourceType,
  179. CoverSrc: media.CoverSrc,
  180. MediaPlayMilliseconds: media.MediaPlayMilliseconds,
  181. PermissionIDs: media.PermissionIDs,
  182. PublishedTime: media.PublishedTime.Format(time.DateTime),
  183. }
  184. } else {
  185. return MediaDTO{
  186. MediaId: media.Id,
  187. AuthorId: media.AuthorID,
  188. AuthorName: media.AuthorName,
  189. MediaType: string(media.MediaType),
  190. Src: media.Src,
  191. MediaName: media.MediaName,
  192. SourceType: media.SourceType,
  193. CoverSrc: media.CoverSrc,
  194. MediaPlayMilliseconds: media.MediaPlayMilliseconds,
  195. PermissionIDs: media.PermissionIDs,
  196. PublishedTime: media.PublishedTime.Format(time.DateOnly),
  197. }
  198. }
  199. }
  200. func GetMediaById(mediaType string, mediaId int) (mediaDTO MediaDTO, err error) {
  201. media, err := mediaDao.GetMediaById(mediaType, mediaId)
  202. if err != nil {
  203. logger.Error("查询媒体信息失败:%v", err)
  204. return
  205. }
  206. mediaDTO = convertMediaDTO(media, true)
  207. return
  208. }
  209. func GetPermissionsByIds(ids []int) (permissionDTOs []reportService.PermissionDTO, err error) {
  210. return reportService.GetFirstPermissionsByIds(ids)
  211. }
  212. func matchAllByCondition(sorts []string, key string, column string, value string) (request *es.ESQueryRequest) {
  213. req := new(es.ESQueryRequest)
  214. //return req.CreateESQueryRequest(htConfig.GetMediaIndex(), ESColumn, key, 0, 1, sorts, es.MatchAllByCondition).ByCondition(column, value)
  215. return req.CreateESQueryRequest(htConfig.GetMediaIndex(), ESColumn, key, 0, 1, sorts, es.MatchAll)
  216. }
  217. func matchByCondition(sorts []string, key string, column string, value string, from int, to int) (request *es.ESQueryRequest) {
  218. req := new(es.ESQueryRequest)
  219. //return req.CreateESQueryRequest(htConfig.GetMediaIndex(), ESColumn, key, from, to, sorts, es.MatchAllByCondition).ByCondition(column, value)
  220. return req.CreateESQueryRequest(htConfig.GetMediaIndex(), ESColumn, key, from, to, sorts, es.MatchAll)
  221. }
  222. func matchRangeByCondition(key string, from int, to int, max int64, sorts []string, column string, value string) (request *es.ESQueryRequest) {
  223. req := new(es.ESQueryRequest)
  224. //return req.CreateESQueryRequest(htConfig.GetMediaIndex(), ESColumn, key, from, to, sorts, es.RangeByCondition).Range(0, max, ESRangeColumn).ByCondition(column, value)
  225. return req.CreateESQueryRequest(htConfig.GetMediaIndex(), ESColumn, key, from, to, sorts, es.Range).Range(0, max, ESRangeColumn)
  226. }
  227. func matchRangeWithDocIds(key string, from int, to int, max int64, sorts []string, docIds []string) (request *es.ESQueryRequest) {
  228. req := new(es.ESQueryRequest)
  229. //return req.CreateESQueryRequest(htConfig.GetMediaIndex(), ESColumn, key, from, to, sorts, es.RangeByCondition).Range(0, max, ESRangeColumn).ByCondition(column, value)
  230. return req.CreateESQueryRequest(htConfig.GetMediaIndex(), ESColumn, key, from, to, sorts, es.Range).Range(0, max, ESRangeColumn).WithDocs(docIds)
  231. }
  232. func GetImageSrc(id int) (src string, err error) {
  233. return mediaDao.GetImageSrc(id)
  234. }