summaryrefslogtreecommitdiff
path: root/internal/cache/db.go
diff options
context:
space:
mode:
Diffstat (limited to 'internal/cache/db.go')
-rw-r--r--internal/cache/db.go54
1 files changed, 54 insertions, 0 deletions
diff --git a/internal/cache/db.go b/internal/cache/db.go
index 8a113bd30..a5325f6ef 100644
--- a/internal/cache/db.go
+++ b/internal/cache/db.go
@@ -139,6 +139,12 @@ type GTSCaches struct {
// Status provides access to the gtsmodel Status database cache.
Status StructCache[*gtsmodel.Status]
+ // StatusBookmark ...
+ StatusBookmark StructCache[*gtsmodel.StatusBookmark]
+
+ // StatusBookmarkIDs ...
+ StatusBookmarkIDs SliceCache[string]
+
// StatusFave provides access to the gtsmodel StatusFave database cache.
StatusFave StructCache[*gtsmodel.StatusFave]
@@ -1102,6 +1108,54 @@ func (c *Caches) initStatus() {
})
}
+func (c *Caches) initStatusBookmark() {
+ // Calculate maximum cache size.
+ cap := calculateResultCacheMax(
+ sizeofStatusBookmark(), // model in-mem size.
+ config.GetCacheStatusBookmarkMemRatio(),
+ )
+
+ log.Infof(nil, "cache size = %d", cap)
+
+ copyF := func(s1 *gtsmodel.StatusBookmark) *gtsmodel.StatusBookmark {
+ s2 := new(gtsmodel.StatusBookmark)
+ *s2 = *s1
+
+ // Don't include ptr fields that
+ // will be populated separately.
+ s2.Account = nil
+ s2.TargetAccount = nil
+ s2.Status = nil
+
+ return s2
+ }
+
+ c.GTS.StatusBookmark.Init(structr.CacheConfig[*gtsmodel.StatusBookmark]{
+ Indices: []structr.IndexConfig{
+ {Fields: "ID"},
+ {Fields: "AccountID,StatusID"},
+ {Fields: "AccountID", Multiple: true},
+ {Fields: "TargetAccountID", Multiple: true},
+ {Fields: "StatusID", Multiple: true},
+ },
+ MaxSize: cap,
+ IgnoreErr: ignoreErrors,
+ Copy: copyF,
+ Invalidate: c.OnInvalidateStatusBookmark,
+ })
+}
+
+func (c *Caches) initStatusBookmarkIDs() {
+ // Calculate maximum cache size.
+ cap := calculateSliceCacheMax(
+ config.GetCacheStatusBookmarkIDsMemRatio(),
+ )
+
+ log.Infof(nil, "cache size = %d", cap)
+
+ c.GTS.StatusBookmarkIDs.Init(0, cap)
+}
+
func (c *Caches) initStatusFave() {
// Calculate maximum cache size.
cap := calculateResultCacheMax(