mirror of
https://github.com/seaweedfs/seaweedfs.git
synced 2024-01-19 02:48:24 +00:00
needle read into a writer
This commit is contained in:
parent
663bc5dc23
commit
685643d506
|
@ -9,6 +9,32 @@ import (
|
||||||
"io"
|
"io"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
// ReadNeedleDataInto uses a needle without n.Data to read the content into an io.Writer
|
||||||
|
func (n *Needle) ReadNeedleDataInto(r backend.BackendStorageFile, offset int64, buf []byte, writer io.Writer, expectedChecksumValue uint32) (err error) {
|
||||||
|
crc := CRC(0)
|
||||||
|
for x := 0; ; x += len(buf) {
|
||||||
|
count, err := n.ReadNeedleData(r, offset, buf, int64(x))
|
||||||
|
if err != nil {
|
||||||
|
if err == io.EOF {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
return fmt.Errorf("ReadNeedleData: %v", err)
|
||||||
|
}
|
||||||
|
if count > 0 {
|
||||||
|
crc = crc.Update(buf[0:count])
|
||||||
|
if _, err = writer.Write(buf[0:count]); err != nil {
|
||||||
|
return fmt.Errorf("ReadNeedleData write: %v", err)
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if expectedChecksumValue != crc.Value() {
|
||||||
|
return fmt.Errorf("ReadNeedleData checksum %v expected %v", crc.Value(), expectedChecksumValue)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
// ReadNeedleData uses a needle without n.Data to read the content
|
// ReadNeedleData uses a needle without n.Data to read the content
|
||||||
// volumeOffset: the offset within the volume
|
// volumeOffset: the offset within the volume
|
||||||
// needleOffset: the offset within the needle Data
|
// needleOffset: the offset within the needle Data
|
||||||
|
|
|
@ -3,7 +3,6 @@ package needle
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
"github.com/chrislusf/seaweedfs/weed/storage/backend"
|
"github.com/chrislusf/seaweedfs/weed/storage/backend"
|
||||||
"github.com/stretchr/testify/assert"
|
|
||||||
"io"
|
"io"
|
||||||
"os"
|
"os"
|
||||||
"testing"
|
"testing"
|
||||||
|
@ -75,24 +74,9 @@ func TestPageRead(t *testing.T) {
|
||||||
fmt.Printf("Checksum value %d\n", checksumValue)
|
fmt.Printf("Checksum value %d\n", checksumValue)
|
||||||
|
|
||||||
buf := make([]byte, 1024)
|
buf := make([]byte, 1024)
|
||||||
crc := CRC(0)
|
if err = n.ReadNeedleDataInto(datBackend, offset, buf, io.Discard, checksumValue); err != nil {
|
||||||
for x := int64(0); ; x += 1024 {
|
t.Fatalf("ReadNeedleDataInto: %v", err)
|
||||||
count, err := n.ReadNeedleData(datBackend, offset, buf, x)
|
|
||||||
if err != nil {
|
|
||||||
if err == io.EOF {
|
|
||||||
break
|
|
||||||
}
|
|
||||||
t.Fatalf("ReadNeedleData: %v", err)
|
|
||||||
}
|
|
||||||
if count > 0 {
|
|
||||||
crc = crc.Update(buf[0:count])
|
|
||||||
} else {
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
fmt.Printf("read checksum value %d\n", crc.Value())
|
|
||||||
|
|
||||||
assert.Equal(t, checksumValue, crc.Value(), "validate checksum value")
|
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue