123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185 |
- package media
- import (
- "encoding/json"
- "eta/eta_mini_ht_api/common/component/es"
- logger "eta/eta_mini_ht_api/common/component/log"
- "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"
- "time"
- )
- const (
- DESC models.Order = "desc"
- ASC models.Order = "asc"
- ESIndex = "media_index"
- 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"`
- SourceType string `json:"sourceType"`
- MediaPlaySeconds int `json:"-"`
- MediaPlayMillisSeconds int `json:"mediaPlayMillisSeconds"`
- PublishedTime string `json:"publishedTime"`
- PermissionIDs string `json:"permissionIds"`
- PermissionNames []string `json:"permissionNames,omitempty"`
- Highlight []string `json:"highlight,omitempty"`
- }
- func SearchMediaList(mediaType string, key string, from int, size int, max int64) (reports []MediaDTO, err error) {
- //同步es
- sorts := append(sortField, "publishedTime:desc")
- request := matchRangeByCondition(key, from, size, max, sorts, ConditionColumn, mediaType)
- 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)
- meida := MediaDTO{}
- err = json.Unmarshal(hit.Source, &meida)
- if err != nil {
- logger.Error("解析媒体数据失败:%v", err)
- continue
- }
- meida.Highlight = content[ESColumn]
- meida.PublishedTime = meida.PublishedTime[:10]
- meida.MediaPlayMillisSeconds = meida.MediaPlaySeconds * 1000
- reports = append(reports, meida)
- }
- return
- }
- func SearchMaxMediaId(mediaType string, key string) (total int64, latestId int64) {
- sort := []string{"mediaId:desc"}
- request := matchAllByCondition(sort, 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(sort, 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 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)
- list = append(list, dto)
- }
- }
- return
- }
- func GetMediaPageByAnalystId(mediaType string, pageInfo page.PageInfo, analystId int) (list []MediaDTO, err error) {
- offset := page.StartIndex(pageInfo.Current, pageInfo.PageSize)
- medias, err := mediaDao.GetMediaPageByAnalystId(pageInfo.LatestId, pageInfo.PageSize, offset, mediaType, analystId)
- if err == nil && medias != nil {
- for _, media := range medias {
- dto := convertMediaDTO(media)
- list = append(list, dto)
- }
- }
- return
- }
- func convertMediaDTO(media mediaDao.Media) MediaDTO {
- return MediaDTO{
- MediaId: media.Id,
- AuthorId: media.AuthorID,
- AuthorName: media.AuthorName,
- MediaType: string(media.MediaType),
- Src: media.Src,
- MediaName: media.MediaName,
- SourceType: media.SourceType,
- MediaPlaySeconds: media.MediaPlayMilliseconds,
- MediaPlayMillisSeconds: 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)
- 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(ESIndex, ESColumn, key, 0, 1, sorts, es.MatchAllByCondition).ByCondition(column, value)
- }
- func matchByCondition(sorts []string, key string, column string, value string, from int, to int) (request *es.ESQueryRequest) {
- req := new(es.ESQueryRequest)
- return req.CreateESQueryRequest(ESIndex, ESColumn, key, from, to, sorts, es.MatchAllByCondition).ByCondition(column, value)
- }
- 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(ESIndex, ESColumn, key, from, to, sorts, es.RangeByCondition).Range(0, max, ESRangeColumn).ByCondition(column, value)
- }
|