minor optimization

1 files changed, 8 insertions, 1 deletions

diff --git a/weed/storage/volume_read.go b/weed/storage/volume_read.go
index b38ec32d3..d4d795fee 100644
--- a/weed/storage/volume_read.go
+++ b/weed/storage/volume_read.go
@@ -107,7 +107,7 @@ func (v *Volume) readNeedleDataInto(n *needle.Needle, readOption *ReadOption, wr
 		readOption.IsOutOfRange = false
 		err = n.ReadNeedleMeta(v.DataBackend, nv.Offset.ToActualOffset(), readSize, v.Version())
 	}
-	buf := mem.Allocate(1024 * 1024)
+	buf := mem.Allocate(min(1024*1024, int(size)))
 	defer mem.Free(buf)
 	actualOffset := nv.Offset.ToActualOffset()
 	if readOption.IsOutOfRange {
@@ -117,6 +117,13 @@ func (v *Volume) readNeedleDataInto(n *needle.Needle, readOption *ReadOption, wr
 	return n.ReadNeedleDataInto(v.DataBackend, actualOffset, buf, writer, offset, size)
 }
 
+func min(x, y int) int {
+	if x < y {
+		return x
+	}
+	return y
+}
+
 // read fills in Needle content by looking up n.Id from NeedleMapper
 func (v *Volume) ReadNeedleBlob(offset int64, size Size) ([]byte, error) {
 	v.dataFileAccessLock.RLock()