diff options
Diffstat (limited to 'internal/processing')
| -rw-r--r-- | internal/processing/media/getfile.go | 12 | ||||
| -rw-r--r-- | internal/processing/media/util.go | 14 | 
2 files changed, 23 insertions, 3 deletions
diff --git a/internal/processing/media/getfile.go b/internal/processing/media/getfile.go index 1faa8702f..c74951e38 100644 --- a/internal/processing/media/getfile.go +++ b/internal/processing/media/getfile.go @@ -180,9 +180,15 @@ func (p *processor) getAttachmentContent(ctx context.Context, requestingAccount  				return nil, 0, err  			} -			// everything read from the readCloser by the media manager will be written into the bufferedWriter -			teeReader := io.TeeReader(readCloser, bufferedWriter) -			return teeReader, fileSize, nil +			// Make a TeeReader so that everything read from the readCloser by the media manager will be written into the bufferedWriter. +			// We wrap this in a teeReadCloser which implements io.ReadCloser, so that whoever uses the teeReader can close the readCloser +			// when they're done with it. +			trc := teeReadCloser{ +				teeReader: io.TeeReader(readCloser, bufferedWriter), +				close:     readCloser.Close, +			} + +			return trc, fileSize, nil  		}  		// close the pipewriter after data has been piped into it, so the reader on the other side doesn't block; diff --git a/internal/processing/media/util.go b/internal/processing/media/util.go index 37dc87979..9739e70b7 100644 --- a/internal/processing/media/util.go +++ b/internal/processing/media/util.go @@ -20,6 +20,7 @@ package media  import (  	"fmt" +	"io"  	"strconv"  	"strings"  ) @@ -61,3 +62,16 @@ func parseFocus(focus string) (focusx, focusy float32, err error) {  	focusy = float32(fy)  	return  } + +type teeReadCloser struct { +	teeReader io.Reader +	close     func() error +} + +func (t teeReadCloser) Read(p []byte) (n int, err error) { +	return t.teeReader.Read(p) +} + +func (t teeReadCloser) Close() error { +	return t.close() +}  | 
