parent
4b97cf4b9d
commit
5a3eb47583
@ -0,0 +1,95 @@ |
|||||||
|
package main |
||||||
|
|
||||||
|
import ( |
||||||
|
"io/ioutil" |
||||||
|
"log" |
||||||
|
"os" |
||||||
|
"strconv" |
||||||
|
|
||||||
|
"github.com/codegangsta/cli" |
||||||
|
"github.com/minio-io/minio/erasure" |
||||||
|
) |
||||||
|
|
||||||
|
func main() { |
||||||
|
app := cli.NewApp() |
||||||
|
app.Name = "minio-encode" |
||||||
|
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", |
||||||
|
}, |
||||||
|
cli.IntFlag{ |
||||||
|
Name: "k", |
||||||
|
Value: 10, |
||||||
|
Usage: "k value of encoder parameters", |
||||||
|
}, |
||||||
|
cli.IntFlag{ |
||||||
|
Name: "m", |
||||||
|
Value: 5, |
||||||
|
Usage: "m value of encoder parameters", |
||||||
|
}, |
||||||
|
} |
||||||
|
app.Action = func(c *cli.Context) { |
||||||
|
// check if minio-encode called without parameters
|
||||||
|
if len(c.Args()) == 1 { |
||||||
|
cli.ShowAppHelp(c) |
||||||
|
} |
||||||
|
|
||||||
|
// 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") |
||||||
|
} |
||||||
|
|
||||||
|
k := c.Int("k") |
||||||
|
m := c.Int("m") |
||||||
|
|
||||||
|
// get chunks
|
||||||
|
chunks := make([][]byte, k+m) |
||||||
|
for i := 0; i < k+m; i++ { |
||||||
|
var err error |
||||||
|
chunks[i], err = ioutil.ReadFile(inputFilePath + "." + strconv.Itoa(i)) |
||||||
|
if err != nil { |
||||||
|
log.Fatal(err) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// get length
|
||||||
|
lengthBytes, err := ioutil.ReadFile(inputFilePath + ".length") |
||||||
|
if err != nil { |
||||||
|
log.Fatal(err) |
||||||
|
} |
||||||
|
lengthString := string(lengthBytes) |
||||||
|
length, err := strconv.Atoi(lengthString) |
||||||
|
if err != nil { |
||||||
|
log.Fatal(err) |
||||||
|
} |
||||||
|
|
||||||
|
// set up encoder
|
||||||
|
erasureParameters, _ := erasure.ValidateParams(k, m, 8, erasure.CAUCHY) |
||||||
|
encoder := erasure.NewEncoder(erasureParameters) |
||||||
|
|
||||||
|
// decode data
|
||||||
|
decodedData, err := encoder.Decode(chunks, length) |
||||||
|
if err != nil { |
||||||
|
log.Fatal(err) |
||||||
|
} |
||||||
|
|
||||||
|
// write decode data out
|
||||||
|
ioutil.WriteFile(outputFilePath, decodedData, 0600) |
||||||
|
} |
||||||
|
app.Run(os.Args) |
||||||
|
} |
Loading…
Reference in new issue