minio/mc
1
0
Fork 0
mirror of https://github.com/minio/mc.git synced 2025-11-14 23:42:27 +03:00
Code Activity

Now progress bar is support for Put object

Browse Source
This commit is contained in:
Harshavardhana
2015-03-31 22:35:59 -07:00
parent 4111a974bd
commit c7221386b2
2 changed files with 33 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
cmd-cp-recursive.go
View File

@@ -19,11 +19,13 @@ package main
import (
"errors"
"fmt"
"io"
"os"
"path"
"path/filepath"
"strings"
"github.com/cheggaaa/pb"
"github.com/minio-io/cli"
"github.com/minio-io/mc/pkg/client"
"github.com/minio-io/mc/pkg/client/s3"
@@ -56,13 +58,24 @@ func (w *walk) putWalk(p string, i os.FileInfo, err error) error {
info(msg)
return nil
}
err = w.s3.Put(bucketname, key, i.Size(), bodyFile)
var bar *pb.ProgressBar
if !w.args.quiet {
// get progress bar
bar = startBar(i.Size())
}
newreader := io.Reader(bodyFile)
if !w.args.quiet {
bar.Start()
newreader = io.TeeReader(bodyFile, bar)
}
err = w.s3.Put(bucketname, key, i.Size(), newreader)
if err != nil {
return err
}
msg := fmt.Sprintf("%s uploaded -- to bucket:%s/%s/%s",
key, w.args.destination.host, bucketname, key)
info(msg)
if !w.args.quiet {
bar.Finish()
info("Success!")
}
return nil
}
@@ -77,7 +90,7 @@ func isBucketExist(bucketName string, v []*client.Bucket) bool {
}
func sourceValidate(input string) error {
if s3.IsValidBucketName(input) {
if !s3.IsValidBucketName(input) {
return fmt.Errorf("Invalid input bucket name [%s]", input)
}
st, err := os.Stat(input)

19
cmd-cp.go
View File

@@ -79,6 +79,11 @@ func firstMode(c *cli.Context, args *cmdArgs) error {
if err != nil {
return err
}
var bar *pb.ProgressBar
if !args.quiet {
// get progress bar
bar = startBar(size)
}
// http://<bucket>.<hostname> is specified without key
if args.destination.key == "" {
args.destination.key = args.source.key
@@ -87,13 +92,19 @@ func firstMode(c *cli.Context, args *cmdArgs) error {
if err != nil {
return err
}
err = s3c.Put(args.destination.bucket, args.destination.key, size, source)
newreader := io.Reader(source)
if !args.quiet {
bar.Start()
newreader = io.TeeReader(source, bar)
}
err = s3c.Put(args.destination.bucket, args.destination.key, size, newreader)
if err != nil {
return err
}
msg := fmt.Sprintf("%s uploaded -- to bucket:(http://%s/%s)", args.source.key,
args.destination.bucket, args.destination.key)
info(msg)
if !args.quiet {
bar.Finish()
info("Success!")
}
return nil
}