-
Notifications
You must be signed in to change notification settings - Fork 2
/
diskstorage.go
124 lines (100 loc) · 3.42 KB
/
diskstorage.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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
package diskstorage
import (
"bytes"
"context"
"errors"
"fmt"
"io"
"net/http"
"os"
)
//go:generate rm -f ./diskstorage_mock.go
//go:generate mockgen -destination diskstorage_mock.go -package diskstorage -mock_names DiskStorage=GoMockDiskStorage -source diskstorage.go
// DiskStorage defines the methods needed to store and manage files on disk.
type DiskStorage interface {
// GetBuffer returns the buffer value.
GetBuffer() *bytes.Buffer
// WriteFile saves the buffer with given path and permission to disk.
WriteFile(filePath string, perm os.FileMode) error
// Write saves the given byte slice to the buffer.
Write(chunk []byte) error
// Download downloads file by given target URL and save to file path.
Download(ctx context.Context, targetURL, filePath string) error
// Delete removes the file at the given file path.
Delete(filePath string) error
// ResetBuffer the buffer value.
ResetBuffer()
}
// compile time interface implementation check.
var _ DiskStorage = (*diskStorage)(nil)
var (
// errFailedSetBuffer is an error message when failed to set buffer.
errFailedSetBuffer = errors.New("failed to set diskstorage.buffer")
// errFailedSetHTTPClient is an error message when failed to set http client.
errFailedSetHTTPClient = errors.New("failed to set diskstorage.http_client")
// errInternal is an error message for internal error.
errInternal = errors.New("internal error")
)
type HTTPClient interface {
Do(req *http.Request) (*http.Response, error)
}
type diskStorage struct {
buffer *bytes.Buffer
httpClient HTTPClient
}
func wrapErr(err1 error, err2 error) error {
return fmt.Errorf("%v: %w", err1, err2)
}
// New returns a new DiskStorage instance with an empty internal buffer.
func New(opts ...Option) (DiskStorage, error) {
d := new(diskStorage)
for _, opt := range append(defaultOptions, opts...) {
if err := opt(d); err != nil {
return nil, fmt.Errorf("failed to apply option: %w", wrapErr(err, errInternal))
}
}
return d, nil
}
// GetBuffer returns the buffer value.
func (d *diskStorage) GetBuffer() *bytes.Buffer {
return d.buffer
}
// WriteFile saves the buffer with given path and permission to disk.
func (d *diskStorage) WriteFile(filePath string, perm os.FileMode) error {
return os.WriteFile(filePath, d.buffer.Bytes(), perm)
}
// Write saves the given byte slice to the buffer.
func (d *diskStorage) Write(chunk []byte) error {
_, err := d.buffer.Write(chunk)
return err
}
// Download downloads file by given target URL and save to file path.
func (d *diskStorage) Download(ctx context.Context, targetURL, filePath string) error {
out, err := os.Create(filePath)
if err != nil {
return fmt.Errorf("failed to create file: %w", wrapErr(err, errInternal))
}
defer func() { _ = out.Close() }()
req, err := http.NewRequestWithContext(ctx, http.MethodGet, targetURL, nil)
if err != nil {
return fmt.Errorf("failed create http request: %w", wrapErr(err, errInternal))
}
resp, err := d.httpClient.Do(req)
if err != nil {
return fmt.Errorf("failed to send http request: %w", err)
}
defer func() { _ = resp.Body.Close() }()
_, err = io.Copy(out, resp.Body)
if err != nil {
return fmt.Errorf("failed writing to file: %w", wrapErr(err, errInternal))
}
return nil
}
// Delete removes the file at the given file path.
func (d *diskStorage) Delete(filePath string) error {
return os.Remove(filePath)
}
// ResetBuffer the buffer value.
func (d *diskStorage) ResetBuffer() {
d.buffer.Reset()
}