135 lines
3.3 KiB
Go
135 lines
3.3 KiB
Go
package mediafiller
|
|
|
|
import (
|
|
"math/rand"
|
|
"service/dbstruct"
|
|
"service/library/logger"
|
|
"time"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
func init() {
|
|
rand.Seed(time.Now().Unix())
|
|
}
|
|
|
|
var defaultMediaFiller *MediaFiller
|
|
|
|
type GetImageByIdsFunc func(ctx *gin.Context, ids []int64) ([]*dbstruct.Image, error)
|
|
type GetVideoByIdsFunc func(ctx *gin.Context, ids []int64) ([]*dbstruct.Video, error)
|
|
|
|
type MediaFiller struct {
|
|
fileServerDomainName string //文件服务器域名
|
|
getImageByIds GetImageByIdsFunc //获取图片数据源
|
|
getVideoByIds GetVideoByIdsFunc //获取视频数据源
|
|
}
|
|
|
|
func InitDefaultMediaFiller() {
|
|
defaultMediaFiller = &MediaFiller{}
|
|
}
|
|
|
|
func SetGetImageByIdsFunc(getImageByIdsFunc GetImageByIdsFunc) {
|
|
defaultMediaFiller.getImageByIds = getImageByIdsFunc
|
|
}
|
|
|
|
func SetGetVideoByIdsFunc(getVideoByIdsFunc GetVideoByIdsFunc) {
|
|
defaultMediaFiller.getVideoByIds = getVideoByIdsFunc
|
|
}
|
|
|
|
func SetFileServerDomainName(fileServerDomainName string) {
|
|
defaultMediaFiller.fileServerDomainName = fileServerDomainName
|
|
}
|
|
|
|
func (p *MediaFiller) GetFileServerDomain() string {
|
|
// cdn测试
|
|
//if rand.Intn(100) < 20 {
|
|
return "https://filecdn01.tiefen.fun/"
|
|
//}
|
|
return defaultMediaFiller.fileServerDomainName
|
|
}
|
|
|
|
func FillEntity(ctx *gin.Context, entity MediaFillable) error {
|
|
videoIds := entity.GetVideoIds()
|
|
videoMap, err := getVideoMapByIds(ctx, videoIds)
|
|
if err != nil {
|
|
logger.Error("mediafiller component getVideoMapByIds failed : %v", err)
|
|
return err
|
|
}
|
|
|
|
imageIds := entity.GetImageIds()
|
|
// 获取视频封面
|
|
for _, v := range videoMap {
|
|
imageIds = append(imageIds, v.CoverId)
|
|
}
|
|
imageMap, err := getImageMapByIds(ctx, imageIds)
|
|
if err != nil {
|
|
logger.Error("mediafiller component getImageMapByIds failed : %v", err)
|
|
return err
|
|
}
|
|
|
|
images := make([]*dbstruct.ToCImage, 0)
|
|
videos := make([]*dbstruct.ToCVideo, 0)
|
|
|
|
for _, imageId := range imageIds {
|
|
if image, ok := imageMap[imageId]; ok {
|
|
images = append(images, transToCImage(image))
|
|
}
|
|
}
|
|
for _, videoId := range videoIds {
|
|
if video, ok := videoMap[videoId]; ok {
|
|
videos = append(videos, transToCVideo(video, imageMap[video.CoverId]))
|
|
}
|
|
}
|
|
entity.SetImages(images)
|
|
entity.SetVideos(videos)
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
func FillList(ctx *gin.Context, list []MediaFillable) error {
|
|
imageIds := make([]int64, 0)
|
|
videoIds := make([]int64, 0)
|
|
|
|
for _, v := range list {
|
|
imageIds = append(imageIds, v.GetImageIds()...)
|
|
videoIds = append(videoIds, v.GetVideoIds()...)
|
|
}
|
|
|
|
videoMap, err := getVideoMapByIds(ctx, videoIds)
|
|
if err != nil {
|
|
logger.Error("mediafiller component getVideoMapByIds failed : %v", err)
|
|
return err
|
|
}
|
|
|
|
// 获取视频封面
|
|
for _, v := range videoMap {
|
|
imageIds = append(imageIds, v.CoverId)
|
|
}
|
|
imageMap, err := getImageMapByIds(ctx, imageIds)
|
|
if err != nil {
|
|
logger.Error("mediafiller component getImageMapByIds failed : %v", err)
|
|
return err
|
|
}
|
|
|
|
for _, v := range list {
|
|
|
|
images := make([]*dbstruct.ToCImage, 0)
|
|
videos := make([]*dbstruct.ToCVideo, 0)
|
|
for _, imageId := range v.GetImageIds() {
|
|
if image, ok := imageMap[imageId]; ok {
|
|
images = append(images, transToCImage(image))
|
|
}
|
|
}
|
|
for _, videoId := range v.GetVideoIds() {
|
|
if video, ok := videoMap[videoId]; ok {
|
|
videos = append(videos, transToCVideo(video, imageMap[video.CoverId]))
|
|
}
|
|
}
|
|
v.SetImages(images)
|
|
v.SetVideos(videos)
|
|
}
|
|
|
|
return nil
|
|
}
|