mirror of
https://github.com/seaweedfs/seaweedfs.git
synced 2026-10-23 06:20:48 +02:00
refactoring
This commit is contained in:
1 parent
79a49ada39
commit
11cc489ca5
4 files changed
+72
-52
No files matched your search
@@ -7,22 +7,25 @@ import (
|
||||
"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 - ((16 + n.Size + 4) % 8)
|
||||
util.Uint32toBytes(header[0:4], n.Checksum.Value())
|
||||
w.Write(header[0 : 4+rest])
|
||||
func (n *Needle) Append(w io.Writer, version Version) uint32 {
|
||||
if version == Version1 {
|
||||
header := make([]byte, NeedleHeaderSize)
|
||||
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 := NeedlePaddingSize - ((NeedleHeaderSize + n.Size + 4) % NeedlePaddingSize)
|
||||
util.Uint32toBytes(header[0:4], n.Checksum.Value())
|
||||
w.Write(header[0 : 4+rest])
|
||||
} else if version == Version2 {
|
||||
}
|
||||
return n.Size
|
||||
}
|
||||
func (n *Needle) Read(r io.Reader, size uint32, version Version) (int, error) {
|
||||
if version == Version1 {
|
||||
bytes := make([]byte, 16+size+4)
|
||||
bytes := make([]byte, NeedleHeaderSize+size+4)
|
||||
ret, e := r.Read(bytes)
|
||||
n.Cookie = util.BytesToUint32(bytes[0:4])
|
||||
n.Id = util.BytesToUint64(bytes[4:12])
|
||||
@@ -37,10 +40,10 @@ func (n *Needle) Read(r io.Reader, size uint32, version Version) (int, error) {
|
||||
}
|
||||
return 0, errors.New("Unsupported Version!")
|
||||
}
|
||||
func ReadNeedle(r *os.File, version Version) (n *Needle, bytesTillNextFile uint32) {
|
||||
func ReadNeedleHeader(r *os.File, version Version) (n *Needle, bodyLength uint32) {
|
||||
n = new(Needle)
|
||||
if version == Version1 {
|
||||
bytes := make([]byte, 16)
|
||||
bytes := make([]byte, NeedleHeaderSize)
|
||||
count, e := r.Read(bytes)
|
||||
if count <= 0 || e != nil {
|
||||
return nil, 0
|
||||
@@ -48,8 +51,21 @@ func ReadNeedle(r *os.File, version Version) (n *Needle, bytesTillNextFile uint3
|
||||
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)
|
||||
bytesTillNextFile = n.Size + 4 + rest
|
||||
rest := NeedlePaddingSize - ((n.Size + 16 + 4) % NeedlePaddingSize)
|
||||
bodyLength = n.Size + 4 + rest
|
||||
} else if version == Version2 {
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
//n should be a needle already read the header
|
||||
//the input stream will read until next file entry
|
||||
func (n *Needle) ReadNeedleBody(r *os.File, version Version, bodyLength uint32) {
|
||||
if version == Version1 {
|
||||
bytes := make([]byte, bodyLength)
|
||||
r.Read(bytes)
|
||||
n.Data = bytes[:n.Size]
|
||||
n.Checksum = NewCRC(n.Data)
|
||||
} else if version == Version2 {
|
||||
}
|
||||
return
|
||||
|
||||
Reference in new issue
Block a user