✨ Count post views

pkg/internal/services/post_views.go

@@ -0,0 +1,49 @@
+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),
+		})
+	}
+}