go/pocketbase
1
0
Fork 0
mirror of https://github.com/pocketbase/pocketbase.git synced 2024-11-24 17:07:00 +02:00
Code Issues Releases Activity

Allows files to be read through FileSystem interface.

Browse Source 
The functionality is needed while Pocketbase is used in embedded mode
This commit is contained in:
Andrei Varabyeu 2023-01-12 11:42:32 +01:00 committed by Gani Georgiev
parent 59e4939e1d
commit a5ceee33df
2 changed files with 28 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tools/filesystem/filesystem.go
View File

@ -285,6 +285,16 @@ var manualExtensionContentTypes = map[string]string{
".css": "text/css", // (see https://github.com/gabriel-vasile/mimetype/pull/113)
}
// / GetFile returns a file content reader for given file key
// / NB! Make sure to call `Close()` after you are done working with it.
func (s *System) GetFile(fileKey string) (io.ReadCloser, error) {
br, readErr := s.bucket.NewReader(s.ctx, fileKey, nil)
if readErr != nil {
return nil, readErr
}
return br, nil
}
// Serve serves the file at fileKey location to an HTTP response.
func (s *System) Serve(res http.ResponseWriter, req *http.Request, fileKey string, name string) error {
br, readErr := s.bucket.NewReader(s.ctx, fileKey, nil)

18
tools/filesystem/filesystem_test.go
View File

@ -348,7 +348,25 @@ func TestFileSystemServe(t *testing.T) {
}
}
}
func TestFileSystemGetFile(t *testing.T) {
dir := createTestDir(t)
defer os.RemoveAll(dir)
fs, err := filesystem.NewLocal(dir)
if err != nil {
t.Fatal(err)
}
defer fs.Close()
f, fErr := fs.GetFile("image.png")
if fErr != nil {
t.Fatal(fErr)
}
defer f.Close()
if f == nil {
t.Fatal("File is supposed to be found")
}
}
func TestFileSystemServeSingleRange(t *testing.T) {
dir := createTestDir(t)
defer os.RemoveAll(dir)