-
Notifications
You must be signed in to change notification settings - Fork 2
/
storage.go
52 lines (40 loc) · 970 Bytes
/
storage.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
package storage
import (
"io/fs"
"os"
"path/filepath"
)
type (
File = fs.File
FileMode = fs.FileMode
)
// FileSystem represents a disk file system
type FileSystem struct {
dir string
}
// NewStorage creates a new storage
func New(dir string) *FileSystem {
return &FileSystem{dir: dir}
}
// Open opens the named file for reading
func (storage *FileSystem) Open(name string) (File, error) {
name = filepath.Join(storage.dir, name)
if err := storage.mkdir(name); err != nil {
return nil, err
}
return os.Open(name)
}
// OpenFile is the generalized open call
func (storage *FileSystem) OpenFile(name string, flag int, perm FileMode) (File, error) {
name = filepath.Join(storage.dir, name)
if err := storage.mkdir(name); err != nil {
return nil, err
}
return os.OpenFile(name, flag, perm)
}
func (storage *FileSystem) mkdir(name string) error {
if path := filepath.Dir(name); path != "" {
return os.MkdirAll(path, 0700)
}
return nil
}