50 lines
1.3 KiB
Go
50 lines
1.3 KiB
Go
|
package services
|
||
|
|
|
||
|
|
import (
|
||
|
|
"git.solsynth.dev/hypernet/interactive/pkg/internal/database"
|
||
|
|
"git.solsynth.dev/hypernet/interactive/pkg/internal/models"
|
||
|
|
"gorm.io/gorm"
|
||
|
|
)
|
||
|
|
|
||
|
|
var postViewQueue []models.PostView
|
||
|
|
|
||
|
|
func AddPostView(post models.Post, account uint) {
|
||
|
|
postViewQueue = append(postViewQueue, models.PostView{
|
||
|
|
AccountID: account,
|
||
|
|
PostID: post.ID,
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
func AddPostViews(posts []models.Post, account uint) {
|
||
|
|
for _, post := range posts {
|
||
|
|
postViewQueue = append(postViewQueue, models.PostView{
|
||
|
|
AccountID: account,
|
||
|
|
PostID: post.ID,
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func FlushPostViews() {
|
||
|
|
if len(postViewQueue) == 0 {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
workingQueue := make([]models.PostView, len(postViewQueue))
|
||
|
|
copy(workingQueue, postViewQueue)
|
||
|
|
clear(postViewQueue)
|
||
|
|
updateRequiredPost := make(map[uint]bool)
|
||
|
|
for _, item := range workingQueue {
|
||
|
|
updateRequiredPost[item.PostID] = true
|
||
|
|
}
|
||
|
|
_ = database.C.CreateInBatches(workingQueue, 1000).Error
|
||
|
|
for k := range updateRequiredPost {
|
||
|
|
var count int64
|
||
|
|
if err := database.C.Model(&models.PostView{}).Where("post_id = ?", k).Count(&count).Error; err != nil {
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
database.C.Model(&models.Post{}).Where("id = ?", k).Updates(map[string]any{
|
||
|
|
"total_views": count,
|
||
|
|
"total_aggressive_views": gorm.Expr("total_aggressive_views + ?", count),
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|