mirror of
https://github.com/superseriousbusiness/gotosocial
synced 2024-12-25 12:13:11 +00:00
1072 lines
25 KiB
Go
1072 lines
25 KiB
Go
|
// GoToSocial
|
||
|
// Copyright (C) GoToSocial Authors admin@gotosocial.org
|
||
|
// SPDX-License-Identifier: AGPL-3.0-or-later
|
||
|
//
|
||
|
// This program is free software: you can redistribute it and/or modify
|
||
|
// it under the terms of the GNU Affero General Public License as published by
|
||
|
// the Free Software Foundation, either version 3 of the License, or
|
||
|
// (at your option) any later version.
|
||
|
//
|
||
|
// This program is distributed in the hope that it will be useful,
|
||
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||
|
// GNU Affero General Public License for more details.
|
||
|
//
|
||
|
// You should have received a copy of the GNU Affero General Public License
|
||
|
// along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||
|
|
||
|
package cache
|
||
|
|
||
|
import (
|
||
|
"time"
|
||
|
|
||
|
"codeberg.org/gruf/go-cache/v3/simple"
|
||
|
"codeberg.org/gruf/go-cache/v3/ttl"
|
||
|
"codeberg.org/gruf/go-structr"
|
||
|
"github.com/superseriousbusiness/gotosocial/internal/cache/domain"
|
||
|
"github.com/superseriousbusiness/gotosocial/internal/config"
|
||
|
"github.com/superseriousbusiness/gotosocial/internal/gtsmodel"
|
||
|
"github.com/superseriousbusiness/gotosocial/internal/log"
|
||
|
)
|
||
|
|
||
|
type GTSCaches struct {
|
||
|
// Account provides access to the gtsmodel Account database cache.
|
||
|
Account structr.Cache[*gtsmodel.Account]
|
||
|
|
||
|
// AccountNote provides access to the gtsmodel Note database cache.
|
||
|
AccountNote structr.Cache[*gtsmodel.AccountNote]
|
||
|
|
||
|
// Application provides access to the gtsmodel Application database cache.
|
||
|
Application structr.Cache[*gtsmodel.Application]
|
||
|
|
||
|
// Block provides access to the gtsmodel Block (account) database cache.
|
||
|
Block structr.Cache[*gtsmodel.Block]
|
||
|
|
||
|
// FollowIDs provides access to the block IDs database cache.
|
||
|
BlockIDs *SliceCache[string]
|
||
|
|
||
|
// BoostOfIDs provides access to the boost of IDs list database cache.
|
||
|
BoostOfIDs *SliceCache[string]
|
||
|
|
||
|
// DomainAllow provides access to the domain allow database cache.
|
||
|
DomainAllow *domain.Cache
|
||
|
|
||
|
// DomainBlock provides access to the domain block database cache.
|
||
|
DomainBlock *domain.Cache
|
||
|
|
||
|
// Emoji provides access to the gtsmodel Emoji database cache.
|
||
|
Emoji structr.Cache[*gtsmodel.Emoji]
|
||
|
|
||
|
// EmojiCategory provides access to the gtsmodel EmojiCategory database cache.
|
||
|
EmojiCategory structr.Cache[*gtsmodel.EmojiCategory]
|
||
|
|
||
|
// Follow provides access to the gtsmodel Follow database cache.
|
||
|
Follow structr.Cache[*gtsmodel.Follow]
|
||
|
|
||
|
// FollowIDs provides access to the follower / following IDs database cache.
|
||
|
// THIS CACHE IS KEYED AS THE FOLLOWING {prefix}{accountID} WHERE PREFIX IS:
|
||
|
// - '>' for following IDs
|
||
|
// - 'l>' for local following IDs
|
||
|
// - '<' for follower IDs
|
||
|
// - 'l<' for local follower IDs
|
||
|
FollowIDs *SliceCache[string]
|
||
|
|
||
|
// FollowRequest provides access to the gtsmodel FollowRequest database cache.
|
||
|
FollowRequest structr.Cache[*gtsmodel.FollowRequest]
|
||
|
|
||
|
// FollowRequestIDs provides access to the follow requester / requesting IDs database
|
||
|
// cache. THIS CACHE IS KEYED AS THE FOLLOWING {prefix}{accountID} WHERE PREFIX IS:
|
||
|
// - '>' for following IDs
|
||
|
// - '<' for follower IDs
|
||
|
FollowRequestIDs *SliceCache[string]
|
||
|
|
||
|
// Instance provides access to the gtsmodel Instance database cache.
|
||
|
Instance structr.Cache[*gtsmodel.Instance]
|
||
|
|
||
|
// InReplyToIDs provides access to the status in reply to IDs list database cache.
|
||
|
InReplyToIDs *SliceCache[string]
|
||
|
|
||
|
// List provides access to the gtsmodel List database cache.
|
||
|
List structr.Cache[*gtsmodel.List]
|
||
|
|
||
|
// ListEntry provides access to the gtsmodel ListEntry database cache.
|
||
|
ListEntry structr.Cache[*gtsmodel.ListEntry]
|
||
|
|
||
|
// Marker provides access to the gtsmodel Marker database cache.
|
||
|
Marker structr.Cache[*gtsmodel.Marker]
|
||
|
|
||
|
// Media provides access to the gtsmodel Media database cache.
|
||
|
Media structr.Cache[*gtsmodel.MediaAttachment]
|
||
|
|
||
|
// Mention provides access to the gtsmodel Mention database cache.
|
||
|
Mention structr.Cache[*gtsmodel.Mention]
|
||
|
|
||
|
// Notification provides access to the gtsmodel Notification database cache.
|
||
|
Notification structr.Cache[*gtsmodel.Notification]
|
||
|
|
||
|
// Poll provides access to the gtsmodel Poll database cache.
|
||
|
Poll structr.Cache[*gtsmodel.Poll]
|
||
|
|
||
|
// PollVote provides access to the gtsmodel PollVote database cache.
|
||
|
PollVote structr.Cache[*gtsmodel.PollVote]
|
||
|
|
||
|
// PollVoteIDs provides access to the poll vote IDs list database cache.
|
||
|
PollVoteIDs *SliceCache[string]
|
||
|
|
||
|
// Report provides access to the gtsmodel Report database cache.
|
||
|
Report structr.Cache[*gtsmodel.Report]
|
||
|
|
||
|
// Status provides access to the gtsmodel Status database cache.
|
||
|
Status structr.Cache[*gtsmodel.Status]
|
||
|
|
||
|
// StatusFave provides access to the gtsmodel StatusFave database cache.
|
||
|
StatusFave structr.Cache[*gtsmodel.StatusFave]
|
||
|
|
||
|
// StatusFaveIDs provides access to the status fave IDs list database cache.
|
||
|
StatusFaveIDs *SliceCache[string]
|
||
|
|
||
|
// Tag provides access to the gtsmodel Tag database cache.
|
||
|
Tag structr.Cache[*gtsmodel.Tag]
|
||
|
|
||
|
// Tombstone provides access to the gtsmodel Tombstone database cache.
|
||
|
Tombstone structr.Cache[*gtsmodel.Tombstone]
|
||
|
|
||
|
// ThreadMute provides access to the gtsmodel ThreadMute database cache.
|
||
|
ThreadMute structr.Cache[*gtsmodel.ThreadMute]
|
||
|
|
||
|
// User provides access to the gtsmodel User database cache.
|
||
|
User structr.Cache[*gtsmodel.User]
|
||
|
|
||
|
// Webfinger provides access to the webfinger URL cache.
|
||
|
// TODO: move out of GTS caches since unrelated to DB.
|
||
|
Webfinger *ttl.Cache[string, string] // TTL=24hr, sweep=5min
|
||
|
}
|
||
|
|
||
|
// NOTE:
|
||
|
// all of the below init functions
|
||
|
// are receivers to the main cache
|
||
|
// struct type, not the database cache
|
||
|
// struct type, in order to get access
|
||
|
// to the full suite of caches for
|
||
|
// our invalidate function hooks.
|
||
|
|
||
|
func (c *Caches) initAccount() {
|
||
|
// Calculate maximum cache size.
|
||
|
cap := calculateResultCacheMax(
|
||
|
sizeofAccount(), // model in-mem size.
|
||
|
config.GetCacheAccountMemRatio(),
|
||
|
)
|
||
|
|
||
|
log.Infof(nil, "cache size = %d", cap)
|
||
|
|
||
|
copyF := func(a1 *gtsmodel.Account) *gtsmodel.Account {
|
||
|
a2 := new(gtsmodel.Account)
|
||
|
*a2 = *a1
|
||
|
|
||
|
// Don't include ptr fields that
|
||
|
// will be populated separately.
|
||
|
// See internal/db/bundb/account.go.
|
||
|
a2.AvatarMediaAttachment = nil
|
||
|
a2.HeaderMediaAttachment = nil
|
||
|
a2.Emojis = nil
|
||
|
|
||
|
return a2
|
||
|
}
|
||
|
|
||
|
c.GTS.Account.Init(structr.Config[*gtsmodel.Account]{
|
||
|
Indices: []structr.IndexConfig{
|
||
|
{Fields: "ID"},
|
||
|
{Fields: "URI"},
|
||
|
{Fields: "URL"},
|
||
|
{Fields: "Username,Domain", AllowZero: true},
|
||
|
{Fields: "PublicKeyURI"},
|
||
|
{Fields: "InboxURI"},
|
||
|
{Fields: "OutboxURI"},
|
||
|
{Fields: "FollowersURI"},
|
||
|
{Fields: "FollowingURI"},
|
||
|
},
|
||
|
MaxSize: cap,
|
||
|
IgnoreErr: ignoreErrors,
|
||
|
CopyValue: copyF,
|
||
|
Invalidate: c.OnInvalidateAccount,
|
||
|
})
|
||
|
}
|
||
|
|
||
|
func (c *Caches) initAccountNote() {
|
||
|
// Calculate maximum cache size.
|
||
|
cap := calculateResultCacheMax(
|
||
|
sizeofAccountNote(), // model in-mem size.
|
||
|
config.GetCacheAccountNoteMemRatio(),
|
||
|
)
|
||
|
|
||
|
log.Infof(nil, "cache size = %d", cap)
|
||
|
|
||
|
copyF := func(n1 *gtsmodel.AccountNote) *gtsmodel.AccountNote {
|
||
|
n2 := new(gtsmodel.AccountNote)
|
||
|
*n2 = *n1
|
||
|
|
||
|
// Don't include ptr fields that
|
||
|
// will be populated separately.
|
||
|
// See internal/db/bundb/relationship_note.go.
|
||
|
n2.Account = nil
|
||
|
n2.TargetAccount = nil
|
||
|
|
||
|
return n2
|
||
|
}
|
||
|
|
||
|
c.GTS.AccountNote.Init(structr.Config[*gtsmodel.AccountNote]{
|
||
|
Indices: []structr.IndexConfig{
|
||
|
{Fields: "ID"},
|
||
|
{Fields: "AccountID,TargetAccountID"},
|
||
|
},
|
||
|
MaxSize: cap,
|
||
|
IgnoreErr: ignoreErrors,
|
||
|
CopyValue: copyF,
|
||
|
})
|
||
|
}
|
||
|
|
||
|
func (c *Caches) initApplication() {
|
||
|
// Calculate maximum cache size.
|
||
|
cap := calculateResultCacheMax(
|
||
|
sizeofApplication(), // model in-mem size.
|
||
|
config.GetCacheApplicationMemRatio(),
|
||
|
)
|
||
|
|
||
|
log.Infof(nil, "cache size = %d", cap)
|
||
|
|
||
|
copyF := func(a1 *gtsmodel.Application) *gtsmodel.Application {
|
||
|
a2 := new(gtsmodel.Application)
|
||
|
*a2 = *a1
|
||
|
return a2
|
||
|
}
|
||
|
|
||
|
c.GTS.Application.Init(structr.Config[*gtsmodel.Application]{
|
||
|
Indices: []structr.IndexConfig{
|
||
|
{Fields: "ID"},
|
||
|
{Fields: "ClientID"},
|
||
|
},
|
||
|
MaxSize: cap,
|
||
|
IgnoreErr: ignoreErrors,
|
||
|
CopyValue: copyF,
|
||
|
})
|
||
|
}
|
||
|
|
||
|
func (c *Caches) initBlock() {
|
||
|
// Calculate maximum cache size.
|
||
|
cap := calculateResultCacheMax(
|
||
|
sizeofBlock(), // model in-mem size.
|
||
|
config.GetCacheBlockMemRatio(),
|
||
|
)
|
||
|
|
||
|
log.Infof(nil, "cache size = %d", cap)
|
||
|
|
||
|
copyF := func(b1 *gtsmodel.Block) *gtsmodel.Block {
|
||
|
b2 := new(gtsmodel.Block)
|
||
|
*b2 = *b1
|
||
|
|
||
|
// Don't include ptr fields that
|
||
|
// will be populated separately.
|
||
|
// See internal/db/bundb/relationship_block.go.
|
||
|
b2.Account = nil
|
||
|
b2.TargetAccount = nil
|
||
|
|
||
|
return b2
|
||
|
}
|
||
|
|
||
|
c.GTS.Block.Init(structr.Config[*gtsmodel.Block]{
|
||
|
Indices: []structr.IndexConfig{
|
||
|
{Fields: "ID"},
|
||
|
{Fields: "URI"},
|
||
|
{Fields: "AccountID,TargetAccountID"},
|
||
|
{Fields: "AccountID", Multiple: true},
|
||
|
{Fields: "TargetAccountID", Multiple: true},
|
||
|
},
|
||
|
MaxSize: cap,
|
||
|
IgnoreErr: ignoreErrors,
|
||
|
CopyValue: copyF,
|
||
|
Invalidate: c.OnInvalidateBlock,
|
||
|
})
|
||
|
}
|
||
|
|
||
|
func (c *Caches) initBlockIDs() {
|
||
|
// Calculate maximum cache size.
|
||
|
cap := calculateSliceCacheMax(
|
||
|
config.GetCacheBlockIDsMemRatio(),
|
||
|
)
|
||
|
|
||
|
log.Infof(nil, "cache size = %d", cap)
|
||
|
|
||
|
c.GTS.BlockIDs = &SliceCache[string]{Cache: simple.New[string, []string](
|
||
|
0,
|
||
|
cap,
|
||
|
)}
|
||
|
}
|
||
|
|
||
|
func (c *Caches) initBoostOfIDs() {
|
||
|
// Calculate maximum cache size.
|
||
|
cap := calculateSliceCacheMax(
|
||
|
config.GetCacheBoostOfIDsMemRatio(),
|
||
|
)
|
||
|
|
||
|
log.Infof(nil, "cache size = %d", cap)
|
||
|
|
||
|
c.GTS.BoostOfIDs = &SliceCache[string]{Cache: simple.New[string, []string](
|
||
|
0,
|
||
|
cap,
|
||
|
)}
|
||
|
}
|
||
|
|
||
|
func (c *Caches) initDomainAllow() {
|
||
|
c.GTS.DomainAllow = new(domain.Cache)
|
||
|
}
|
||
|
|
||
|
func (c *Caches) initDomainBlock() {
|
||
|
c.GTS.DomainBlock = new(domain.Cache)
|
||
|
}
|
||
|
|
||
|
func (c *Caches) initEmoji() {
|
||
|
// Calculate maximum cache size.
|
||
|
cap := calculateResultCacheMax(
|
||
|
sizeofEmoji(), // model in-mem size.
|
||
|
config.GetCacheEmojiMemRatio(),
|
||
|
)
|
||
|
|
||
|
log.Infof(nil, "cache size = %d", cap)
|
||
|
|
||
|
copyF := func(e1 *gtsmodel.Emoji) *gtsmodel.Emoji {
|
||
|
e2 := new(gtsmodel.Emoji)
|
||
|
*e2 = *e1
|
||
|
|
||
|
// Don't include ptr fields that
|
||
|
// will be populated separately.
|
||
|
// See internal/db/bundb/emoji.go.
|
||
|
e2.Category = nil
|
||
|
|
||
|
return e2
|
||
|
}
|
||
|
|
||
|
c.GTS.Emoji.Init(structr.Config[*gtsmodel.Emoji]{
|
||
|
Indices: []structr.IndexConfig{
|
||
|
{Fields: "ID"},
|
||
|
{Fields: "URI"},
|
||
|
{Fields: "Shortcode,Domain", AllowZero: true},
|
||
|
{Fields: "ImageStaticURL"},
|
||
|
{Fields: "CategoryID", Multiple: true},
|
||
|
},
|
||
|
MaxSize: cap,
|
||
|
IgnoreErr: ignoreErrors,
|
||
|
CopyValue: copyF,
|
||
|
})
|
||
|
}
|
||
|
|
||
|
func (c *Caches) initEmojiCategory() {
|
||
|
// Calculate maximum cache size.
|
||
|
cap := calculateResultCacheMax(
|
||
|
sizeofEmojiCategory(), // model in-mem size.
|
||
|
config.GetCacheEmojiCategoryMemRatio(),
|
||
|
)
|
||
|
|
||
|
log.Infof(nil, "cache size = %d", cap)
|
||
|
|
||
|
copyF := func(c1 *gtsmodel.EmojiCategory) *gtsmodel.EmojiCategory {
|
||
|
c2 := new(gtsmodel.EmojiCategory)
|
||
|
*c2 = *c1
|
||
|
return c2
|
||
|
}
|
||
|
|
||
|
c.GTS.EmojiCategory.Init(structr.Config[*gtsmodel.EmojiCategory]{
|
||
|
Indices: []structr.IndexConfig{
|
||
|
{Fields: "ID"},
|
||
|
{Fields: "Name"},
|
||
|
},
|
||
|
MaxSize: cap,
|
||
|
IgnoreErr: ignoreErrors,
|
||
|
CopyValue: copyF,
|
||
|
Invalidate: c.OnInvalidateEmojiCategory,
|
||
|
})
|
||
|
}
|
||
|
|
||
|
func (c *Caches) initFollow() {
|
||
|
// Calculate maximum cache size.
|
||
|
cap := calculateResultCacheMax(
|
||
|
sizeofFollow(), // model in-mem size.
|
||
|
config.GetCacheFollowMemRatio(),
|
||
|
)
|
||
|
|
||
|
log.Infof(nil, "cache size = %d", cap)
|
||
|
|
||
|
copyF := func(f1 *gtsmodel.Follow) *gtsmodel.Follow {
|
||
|
f2 := new(gtsmodel.Follow)
|
||
|
*f2 = *f1
|
||
|
|
||
|
// Don't include ptr fields that
|
||
|
// will be populated separately.
|
||
|
// See internal/db/bundb/relationship_follow.go.
|
||
|
f2.Account = nil
|
||
|
f2.TargetAccount = nil
|
||
|
|
||
|
return f2
|
||
|
}
|
||
|
|
||
|
c.GTS.Follow.Init(structr.Config[*gtsmodel.Follow]{
|
||
|
Indices: []structr.IndexConfig{
|
||
|
{Fields: "ID"},
|
||
|
{Fields: "URI"},
|
||
|
{Fields: "AccountID,TargetAccountID"},
|
||
|
{Fields: "AccountID", Multiple: true},
|
||
|
{Fields: "TargetAccountID", Multiple: true},
|
||
|
},
|
||
|
MaxSize: cap,
|
||
|
IgnoreErr: ignoreErrors,
|
||
|
CopyValue: copyF,
|
||
|
Invalidate: c.OnInvalidateFollow,
|
||
|
})
|
||
|
}
|
||
|
|
||
|
func (c *Caches) initFollowIDs() {
|
||
|
// Calculate maximum cache size.
|
||
|
cap := calculateSliceCacheMax(
|
||
|
config.GetCacheFollowIDsMemRatio(),
|
||
|
)
|
||
|
|
||
|
log.Infof(nil, "cache size = %d", cap)
|
||
|
|
||
|
c.GTS.FollowIDs = &SliceCache[string]{Cache: simple.New[string, []string](
|
||
|
0,
|
||
|
cap,
|
||
|
)}
|
||
|
}
|
||
|
|
||
|
func (c *Caches) initFollowRequest() {
|
||
|
// Calculate maximum cache size.
|
||
|
cap := calculateResultCacheMax(
|
||
|
sizeofFollowRequest(), // model in-mem size.
|
||
|
config.GetCacheFollowRequestMemRatio(),
|
||
|
)
|
||
|
|
||
|
log.Infof(nil, "cache size = %d", cap)
|
||
|
|
||
|
copyF := func(f1 *gtsmodel.FollowRequest) *gtsmodel.FollowRequest {
|
||
|
f2 := new(gtsmodel.FollowRequest)
|
||
|
*f2 = *f1
|
||
|
|
||
|
// Don't include ptr fields that
|
||
|
// will be populated separately.
|
||
|
// See internal/db/bundb/relationship_follow_req.go.
|
||
|
f2.Account = nil
|
||
|
f2.TargetAccount = nil
|
||
|
|
||
|
return f2
|
||
|
}
|
||
|
|
||
|
c.GTS.FollowRequest.Init(structr.Config[*gtsmodel.FollowRequest]{
|
||
|
Indices: []structr.IndexConfig{
|
||
|
{Fields: "ID"},
|
||
|
{Fields: "URI"},
|
||
|
{Fields: "AccountID,TargetAccountID"},
|
||
|
{Fields: "AccountID", Multiple: true},
|
||
|
{Fields: "TargetAccountID", Multiple: true},
|
||
|
},
|
||
|
MaxSize: cap,
|
||
|
IgnoreErr: ignoreErrors,
|
||
|
CopyValue: copyF,
|
||
|
Invalidate: c.OnInvalidateFollowRequest,
|
||
|
})
|
||
|
}
|
||
|
|
||
|
func (c *Caches) initFollowRequestIDs() {
|
||
|
// Calculate maximum cache size.
|
||
|
cap := calculateSliceCacheMax(
|
||
|
config.GetCacheFollowRequestIDsMemRatio(),
|
||
|
)
|
||
|
|
||
|
log.Infof(nil, "cache size = %d", cap)
|
||
|
|
||
|
c.GTS.FollowRequestIDs = &SliceCache[string]{Cache: simple.New[string, []string](
|
||
|
0,
|
||
|
cap,
|
||
|
)}
|
||
|
}
|
||
|
|
||
|
func (c *Caches) initInReplyToIDs() {
|
||
|
// Calculate maximum cache size.
|
||
|
cap := calculateSliceCacheMax(
|
||
|
config.GetCacheInReplyToIDsMemRatio(),
|
||
|
)
|
||
|
|
||
|
log.Infof(nil, "cache size = %d", cap)
|
||
|
|
||
|
c.GTS.InReplyToIDs = &SliceCache[string]{Cache: simple.New[string, []string](
|
||
|
0,
|
||
|
cap,
|
||
|
)}
|
||
|
}
|
||
|
|
||
|
func (c *Caches) initInstance() {
|
||
|
// Calculate maximum cache size.
|
||
|
cap := calculateResultCacheMax(
|
||
|
sizeofInstance(), // model in-mem size.
|
||
|
config.GetCacheInstanceMemRatio(),
|
||
|
)
|
||
|
|
||
|
log.Infof(nil, "cache size = %d", cap)
|
||
|
|
||
|
copyF := func(i1 *gtsmodel.Instance) *gtsmodel.Instance {
|
||
|
i2 := new(gtsmodel.Instance)
|
||
|
*i2 = *i1
|
||
|
|
||
|
// Don't include ptr fields that
|
||
|
// will be populated separately.
|
||
|
// See internal/db/bundb/instance.go.
|
||
|
i2.DomainBlock = nil
|
||
|
i2.ContactAccount = nil
|
||
|
|
||
|
return i1
|
||
|
}
|
||
|
|
||
|
c.GTS.Instance.Init(structr.Config[*gtsmodel.Instance]{
|
||
|
Indices: []structr.IndexConfig{
|
||
|
{Fields: "ID"},
|
||
|
{Fields: "Domain"},
|
||
|
},
|
||
|
MaxSize: cap,
|
||
|
IgnoreErr: ignoreErrors,
|
||
|
CopyValue: copyF,
|
||
|
})
|
||
|
}
|
||
|
|
||
|
func (c *Caches) initList() {
|
||
|
// Calculate maximum cache size.
|
||
|
cap := calculateResultCacheMax(
|
||
|
sizeofList(), // model in-mem size.
|
||
|
config.GetCacheListMemRatio(),
|
||
|
)
|
||
|
|
||
|
log.Infof(nil, "cache size = %d", cap)
|
||
|
|
||
|
copyF := func(l1 *gtsmodel.List) *gtsmodel.List {
|
||
|
l2 := new(gtsmodel.List)
|
||
|
*l2 = *l1
|
||
|
|
||
|
// Don't include ptr fields that
|
||
|
// will be populated separately.
|
||
|
// See internal/db/bundb/list.go.
|
||
|
l2.Account = nil
|
||
|
l2.ListEntries = nil
|
||
|
|
||
|
return l2
|
||
|
}
|
||
|
|
||
|
c.GTS.List.Init(structr.Config[*gtsmodel.List]{
|
||
|
Indices: []structr.IndexConfig{
|
||
|
{Fields: "ID"},
|
||
|
},
|
||
|
MaxSize: cap,
|
||
|
IgnoreErr: ignoreErrors,
|
||
|
CopyValue: copyF,
|
||
|
Invalidate: c.OnInvalidateList,
|
||
|
})
|
||
|
}
|
||
|
|
||
|
func (c *Caches) initListEntry() {
|
||
|
// Calculate maximum cache size.
|
||
|
cap := calculateResultCacheMax(
|
||
|
sizeofListEntry(), // model in-mem size.
|
||
|
config.GetCacheListEntryMemRatio(),
|
||
|
)
|
||
|
|
||
|
log.Infof(nil, "cache size = %d", cap)
|
||
|
|
||
|
copyF := func(l1 *gtsmodel.ListEntry) *gtsmodel.ListEntry {
|
||
|
l2 := new(gtsmodel.ListEntry)
|
||
|
*l2 = *l1
|
||
|
|
||
|
// Don't include ptr fields that
|
||
|
// will be populated separately.
|
||
|
// See internal/db/bundb/list.go.
|
||
|
l2.Follow = nil
|
||
|
|
||
|
return l2
|
||
|
}
|
||
|
|
||
|
c.GTS.ListEntry.Init(structr.Config[*gtsmodel.ListEntry]{
|
||
|
Indices: []structr.IndexConfig{
|
||
|
{Fields: "ID"},
|
||
|
{Fields: "ListID", Multiple: true},
|
||
|
{Fields: "FollowID", Multiple: true},
|
||
|
},
|
||
|
MaxSize: cap,
|
||
|
IgnoreErr: ignoreErrors,
|
||
|
CopyValue: copyF,
|
||
|
})
|
||
|
}
|
||
|
|
||
|
func (c *Caches) initMarker() {
|
||
|
// Calculate maximum cache size.
|
||
|
cap := calculateResultCacheMax(
|
||
|
sizeofMarker(), // model in-mem size.
|
||
|
config.GetCacheMarkerMemRatio(),
|
||
|
)
|
||
|
|
||
|
log.Infof(nil, "cache size = %d", cap)
|
||
|
|
||
|
copyF := func(m1 *gtsmodel.Marker) *gtsmodel.Marker {
|
||
|
m2 := new(gtsmodel.Marker)
|
||
|
*m2 = *m1
|
||
|
return m2
|
||
|
}
|
||
|
|
||
|
c.GTS.Marker.Init(structr.Config[*gtsmodel.Marker]{
|
||
|
Indices: []structr.IndexConfig{
|
||
|
{Fields: "AccountID,Name"},
|
||
|
},
|
||
|
MaxSize: cap,
|
||
|
IgnoreErr: ignoreErrors,
|
||
|
CopyValue: copyF,
|
||
|
})
|
||
|
}
|
||
|
|
||
|
func (c *Caches) initMedia() {
|
||
|
// Calculate maximum cache size.
|
||
|
cap := calculateResultCacheMax(
|
||
|
sizeofMedia(), // model in-mem size.
|
||
|
config.GetCacheMediaMemRatio(),
|
||
|
)
|
||
|
|
||
|
log.Infof(nil, "cache size = %d", cap)
|
||
|
|
||
|
copyF := func(m1 *gtsmodel.MediaAttachment) *gtsmodel.MediaAttachment {
|
||
|
m2 := new(gtsmodel.MediaAttachment)
|
||
|
*m2 = *m1
|
||
|
return m2
|
||
|
}
|
||
|
|
||
|
c.GTS.Media.Init(structr.Config[*gtsmodel.MediaAttachment]{
|
||
|
Indices: []structr.IndexConfig{
|
||
|
{Fields: "ID"},
|
||
|
},
|
||
|
MaxSize: cap,
|
||
|
IgnoreErr: ignoreErrors,
|
||
|
CopyValue: copyF,
|
||
|
Invalidate: c.OnInvalidateMedia,
|
||
|
})
|
||
|
}
|
||
|
|
||
|
func (c *Caches) initMention() {
|
||
|
// Calculate maximum cache size.
|
||
|
cap := calculateResultCacheMax(
|
||
|
sizeofMention(), // model in-mem size.
|
||
|
config.GetCacheMentionMemRatio(),
|
||
|
)
|
||
|
|
||
|
log.Infof(nil, "cache size = %d", cap)
|
||
|
|
||
|
copyF := func(m1 *gtsmodel.Mention) *gtsmodel.Mention {
|
||
|
m2 := new(gtsmodel.Mention)
|
||
|
*m2 = *m1
|
||
|
|
||
|
// Don't include ptr fields that
|
||
|
// will be populated separately.
|
||
|
// See internal/db/bundb/mention.go.
|
||
|
m2.Status = nil
|
||
|
m2.OriginAccount = nil
|
||
|
m2.TargetAccount = nil
|
||
|
|
||
|
return m2
|
||
|
}
|
||
|
|
||
|
c.GTS.Mention.Init(structr.Config[*gtsmodel.Mention]{
|
||
|
Indices: []structr.IndexConfig{
|
||
|
{Fields: "ID"},
|
||
|
},
|
||
|
MaxSize: cap,
|
||
|
IgnoreErr: ignoreErrors,
|
||
|
CopyValue: copyF,
|
||
|
})
|
||
|
}
|
||
|
|
||
|
func (c *Caches) initNotification() {
|
||
|
// Calculate maximum cache size.
|
||
|
cap := calculateResultCacheMax(
|
||
|
sizeofNotification(), // model in-mem size.
|
||
|
config.GetCacheNotificationMemRatio(),
|
||
|
)
|
||
|
|
||
|
log.Infof(nil, "cache size = %d", cap)
|
||
|
|
||
|
copyF := func(n1 *gtsmodel.Notification) *gtsmodel.Notification {
|
||
|
n2 := new(gtsmodel.Notification)
|
||
|
*n2 = *n1
|
||
|
|
||
|
// Don't include ptr fields that
|
||
|
// will be populated separately.
|
||
|
// See internal/db/bundb/notification.go.
|
||
|
n2.Status = nil
|
||
|
n2.OriginAccount = nil
|
||
|
n2.TargetAccount = nil
|
||
|
|
||
|
return n2
|
||
|
}
|
||
|
|
||
|
c.GTS.Notification.Init(structr.Config[*gtsmodel.Notification]{
|
||
|
Indices: []structr.IndexConfig{
|
||
|
{Fields: "ID"},
|
||
|
{Fields: "NotificationType,TargetAccountID,OriginAccountID,StatusID", AllowZero: true},
|
||
|
},
|
||
|
MaxSize: cap,
|
||
|
IgnoreErr: ignoreErrors,
|
||
|
CopyValue: copyF,
|
||
|
})
|
||
|
}
|
||
|
|
||
|
func (c *Caches) initPoll() {
|
||
|
// Calculate maximum cache size.
|
||
|
cap := calculateResultCacheMax(
|
||
|
sizeofPoll(), // model in-mem size.
|
||
|
config.GetCachePollMemRatio(),
|
||
|
)
|
||
|
|
||
|
log.Infof(nil, "cache size = %d", cap)
|
||
|
|
||
|
copyF := func(p1 *gtsmodel.Poll) *gtsmodel.Poll {
|
||
|
p2 := new(gtsmodel.Poll)
|
||
|
*p2 = *p1
|
||
|
|
||
|
// Don't include ptr fields that
|
||
|
// will be populated separately.
|
||
|
// See internal/db/bundb/poll.go.
|
||
|
p2.Status = nil
|
||
|
|
||
|
// Don't include ephemeral fields
|
||
|
// which are only expected to be
|
||
|
// set on ONE poll instance.
|
||
|
p2.Closing = false
|
||
|
|
||
|
return p2
|
||
|
}
|
||
|
|
||
|
c.GTS.Poll.Init(structr.Config[*gtsmodel.Poll]{
|
||
|
Indices: []structr.IndexConfig{
|
||
|
{Fields: "ID"},
|
||
|
{Fields: "StatusID"},
|
||
|
},
|
||
|
MaxSize: cap,
|
||
|
IgnoreErr: ignoreErrors,
|
||
|
CopyValue: copyF,
|
||
|
Invalidate: c.OnInvalidatePoll,
|
||
|
})
|
||
|
}
|
||
|
|
||
|
func (c *Caches) initPollVote() {
|
||
|
// Calculate maximum cache size.
|
||
|
cap := calculateResultCacheMax(
|
||
|
sizeofPollVote(), // model in-mem size.
|
||
|
config.GetCachePollVoteMemRatio(),
|
||
|
)
|
||
|
|
||
|
log.Infof(nil, "cache size = %d", cap)
|
||
|
|
||
|
copyF := func(v1 *gtsmodel.PollVote) *gtsmodel.PollVote {
|
||
|
v2 := new(gtsmodel.PollVote)
|
||
|
*v2 = *v1
|
||
|
|
||
|
// Don't include ptr fields that
|
||
|
// will be populated separately.
|
||
|
// See internal/db/bundb/poll.go.
|
||
|
v2.Account = nil
|
||
|
v2.Poll = nil
|
||
|
|
||
|
return v2
|
||
|
}
|
||
|
|
||
|
c.GTS.PollVote.Init(structr.Config[*gtsmodel.PollVote]{
|
||
|
Indices: []structr.IndexConfig{
|
||
|
{Fields: "ID"},
|
||
|
{Fields: "PollID", Multiple: true},
|
||
|
{Fields: "PollID,AccountID"},
|
||
|
},
|
||
|
MaxSize: cap,
|
||
|
IgnoreErr: ignoreErrors,
|
||
|
CopyValue: copyF,
|
||
|
Invalidate: c.OnInvalidatePollVote,
|
||
|
})
|
||
|
}
|
||
|
|
||
|
func (c *Caches) initPollVoteIDs() {
|
||
|
// Calculate maximum cache size.
|
||
|
cap := calculateSliceCacheMax(
|
||
|
config.GetCachePollVoteIDsMemRatio(),
|
||
|
)
|
||
|
|
||
|
log.Infof(nil, "cache size = %d", cap)
|
||
|
|
||
|
c.GTS.PollVoteIDs = &SliceCache[string]{Cache: simple.New[string, []string](
|
||
|
0,
|
||
|
cap,
|
||
|
)}
|
||
|
}
|
||
|
|
||
|
func (c *Caches) initReport() {
|
||
|
// Calculate maximum cache size.
|
||
|
cap := calculateResultCacheMax(
|
||
|
sizeofReport(), // model in-mem size.
|
||
|
config.GetCacheReportMemRatio(),
|
||
|
)
|
||
|
|
||
|
log.Infof(nil, "cache size = %d", cap)
|
||
|
|
||
|
copyF := func(r1 *gtsmodel.Report) *gtsmodel.Report {
|
||
|
r2 := new(gtsmodel.Report)
|
||
|
*r2 = *r1
|
||
|
|
||
|
// Don't include ptr fields that
|
||
|
// will be populated separately.
|
||
|
// See internal/db/bundb/report.go.
|
||
|
r2.Account = nil
|
||
|
r2.TargetAccount = nil
|
||
|
r2.Statuses = nil
|
||
|
r2.Rules = nil
|
||
|
r2.ActionTakenByAccount = nil
|
||
|
|
||
|
return r2
|
||
|
}
|
||
|
|
||
|
c.GTS.Report.Init(structr.Config[*gtsmodel.Report]{
|
||
|
Indices: []structr.IndexConfig{
|
||
|
{Fields: "ID"},
|
||
|
},
|
||
|
MaxSize: cap,
|
||
|
IgnoreErr: ignoreErrors,
|
||
|
CopyValue: copyF,
|
||
|
})
|
||
|
}
|
||
|
|
||
|
func (c *Caches) initStatus() {
|
||
|
// Calculate maximum cache size.
|
||
|
cap := calculateResultCacheMax(
|
||
|
sizeofStatus(), // model in-mem size.
|
||
|
config.GetCacheStatusMemRatio(),
|
||
|
)
|
||
|
|
||
|
log.Infof(nil, "cache size = %d", cap)
|
||
|
|
||
|
copyF := func(s1 *gtsmodel.Status) *gtsmodel.Status {
|
||
|
s2 := new(gtsmodel.Status)
|
||
|
*s2 = *s1
|
||
|
|
||
|
// Don't include ptr fields that
|
||
|
// will be populated separately.
|
||
|
// See internal/db/bundb/status.go.
|
||
|
s2.Account = nil
|
||
|
s2.InReplyTo = nil
|
||
|
s2.InReplyToAccount = nil
|
||
|
s2.BoostOf = nil
|
||
|
s2.BoostOfAccount = nil
|
||
|
s2.Poll = nil
|
||
|
s2.Attachments = nil
|
||
|
s2.Tags = nil
|
||
|
s2.Mentions = nil
|
||
|
s2.Emojis = nil
|
||
|
s2.CreatedWithApplication = nil
|
||
|
|
||
|
return s2
|
||
|
}
|
||
|
|
||
|
c.GTS.Status.Init(structr.Config[*gtsmodel.Status]{
|
||
|
Indices: []structr.IndexConfig{
|
||
|
{Fields: "ID"},
|
||
|
{Fields: "URI"},
|
||
|
{Fields: "URL"},
|
||
|
{Fields: "PollID"},
|
||
|
{Fields: "BoostOfID,AccountID"},
|
||
|
{Fields: "ThreadID", Multiple: true},
|
||
|
},
|
||
|
MaxSize: cap,
|
||
|
IgnoreErr: ignoreErrors,
|
||
|
CopyValue: copyF,
|
||
|
Invalidate: c.OnInvalidateStatus,
|
||
|
})
|
||
|
}
|
||
|
|
||
|
func (c *Caches) initStatusFave() {
|
||
|
// Calculate maximum cache size.
|
||
|
cap := calculateResultCacheMax(
|
||
|
sizeofStatusFave(), // model in-mem size.
|
||
|
config.GetCacheStatusFaveMemRatio(),
|
||
|
)
|
||
|
|
||
|
log.Infof(nil, "cache size = %d", cap)
|
||
|
|
||
|
copyF := func(f1 *gtsmodel.StatusFave) *gtsmodel.StatusFave {
|
||
|
f2 := new(gtsmodel.StatusFave)
|
||
|
*f2 = *f1
|
||
|
|
||
|
// Don't include ptr fields that
|
||
|
// will be populated separately.
|
||
|
// See internal/db/bundb/statusfave.go.
|
||
|
f2.Account = nil
|
||
|
f2.TargetAccount = nil
|
||
|
f2.Status = nil
|
||
|
|
||
|
return f2
|
||
|
}
|
||
|
|
||
|
c.GTS.StatusFave.Init(structr.Config[*gtsmodel.StatusFave]{
|
||
|
Indices: []structr.IndexConfig{
|
||
|
{Fields: "ID"},
|
||
|
{Fields: "AccountID,StatusID"},
|
||
|
{Fields: "StatusID", Multiple: true},
|
||
|
},
|
||
|
MaxSize: cap,
|
||
|
IgnoreErr: ignoreErrors,
|
||
|
CopyValue: copyF,
|
||
|
Invalidate: c.OnInvalidateStatusFave,
|
||
|
})
|
||
|
}
|
||
|
|
||
|
func (c *Caches) initStatusFaveIDs() {
|
||
|
// Calculate maximum cache size.
|
||
|
cap := calculateSliceCacheMax(
|
||
|
config.GetCacheStatusFaveIDsMemRatio(),
|
||
|
)
|
||
|
|
||
|
log.Infof(nil, "cache size = %d", cap)
|
||
|
|
||
|
c.GTS.StatusFaveIDs = &SliceCache[string]{Cache: simple.New[string, []string](
|
||
|
0,
|
||
|
cap,
|
||
|
)}
|
||
|
}
|
||
|
|
||
|
func (c *Caches) initTag() {
|
||
|
// Calculate maximum cache size.
|
||
|
cap := calculateResultCacheMax(
|
||
|
sizeofTag(), // model in-mem size.
|
||
|
config.GetCacheTagMemRatio(),
|
||
|
)
|
||
|
|
||
|
log.Infof(nil, "cache size = %d", cap)
|
||
|
|
||
|
copyF := func(m1 *gtsmodel.Tag) *gtsmodel.Tag {
|
||
|
m2 := new(gtsmodel.Tag)
|
||
|
*m2 = *m1
|
||
|
return m2
|
||
|
}
|
||
|
|
||
|
c.GTS.Tag.Init(structr.Config[*gtsmodel.Tag]{
|
||
|
Indices: []structr.IndexConfig{
|
||
|
{Fields: "ID"},
|
||
|
{Fields: "Name"},
|
||
|
},
|
||
|
MaxSize: cap,
|
||
|
IgnoreErr: ignoreErrors,
|
||
|
CopyValue: copyF,
|
||
|
})
|
||
|
}
|
||
|
|
||
|
func (c *Caches) initThreadMute() {
|
||
|
cap := calculateResultCacheMax(
|
||
|
sizeOfThreadMute(), // model in-mem size.
|
||
|
config.GetCacheThreadMuteMemRatio(),
|
||
|
)
|
||
|
|
||
|
log.Infof(nil, "cache size = %d", cap)
|
||
|
|
||
|
copyF := func(t1 *gtsmodel.ThreadMute) *gtsmodel.ThreadMute {
|
||
|
t2 := new(gtsmodel.ThreadMute)
|
||
|
*t2 = *t1
|
||
|
return t2
|
||
|
}
|
||
|
|
||
|
c.GTS.ThreadMute.Init(structr.Config[*gtsmodel.ThreadMute]{
|
||
|
Indices: []structr.IndexConfig{
|
||
|
{Fields: "ID"},
|
||
|
{Fields: "ThreadID", Multiple: true},
|
||
|
{Fields: "AccountID", Multiple: true},
|
||
|
{Fields: "ThreadID,AccountID"},
|
||
|
},
|
||
|
MaxSize: cap,
|
||
|
IgnoreErr: ignoreErrors,
|
||
|
CopyValue: copyF,
|
||
|
})
|
||
|
}
|
||
|
|
||
|
func (c *Caches) initTombstone() {
|
||
|
// Calculate maximum cache size.
|
||
|
cap := calculateResultCacheMax(
|
||
|
sizeofTombstone(), // model in-mem size.
|
||
|
config.GetCacheTombstoneMemRatio(),
|
||
|
)
|
||
|
|
||
|
log.Infof(nil, "cache size = %d", cap)
|
||
|
|
||
|
copyF := func(t1 *gtsmodel.Tombstone) *gtsmodel.Tombstone {
|
||
|
t2 := new(gtsmodel.Tombstone)
|
||
|
*t2 = *t1
|
||
|
return t2
|
||
|
}
|
||
|
|
||
|
c.GTS.Tombstone.Init(structr.Config[*gtsmodel.Tombstone]{
|
||
|
Indices: []structr.IndexConfig{
|
||
|
{Fields: "ID"},
|
||
|
{Fields: "URI"},
|
||
|
},
|
||
|
MaxSize: cap,
|
||
|
IgnoreErr: ignoreErrors,
|
||
|
CopyValue: copyF,
|
||
|
})
|
||
|
}
|
||
|
|
||
|
func (c *Caches) initUser() {
|
||
|
// Calculate maximum cache size.
|
||
|
cap := calculateResultCacheMax(
|
||
|
sizeofUser(), // model in-mem size.
|
||
|
config.GetCacheUserMemRatio(),
|
||
|
)
|
||
|
|
||
|
log.Infof(nil, "cache size = %d", cap)
|
||
|
|
||
|
copyF := func(u1 *gtsmodel.User) *gtsmodel.User {
|
||
|
u2 := new(gtsmodel.User)
|
||
|
*u2 = *u1
|
||
|
|
||
|
// Don't include ptr fields that
|
||
|
// will be populated separately.
|
||
|
// See internal/db/bundb/user.go.
|
||
|
u2.Account = nil
|
||
|
|
||
|
return u2
|
||
|
}
|
||
|
|
||
|
c.GTS.User.Init(structr.Config[*gtsmodel.User]{
|
||
|
Indices: []structr.IndexConfig{
|
||
|
{Fields: "ID"},
|
||
|
{Fields: "AccountID"},
|
||
|
{Fields: "Email"},
|
||
|
{Fields: "ConfirmationToken"},
|
||
|
{Fields: "ExternalID"},
|
||
|
},
|
||
|
MaxSize: cap,
|
||
|
IgnoreErr: ignoreErrors,
|
||
|
CopyValue: copyF,
|
||
|
Invalidate: c.OnInvalidateUser,
|
||
|
})
|
||
|
}
|
||
|
|
||
|
func (c *Caches) initWebfinger() {
|
||
|
// Calculate maximum cache size.
|
||
|
cap := calculateCacheMax(
|
||
|
sizeofURIStr, sizeofURIStr,
|
||
|
config.GetCacheWebfingerMemRatio(),
|
||
|
)
|
||
|
|
||
|
log.Infof(nil, "cache size = %d", cap)
|
||
|
|
||
|
c.GTS.Webfinger = new(ttl.Cache[string, string])
|
||
|
c.GTS.Webfinger.Init(
|
||
|
0,
|
||
|
cap,
|
||
|
24*time.Hour,
|
||
|
)
|
||
|
}
|