Adding read header for erasure

This commit is contained in:
Frederick F. Kautz IV
2015-03-07 15:53:41 -08:00
parent 00381c887a
commit ca1a4b616c
2 changed files with 43 additions and 1 deletions

View File

@@ -22,6 +22,11 @@ import (
"encoding/gob"
"errors"
"io"
"strconv"
)
const (
Version = uint32(1)
)
// DataHeader represents the structure serialized to gob.
@@ -84,7 +89,7 @@ func Write(target io.Writer, key string, part uint8, length uint32, k, m uint8,
encoder.Encode(header)
// write version
binary.Write(target, binary.LittleEndian, uint32(1))
binary.Write(target, binary.LittleEndian, uint32(Version))
// write encoded header
if _, err := io.Copy(target, &headerBuffer); err != nil {
@@ -96,3 +101,18 @@ func Write(target io.Writer, key string, part uint8, length uint32, k, m uint8,
}
return nil
}
// Read an erasure block
func ReadHeader(reader io.Reader) (dataHeader DataHeader, err error) {
versionArray := make([]byte, 4)
if err := binary.Read(reader, binary.LittleEndian, versionArray); err != nil {
return dataHeader, err
}
version := binary.LittleEndian.Uint32(versionArray)
if version > Version || version == 0 {
return dataHeader, errors.New("Unknown version: " + strconv.FormatUint(uint64(version), 10))
}
decoder := gob.NewDecoder(reader)
err = decoder.Decode(&dataHeader)
return dataHeader, err
}