diff options
Diffstat (limited to 'internal/media')
-rw-r--r-- | internal/media/processingmedia.go | 8 | ||||
-rw-r--r-- | internal/media/prune.go | 4 |
2 files changed, 6 insertions, 6 deletions
diff --git a/internal/media/processingmedia.go b/internal/media/processingmedia.go index c00c203a6..a315fa387 100644 --- a/internal/media/processingmedia.go +++ b/internal/media/processingmedia.go @@ -123,13 +123,13 @@ func (p *ProcessingMedia) load(ctx context.Context) (*gtsmodel.MediaAttachment, } if p.recache { - // Existing attachment we're recaching, so only need to update. - err = p.mgr.state.DB.UpdateByID(ctx, p.media, p.media.ID) + // Existing attachment we're recaching, so only update. + err = p.mgr.state.DB.UpdateAttachment(ctx, p.media) return err } - // New attachment, first time caching. - err = p.mgr.state.DB.Put(ctx, p.media) + // First time caching this attachment, insert it. + err = p.mgr.state.DB.PutAttachment(ctx, p.media) return err }) diff --git a/internal/media/prune.go b/internal/media/prune.go index 9e73fd2a3..aeddcef4c 100644 --- a/internal/media/prune.go +++ b/internal/media/prune.go @@ -320,7 +320,7 @@ func (m *manager) deleteAttachment(ctx context.Context, attachment *gtsmodel.Med } // Delete attachment completely. - return m.state.DB.DeleteByID(ctx, attachment.ID, attachment) + return m.state.DB.DeleteAttachment(ctx, attachment.ID) } func (m *manager) uncacheAttachment(ctx context.Context, attachment *gtsmodel.MediaAttachment) error { @@ -332,7 +332,7 @@ func (m *manager) uncacheAttachment(ctx context.Context, attachment *gtsmodel.Me attachment.UpdatedAt = time.Now() cached := false attachment.Cached = &cached - return m.state.DB.UpdateByID(ctx, attachment, attachment.ID, "updated_at", "cached") + return m.state.DB.UpdateAttachment(ctx, attachment, "updated_at", "cached") } func (m *manager) removeFiles(ctx context.Context, keys ...string) (int, error) { |