Adding input and output file parameters to minio-encode.

This commit is contained in:
Frederick F. Kautz IV 2014-11-22 19:25:33 -08:00
parent eb36c9e4d1
commit 6faa55e25a

View File

@ -14,17 +14,58 @@ func main() {
app := cli.NewApp() app := cli.NewApp()
app.Name = "minio-encode" app.Name = "minio-encode"
app.Usage = "erasure encode a byte stream" app.Usage = "erasure encode a byte stream"
app.Flags = []cli.Flag{
cli.StringFlag{
Name: "input,i",
Value: "",
Usage: "Input file",
},
cli.StringFlag{
Name: "output,o",
Value: "",
Usage: "Output file",
},
}
app.Action = func(c *cli.Context) { app.Action = func(c *cli.Context) {
erasureParameters, _ := erasure.ValidateParams(10, 5, 8, erasure.CAUCHY) // check if minio-encode called without parameters
if len(c.Args()) == 1 {
encoder := erasure.NewEncoder(erasureParameters) cli.ShowAppHelp(c)
input, err := ioutil.ReadAll(os.Stdin)
if err != nil {
log.Fatal("Error reading stdin")
} }
// get input path
if c.String("input") == "" {
log.Fatal("No input specified")
}
inputFilePath := c.String("input")
// get output path
outputFilePath := inputFilePath
if c.String("output") != "" {
outputFilePath = c.String("output")
}
// get file
inputFile, err := os.Open(inputFilePath)
if err != nil {
log.Fatal(err)
}
// read file
input, err := ioutil.ReadAll(inputFile)
if err != nil {
log.Fatal(err)
}
// set up encoder
erasureParameters, _ := erasure.ValidateParams(10, 5, 8, erasure.CAUCHY)
encoder := erasure.NewEncoder(erasureParameters)
// encode data
encodedData, _ := encoder.Encode(input) encodedData, _ := encoder.Encode(input)
// write encoded data out
for key, data := range encodedData { for key, data := range encodedData {
ioutil.WriteFile("output."+strconv.Itoa(key), data, 0600) ioutil.WriteFile(outputFilePath+"."+strconv.Itoa(key), data, 0600)
} }
} }
app.Run(os.Args) app.Run(os.Args)