GoFilm/server/logic/IndexLogic.go

package logic

import (
	"fmt"
	"github.com/gin-gonic/gin"
	"regexp"
	"server/config"
	"server/model"
	"server/plugin/db"
	"server/plugin/spider"
	"strings"
)

/*
*
 IndexController数据处理
*/

type IndexLogic struct {
}

var IL *IndexLogic

// IndexPage 首页数据处理
func (i *IndexLogic) IndexPage() map[string]interface{} {
	//声明返回值
	//Info := make(map[string]interface{})
	// 首页请求时长较高, 采用redis进行缓存, 在定时任务更新影片时清除对应缓存
	// 判断是否存在缓存数据, 存在则直接将数据返回
	Info := model.GetCacheData(config.IndexCacheKey)
	if Info != nil {
		return Info
	}
	Info = make(map[string]interface{})
	// 1. 首页分类数据处理 导航分类数据处理, 只提供 电影 电视剧 综艺 动漫 四大顶级分类和其子分类
	tree := model.CategoryTree{Category: &model.Category{Id: 0, Name: "分类信息"}}
	sysTree := model.GetCategoryTree()
	//  由于采集源数据格式不一,因此采用名称匹配
	for _, c := range sysTree.Children {
		switch c.Category.Name {
		case "电影", "电影片", "连续剧", "电视剧", "综艺", "综艺片", "动漫", "动漫片":
			tree.Children = append(tree.Children, c)
		}
	}
	Info["category"] = tree
	// 2. 提供用于首页展示的顶级分类影片信息, 每分类 14条数据
	var list []map[string]interface{}
	for _, c := range tree.Children {
		page := model.Page{PageSize: 14, Current: 1}
		movies := model.GetMovieListByPid(c.Id, &page)
		// 获取当前分类的本月热门影片
		HotMovies := model.GetHotMovieByPid(c.Id, &page)
		item := map[string]interface{}{"nav": c, "movies": movies, "hot": HotMovies}
		list = append(list, item)
	}
	Info["content"] = list
	// 不存在首页数据缓存时将查询数据缓存到redis中
	model.DataCache(config.IndexCacheKey, Info)
	return Info
}

// GetFilmDetail 影片详情信息页面处理
func (i *IndexLogic) GetFilmDetail(id int) model.MovieDetail {
	// 通过Id 获取影片search信息
	search := model.SearchInfo{}
	db.Mdb.Where("mid", id).First(&search)
	// 获取redis中的完整影视信息 MovieDetail:Cid11:Id24676
	movieDetail := model.GetDetailByKey(fmt.Sprintf(config.MovieDetailKey, search.Cid, search.Mid))
	//查找其他站点是否存在影片对应的播放源
	multipleSource(&movieDetail)
	return movieDetail
}

// GetCategoryInfo 分类信息获取, 组装导航栏需要的信息
func (i *IndexLogic) GetCategoryInfo() gin.H {
	// 组装nav导航所需的信息
	nav := gin.H{}
	// 1.获取所有分类信息
	tree := model.GetCategoryTree()
	// 2. 过滤出主页四大分类的tree信息
	for _, t := range tree.Children {
		switch t.Category.Name {
		case "动漫", "动漫片":
			nav["cartoon"] = t
		case "电影", "电影片":
			nav["film"] = t
		case "连续剧", "电视剧":
			nav["tv"] = t
		case "综艺", "综艺片":
			nav["variety"] = t
		}
	}
	// 获取所有的分类
	return nav
}

// SearchFilmInfo 获取关键字匹配的影片信息
func (i *IndexLogic) SearchFilmInfo(key string, page *model.Page) []model.MovieBasicInfo {
	// 1. 从mysql中获取满足条件的数据, 每页10条
	sl := model.SearchFilmKeyword(key, page)
	// 2. 获取redis中的basicMovieInfo信息
	var bl []model.MovieBasicInfo
	for _, s := range sl {
		bl = append(bl, model.GetBasicInfoByKey(fmt.Sprintf(config.MovieBasicInfoKey, s.Cid, s.Mid)))
	}
	return bl
}

// GetFilmCategory 根据Pid或Cid获取指定的分页数据
func (i *IndexLogic) GetFilmCategory(id int64, idType string, page *model.Page) []model.MovieBasicInfo {
	// 1. 根据不同类型进不同的查找
	var basicList []model.MovieBasicInfo
	switch idType {
	case "pid":
		basicList = model.GetMovieListByPid(id, page)
	case "cid":
		basicList = model.GetMovieListByCid(id, page)
	}
	return basicList
}

// GetPidCategory 获取pid对应的分类信息
func (i *IndexLogic) GetPidCategory(pid int64) *model.CategoryTree {
	tree := model.GetCategoryTree()
	for _, t := range tree.Children {
		if t.Id == pid {
			return t
		}
	}
	return nil
}

// RelateMovie 根据当前影片信息匹配相关的影片
func (i *IndexLogic) RelateMovie(detail model.MovieDetail, page *model.Page) []model.MovieBasicInfo {
	/*
		根据当前影片信息匹配相关的影片
		1. 分类Cid,
		2. 影片名Name
		3. 剧情内容标签class_tag
		4. 地区 area
		5. 语言 Language
	*/
	search := model.SearchInfo{
		Cid:      detail.Cid,
		Name:     detail.Name,
		ClassTag: detail.ClassTag,
		Area:     detail.Area,
		Language: detail.Language,
	}
	return model.GetRelateMovieBasicInfo(search, page)
}

/*
		将多个站点的对应影视播放源追加到主站点播放列表中
	 1. 将主站点影片的name 和 subtitle 进行处理添加到用于匹配对应播放源的map中
	 2. 仅对主站点影片name进行映射关系处理并将结果添加到map中
	    例如: xxx第一季  xxx
*/
func multipleSource(detail *model.MovieDetail) {
	// 整合多播放源, 初始化存储key map
	names := make(map[string]int)
	// 1. 判断detail的dbId是否存在, 存在则添加到names中作为匹配条件
	if detail.DbId > 0 {
		names[model.GenerateHashKey(detail.DbId)] = 0
	}
	// 2. 对name进行去除特殊格式处理
	names[model.GenerateHashKey(detail.Name)] = 0
	// 3. 对包含第一季的name进行处理
	names[model.GenerateHashKey(regexp.MustCompile(`第一季$`).ReplaceAllString(detail.Name, ""))] = 0

	// 4. 将subtitle进行切分,放入names中
	if len(detail.SubTitle) > 0 && strings.Contains(detail.SubTitle, ",") {
		for _, v := range strings.Split(detail.SubTitle, ",") {
			names[model.GenerateHashKey(v)] = 0
		}
	}
	if len(detail.SubTitle) > 0 && strings.Contains(detail.SubTitle, "/") {
		for _, v := range strings.Split(detail.SubTitle, "/") {
			names[model.GenerateHashKey(v)] = 0
		}
	}
	// 遍历站点列表
	for _, s := range spider.SiteList {
		for k, _ := range names {
			pl := model.GetMultiplePlay(s.Name, k)
			if len(pl) > 0 {
				// 如果当前站点已经匹配到数据则直接退出当前循环
				detail.PlayList = append(detail.PlayList, pl)
				break
			}
		}
	}

}