123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244 |
- package media
- import (
- "encoding/json"
- "eta/eta_mini_ht_api/common/component/config"
- "eta/eta_mini_ht_api/common/component/es"
- logger "eta/eta_mini_ht_api/common/component/log"
- "eta/eta_mini_ht_api/common/contants"
- "eta/eta_mini_ht_api/common/utils/page"
- reportService "eta/eta_mini_ht_api/domian/report"
- "eta/eta_mini_ht_api/models"
- mediaDao "eta/eta_mini_ht_api/models/media"
- "sort"
- "strconv"
- "time"
- )
- var (
- htConfig = config.GetConfig(contants.HT).(*config.HTBizConfig)
- )
- const (
- DESC models.Order = "desc"
- ASC models.Order = "asc"
- ESColumn = "mediaName"
- ESRangeColumn = "mediaId"
- ConditionColumn = "mediaType"
- )
- var (
- sortField = []string{"_score:desc"}
- )
- func elastic() *es.ESClient {
- return es.GetInstance()
- }
- type MediaDTO struct {
- MediaId int `json:"mediaId"`
- AuthorId int `json:"authorId"`
- AuthorName string `json:"authorName,omitempty"`
- MediaType string `json:"mediaType"`
- Src string `json:"src"`
- MediaName string `json:"mediaName"`
- MediaTitle string `json:"mediaTitle,omitempty"`
- CoverSrc string `json:"coverSrc"`
- SourceType string `json:"sourceType"`
- MediaPlayMilliseconds int `json:"mediaPlayMilliseconds"`
- PublishedTime string `json:"publishedTime"`
- PermissionIDs string `json:"permissionIds"`
- PermissionNames []string `json:"permissionNames,omitempty"`
- Highlight []string `json:"highlight,omitempty"`
- Login bool `json:"login"`
- RiskLevelStatus string `json:"riskLevelStatus"`
- }
- func SearchMediaList(_ string, key string, mediaIds []int, from int, size int, max int64) (reports []MediaDTO, err error) {
- //同步es
- var docIds []string
- for _, id := range mediaIds {
- docIds = append(docIds, strconv.Itoa(id))
- }
- sorts := append(sortField, "publishedTime:desc")
- request := matchRangeWithDocIds(key, from, size, max, sorts, docIds)
- re, err := elastic().Search(request)
- if err != nil {
- logger.Error("es搜索失败:%v", err)
- }
- hits := elastic().GetSource(re.Hits)
- for _, hit := range hits {
- var content map[string][]string
- err = json.Unmarshal(hit.Highlight, &content)
- media := MediaDTO{}
- err = json.Unmarshal(hit.Source, &media)
- if err != nil {
- logger.Error("解析媒体数据失败:%v", err)
- continue
- }
- media.Highlight = content[ESColumn]
- media.PublishedTime = media.PublishedTime[:10]
- media.MediaTitle = media.Highlight[0]
- reports = append(reports, media)
- }
- return
- }
- func SearchMaxMediaId(mediaType string, key string) (total int64, latestId int64) {
- mediaSort := []string{"mediaId:desc"}
- request := matchAllByCondition(mediaSort, key, ConditionColumn, mediaType)
- //同步es
- //re, err := elastic().Search(request)
- re, err := elastic().Count(request)
- if err != nil {
- logger.Error("es搜索失败:%v", err)
- }
- count := re.Count
- total = int64(count)
- if total > 0 {
- request = matchByCondition(mediaSort, key, ConditionColumn, mediaType, 0, count)
- re, err = elastic().Search(request)
- if err != nil {
- logger.Error("es搜索异常:%v", err)
- }
- hits := elastic().GetSource(re.Hits)
- data := hits[0].Source
- media := MediaDTO{}
- err = json.Unmarshal(data, &media)
- if err != nil {
- logger.Error("获取当前最大媒体id失败:%v", err)
- return
- }
- total = int64(len(hits))
- latestId = int64(media.MediaId)
- return
- }
- return
- }
- func GetMediaPermissionMappingByPermissionIds(mediaType string, permissionIds []int) (total int64, latestId int64, ids []int) {
- ids, err := mediaDao.GetMediaPermissionMappingByPermissionId(mediaType, permissionIds)
- if err != nil {
- logger.Error("获取当前最大媒体id失败:%v", err)
- return 0, 0, ids
- }
- sort.Slice(ids, func(i, j int) bool {
- return ids[i] > ids[j]
- })
- return int64(len(ids)), int64(ids[0]), ids
- }
- func GetAnalystMediaPermissionMappingByPermissionIds(mediaType string, permissionIds []int, analystId int) (total int64, latestId int64, ids []int) {
- ids, err := mediaDao.GetMediaPermissionMappingByPermissionId(mediaType, permissionIds)
- if err != nil {
- logger.Error("获取当前最大媒体id失败:%v", err)
- return 0, 0, ids
- }
- ids, err = mediaDao.GetAnalystMediaRangeReportIds(mediaType, ids, analystId)
- if err != nil {
- logger.Error("根据研究员过滤媒体列表id失败:%v", err)
- return 0, 0, ids
- }
- if len(ids) == 0 {
- logger.Info("根据研究员过滤媒体列表id为空")
- return 0, 0, ids
- }
- sort.Slice(ids, func(i, j int) bool {
- return ids[i] > ids[j]
- })
- return int64(len(ids)), int64(ids[0]), ids
- }
- func GetTotalPageCount(mediaType string) (count int64, latestId int64) {
- return mediaDao.GetCountByMediaType(mediaType)
- }
- func GetTotalPageCountByAnalystId(mediaType string, analystId int) (count int64, latestId int64) {
- return mediaDao.GetCountByMediaTypeByAnalystId(mediaType, analystId)
- }
- func GetMediaPageByIds(mediaType string, pageInfo page.PageInfo, mediaIds []int) (list []MediaDTO, err error) {
- offset := page.StartIndex(pageInfo.Current, pageInfo.PageSize)
- medias, err := mediaDao.GetMediaPageByIds(pageInfo.LatestId, pageInfo.PageSize, offset, mediaType, mediaIds)
- if err == nil && medias != nil {
- for _, media := range medias {
- dto := convertMediaDTO(media, false)
- list = append(list, dto)
- }
- }
- return
- }
- func GetMediaPageByAnalystId(mediaType string, pageInfo page.PageInfo, analystId int, mediaIds []int) (list []MediaDTO, err error) {
- offset := page.StartIndex(pageInfo.Current, pageInfo.PageSize)
- medias, err := mediaDao.GetMediaPageByAnalystId(pageInfo.LatestId, pageInfo.PageSize, offset, mediaType, analystId, mediaIds)
- if err == nil && medias != nil {
- for _, media := range medias {
- dto := convertMediaDTO(media, false)
- list = append(list, dto)
- }
- }
- return
- }
- func convertMediaDTO(media mediaDao.Media, fullTime bool) MediaDTO {
- if fullTime {
- return MediaDTO{
- MediaId: media.Id,
- AuthorId: media.AuthorID,
- AuthorName: media.AuthorName,
- MediaType: string(media.MediaType),
- Src: media.Src,
- MediaName: media.MediaName,
- SourceType: media.SourceType,
- CoverSrc: media.CoverSrc,
- MediaPlayMilliseconds: media.MediaPlayMilliseconds,
- PermissionIDs: media.PermissionIDs,
- PublishedTime: media.PublishedTime.Format(time.DateTime),
- }
- } else {
- return MediaDTO{
- MediaId: media.Id,
- AuthorId: media.AuthorID,
- AuthorName: media.AuthorName,
- MediaType: string(media.MediaType),
- Src: media.Src,
- MediaName: media.MediaName,
- SourceType: media.SourceType,
- CoverSrc: media.CoverSrc,
- MediaPlayMilliseconds: media.MediaPlayMilliseconds,
- PermissionIDs: media.PermissionIDs,
- PublishedTime: media.PublishedTime.Format(time.DateOnly),
- }
- }
- }
- func GetMediaById(mediaType string, mediaId int) (mediaDTO MediaDTO, err error) {
- media, err := mediaDao.GetMediaById(mediaType, mediaId)
- if err != nil {
- logger.Error("查询媒体信息失败:%v", err)
- return
- }
- mediaDTO = convertMediaDTO(media, true)
- return
- }
- func GetPermissionsByIds(ids []int) (permissionDTOs []reportService.PermissionDTO, err error) {
- return reportService.GetFirstPermissionsByIds(ids)
- }
- func matchAllByCondition(sorts []string, key string, column string, value string) (request *es.ESQueryRequest) {
- req := new(es.ESQueryRequest)
- //return req.CreateESQueryRequest(htConfig.GetMediaIndex(), ESColumn, key, 0, 1, sorts, es.MatchAllByCondition).ByCondition(column, value)
- return req.CreateESQueryRequest(htConfig.GetMediaIndex(), ESColumn, key, 0, 1, sorts, es.MatchAll)
- }
- func matchByCondition(sorts []string, key string, column string, value string, from int, to int) (request *es.ESQueryRequest) {
- req := new(es.ESQueryRequest)
- //return req.CreateESQueryRequest(htConfig.GetMediaIndex(), ESColumn, key, from, to, sorts, es.MatchAllByCondition).ByCondition(column, value)
- return req.CreateESQueryRequest(htConfig.GetMediaIndex(), ESColumn, key, from, to, sorts, es.MatchAll)
- }
- func matchRangeByCondition(key string, from int, to int, max int64, sorts []string, column string, value string) (request *es.ESQueryRequest) {
- req := new(es.ESQueryRequest)
- //return req.CreateESQueryRequest(htConfig.GetMediaIndex(), ESColumn, key, from, to, sorts, es.RangeByCondition).Range(0, max, ESRangeColumn).ByCondition(column, value)
- return req.CreateESQueryRequest(htConfig.GetMediaIndex(), ESColumn, key, from, to, sorts, es.Range).Range(0, max, ESRangeColumn)
- }
- func matchRangeWithDocIds(key string, from int, to int, max int64, sorts []string, docIds []string) (request *es.ESQueryRequest) {
- req := new(es.ESQueryRequest)
- //return req.CreateESQueryRequest(htConfig.GetMediaIndex(), ESColumn, key, from, to, sorts, es.RangeByCondition).Range(0, max, ESRangeColumn).ByCondition(column, value)
- return req.CreateESQueryRequest(htConfig.GetMediaIndex(), ESColumn, key, from, to, sorts, es.Range).Range(0, max, ESRangeColumn).WithDocs(docIds)
- }
- func GetImageSrc(id int) (src string, err error) {
- return mediaDao.GetImageSrc(id)
- }
|