feat:添加响应结构体方便前端渲染

This commit is contained in:
Leon
2025-12-15 17:34:34 +08:00
parent 9713a23e7b
commit ab488d7b74
3 changed files with 58 additions and 21 deletions

29
internal/feed/entity.go Normal file
View File

@@ -0,0 +1,29 @@
package feed
type FeedAuthor struct {
ID uint `json:"id"`
Username string `json:"username"`
}
type FeedVideoItem struct {
ID uint `json:"id"`
Author FeedAuthor `json:"author"`
Title string `json:"title"`
Description string `json:"description,omitempty"`
PlayURL string `json:"play_url"`
CoverURL string `json:"cover_url"`
CreateTime int64 `json:"create_time"`
LikesCount int64 `json:"likes_count"`
IsLiked bool `json:"is_liked"`
}
type ListLatestRequest struct {
Limit int `json:"limit"`
LatestTime int64 `json:"latest_time"`
}
type ListLatestResponse struct {
VideoList []FeedVideoItem `json:"video_list"`
NextTime int64 `json:"next_time"`
HasMore bool `json:"has_more"`
}

View File

@@ -6,11 +6,6 @@ import (
"github.com/gin-gonic/gin" "github.com/gin-gonic/gin"
) )
type ListLatestRequest struct {
Limit int `json:"limit"`
LatestTime int64 `json:"latest_time"`
}
type FeedHandler struct { type FeedHandler struct {
service *FeedService service *FeedService
} }
@@ -32,10 +27,10 @@ func (f *FeedHandler) ListLatest(c *gin.Context) {
if req.LatestTime > 0 { if req.LatestTime > 0 {
latestTime = time.Unix(req.LatestTime, 0) latestTime = time.Unix(req.LatestTime, 0)
} }
feeds, err := f.service.ListLatest(c.Request.Context(), req.Limit, latestTime) feedItems, err := f.service.ListLatest(c.Request.Context(), req.Limit, latestTime)
if err != nil { if err != nil {
c.JSON(500, gin.H{"error": err.Error()}) c.JSON(500, gin.H{"error": err.Error()})
return return
} }
c.JSON(200, feeds) c.JSON(200, feedItems)
} }

View File

@@ -7,23 +7,18 @@ import (
) )
type FeedService struct { type FeedService struct {
repo *FeedRepository repo *FeedRepository
likeRepo *video.LikeRepository
} }
type FeedResponse struct { func NewFeedService(repo *FeedRepository, likeRepo *video.LikeRepository) *FeedService {
VideoList []video.Video `json:"video_list"` return &FeedService{repo: repo, likeRepo: likeRepo}
NextTime time.Time `json:"next_time"`
HasMore bool `json:"has_more"`
} }
func NewFeedService(repo *FeedRepository) *FeedService { func (f *FeedService) ListLatest(ctx context.Context, limit int, latestBefore time.Time) (ListLatestResponse, error) {
return &FeedService{repo: repo}
}
func (f *FeedService) ListLatest(ctx context.Context, limit int, latestBefore time.Time) (FeedResponse, error) {
videos, err := f.repo.ListLatest(ctx, limit, latestBefore) videos, err := f.repo.ListLatest(ctx, limit, latestBefore)
if err != nil { if err != nil {
return FeedResponse{}, err return ListLatestResponse{}, err
} }
var nextTime time.Time var nextTime time.Time
if len(videos) > 0 { if len(videos) > 0 {
@@ -32,9 +27,27 @@ func (f *FeedService) ListLatest(ctx context.Context, limit int, latestBefore ti
nextTime = time.Time{} nextTime = time.Time{}
} }
hasMore := len(videos) == limit hasMore := len(videos) == limit
resp := FeedResponse{ feedVideos := make([]FeedVideoItem, 0, len(videos))
VideoList: videos, for _, video := range videos {
NextTime: nextTime, isLiked, err := f.likeRepo.IsLiked(ctx, video.ID, video.AuthorID)
if err != nil {
return ListLatestResponse{}, err
}
feedVideos = append(feedVideos, FeedVideoItem{
ID: video.ID,
Author: FeedAuthor{ID: video.AuthorID, Username: video.Username},
Title: video.Title,
Description: video.Description,
PlayURL: video.PlayURL,
CoverURL: video.CoverURL,
CreateTime: video.CreateTime.Unix(),
LikesCount: video.LikesCount,
IsLiked: isLiked,
})
}
resp := ListLatestResponse{
VideoList: feedVideos,
NextTime: nextTime.Unix(),
HasMore: hasMore, HasMore: hasMore,
} }
return resp, nil return resp, nil