summaryrefslogtreecommitdiff
path: root/internal/cache
diff options
context:
space:
mode:
Diffstat (limited to 'internal/cache')
-rw-r--r--internal/cache/cache.go1
-rw-r--r--internal/cache/db.go35
-rw-r--r--internal/cache/invalidate.go5
-rw-r--r--internal/cache/size.go19
4 files changed, 58 insertions, 2 deletions
diff --git a/internal/cache/cache.go b/internal/cache/cache.go
index a4f9f2044..1a66fcd6b 100644
--- a/internal/cache/cache.go
+++ b/internal/cache/cache.go
@@ -105,6 +105,7 @@ func (c *Caches) Init() {
c.initStatus()
c.initStatusBookmark()
c.initStatusBookmarkIDs()
+ c.initStatusEdit()
c.initStatusFave()
c.initStatusFaveIDs()
c.initTag()
diff --git a/internal/cache/db.go b/internal/cache/db.go
index aac11236a..dc47bc31c 100644
--- a/internal/cache/db.go
+++ b/internal/cache/db.go
@@ -226,6 +226,9 @@ type DBCaches struct {
// StatusBookmarkIDs provides access to the status bookmark IDs list database cache.
StatusBookmarkIDs SliceCache[string]
+ // StatusEdit provides access to the gtsmodel StatusEdit database cache.
+ StatusEdit StructCache[*gtsmodel.StatusEdit]
+
// StatusFave provides access to the gtsmodel StatusFave database cache.
StatusFave StructCache[*gtsmodel.StatusFave]
@@ -1385,6 +1388,38 @@ func (c *Caches) initStatusBookmarkIDs() {
c.DB.StatusBookmarkIDs.Init(0, cap)
}
+func (c *Caches) initStatusEdit() {
+ // Calculate maximum cache size.
+ cap := calculateResultCacheMax(
+ sizeofStatusEdit(), // model in-mem size.
+ config.GetCacheStatusEditMemRatio(),
+ )
+
+ log.Infof(nil, "cache size = %d", cap)
+
+ copyF := func(s1 *gtsmodel.StatusEdit) *gtsmodel.StatusEdit {
+ s2 := new(gtsmodel.StatusEdit)
+ *s2 = *s1
+
+ // Don't include ptr fields that
+ // will be populated separately.
+ s2.Attachments = nil
+
+ return s2
+ }
+
+ c.DB.StatusEdit.Init(structr.CacheConfig[*gtsmodel.StatusEdit]{
+ Indices: []structr.IndexConfig{
+ {Fields: "ID"},
+ {Fields: "StatusID", Multiple: true},
+ },
+ MaxSize: cap,
+ IgnoreErr: ignoreErrors,
+ Copy: copyF,
+ Invalidate: c.OnInvalidateStatusEdit,
+ })
+}
+
func (c *Caches) initStatusFave() {
// Calculate maximum cache size.
cap := calculateResultCacheMax(
diff --git a/internal/cache/invalidate.go b/internal/cache/invalidate.go
index 9b42e88f6..42d7b7399 100644
--- a/internal/cache/invalidate.go
+++ b/internal/cache/invalidate.go
@@ -273,6 +273,11 @@ func (c *Caches) OnInvalidateStatusBookmark(bookmark *gtsmodel.StatusBookmark) {
c.DB.StatusBookmarkIDs.Invalidate(bookmark.StatusID)
}
+func (c *Caches) OnInvalidateStatusEdit(edit *gtsmodel.StatusEdit) {
+ // Invalidate cache of related status model.
+ c.DB.Status.Invalidate("ID", edit.StatusID)
+}
+
func (c *Caches) OnInvalidateStatusFave(fave *gtsmodel.StatusFave) {
// Invalidate status fave ID list for this status.
c.DB.StatusFaveIDs.Invalidate(fave.StatusID)
diff --git a/internal/cache/size.go b/internal/cache/size.go
index 26f4096ed..988755099 100644
--- a/internal/cache/size.go
+++ b/internal/cache/size.go
@@ -505,7 +505,6 @@ func sizeofMedia() uintptr {
URL: exampleURI,
RemoteURL: exampleURI,
CreatedAt: exampleTime,
- UpdatedAt: exampleTime,
Type: gtsmodel.FileTypeImage,
AccountID: exampleID,
Description: exampleText,
@@ -532,7 +531,6 @@ func sizeofMention() uintptr {
ID: exampleURI,
StatusID: exampleURI,
CreatedAt: exampleTime,
- UpdatedAt: exampleTime,
OriginAccountID: exampleURI,
OriginAccountURI: exampleURI,
TargetAccountID: exampleID,
@@ -674,6 +672,23 @@ func sizeofStatusBookmark() uintptr {
}))
}
+func sizeofStatusEdit() uintptr {
+ return uintptr(size.Of(&gtsmodel.StatusEdit{
+ ID: exampleID,
+ Content: exampleText,
+ ContentWarning: exampleUsername, // similar length
+ Text: exampleText,
+ Language: "en",
+ Sensitive: func() *bool { ok := false; return &ok }(),
+ AttachmentIDs: []string{exampleID, exampleID, exampleID},
+ Attachments: nil,
+ PollOptions: []string{exampleTextSmall, exampleTextSmall, exampleTextSmall, exampleTextSmall},
+ PollVotes: []int{69, 420, 1337, 1969},
+ StatusID: exampleID,
+ CreatedAt: exampleTime,
+ }))
+}
+
func sizeofStatusFave() uintptr {
return uintptr(size.Of(&gtsmodel.StatusFave{
ID: exampleID,