Optimize user context filter query

This commit is contained in:
LittleSheep 2024-12-15 23:41:02 +08:00
parent 79a61cc732
commit 76e7dd7a07

View File

@ -4,6 +4,11 @@ import (
"context" "context"
"errors" "errors"
"fmt" "fmt"
"regexp"
"strconv"
"strings"
"time"
localCache "git.solsynth.dev/hypernet/interactive/pkg/internal/cache" localCache "git.solsynth.dev/hypernet/interactive/pkg/internal/cache"
"git.solsynth.dev/hypernet/interactive/pkg/internal/gap" "git.solsynth.dev/hypernet/interactive/pkg/internal/gap"
"git.solsynth.dev/hypernet/nexus/pkg/proto" "git.solsynth.dev/hypernet/nexus/pkg/proto"
@ -13,10 +18,6 @@ import (
"github.com/eko/gocache/lib/v4/marshaler" "github.com/eko/gocache/lib/v4/marshaler"
"github.com/eko/gocache/lib/v4/store" "github.com/eko/gocache/lib/v4/store"
"gorm.io/datatypes" "gorm.io/datatypes"
"regexp"
"strconv"
"strings"
"time"
"git.solsynth.dev/hypernet/interactive/pkg/internal/database" "git.solsynth.dev/hypernet/interactive/pkg/internal/database"
"git.solsynth.dev/hypernet/interactive/pkg/internal/models" "git.solsynth.dev/hypernet/interactive/pkg/internal/models"
@ -39,7 +40,6 @@ func FilterPostWithUserContext(tx *gorm.DB, user *authm.Account) *gorm.DB {
type userContextState struct { type userContextState struct {
Allowlist []uint Allowlist []uint
Blocklist []uint
InvisibleList []uint InvisibleList []uint
} }
@ -47,14 +47,13 @@ func FilterPostWithUserContext(tx *gorm.DB, user *authm.Account) *gorm.DB {
marshal := marshaler.New(cacheManager) marshal := marshaler.New(cacheManager)
ctx := context.Background() ctx := context.Background()
var allowlist, blocklist, invisibleList []uint var allowlist, invisibleList []uint
statusCacheKey := fmt.Sprintf("post-user-context-query#%d", user.ID) statusCacheKey := fmt.Sprintf("post-user-context-query#%d", user.ID)
statusCache, err := marshal.Get(ctx, statusCacheKey, new(userContextState)) statusCache, err := marshal.Get(ctx, statusCacheKey, new(userContextState))
if err == nil { if err == nil {
state := statusCache.(*userContextState) state := statusCache.(*userContextState)
allowlist = state.Allowlist allowlist = state.Allowlist
blocklist = state.Blocklist
invisibleList = state.InvisibleList invisibleList = state.InvisibleList
} else { } else {
userFriends, _ := authkit.ListRelative(gap.Nx, user.ID, int32(authm.RelationshipFriend), true) userFriends, _ := authkit.ListRelative(gap.Nx, user.ID, int32(authm.RelationshipFriend), true)
@ -86,14 +85,6 @@ func FilterPostWithUserContext(tx *gorm.DB, user *authm.Account) *gorm.DB {
}), func(item models.Publisher, index int) uint { }), func(item models.Publisher, index int) uint {
return uint(item.ID) return uint(item.ID)
}) })
blocklist = lo.Map(lo.Filter(publishers, func(item models.Publisher, index int) bool {
if item.AccountID == nil {
return false
}
return lo.Contains(userGotBlockList, *item.AccountID)
}), func(item models.Publisher, index int) uint {
return uint(item.ID)
})
invisibleList = lo.Map(lo.Filter(publishers, func(item models.Publisher, index int) bool { invisibleList = lo.Map(lo.Filter(publishers, func(item models.Publisher, index int) bool {
if item.AccountID == nil { if item.AccountID == nil {
return false return false
@ -108,7 +99,6 @@ func FilterPostWithUserContext(tx *gorm.DB, user *authm.Account) *gorm.DB {
statusCacheKey, statusCacheKey,
userContextState{ userContextState{
Allowlist: allowlist, Allowlist: allowlist,
Blocklist: blocklist,
InvisibleList: invisibleList, InvisibleList: invisibleList,
}, },
store.WithExpiration(5*time.Minute), store.WithExpiration(5*time.Minute),
@ -118,14 +108,13 @@ func FilterPostWithUserContext(tx *gorm.DB, user *authm.Account) *gorm.DB {
tx = tx.Where( tx = tx.Where(
"publisher_id = ? OR visibility != ? OR "+ "publisher_id = ? OR visibility != ? OR "+
"(visibility = ? AND publisher_id IN ? AND publisher_id NOT IN ?) OR "+ "(visibility = ? AND publisher_id IN ?) OR "+
"(visibility = ? AND ?) OR "+ "(visibility = ? AND ?) OR "+
"(visibility = ? AND NOT ?)", "(visibility = ? AND NOT ?)",
user.ID, user.ID,
NoneVisibility, NoneVisibility,
FriendsVisibility, FriendsVisibility,
allowlist, allowlist,
blocklist,
SelectedVisibility, SelectedVisibility,
datatypes.JSONQuery("visible_users").HasKey(strconv.Itoa(int(user.ID))), datatypes.JSONQuery("visible_users").HasKey(strconv.Itoa(int(user.ID))),
FilteredVisibility, FilteredVisibility,