larryh/corso
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add progress bar to onedrive restore (#1170)

Browse Source 
## Type of change

- [x] 🐹 Trivial/Minor

## Issue(s)

* #903

## Test Plan

- [x] 💪 Manual
This commit is contained in:
Keepers 2022-10-14 13:51:58 -06:00 committed by GitHub
parent d2d5cd7f42
commit f69e27b6fd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/internal/connector/onedrive/restore.go
View File

@ -10,6 +10,7 @@ import (
"github.com/alcionai/corso/src/internal/connector/graph"
"github.com/alcionai/corso/src/internal/connector/support"
"github.com/alcionai/corso/src/internal/data"
"github.com/alcionai/corso/src/internal/observe"
"github.com/alcionai/corso/src/pkg/backup/details"
"github.com/alcionai/corso/src/pkg/control"
"github.com/alcionai/corso/src/pkg/logger"
@ -250,8 +251,13 @@ func restoreItem(
return nil, errors.Wrapf(err, "failed to create item upload session %s", itemName)
}
iReader := itemData.ToReader()
progReader, closer := observe.ItemProgress(iReader, itemName, ss.Size())
go closer()
// Upload the stream data
written, err := io.CopyBuffer(w, itemData.ToReader(), copyBuffer)
written, err := io.CopyBuffer(w, progReader, copyBuffer)
if err != nil {
return nil, errors.Wrapf(err, "failed to upload data: item %s", itemName)
}