paopao-ce/internal/dao/sakila/timeline.go

// Copyright 2023 ROC. All rights reserved.
// Use of this source code is governed by a MIT style
// license that can be found in the LICENSE file.

package sakila

import (
	"github.com/bitbus/sqlx"
	"github.com/rocboss/paopao-ce/internal/core"
	"github.com/rocboss/paopao-ce/internal/core/ms"
	"github.com/rocboss/paopao-ce/internal/dao/sakila/auto/cc"
	"github.com/sirupsen/logrus"
)

var (
	_ core.IndexPostsService = (*shipIndexSrv)(nil)
	_ core.IndexPostsService = (*simpleIndexPostsSrv)(nil)
)

type shipIndexSrv struct {
	*sqlxSrv
	ams core.AuthorizationManageService
	ths core.TweetHelpService
	q   *cc.ShipIndex
}

type simpleIndexPostsSrv struct {
	*sqlxSrv
	ths core.TweetHelpService
	q   *cc.SimpleIndex
}

// IndexPosts 根据userId查询广场推文列表，简单做到不同用户的主页都是不同的；
func (s *shipIndexSrv) IndexPosts(user *ms.User, offset int, limit int) (res *ms.IndexTweetList, err error) {
	var posts []*ms.Post
	res = &ms.IndexTweetList{}
	switch {
	case user == nil:
		if err = s.q.IndexByGuest.Select(&posts, limit, offset); err == nil {
			err = s.q.IndexCountByGuest.Get(&res.Total)
		}
	case user != nil && user.IsAdmin:
		if err = s.q.IndexByAdmin.Select(&posts, limit, offset); err == nil {
			err = s.q.IndexCountByAdmin.Get(&res.Total)
		}
	default:
		friendIds, _ := s.ams.BeFriendIds(user.ID)
		friendIds = append(friendIds, user.ID)
		err = s.db.InSelect(&posts, s.q.IndexBySelf, user.ID, friendIds, limit, offset)
		if err == nil {
			err = s.db.InGet(&res.Total, s.q.IndexCountBySelf, user.ID, friendIds)
		}
	}
	if err != nil {
		logrus.Debugf("shipIndex.IndexPosts err: %s", err)
		return
	}
	if res.Tweets, err = s.ths.MergePosts(posts); err != nil {
		logrus.Debugf("shipIndex.IndexPosts merge posts err: %s", err)
		return
	}
	return
}

// simpleCacheIndexGetPosts simpleCacheIndex 专属获取广场推文列表函数
func (s *simpleIndexPostsSrv) IndexPosts(_user *ms.User, offset int, limit int) (res *ms.IndexTweetList, err error) {
	var posts []*ms.Post
	res = &ms.IndexTweetList{}
	if err = s.q.Index.Select(&posts, limit, offset); err == nil {
		err = s.q.IndexCount.Get(&res.Total)
	}
	if err != nil {
		logrus.Debugf("simpleIndexPostsSrv.IndexPosts err: %s", err)
		return
	}
	if res.Tweets, err = s.ths.MergePosts(posts); err != nil {
		logrus.Debugf("shipIndex.IndexPosts merge posts err: %s", err)
		return
	}
	return
}

func newShipIndexService(db *sqlx.DB, ams core.AuthorizationManageService, ths core.TweetHelpService) core.IndexPostsService {
	return &shipIndexSrv{
		ams:     ams,
		ths:     ths,
		sqlxSrv: newSqlxSrv(db),
		q:       ccBuild(db, cc.BuildShipIndex),
	}
}

func newSimpleIndexPostsService(db *sqlx.DB, ths core.TweetHelpService) core.IndexPostsService {
	return &simpleIndexPostsSrv{
		ths:     ths,
		sqlxSrv: newSqlxSrv(db),
		q:       ccBuild(db, cc.BuildSimpleIndex),
	}
}