aboutsummaryrefslogtreecommitdiffstats
path: root/drive/upload.go
diff options
context:
space:
mode:
Diffstat (limited to 'drive/upload.go')
-rw-r--r--drive/upload.go7
1 files changed, 5 insertions, 2 deletions
diff --git a/drive/upload.go b/drive/upload.go
index 03cb5ad..0c1df0c 100644
--- a/drive/upload.go
+++ b/drive/upload.go
@@ -12,13 +12,13 @@ import (
type UploadFileArgs struct {
Out io.Writer
+ Progress io.Writer
Path string
Name string
Parents []string
Mime string
Recursive bool
Share bool
- NoProgress bool
ChunkSize int64
}
@@ -60,7 +60,10 @@ func (self *Drive) Upload(args UploadFileArgs) (err error) {
// Chunk size option
chunkSize := googleapi.ChunkSize(int(args.ChunkSize))
- f, err := self.service.Files.Create(dstFile).Media(srcFile, chunkSize).Do()
+ // Wrap file in progress reader
+ srcReader := getProgressReader(srcFile, args.Progress, srcFileInfo.Size())
+
+ f, err := self.service.Files.Create(dstFile).Media(srcReader, chunkSize).Do()
if err != nil {
return fmt.Errorf("Failed to upload file: %s", err)
}