Skip to content

Commit 7f9f58e

Browse files
committed
feat: dfdaemon upload and object storage add middlewares
Signed-off-by: Gaius <[email protected]>
1 parent d86005c commit 7f9f58e

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

client/daemon/daemon.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -219,7 +219,7 @@ func New(opt *config.DaemonOption, d dfpath.Dfpath) (Daemon, error) {
219219
return nil, err
220220
}
221221

222-
uploadManager, err := upload.NewUploadManager(storageManager,
222+
uploadManager, err := upload.NewUploadManager(opt, storageManager, d.LogDir(),
223223
upload.WithLimiter(rate.NewLimiter(opt.Upload.RateLimit.Limit, int(opt.Upload.RateLimit.Limit))))
224224
if err != nil {
225225
return nil, err

client/daemon/upload/upload_manager_test.go

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,7 @@ import (
3030
testifyassert "github.com/stretchr/testify/assert"
3131
"golang.org/x/time/rate"
3232

33+
"d7y.io/dragonfly/v2/client/config"
3334
"d7y.io/dragonfly/v2/client/daemon/storage"
3435
"d7y.io/dragonfly/v2/client/daemon/test"
3536
mock_storage "d7y.io/dragonfly/v2/client/daemon/test/mock/storage"
@@ -55,7 +56,7 @@ func TestUploadManager_Serve(t *testing.T) {
5556
io.NopCloser(nil), nil
5657
})
5758

58-
um, err := NewUploadManager(mockStorageManager, WithLimiter(rate.NewLimiter(16*1024, 16*1024)))
59+
um, err := NewUploadManager(config.NewDaemonConfig(), mockStorageManager, os.TempDir(), WithLimiter(rate.NewLimiter(16*1024, 16*1024)))
5960
assert.Nil(err, "NewUploadManager")
6061

6162
listen, err := net.Listen("tcp4", "127.0.0.1:0")

0 commit comments

Comments
 (0)