Chris Lu
12 years ago
4 changed files with 73 additions and 54 deletions
-
42weed-fs/src/pkg/storage/needle.go
-
52weed-fs/src/pkg/storage/needle_read_write.go
-
3weed-fs/src/pkg/storage/volume.go
-
12weed-fs/src/pkg/storage/volume_version.go
@ -0,0 +1,52 @@ |
|||
package storage |
|||
|
|||
import ( |
|||
"errors" |
|||
"io" |
|||
"os" |
|||
"pkg/util" |
|||
) |
|||
|
|||
func (n *Needle) Append(w io.Writer) uint32 { |
|||
header := make([]byte, 16) |
|||
util.Uint32toBytes(header[0:4], n.Cookie) |
|||
util.Uint64toBytes(header[4:12], n.Id) |
|||
n.Size = uint32(len(n.Data)) |
|||
util.Uint32toBytes(header[12:16], n.Size) |
|||
w.Write(header) |
|||
w.Write(n.Data) |
|||
rest := 8 - ((n.Size + 16 + 4) % 8) |
|||
util.Uint32toBytes(header[0:4], n.Checksum.Value()) |
|||
w.Write(header[0 : rest+4]) |
|||
return n.Size |
|||
} |
|||
func (n *Needle) Read(r io.Reader, size uint32, version Version) (int, error) { |
|||
if version == Version1 { |
|||
bytes := make([]byte, size+16+4) |
|||
ret, e := r.Read(bytes) |
|||
n.Cookie = util.BytesToUint32(bytes[0:4]) |
|||
n.Id = util.BytesToUint64(bytes[4:12]) |
|||
n.Size = util.BytesToUint32(bytes[12:16]) |
|||
n.Data = bytes[16 : 16+size] |
|||
checksum := util.BytesToUint32(bytes[16+size : 16+size+4]) |
|||
if checksum != NewCRC(n.Data).Value() { |
|||
return 0, errors.New("CRC error! Data On Disk Corrupted!") |
|||
} |
|||
return ret, e |
|||
}else if version == Version2 { |
|||
} |
|||
return 0, errors.New("Unsupported Version!") |
|||
} |
|||
func ReadNeedle(r *os.File) (*Needle, uint32) { |
|||
n := new(Needle) |
|||
bytes := make([]byte, 16) |
|||
count, e := r.Read(bytes) |
|||
if count <= 0 || e != nil { |
|||
return nil, 0 |
|||
} |
|||
n.Cookie = util.BytesToUint32(bytes[0:4]) |
|||
n.Id = util.BytesToUint64(bytes[4:12]) |
|||
n.Size = util.BytesToUint32(bytes[12:16]) |
|||
rest := 8 - ((n.Size + 16 + 4) % 8) |
|||
return n, n.Size + 4 + rest |
|||
} |
@ -0,0 +1,12 @@ |
|||
package storage |
|||
|
|||
import ( |
|||
) |
|||
|
|||
type Version uint8 |
|||
|
|||
const ( |
|||
CurrentVersion = Version(1) |
|||
Version1 = Version(1) |
|||
Version2 = Version(2) |
|||
) |
Write
Preview
Loading…
Cancel
Save
Reference in new issue