mirrors/mostlymatter
1
0
Форкнуть 0
Код Задачи Запросы на слияние Действия Пакеты Проекты Релизы Вики Активность

Fix file streaming to close files. (#9168)

Просмотр исходного кода
Этот коммит содержится в:
Christopher Speller
2018-07-26 10:52:24 -07:00
коммит произвёл GitHub
родитель bae26ec268
Коммит 90e5e279c1
3 изменённых файлов: 26 добавлений и 7 удалений
Скачать Patch файл Скачать Diff файл Показать все файлы Свернуть все файлы

29
api4/file.go
Просмотреть файл

@@ -160,6 +160,7 @@ func getFile(c *Context, w http.ResponseWriter, r *http.Request) {
c.Err.StatusCode = http.StatusNotFound
return
}
defer fileReader.Close()
err = writeFileResponse(info.Name, info.MimeType, info.Size, fileReader, forceDownload, w, r)
if err != nil {
@@ -195,10 +196,16 @@ func getFileThumbnail(c *Context, w http.ResponseWriter, r *http.Request) {
return
}
if fileReader, err := c.App.FileReader(info.ThumbnailPath); err != nil {
fileReader, err := c.App.FileReader(info.ThumbnailPath)
if err != nil {
c.Err = err
c.Err.StatusCode = http.StatusNotFound
} else if err := writeFileResponse(info.Name, THUMBNAIL_IMAGE_TYPE, 0, fileReader, forceDownload, w, r); err != nil {
return
}
defer fileReader.Close()
err = writeFileResponse(info.Name, THUMBNAIL_IMAGE_TYPE, 0, fileReader, forceDownload, w, r)
if err != nil {
c.Err = err
return
}
@@ -264,10 +271,15 @@ func getFilePreview(c *Context, w http.ResponseWriter, r *http.Request) {
return
}
if fileReader, err := c.App.FileReader(info.PreviewPath); err != nil {
fileReader, err := c.App.FileReader(info.PreviewPath)
if err != nil {
c.Err = err
c.Err.StatusCode = http.StatusNotFound
} else if err := writeFileResponse(info.Name, PREVIEW_IMAGE_TYPE, 0, fileReader, forceDownload, w, r); err != nil {
}
defer fileReader.Close()
err = writeFileResponse(info.Name, PREVIEW_IMAGE_TYPE, 0, fileReader, forceDownload, w, r)
if err != nil {
c.Err = err
return
}
@@ -325,10 +337,15 @@ func getPublicFile(c *Context, w http.ResponseWriter, r *http.Request) {
return
}
if fileReader, err := c.App.FileReader(info.Path); err != nil {
fileReader, err := c.App.FileReader(info.Path)
if err != nil {
c.Err = err
c.Err.StatusCode = http.StatusNotFound
} else if err := writeFileResponse(info.Name, info.MimeType, info.Size, fileReader, true, w, r); err != nil {
}
defer fileReader.Close()
err = writeFileResponse(info.Name, info.MimeType, info.Size, fileReader, true, w, r)
if err != nil {
c.Err = err
return
}

3
app/file.go
Просмотреть файл

@@ -70,7 +70,8 @@ func (a *App) ReadFile(path string) ([]byte, *model.AppError) {
return backend.ReadFile(path)
}
func (a *App) FileReader(path string) (io.Reader, *model.AppError) {
// Caller must close the first return value
func (a *App) FileReader(path string) (io.ReadCloser, *model.AppError) {
backend, err := a.FileBackend()
if err != nil {
return nil, err

1
utils/file_backend_s3.go
Просмотреть файл

@@ -82,6 +82,7 @@ func (b *S3FileBackend) TestConnection() *model.AppError {
return nil
}
// Caller must close the first return value
func (b *S3FileBackend) Reader(path string) (io.ReadCloser, *model.AppError) {
s3Clnt, err := b.s3New()
if err != nil {