diff --git a/cmd/erasure-demo/get.go b/cmd/erasure-demo/get.go index 9718ea3b7..df818335f 100644 --- a/cmd/erasure-demo/get.go +++ b/cmd/erasure-demo/get.go @@ -31,8 +31,14 @@ func get(c *cli.Context) { } case "erasure": { - if objectReader, err = erasureGet(config, objectName); err != nil { - log.Fatal(err) + if len(objectName) == 0 { + if objectReader, err = erasureGetList(config); err != nil { + log.Fatal(err) + } + } else { + if objectReader, err = erasureGet(config, objectName); err != nil { + log.Fatal(err) + } } } default: diff --git a/cmd/erasure-demo/put.go b/cmd/erasure-demo/put.go index 01d69145e..c025e1731 100644 --- a/cmd/erasure-demo/put.go +++ b/cmd/erasure-demo/put.go @@ -3,6 +3,7 @@ package main import ( "log" "os" + "path" "github.com/codegangsta/cli" ) @@ -12,12 +13,13 @@ func put(c *cli.Context) { if err != nil { log.Fatal(err) } - objectPath := c.Args().Get(0) - var filePath string + var filePath, objectName string switch len(c.Args()) { case 1: - filePath = objectPath + objectName = path.Base(c.Args().Get(0)) + filePath = c.Args().Get(0) case 2: + objectName = c.Args().Get(0) filePath = c.Args().Get(1) default: log.Fatal("Please specify a valid object name \n # erasure-demo put [OBJECTNAME] [FILENAME]") @@ -29,13 +31,13 @@ func put(c *cli.Context) { switch config.storageDriver { case "fs": { - if err := fsPut(config, c.Args().Get(0), inputFile); err != nil { + if err := fsPut(config, objectName, inputFile); err != nil { log.Fatal(err) } } case "erasure": { - if err := erasurePut(config, c.Args().Get(0), inputFile); err != nil { + if err := erasurePut(config, objectName, inputFile); err != nil { log.Fatal(err) } }