Improve random post picker
This commit is contained in:
@@ -7,6 +7,7 @@ import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"math/rand"
|
||||
"net/http"
|
||||
"strings"
|
||||
|
||||
@@ -106,19 +107,33 @@ func (ps *PostService) InsertNewAccounts(newAccounts []domain.Account) int {
|
||||
|
||||
func (ps *PostService) GetRandomPost() domain.Post {
|
||||
var post domain.Post
|
||||
orderExpr := "RANDOM()" // sqlite
|
||||
if config.Config.DBEngine != "sqlite" {
|
||||
orderExpr = "RAND()" // mariadb/mysql
|
||||
var postIDs []uint
|
||||
|
||||
// AI Enhanced
|
||||
// Step 1: Fetch eligible post IDs (with at least one approved attachment)
|
||||
ps.db.
|
||||
Model(&domain.Post{}).
|
||||
Where("exists (select 1 from media_attachments where post_id = posts.id and approved = ?)", true).
|
||||
Pluck("id", &postIDs)
|
||||
|
||||
if len(postIDs) == 0 {
|
||||
return domain.Post{} // No eligible posts
|
||||
}
|
||||
|
||||
// Step 2: Pick a random ID in Go
|
||||
randomID := postIDs[rand.Intn(len(postIDs))]
|
||||
|
||||
// Step 3: Load the full post with related data
|
||||
ps.db.
|
||||
Preload("Account").
|
||||
Preload("Attachments", "Approved = ?", true).
|
||||
Where("exists (select 1 from media_attachments where post_id = posts.id and approved = ?)", true).
|
||||
Order(orderExpr).
|
||||
First(&post)
|
||||
First(&post, randomID)
|
||||
|
||||
// Step 4: Keep only the first attachment if any
|
||||
if len(post.Attachments) > 0 {
|
||||
post.Attachments = []domain.MediaAttachment{post.Attachments[0]}
|
||||
}
|
||||
|
||||
return post
|
||||
}
|
||||
|
||||
|
Reference in New Issue
Block a user