Compare commits
2 Commits
47cced4e75
...
6bedb3a17d
Author | SHA1 | Date | |
---|---|---|---|
6bedb3a17d | |||
bce86224bb |
@ -1,12 +1,10 @@
|
|||||||
package api
|
package api
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
|
||||||
"git.solsynth.dev/hypernet/nexus/pkg/nex/sec"
|
"git.solsynth.dev/hypernet/nexus/pkg/nex/sec"
|
||||||
authm "git.solsynth.dev/hypernet/passport/pkg/authkit/models"
|
authm "git.solsynth.dev/hypernet/passport/pkg/authkit/models"
|
||||||
|
|
||||||
"git.solsynth.dev/hypernet/messaging/pkg/internal/database"
|
"git.solsynth.dev/hypernet/messaging/pkg/internal/database"
|
||||||
"git.solsynth.dev/hypernet/messaging/pkg/internal/models"
|
|
||||||
"github.com/gofiber/fiber/v2"
|
"github.com/gofiber/fiber/v2"
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -16,35 +14,6 @@ func getWhatsNew(c *fiber.Ctx) error {
|
|||||||
}
|
}
|
||||||
user := c.Locals("user").(authm.Account)
|
user := c.Locals("user").(authm.Account)
|
||||||
|
|
||||||
var lookupRange []uint
|
|
||||||
var lookupPivots []int
|
|
||||||
var ignoreRange []uint
|
|
||||||
var channelMembers []models.ChannelMember
|
|
||||||
if err := database.C.Where("account_id = ?", user.ID).
|
|
||||||
Select("id", "channel_id", "reading_anchor").
|
|
||||||
Find(&channelMembers).Error; err != nil {
|
|
||||||
return fiber.NewError(fiber.StatusInternalServerError, fmt.Sprintf("unable to get channel identity of you: %v", err))
|
|
||||||
} else {
|
|
||||||
for _, member := range channelMembers {
|
|
||||||
if member.ReadingAnchor == nil {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
lookupRange = append(lookupRange, member.ChannelID)
|
|
||||||
lookupPivots = append(lookupPivots, *member.ReadingAnchor)
|
|
||||||
ignoreRange = append(ignoreRange, member.ID)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
tx := database.C
|
|
||||||
tx = tx.Where("channel_id IN ?", lookupRange)
|
|
||||||
tx = tx.Where("sender_id NOT IN ?", ignoreRange)
|
|
||||||
|
|
||||||
countTx := tx
|
|
||||||
var count int64
|
|
||||||
if err := countTx.Model(&models.Event{}).Count(&count).Error; err != nil {
|
|
||||||
return fiber.NewError(fiber.StatusInternalServerError, err.Error())
|
|
||||||
}
|
|
||||||
|
|
||||||
var result []struct {
|
var result []struct {
|
||||||
ChannelID uint `json:"channel_id"`
|
ChannelID uint `json:"channel_id"`
|
||||||
UnreadMessageCount int `json:"count"`
|
UnreadMessageCount int `json:"count"`
|
||||||
@ -52,7 +21,7 @@ func getWhatsNew(c *fiber.Ctx) error {
|
|||||||
if err := database.C.Table("channel_members cm").
|
if err := database.C.Table("channel_members cm").
|
||||||
Select("cm.channel_id, COUNT(m.id) AS unread_message_count").
|
Select("cm.channel_id, COUNT(m.id) AS unread_message_count").
|
||||||
Joins("JOIN messages m ON m.channel_id = cm.channel_id").
|
Joins("JOIN messages m ON m.channel_id = cm.channel_id").
|
||||||
Where("m.id > cm.reading_anchor AND cm.account_id = ?", 1).
|
Where("m.id > cm.reading_anchor AND cm.account_id = ?", user.ID).
|
||||||
Group("cm.channel_id").
|
Group("cm.channel_id").
|
||||||
Scan(&result).Error; err != nil {
|
Scan(&result).Error; err != nil {
|
||||||
return fiber.NewError(fiber.StatusInternalServerError, err.Error())
|
return fiber.NewError(fiber.StatusInternalServerError, err.Error())
|
||||||
|
@ -4,7 +4,6 @@ import (
|
|||||||
"git.solsynth.dev/hypernet/messaging/pkg/internal/database"
|
"git.solsynth.dev/hypernet/messaging/pkg/internal/database"
|
||||||
"git.solsynth.dev/hypernet/messaging/pkg/internal/models"
|
"git.solsynth.dev/hypernet/messaging/pkg/internal/models"
|
||||||
"github.com/rs/zerolog/log"
|
"github.com/rs/zerolog/log"
|
||||||
"github.com/samber/lo"
|
|
||||||
"gorm.io/gorm"
|
"gorm.io/gorm"
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -22,19 +21,15 @@ func FlushReadingAnchor() {
|
|||||||
if len(readingAnchorQueue) == 0 {
|
if len(readingAnchorQueue) == 0 {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
idSet := lo.Uniq(lo.Map(lo.Keys(readingAnchorQueue), func(item uint, _ int) uint {
|
|
||||||
return item
|
|
||||||
}))
|
|
||||||
var pairs []map[string]any
|
|
||||||
for k, v := range readingAnchorQueue {
|
for k, v := range readingAnchorQueue {
|
||||||
pairs = append(pairs, map[string]any{
|
|
||||||
"id": k,
|
|
||||||
"reading_anchor": gorm.Expr("GREATEST(reading_anchor, ?)", v),
|
|
||||||
})
|
|
||||||
}
|
|
||||||
if err := database.C.Model(&models.ChannelMember{}).
|
if err := database.C.Model(&models.ChannelMember{}).
|
||||||
Where("id IN ?", idSet).
|
Where("id = ?", k).
|
||||||
Updates(pairs).Error; err != nil {
|
Updates(map[string]any{
|
||||||
|
"reading_anchor": gorm.Expr("GREATEST(reading_anchor, ?)", v),
|
||||||
|
}).Error; err != nil {
|
||||||
log.Error().Err(err).Msg("An error occurred when flushing reading anchor...")
|
log.Error().Err(err).Msg("An error occurred when flushing reading anchor...")
|
||||||
|
return
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
clear(readingAnchorQueue)
|
||||||
}
|
}
|
||||||
|
Loading…
x
Reference in New Issue
Block a user