go-common/app/interface/openplatform/article/service/article.go

1157 lines
28 KiB
Go
Raw Normal View History

2019-04-22 10:49:16 +00:00
package service
import (
"context"
"errors"
"sort"
"strconv"
"sync"
"go-common/app/interface/openplatform/article/dao"
artmdl "go-common/app/interface/openplatform/article/model"
account "go-common/app/service/main/account/model"
"go-common/library/ecode"
"go-common/library/log"
"go-common/library/sync/errgroup"
)
var _moreNum = 3
// AddArticleCache adds artmdl.
func (s *Service) AddArticleCache(c context.Context, aid int64) (err error) {
var a *artmdl.Article
if a, err = s.dao.Article(c, aid); err != nil {
dao.PromError("article:新增文章缓存获取文章")
return
}
if a == nil {
dao.PromError("article:新增文章缓存文章未找到")
log.Error("s.Article(%d) is blank", aid)
return
}
group, errCtx := errgroup.WithContext(c)
group.Go(func() error {
return s.dao.AddArticlesMetaCache(errCtx, a.Meta)
})
group.Go(func() error {
return s.dao.AddArticleContentCache(errCtx, aid, a.Content)
})
group.Go(func() error {
if a.Keywords == "" {
return s.dao.AddArticleKeywordsCache(errCtx, aid, a.Summary)
}
return s.dao.AddArticleKeywordsCache(errCtx, aid, a.Keywords)
})
group.Go(func() (err error) {
return s.addUpperCache(errCtx, a.Author.Mid, aid, int64(a.PublishTime))
})
group.Go(func() (err error) {
return s.addArtSortCache(errCtx, a.Meta)
})
group.Go(func() (err error) {
return s.AddCacheHotspotArt(c, s.metaToSearch(c, a.Meta))
})
group.Go(func() (err error) {
if lid, _ := s.rebuildArticleListCache(c, aid); lid > 0 {
s.updateListInfo(c, lid)
return s.RebuildListCache(c, lid)
}
return
})
if err = group.Wait(); err != nil {
log.Errorv(c, log.KV("log", "AddArticleCache"), log.KV("error", err), log.KV("msg", "group.Wait()"))
dao.PromError("article:添加文章缓存")
}
return
}
// UpdateArticleCache adds artmdl.
func (s *Service) UpdateArticleCache(c context.Context, aid, oldCid int64) (err error) {
var a *artmdl.Article
if a, err = s.dao.Article(c, aid); err != nil {
dao.PromError("article:更新文章缓存获取文章")
return
}
if a == nil {
return
}
group, errCtx := errgroup.WithContext(c)
group.Go(func() error {
return s.dao.AddArticlesMetaCache(errCtx, a.Meta)
})
group.Go(func() error {
return s.dao.AddArticleContentCache(errCtx, aid, a.Content)
})
group.Go(func() error {
if a.Keywords == "" {
return s.dao.AddArticleKeywordsCache(errCtx, aid, a.Summary)
}
return s.dao.AddArticleKeywordsCache(errCtx, aid, a.Keywords)
})
group.Go(func() error {
if a.AttrVal(artmdl.AttrBitNoDistribute) {
return s.dao.DelUpperCache(errCtx, a.Meta.Author.Mid, aid)
}
return s.addUpperCache(errCtx, a.Meta.Author.Mid, aid, int64(a.Meta.PublishTime))
})
group.Go(func() (err error) {
if err = s.DelCacheHotspotArt(c, aid); err != nil {
return
}
if !a.AttrVal(artmdl.AttrBitNoDistribute) {
return s.AddCacheHotspotArt(c, s.metaToSearch(c, a.Meta))
}
return
})
group.Go(func() (err error) {
var lid int64
if lid, err = s.rebuildArticleListCache(c, aid); lid > 0 {
s.updateListInfo(c, lid)
err = s.RebuildListCache(c, lid)
}
return
})
group.Go(func() (err error) {
var root, oldRoot int64
if root, err = s.CategoryToRoot(a.Category.ID); err != nil {
dao.PromError("article:更新文章缓存查找分类")
log.Error("s.CategoryToRoot(%d,%d) error(%+v)", aid, a.Category.ID, err)
return
}
if oldCid == a.Category.ID {
return nil
}
if oldRoot, err = s.CategoryToRoot(oldCid); err != nil {
dao.PromError("article:更新文章缓存查找分类")
log.Error("s.CategoryToRoot(%d,%d) error(%+v)", aid, oldCid, err)
return
}
cids := []int64{oldCid}
if root != oldRoot {
cids = append(cids, oldRoot)
}
if err = s.delArtSortCacheFromCid(errCtx, aid, cids...); err != nil {
return
}
return s.addArtSortCache(errCtx, a.Meta)
})
if err = group.Wait(); err != nil {
log.Errorv(c, log.KV("log", "UpdateArticleCache"), log.KV("error", err), log.KV("msg", "group.Wait()"))
dao.PromError("article:更新文章缓存")
}
return
}
func (s *Service) addUpperCache(c context.Context, mid, aid, ptime int64) (err error) {
var (
exists map[int64]bool
arts map[int64][][2]int64
)
if exists, err = s.dao.ExpireUppersCache(c, []int64{mid}); err != nil {
return
}
if exists[mid] {
return s.dao.AddUpperCache(c, mid, aid, ptime)
}
if arts, err = s.dao.UppersPassed(c, []int64{mid}); err != nil {
dao.PromError("article:新增文章缓存获取up过审")
return
}
return s.dao.AddUpperCaches(c, arts)
}
//RootCategory 找到一级分区
func (s *Service) RootCategory(c context.Context, aid int64) (root int64, cid int64, err error) {
var art *artmdl.Meta
if art, err = s.dao.AllArticleMeta(c, aid); err != nil {
return
}
if art == nil {
err = ecode.NothingFound
return
}
cid = art.Category.ID
root, err = s.CategoryToRoot(art.Category.ID)
return
}
// CategoryToRoot 找到一级分区
func (s *Service) CategoryToRoot(cid int64) (res int64, err error) {
for (s.categoriesMap[cid] != nil) && (s.categoriesMap[cid].ParentID != _recommendCategory) {
cid = s.categoriesMap[cid].ParentID
}
if (s.categoriesMap[cid] == nil) || (s.categoriesMap[cid].ParentID != _recommendCategory) {
err = ecode.ArtNoCategory
return
}
res = cid
return
}
// DelArticleCache deletes artmdl.
func (s *Service) DelArticleCache(c context.Context, mid, aid int64) (err error) {
group, errCtx := errgroup.WithContext(c)
group.Go(func() error {
return s.dao.DelUpperCache(errCtx, mid, aid)
})
group.Go(func() error {
return s.dao.DelArticleMetaCache(errCtx, aid)
})
group.Go(func() error {
return s.dao.DelArticleContentCache(errCtx, aid)
})
group.Go(func() error {
return s.DelCacheHotspotArt(c, aid)
})
group.Go(func() error {
return s.dao.DelArticleStatsCache(errCtx, aid)
})
group.Go(func() (err error) {
err = s.delArtSortCache(errCtx, aid)
return
})
group.Go(func() (err error) {
var lid int64
if lid, err = s.rebuildArticleListCache(c, aid); lid > 0 {
s.updateListInfo(c, lid)
err = s.RebuildListCache(c, lid)
}
return
})
if err = group.Wait(); err != nil {
log.Errorv(c, log.KV("log", "DelArticleCache"), log.KV("error", err), log.KV("msg", "group.Wait()"))
dao.PromError("article:删除文章缓存")
}
return
}
// Article get article
func (s *Service) Article(c context.Context, id int64) (res *artmdl.Article, err error) {
var am *artmdl.Meta
if am, err = s.ArticleMeta(c, id); err != nil || am == nil {
return
}
res = &artmdl.Article{Meta: am}
if res.Content, err = s.content(c, id); err != nil {
return
}
res.Keywords = s.keywords(c, id, res.Summary)
if res.Content == "" {
dao.PromError("article:文章内容为空")
log.Error("s.Article(%v) content is blank", id)
}
s.media(c, am)
log.Info("s.Article() aid(%d) title(%s) content length(%d)", res.ID, res.Title, len(res.Content))
return
}
// MediaCategory .
func (s *Service) MediaCategory(c context.Context, mediaID int64, mid int64) (cg *artmdl.Category, err error) {
var (
res *artmdl.MediaResult
cid int64
)
if mediaID == 0 {
return
}
if res, err = s.dao.Media(c, mediaID, mid); err != nil {
log.Error("s.MediaCategory(%d) get media info failed: %v", mediaID, err)
dao.PromError("article:番剧信息获取失败")
return
}
if res.Media.TypeID > 4 || res.Media.TypeID == 0 || len(s.c.Article.Media) < 5 {
err = errors.New("番剧类型错误或者未配置番剧类别")
return
}
cid = s.c.Article.Media[res.Media.TypeID]
cg = s.categoriesMap[cid]
return
}
func (s *Service) media(c context.Context, am *artmdl.Meta) {
var (
res *artmdl.MediaResult
err error
)
if am.Media == nil || am.Media.MediaID == 0 {
return
}
if res, err = s.dao.Media(c, am.Media.MediaID, am.Author.Mid); err != nil {
log.Error("s.media(%d) get media info failed: %v", am.Media.MediaID, err)
dao.PromError("article:番剧信息获取失败")
return
}
am.Media.MediaID = res.Media.MediaID
am.Media.Score = res.Score
am.Media.Title = res.Media.Title
am.Media.Cover = res.Media.Cover
am.Media.Area = res.Media.Area
am.Media.TypeID = res.Media.TypeID
am.Media.TypeName = res.Media.TypeName
return
}
// ArticleMeta gets article's meta.
func (s *Service) ArticleMeta(c context.Context, aid int64) (res *artmdl.Meta, err error) {
var addCache = true
if res, err = s.dao.ArticleMetaCache(c, aid); err != nil {
addCache = false
err = nil
}
if res == nil {
if res, err = s.dao.ArticleMeta(c, aid); err != nil || res == nil {
return
}
}
if s.categoriesMap[res.Category.ID] != nil {
res.Category = s.categoriesMap[res.Category.ID]
res.Categories = s.categoryParents[res.Category.ID]
}
group := &errgroup.Group{}
// get author
group.Go(func() error {
var author *artmdl.Author
if author, _ = s.author(c, res.Author.Mid); author != nil {
res.Author = author
}
return nil
})
// get stats
group.Go(func() error {
var stat *artmdl.Stats
if stat, _ = s.stat(c, aid); stat != nil {
res.Stats = stat
return nil
}
if res.Stats == nil {
res.Stats = new(artmdl.Stats)
}
return nil
})
// get tag
group.Go(func() error {
var tags []*artmdl.Tag
if tags, _ = s.Tags(c, aid, false); len(tags) > 0 {
res.Tags = tags
return nil
}
if len(res.Tags) == 0 {
res.Tags = []*artmdl.Tag{}
}
return nil
})
// get list
group.Go(func() (err error) {
lists, _ := s.dao.ArtsList(c, []int64{aid})
res.List = lists[aid]
return
})
group.Wait()
if addCache {
cache.Save(func() { s.dao.AddArticlesMetaCache(context.TODO(), res) })
}
return
}
func (s *Service) accountInfo(c context.Context, mid int64) (info *account.Card, err error) {
var (
arg = &account.ArgMid{Mid: mid}
)
if info, err = s.accountRPC.Card3(c, arg); err != nil {
dao.PromError("article:获取作者信息")
log.Error("s.accountRPC.Card3(%+v) error(%+v)", arg, err)
return
}
return
}
func (s *Service) author(c context.Context, mid int64) (res *artmdl.Author, err error) {
var (
card *account.Card
arg = &account.ArgMid{Mid: mid}
)
if card, err = s.accountRPC.Card3(c, arg); err != nil {
dao.PromError("article:获取作者信息")
log.Error("s.accountRPC.Info(%+v) error(%+v)", arg, err)
return
}
res = &artmdl.Author{
Mid: mid,
Name: card.Name,
Face: card.Face,
Pendant: artmdl.Pendant{
Pid: int32(card.Pendant.Pid),
Name: card.Pendant.Name,
Image: card.Pendant.Image,
Expire: int32(card.Pendant.Expire),
},
Nameplate: artmdl.Nameplate{
Nid: card.Nameplate.Nid,
Name: card.Nameplate.Name,
Image: card.Nameplate.Image,
ImageSmall: card.Nameplate.ImageSmall,
Level: card.Nameplate.Level,
Condition: card.Nameplate.Condition,
},
Vip: artmdl.VipInfo{
Type: card.Vip.Type,
Status: card.Vip.Status,
},
}
if card.Official.Role == 0 {
res.OfficialVerify.Type = -1
} else {
if card.Official.Role <= 2 {
res.OfficialVerify.Type = 0
} else {
res.OfficialVerify.Type = 1
}
res.OfficialVerify.Desc = card.Official.Title
}
return
}
func (s *Service) authors(c context.Context, mids []int64) (res map[int64]*artmdl.Author, err error) {
res = make(map[int64]*artmdl.Author)
if len(mids) == 0 {
return
}
var (
cards map[int64]*account.Card
arg = &account.ArgMids{Mids: mids}
)
if cards, err = s.accountRPC.Cards3(c, arg); err != nil {
dao.PromError("article:批量获取作者信息")
log.Error("s.accountRPC.Infos(%+v) error(%+v)", arg, err)
return
}
for mid, card := range cards {
au := &artmdl.Author{
Mid: mid,
Name: card.Name,
Face: card.Face,
Pendant: artmdl.Pendant{
Pid: int32(card.Pendant.Pid),
Name: card.Pendant.Name,
Image: card.Pendant.Image,
Expire: int32(card.Pendant.Expire),
},
Nameplate: artmdl.Nameplate{
Nid: card.Nameplate.Nid,
Name: card.Nameplate.Name,
Image: card.Nameplate.Image,
ImageSmall: card.Nameplate.ImageSmall,
Level: card.Nameplate.Level,
Condition: card.Nameplate.Condition,
},
Vip: artmdl.VipInfo{
Type: card.Vip.Type,
Status: card.Vip.Status,
},
}
if card.Official.Role == 0 {
au.OfficialVerify.Type = -1
} else {
if card.Official.Role <= 2 {
au.OfficialVerify.Type = 0
} else {
au.OfficialVerify.Type = 1
}
au.OfficialVerify.Desc = card.Official.Title
}
res[mid] = au
}
return
}
func (s *Service) authorDetail(c context.Context, mid int64) (res *artmdl.Author, err error) {
var (
card *account.Card
arg = &account.ArgMid{Mid: mid}
)
if card, err = s.accountRPC.Card3(c, arg); err != nil {
dao.PromError("article:单个获取作者信息")
log.Error("s.accountRPC.Info(%+v) error(%+v)", arg, err)
return
}
if card == nil {
return
}
res = &artmdl.Author{
Mid: mid,
Name: card.Name,
Face: card.Face,
Pendant: artmdl.Pendant{
Pid: int32(card.Pendant.Pid),
Name: card.Pendant.Name,
Image: card.Pendant.Image,
Expire: int32(card.Pendant.Expire),
},
Nameplate: artmdl.Nameplate{
Nid: card.Nameplate.Nid,
Name: card.Nameplate.Name,
Image: card.Nameplate.Image,
ImageSmall: card.Nameplate.ImageSmall,
Level: card.Nameplate.Level,
Condition: card.Nameplate.Condition,
},
}
if card.Official.Role == 0 {
res.OfficialVerify.Type = -1
} else {
if card.Official.Role <= 2 {
res.OfficialVerify.Type = 0
} else {
res.OfficialVerify.Type = 1
}
res.OfficialVerify.Desc = card.Official.Title
}
return
}
func (s *Service) content(c context.Context, aid int64) (res string, err error) {
var addCache = true
if res, err = s.dao.ArticleContentCache(c, aid); err != nil {
addCache = false
err = nil
} else if res != "" {
return
}
if res, err = s.dao.ArticleContent(c, aid); err != nil {
dao.PromError("article:稿件内容")
return
}
if addCache && res != "" {
cache.Save(func() {
s.dao.AddArticleContentCache(context.TODO(), aid, res)
})
}
return
}
// ListCategories list categories
func (s *Service) ListCategories(c context.Context, ip string) (res artmdl.Categories, err error) {
if len(s.Categories) == 0 {
err = ecode.NothingFound
return
}
res = s.Categories
return
}
// ListCategoriesMap list category map
func (s *Service) ListCategoriesMap(c context.Context, ip string) (res map[int64]*artmdl.Category, err error) {
if len(s.Categories) == 0 {
err = ecode.NothingFound
return
}
res = s.categoriesMap
return
}
// ArticleMetas get article meta
func (s *Service) ArticleMetas(c context.Context, ids []int64) (res map[int64]*artmdl.Meta, err error) {
var (
addCache = true
group *errgroup.Group
cachedMetas, missedMetas map[int64]*artmdl.Meta
missedMetaIDs, resIDs []int64
mutex = &sync.Mutex{}
)
res = make(map[int64]*artmdl.Meta)
// get meta
if cachedMetas, missedMetaIDs, err = s.dao.ArticlesMetaCache(c, ids); err != nil {
addCache = false
err = nil
}
if len(missedMetaIDs) > 0 {
missedMetas, _ = s.dao.ArticleMetas(c, missedMetaIDs)
}
// 合并缓存和回源的数据
for id, artm := range cachedMetas {
res[id] = artm
resIDs = append(resIDs, id)
}
for id, artm := range missedMetas {
res[id] = artm
resIDs = append(resIDs, id)
}
// 更新分类
for id, art := range res {
if art.Category == nil {
continue
}
if s.categoriesMap[art.Category.ID] != nil {
res[id].Category = s.categoriesMap[art.Category.ID]
res[id].Categories = s.categoryParents[art.Category.ID]
}
}
group = &errgroup.Group{}
// get author
group.Go(func() (err error) {
var (
mids []int64
authors map[int64]*artmdl.Author
authorsMap = make(map[int64]bool)
)
for _, art := range res {
authorsMap[art.Author.Mid] = true
}
for id := range authorsMap {
mids = append(mids, id)
}
if authors, err = s.authors(c, mids); err != nil {
dao.PromError("article:稿件获取作者信息")
err = nil
return
}
mutex.Lock()
for _, art := range res {
author := authors[art.Author.Mid]
if author != nil {
art.Author = author
}
}
mutex.Unlock()
return
})
//get stats
group.Go(func() (err error) {
stats, _ := s.stats(c, resIDs)
mutex.Lock()
for id := range res {
s := stats[id]
if s == nil {
s = new(artmdl.Stats)
}
res[id].Stats = s
}
mutex.Unlock()
return
})
// get list
group.Go(func() (err error) {
lists, _ := s.dao.ArtsList(c, resIDs)
mutex.Lock()
for id := range res {
res[id].List = lists[id]
}
mutex.Unlock()
return
})
group.Wait()
if addCache && len(missedMetas) > 0 {
cache.Save(func() {
for _, art := range missedMetas {
s.dao.AddArticlesMetaCache(context.TODO(), art)
}
})
}
return
}
func filterNoDistributeArts(as []*artmdl.Meta) (res []*artmdl.Meta) {
for _, a := range as {
if (a != nil) && !a.AttrVal(artmdl.AttrBitNoDistribute) {
res = append(res, a)
}
}
return
}
func filterNoDistributeArtsMap(as map[int64]*artmdl.Meta) {
for id, a := range as {
if (a != nil) && a.AttrVal(artmdl.AttrBitNoDistribute) {
delete(as, id)
}
}
}
// AddArtContentCache add article content cache
func (s *Service) AddArtContentCache(c context.Context, aid int64, content string) (err error) {
if content == "" {
return
}
err = s.dao.AddArticleContentCache(c, aid, content)
return
}
// ArticleRemainCount returns the number that user could be use to posting new articles.
func (s *Service) ArticleRemainCount(c context.Context, mid int64) (num int, err error) {
if mid <= 0 {
return
}
var count, limit int
if count, err = s.dao.ArticleRemainCount(c, mid); err != nil {
return
}
author, _ := s.dao.Author(c, mid)
if author != nil {
limit = author.Limit
}
if limit == 0 {
limit = s.c.Article.UpperArticleLimit
}
if count > limit {
return
}
num = limit - count
return
}
// AddComplaint add complaint.
func (s *Service) AddComplaint(c context.Context, aid, mid, ctype int64, reason, imageUrls, ip string) (err error) {
var exist, protected bool
if exist, err = s.dao.ComplaintExist(c, aid, mid); (err != nil) || exist {
return
}
if err = s.dao.AddComplaint(c, aid, mid, ctype, reason, imageUrls); err != nil {
return
}
if protected, err = s.dao.ComplaintProtected(c, aid); err != nil || protected {
return
}
err = s.dao.AddComplaintCount(c, aid)
return
}
// MoreArts get author's more articles.
func (s *Service) MoreArts(c context.Context, aid int64) (res []*artmdl.Meta, err error) {
var am *artmdl.Meta
if am, err = s.ArticleMeta(c, aid); err != nil {
dao.PromError("article:获取文章meta")
return
}
if am == nil || am.Author == nil {
return
}
var (
exist bool
beforeAids, afterAids, tmpAids []int64
aidTimes [][2]int64
addCache = true
tmpRes map[int64]*artmdl.Meta
mid = am.Author.Mid
)
if exist, err = s.dao.ExpireUpperCache(c, mid); err != nil {
addCache = false
err = nil
} else if exist {
if beforeAids, afterAids, err = s.dao.MoreArtsCaches(c, mid, int64(am.PublishTime), _moreNum+4); err != nil {
addCache = false
exist = false
}
}
if !exist {
if aidTimes, err = s.dao.UpperPassed(c, mid); err != nil {
return
}
if addCache {
cache.Save(func() {
s.dao.AddUpperCaches(context.TODO(), map[int64][][2]int64{mid: aidTimes})
})
}
for _, aidTime := range aidTimes {
tmpAids = append(tmpAids, aidTime[0])
}
beforeAids, afterAids = splitAids(tmpAids, aid)
}
if len(beforeAids)+len(afterAids) == 0 {
return
}
tmpAids = append([]int64{}, beforeAids...)
tmpAids = append(tmpAids, afterAids...)
if tmpRes, err = s.ArticleMetas(c, tmpAids); err != nil {
return
}
filterNoDistributeArtsMap(tmpRes)
res = fmtMoreArts(beforeAids, afterAids, tmpRes)
return
}
func splitAids(aids []int64, aid int64) (beforeAids []int64, afterAids []int64) {
position := -1
for i, a := range aids {
if a == aid {
position = i
break
}
}
if position == -1 {
return
}
l := len(aids)
if position+_moreNum > l {
beforeAids = aids[position+1 : l]
} else {
beforeAids = aids[position+1 : position+_moreNum]
}
if position-_moreNum < 0 {
afterAids = aids[0:position]
} else {
afterAids = aids[position-_moreNum : position]
}
return
}
// 位置说明: 按照ptime逆序: afterAids -> (aid) -> beforeAids, after取一个 before取2个
func fmtMoreArts(beforeAids, afterAids []int64, tmpRes map[int64]*artmdl.Meta) (res []*artmdl.Meta) {
var before, after []*artmdl.Meta
for _, aid := range beforeAids {
if v := tmpRes[aid]; v != nil {
before = append(before, v)
}
}
sort.Sort(artmdl.Metas(before))
for _, aid := range afterAids {
if v := tmpRes[aid]; v != nil {
after = append(after, v)
}
}
sort.Sort(artmdl.Metas(after))
if len(after) > _moreNum {
after = after[len(after)-_moreNum:]
}
if len(before) > _moreNum {
before = before[:_moreNum]
}
lenAfter := len(after)
lenBefore := len(before)
// 正常逻辑: 前一个后三个
if (lenAfter > 0) && (lenBefore > 1) {
res = []*artmdl.Meta{after[lenAfter-1]}
res = append(res, before[:2]...)
} else if lenAfter == 0 {
// 前面不够
res = before
} else if lenBefore == 0 {
// 后面不够
res = after
} else {
//前面补全后面缺失的
if lenAfter-(_moreNum-lenBefore) > 0 {
res = append(res, after[lenAfter-(_moreNum-lenBefore):]...)
} else {
res = append(res, after...)
}
res = append(res, before...)
}
return
}
// NewArticleCount get new article count
func (s *Service) NewArticleCount(c context.Context, ptime int64) (res int64, err error) {
res, err = s.dao.NewArticleCount(c, ptime)
return
}
// Mores get more articles
func (s *Service) Mores(c context.Context, aid, loginMID int64) (res artmdl.MoreArts, err error) {
var art *artmdl.Meta
group := &errgroup.Group{}
if art, err = s.ArticleMeta(c, aid); (err != nil) || (art == nil) || (!art.IsNormal()) {
err = ecode.NothingFound
return
}
res.Author = &artmdl.AccountCard{Mid: strconv.FormatInt(art.Author.Mid, 10), Face: art.Author.Face, Name: art.Author.Name}
mid := art.Author.Mid
group.Go(func() (e error) {
var profile *account.ProfileStat
if profile, e = s.accountRPC.ProfileWithStat3(c, &account.ArgMid{Mid: mid}); e != nil {
dao.PromError("article:Card3")
log.Error("s.acc.Card3(%d) error %v", mid, e)
return
}
if profile != nil {
res.Author.FromProfileStat(profile)
}
return
})
group.Go(func() (e error) {
if res.Articles, e = s.similars(c, aid, art.Category.ID); e != nil {
dao.PromError("article:similars")
}
if res.Articles != nil && len(res.Articles) > 0 {
return
}
if res.Articles, e = s.MoreArts(c, aid); e != nil {
dao.PromError("article:MoreArts")
}
if res.Articles == nil {
res.Articles = []*artmdl.Meta{}
}
return
})
group.Go(func() (e error) {
if res.Total, e = s.UpperArtsCount(c, mid); e != nil {
dao.PromError("article:获取作者文章数")
}
return
})
group.Go(func() error {
// read count
if stat, e := s.dao.CacheUpStatDaily(c, mid); e != nil {
dao.PromError("article:CacheUpStatDaily")
} else if stat != nil {
res.ReadCount = stat.View
return nil
}
if stat, e := s.dao.UpStat(c, mid); e != nil {
dao.PromError("article:获取作者文章数")
} else {
res.ReadCount = stat.View
}
return nil
})
group.Go(func() (e error) {
if loginMID == 0 {
return
}
if res.Attention, e = s.isAttention(c, loginMID, mid); e != nil {
dao.PromError("article:获取作者文章数")
}
return
})
group.Wait()
return
}
// FeedArticleMetas .
func (s *Service) FeedArticleMetas(c context.Context, ids []int64) (res map[int64]*artmdl.Meta, err error) {
var (
addCache = true
group *errgroup.Group
cachedMetas, missedMetas map[int64]*artmdl.Meta
missedMetaIDs, resIDs []int64
mutex = &sync.Mutex{}
)
res = make(map[int64]*artmdl.Meta)
// get meta
if cachedMetas, missedMetaIDs, err = s.dao.ArticlesMetaCache(c, ids); err != nil {
addCache = false
err = nil
}
if len(missedMetaIDs) > 0 {
missedMetas, _ = s.dao.ArticleMetas(c, missedMetaIDs)
}
// 合并缓存和回源的数据
for id, artm := range cachedMetas {
res[id] = artm
resIDs = append(resIDs, id)
}
for id, artm := range missedMetas {
res[id] = artm
resIDs = append(resIDs, id)
}
// 更新分类
for id, art := range res {
if art.Category == nil {
continue
}
if s.categoriesMap[art.Category.ID] != nil {
res[id].Category = s.categoriesMap[art.Category.ID]
res[id].Categories = s.categoryParents[art.Category.ID]
}
}
group = &errgroup.Group{}
// get author
group.Go(func() (err error) {
var (
mids []int64
authors map[int64]*artmdl.Author
authorsMap = make(map[int64]bool)
)
for _, art := range missedMetas {
authorsMap[art.Author.Mid] = true
}
for id := range authorsMap {
mids = append(mids, id)
}
if authors, err = s.authors(c, mids); err != nil {
dao.PromError("article:稿件获取作者信息")
err = nil
return
}
mutex.Lock()
for _, art := range missedMetas {
author := authors[art.Author.Mid]
if author != nil {
art.Author = author
}
}
mutex.Unlock()
return
})
//get stats
group.Go(func() (err error) {
stats, _ := s.stats(c, resIDs)
mutex.Lock()
for id := range res {
s := stats[id]
if s == nil {
s = new(artmdl.Stats)
}
res[id].Stats = s
}
mutex.Unlock()
return
})
group.Wait()
if addCache && len(missedMetas) > 0 {
cache.Save(func() {
for _, art := range missedMetas {
s.dao.AddArticlesMetaCache(context.TODO(), art)
}
})
}
return
}
// AddCheatFilter .
func (s *Service) AddCheatFilter(c context.Context, aid int64, lv int) (err error) {
return s.dao.AddCheatFilter(c, aid, lv)
}
// DelCheatFilter .
func (s *Service) DelCheatFilter(c context.Context, aid int64) (err error) {
return s.dao.DelCheatFilter(c, aid)
}
// similars .
func (s *Service) similars(c context.Context, aid int64, cid int64) (res []*artmdl.Meta, err error) {
var (
tags []*artmdl.Tag
aidst = make(map[int64]bool)
aidsr []int64
aidsa = make(map[int64]bool)
tmps []int64
aids []int64
id int64
meta *artmdl.Meta
metas map[int64]*artmdl.Meta
nils []int64
)
if tags, err = s.Tags(c, aid, false); err != nil {
return
}
for _, tag := range tags {
var tagArts *artmdl.TagArts
if tagArts, err = s.dao.CacheAidsByTag(c, tag.Tid); err != nil {
return
}
if tagArts == nil {
nils = append(nils, tag.Tid)
continue
}
for _, id = range tagArts.Aids {
aidst[id] = true
}
}
if len(nils) > 0 {
if tmps, err = s.dao.TagArticles(c, nils); err != nil {
return
}
for _, id = range tmps {
aidst[id] = true
}
}
delete(aidst, aid)
aidsr = s.getRecommentsGroups(c, cid, aid)
if len(aidsr) == 0 && len(aidst) == 0 {
return
}
if len(aidsr) == 0 {
for id = range aidst {
aids = append(aids, id)
if len(aids) > 10 {
break
}
}
if metas, err = s.ArticleMetas(c, aids); err != nil {
return
}
for _, meta := range metas {
if artmdl.NoDistributeAttr(meta.Attributes) || artmdl.NoRegionAttr(meta.Attributes) {
continue
}
res = append(res, meta)
if len(res) == 3 {
break
}
}
return
}
for _, id = range aidsr {
aids = append(aids, id)
aidsa[id] = true
}
for id = range aidst {
aidsa[id] = true
}
if metas, err = s.ArticleMetas(c, aids); err != nil {
return
}
for _, meta = range metas {
res = append(res, meta)
delete(aidsa, meta.ID)
break
}
tmps = []int64{}
for id = range aidsa {
tmps = append(tmps, id)
if len(tmps) > 10 {
break
}
}
if metas, err = s.ArticleMetas(c, tmps); err != nil {
return
}
for _, meta := range metas {
if artmdl.NoDistributeAttr(meta.Attributes) || artmdl.NoRegionAttr(meta.Attributes) {
continue
}
res = append(res, meta)
if len(res) == 3 {
break
}
}
return
}
// MediaArticle .
func (s *Service) MediaArticle(c context.Context, mediaID, mid int64) (id int64, err error) {
return s.dao.MediaArticle(c, mediaID, mid)
}
// MediaIDByID .
func (s *Service) MediaIDByID(c context.Context, aid int64) (id int64, err error) {
return s.dao.MediaIDByID(c, aid)
}
func (s *Service) keywords(c context.Context, id int64, summary string) (keywords string) {
var (
err error
addCache = true
)
if keywords, err = s.dao.ArticleKeywordsCache(c, id); keywords != "" {
return
}
if err != nil {
addCache = false
}
if keywords, err = s.dao.ArticleKeywords(c, id); err != nil {
dao.PromError("article:文章关键词")
}
if keywords == "" {
keywords = summary
}
if addCache {
cache.Save(func() {
s.dao.AddArticleKeywordsCache(context.TODO(), id, keywords)
})
}
return
}