HyperNet/Interactive
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

⚡ Improve list query speed

Browse Source
This commit is contained in:
LittleSheep 2024-02-03 00:59:28 +08:00
parent 56a9c765b9
commit 807f7cff46
1 changed files with 11 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
pkg/server/posts_api.go
View File

@ -17,7 +17,7 @@ func listPost(c *fiber.Ctx) error {
offset := c.QueryInt("offset", 0) offset := c.QueryInt("offset", 0)
var count int64 var count int64
var posts []models.Post var posts []*models.Post
if err := database.C. if err := database.C.
Where(&models.Post{RealmID: nil}). Where(&models.Post{RealmID: nil}).
Model(&models.Post{}). Model(&models.Post{}).
@ -35,7 +35,7 @@ func listPost(c *fiber.Ctx) error {
return fiber.NewError(fiber.StatusInternalServerError, err.Error()) return fiber.NewError(fiber.StatusInternalServerError, err.Error())
} }
postIds := lo.Map(posts, func(item models.Post, _ int) uint { postIds := lo.Map(posts, func(item *models.Post, _ int) uint {
return item.ID return item.ID
}) })
@ -58,9 +58,15 @@ FROM %sposts t
GROUP BY post_id) d ON t.id = d.post_id GROUP BY post_id) d ON t.id = d.post_id
WHERE t.id IN (?)`, prefix, prefix, prefix), postIds).Scan(&reactInfo) WHERE t.id IN (?)`, prefix, prefix, prefix), postIds).Scan(&reactInfo)
for idx, info := range reactInfo { postMap := lo.SliceToMap(posts, func(item *models.Post) (uint, *models.Post) {
posts[idx].LikeCount = info.LikeCount return item.ID, item
posts[idx].DislikeCount = info.DislikeCount })
for _, info := range reactInfo {
if post, ok := postMap[info.PostID]; ok {
post.LikeCount = info.LikeCount
post.DislikeCount = info.DislikeCount
}
} }
return c.JSON(fiber.Map{ return c.JSON(fiber.Map{