parent
15552a59f9
commit
856e0100c0
@ -0,0 +1,98 @@ |
||||
/* |
||||
* Mini Object Storage, (C) 2015 Minio, Inc. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
*/ |
||||
|
||||
package erasure1 |
||||
|
||||
import ( |
||||
"bytes" |
||||
"encoding/binary" |
||||
"encoding/gob" |
||||
"errors" |
||||
"io" |
||||
) |
||||
|
||||
// DataHeader represents the structure serialized to gob.
|
||||
type DataHeader struct { |
||||
// object + block stored
|
||||
Key string |
||||
// chunk index of encoded block
|
||||
ChunkIndex uint8 |
||||
// Original Length of the block output
|
||||
OriginalLength uint32 |
||||
// Data Blocks
|
||||
EncoderK uint8 |
||||
// Parity Blocks
|
||||
EncoderM uint8 |
||||
// Matrix Technique
|
||||
EncoderTechnique EncoderTechnique |
||||
} |
||||
|
||||
// EncoderTechnique specified the Matrix type used in encoding
|
||||
type EncoderTechnique int |
||||
|
||||
const ( |
||||
// Vandermonde matrix type
|
||||
Vandermonde EncoderTechnique = iota |
||||
// Cauchy matrix type
|
||||
Cauchy |
||||
) |
||||
|
||||
// validate populated header
|
||||
func validateHeader(header DataHeader) error { |
||||
if header.Key == "" { |
||||
return errors.New("Empty Key") |
||||
} |
||||
|
||||
if header.EncoderTechnique > 1 { |
||||
return errors.New("Invalid encoder technique") |
||||
} |
||||
|
||||
return nil |
||||
} |
||||
|
||||
// WriteData returns error upon any failure
|
||||
func Write(target io.Writer, key string, part uint8, length uint32, k, m uint8, technique EncoderTechnique, data io.Reader) error { |
||||
header := DataHeader{ |
||||
Key: key, |
||||
ChunkIndex: part, |
||||
OriginalLength: length, |
||||
EncoderK: k, |
||||
EncoderM: m, |
||||
EncoderTechnique: technique, |
||||
} |
||||
|
||||
if err := validateHeader(header); err != nil { |
||||
return err |
||||
} |
||||
|
||||
var headerBuffer bytes.Buffer |
||||
// encode header
|
||||
encoder := gob.NewEncoder(&headerBuffer) |
||||
encoder.Encode(header) |
||||
|
||||
// write version
|
||||
binary.Write(target, binary.LittleEndian, uint32(1)) |
||||
|
||||
// write encoded header
|
||||
if _, err := io.Copy(target, &headerBuffer); err != nil { |
||||
return err |
||||
} |
||||
// write data
|
||||
if _, err := io.Copy(target, data); err != nil { |
||||
return err |
||||
} |
||||
return nil |
||||
} |
@ -1,106 +0,0 @@ |
||||
/* |
||||
* Mini Object Storage, (C) 2015 Minio, Inc. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
*/ |
||||
|
||||
package erasure |
||||
|
||||
import ( |
||||
"bytes" |
||||
"encoding/binary" |
||||
"encoding/gob" |
||||
"fmt" |
||||
"io" |
||||
) |
||||
|
||||
// Metadata map
|
||||
type Metadata map[string]string |
||||
|
||||
// DataHeader struct
|
||||
type DataHeader struct { |
||||
Key string |
||||
Part uint8 |
||||
Metadata Metadata |
||||
EncoderParams EncoderParams |
||||
} |
||||
|
||||
// EncoderTechnique type
|
||||
type EncoderTechnique int |
||||
|
||||
// EncoderTechniques
|
||||
const ( |
||||
Vandermonde EncoderTechnique = iota |
||||
Cauchy |
||||
) |
||||
|
||||
// EncoderParams struct
|
||||
type EncoderParams struct { |
||||
Length uint32 |
||||
K uint8 |
||||
M uint8 |
||||
Technique EncoderTechnique |
||||
} |
||||
|
||||
// NewHeader populate new header
|
||||
func NewHeader(key string, part uint8, metadata Metadata, encoderParams EncoderParams) DataHeader { |
||||
header := DataHeader{} |
||||
header.Key = key |
||||
header.Part = part |
||||
header.Metadata = metadata |
||||
header.EncoderParams = EncoderParams{ |
||||
Length: encoderParams.Length, |
||||
K: encoderParams.K, |
||||
M: encoderParams.M, |
||||
Technique: encoderParams.Technique, |
||||
} |
||||
return header |
||||
} |
||||
|
||||
// ValidateHeader validate populated header
|
||||
func ValidateHeader(header DataHeader) bool { |
||||
if header.Key == "" || header.Part < 0 || len(header.Metadata) < 2 { |
||||
return false |
||||
} |
||||
|
||||
if header.EncoderParams.Length < 0 || header.EncoderParams.Technique > 1 { |
||||
return false |
||||
} |
||||
|
||||
return true |
||||
} |
||||
|
||||
// WriteData write data, returns error upon any failure
|
||||
func WriteData(target io.Writer, header DataHeader, data io.Reader) error { |
||||
if !ValidateHeader(header) { |
||||
return fmt.Errorf("Invalid header") |
||||
} |
||||
|
||||
var headerBuffer bytes.Buffer |
||||
// encode header
|
||||
encoder := gob.NewEncoder(&headerBuffer) |
||||
encoder.Encode(header) |
||||
|
||||
// write version
|
||||
binary.Write(target, binary.LittleEndian, uint32(1)) |
||||
|
||||
// write encoded header
|
||||
if _, err := io.Copy(target, &headerBuffer); err != nil { |
||||
return err |
||||
} |
||||
// write data
|
||||
if _, err := io.Copy(target, data); err != nil { |
||||
return err |
||||
} |
||||
return nil |
||||
} |
Loading…
Reference in new issue