2022-08-24 05:55:19 +00:00
|
|
|
package models
|
|
|
|
|
2022-08-31 05:13:57 +00:00
|
|
|
import (
|
|
|
|
"code.nonshy.com/nonshy/website/pkg/log"
|
|
|
|
)
|
2022-08-24 05:55:19 +00:00
|
|
|
|
|
|
|
// ForumStatistics queries for forum-level statistics.
|
|
|
|
type ForumStatistics struct {
|
|
|
|
RecentThread *Thread
|
2022-08-26 02:58:43 +00:00
|
|
|
RecentPost *Comment // latest post on the recent thread
|
2022-08-24 05:55:19 +00:00
|
|
|
Threads uint64
|
|
|
|
Posts uint64
|
|
|
|
Users uint64
|
|
|
|
}
|
|
|
|
|
|
|
|
type ForumStatsMap map[uint64]*ForumStatistics
|
|
|
|
|
|
|
|
// MapForumStatistics looks up statistics for a set of forums.
|
|
|
|
func MapForumStatistics(forums []*Forum) ForumStatsMap {
|
|
|
|
var (
|
|
|
|
result = ForumStatsMap{}
|
|
|
|
IDs = []uint64{}
|
|
|
|
)
|
|
|
|
|
|
|
|
// Collect forum IDs and initialize the map.
|
|
|
|
for _, forum := range forums {
|
|
|
|
IDs = append(IDs, forum.ID)
|
|
|
|
result[forum.ID] = &ForumStatistics{}
|
|
|
|
}
|
|
|
|
|
2022-08-25 04:17:34 +00:00
|
|
|
// Gather all the statistics.
|
|
|
|
result.generateThreadCount(IDs)
|
|
|
|
result.generatePostCount(IDs)
|
|
|
|
result.generateUserCount(IDs)
|
|
|
|
result.generateRecentThreads(IDs)
|
2022-08-26 02:58:43 +00:00
|
|
|
result.generateRecentPosts(IDs)
|
2022-08-24 05:55:19 +00:00
|
|
|
|
|
|
|
return result
|
|
|
|
}
|
|
|
|
|
|
|
|
// Has stats for this thread? (we should..)
|
|
|
|
func (ts ForumStatsMap) Has(threadID uint64) bool {
|
|
|
|
_, ok := ts[threadID]
|
|
|
|
return ok
|
|
|
|
}
|
|
|
|
|
|
|
|
// Get thread stats.
|
|
|
|
func (ts ForumStatsMap) Get(threadID uint64) *ForumStatistics {
|
|
|
|
if stats, ok := ts[threadID]; ok {
|
|
|
|
return stats
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
2022-08-25 04:17:34 +00:00
|
|
|
|
|
|
|
// Compute the count of threads in each of the forum 'IDs'.
|
|
|
|
func (ts ForumStatsMap) generateThreadCount(IDs []uint64) {
|
|
|
|
// Hold the result of the count/group by query.
|
|
|
|
type group struct {
|
|
|
|
ID uint64
|
|
|
|
Threads uint64
|
|
|
|
}
|
|
|
|
var groups = []group{}
|
|
|
|
|
|
|
|
// Count comments grouped by thread IDs.
|
|
|
|
err := DB.Table(
|
|
|
|
"threads",
|
|
|
|
).Select(
|
|
|
|
"forum_id AS id, count(id) AS threads",
|
|
|
|
).Where(
|
|
|
|
"forum_id IN ?",
|
|
|
|
IDs,
|
|
|
|
).Group("forum_id").Scan(&groups)
|
|
|
|
|
|
|
|
if err.Error != nil {
|
|
|
|
log.Error("MapForumStatistics: SQL error: %s", err.Error)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Map the results in.
|
|
|
|
for _, row := range groups {
|
|
|
|
if stats, ok := ts[row.ID]; ok {
|
|
|
|
stats.Threads = row.Threads
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Compute the count of all posts in each of the forum 'IDs'.
|
|
|
|
func (ts ForumStatsMap) generatePostCount(IDs []uint64) {
|
|
|
|
type group struct {
|
|
|
|
ID uint64
|
|
|
|
Posts uint64
|
|
|
|
}
|
|
|
|
var groups = []group{}
|
|
|
|
|
|
|
|
err := DB.Table(
|
|
|
|
"comments",
|
|
|
|
).Joins(
|
|
|
|
"JOIN threads ON (table_name = 'threads' AND table_id = threads.id)",
|
|
|
|
).Joins(
|
|
|
|
"JOIN forums ON (threads.forum_id = forums.id)",
|
|
|
|
).Select(
|
|
|
|
"forums.id AS id, count(comments.id) AS posts",
|
|
|
|
).Where(
|
|
|
|
`table_name = 'threads' AND EXISTS (
|
|
|
|
SELECT 1
|
|
|
|
FROM threads
|
|
|
|
WHERE table_id = threads.id
|
|
|
|
AND threads.forum_id IN ?
|
|
|
|
)`,
|
|
|
|
IDs,
|
|
|
|
).Group("forums.id").Scan(&groups)
|
|
|
|
|
|
|
|
if err.Error != nil {
|
|
|
|
log.Error("SQL error collecting posts for forum: %s", err.Error)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Map the results in.
|
|
|
|
for _, row := range groups {
|
|
|
|
if stats, ok := ts[row.ID]; ok {
|
|
|
|
stats.Posts = row.Posts
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Compute the count of all users in each of the forum 'IDs'.
|
|
|
|
func (ts ForumStatsMap) generateUserCount(IDs []uint64) {
|
|
|
|
type group struct {
|
|
|
|
ForumID uint64
|
|
|
|
Users uint64
|
|
|
|
}
|
|
|
|
var groups = []group{}
|
|
|
|
|
|
|
|
err := DB.Table(
|
|
|
|
"comments",
|
|
|
|
).Joins(
|
|
|
|
"JOIN threads ON (table_name = 'threads' AND table_id = threads.id)",
|
|
|
|
).Joins(
|
|
|
|
"JOIN forums ON (threads.forum_id = forums.id)",
|
|
|
|
).Select(
|
|
|
|
"forums.id AS forum_id, count(distinct(comments.user_id)) AS users",
|
|
|
|
).Where(
|
|
|
|
"forums.id IN ?",
|
|
|
|
IDs,
|
|
|
|
).Group("forums.id").Scan(&groups)
|
|
|
|
|
|
|
|
if err.Error != nil {
|
|
|
|
log.Error("SQL error collecting users for forum: %s", err.Error)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Map the results in.
|
|
|
|
for _, row := range groups {
|
|
|
|
if stats, ok := ts[row.ForumID]; ok {
|
|
|
|
stats.Users = row.Users
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Compute the recent threads for each of the forum 'IDs'.
|
|
|
|
func (ts ForumStatsMap) generateRecentThreads(IDs []uint64) {
|
|
|
|
var threadIDs = []map[string]interface{}{}
|
|
|
|
err := DB.Table(
|
|
|
|
"threads",
|
|
|
|
).Select(
|
|
|
|
"forum_id, id AS thread_id, updated_at",
|
|
|
|
).Where(
|
|
|
|
`updated_at = (SELECT MAX(updated_at)
|
|
|
|
FROM threads t2
|
|
|
|
WHERE threads.forum_id = t2.forum_id)
|
|
|
|
AND threads.forum_id IN ?`,
|
|
|
|
IDs,
|
|
|
|
).Order(
|
|
|
|
"updated_at desc",
|
|
|
|
).Scan(&threadIDs)
|
|
|
|
|
|
|
|
if err.Error != nil {
|
|
|
|
log.Error("Getting most recent thread IDs: %s", err.Error)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Map them easier.
|
|
|
|
var (
|
|
|
|
threadForumMap = map[uint64]uint64{}
|
|
|
|
allThreadIDs = []uint64{}
|
|
|
|
)
|
|
|
|
for _, row := range threadIDs {
|
|
|
|
if row["thread_id"] == nil || row["forum_id"] == nil {
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
|
|
|
|
var (
|
|
|
|
threadID = uint64(row["thread_id"].(int64))
|
|
|
|
forumID = uint64(row["forum_id"].(int64))
|
|
|
|
)
|
|
|
|
|
|
|
|
allThreadIDs = append(allThreadIDs, threadID)
|
|
|
|
threadForumMap[threadID] = forumID
|
|
|
|
}
|
|
|
|
|
|
|
|
// Select and map these threads in.
|
|
|
|
if threadMap, err := GetThreads(allThreadIDs); err == nil {
|
|
|
|
for threadID, thread := range threadMap {
|
|
|
|
if forumID, ok := threadForumMap[threadID]; ok {
|
|
|
|
if stats, ok := ts[forumID]; ok {
|
|
|
|
stats.RecentThread = thread
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2022-08-26 02:58:43 +00:00
|
|
|
|
|
|
|
// Compute the recent post on each recent thread of each forum.
|
|
|
|
func (ts ForumStatsMap) generateRecentPosts(IDs []uint64) {
|
|
|
|
// We already have the RecentThread of each forum - map these Thread IDs to recent comments.
|
|
|
|
var (
|
|
|
|
threadIDs = []uint64{}
|
|
|
|
threadStatsMap = map[uint64]*ForumStatistics{}
|
|
|
|
)
|
|
|
|
for _, stats := range ts {
|
|
|
|
if stats.RecentThread != nil {
|
|
|
|
threadIDs = append(threadIDs, stats.RecentThread.ID)
|
|
|
|
threadStatsMap[stats.RecentThread.ID] = stats
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// The newest posts in these threads.
|
|
|
|
type scanner struct {
|
|
|
|
ThreadID uint64
|
|
|
|
CommentID uint64
|
|
|
|
}
|
|
|
|
var scan []scanner
|
|
|
|
err := DB.Table(
|
|
|
|
"comments",
|
|
|
|
).Select(
|
|
|
|
"table_id AS thread_id, id AS comment_id",
|
|
|
|
).Where(
|
|
|
|
`table_name='threads' AND table_id IN ?
|
|
|
|
AND updated_at = (SELECT MAX(updated_at)
|
|
|
|
FROM comments c2
|
|
|
|
WHERE c2.table_name=comments.table_name
|
|
|
|
AND c2.table_id=comments.table_id
|
|
|
|
)`,
|
|
|
|
threadIDs,
|
|
|
|
).Order(
|
|
|
|
"updated_at desc",
|
|
|
|
).Scan(&scan)
|
|
|
|
if err.Error != nil {
|
|
|
|
log.Error("Getting most recent post IDs: %s", err.Error)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Gather the ThreadID:CommentID map.
|
|
|
|
var (
|
|
|
|
commentIDs = []uint64{}
|
|
|
|
commentStatsMap = map[uint64]*ForumStatistics{}
|
|
|
|
)
|
|
|
|
for _, row := range scan {
|
|
|
|
if stats, ok := threadStatsMap[row.ThreadID]; ok {
|
|
|
|
commentStatsMap[row.CommentID] = stats
|
|
|
|
commentIDs = append(commentIDs, row.CommentID)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Select all these comments and map them in.
|
|
|
|
if commentMap, err := GetComments(commentIDs); err == nil {
|
|
|
|
for commentId, comment := range commentMap {
|
|
|
|
if stats, ok := commentStatsMap[commentId]; ok {
|
|
|
|
stats.RecentPost = comment
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|