diff options
| author |  tobi <31960611+tsmethurst@users.noreply.github.com> | 2022-02-22 13:50:33 +0100 |
|---|---|---|
| committer |  GitHub <noreply@github.com> | 2022-02-22 13:50:33 +0100 |
| commit | 55b83bea7cad893aabda4ab5e05c248fb66e1e51 (patch) | |
| tree | 4f0fe25b706e5635bcdd91e6ad19324fc250760a /internal/media/processingmedia.go | |
| parent | [bugfix] Fix thumbnail image type (#406) (diff) | |
| download | gotosocial-55b83bea7cad893aabda4ab5e05c248fb66e1e51.tar.xz | |
[feature] Add postDataCallbackFunc to allow cleanup (#408)
Diffstat (limited to 'internal/media/processingmedia.go')
| -rw-r--r-- | internal/media/processingmedia.go | 9 |
1 files changed, 8 insertions, 1 deletions
diff --git a/internal/media/processingmedia.go b/internal/media/processingmedia.go index 2df5fe584..5d5cfd249 100644 --- a/internal/media/processingmedia.go +++ b/internal/media/processingmedia.go @@ -48,6 +48,7 @@ type ProcessingMedia struct { attachment *gtsmodel.MediaAttachment data DataFunc + postData PostDataCallbackFunc read bool // bool indicating that data function has been triggered already thumbState int32 // the processing state of the media thumbnail @@ -313,10 +314,15 @@ func (p *ProcessingMedia) store(ctx context.Context) error { } p.read = true + + if p.postData != nil { + return p.postData(ctx) + } + return nil } -func (m *manager) preProcessMedia(ctx context.Context, data DataFunc, accountID string, ai *AdditionalMediaInfo) (*ProcessingMedia, error) { +func (m *manager) preProcessMedia(ctx context.Context, data DataFunc, postData PostDataCallbackFunc, accountID string, ai *AdditionalMediaInfo) (*ProcessingMedia, error) { id, err := id.NewRandomULID() if err != nil { return nil, err @@ -403,6 +409,7 @@ func (m *manager) preProcessMedia(ctx context.Context, data DataFunc, accountID processingMedia := &ProcessingMedia{ attachment: attachment, data: data, + postData: postData, thumbState: int32(received), fullSizeState: int32(received), database: m.db, |