correct cache: fix racing condition

2 files changed, 7 insertions, 9 deletions

diff --git a/go/storage/needle_byte_cache.go b/go/storage/needle_byte_cache.go
index 24a72e244..d39ed23c4 100644
--- a/go/storage/needle_byte_cache.go
+++ b/go/storage/needle_byte_cache.go
@@ -45,21 +45,24 @@ func (block *Block) increaseReference() {
 
 // get bytes from the LRU cache of []byte first, then from the bytes pool
 // when []byte in LRU cache is evicted, it will be put back to the bytes pool
-func getBytesForFileBlock(r *os.File, offset int64, readSize int) (block *Block, isNew bool) {
+func getBytesForFileBlock(r *os.File, offset int64, readSize int) (dataSlice []byte, block *Block, err error) {
 	// check cache, return if found
 	cacheKey := fmt.Sprintf("%d:%d:%d", r.Fd(), offset>>3, readSize)
 	if obj, found := bytesCache.Get(cacheKey); found {
 		block = obj.(*Block)
 		block.increaseReference()
-		return block, false
+		dataSlice = block.Bytes[0:readSize]
+		return dataSlice, block, nil
 	}
 
 	// get the []byte from pool
 	b := bytesPool.Get(readSize)
 	// refCount = 2, one by the bytesCache, one by the actual needle object
 	block = &Block{Bytes: b, refCount: 2}
+	dataSlice = block.Bytes[0:readSize]
+	_, err = r.ReadAt(dataSlice, offset)
 	bytesCache.Add(cacheKey, block)
-	return block, true
+	return dataSlice, block, err
 }
 
 func (n *Needle) ReleaseMemory() {
diff --git a/go/storage/needle_read_write.go b/go/storage/needle_read_write.go
index e2be256be..49ae4fee4 100644
--- a/go/storage/needle_read_write.go
+++ b/go/storage/needle_read_write.go
@@ -139,12 +139,7 @@ func (n *Needle) Append(w io.Writer, version Version) (size uint32, err error) {
 func ReadNeedleBlob(r *os.File, offset int64, size uint32) (dataSlice []byte, block *Block, err error) {
 	padding := NeedlePaddingSize - ((NeedleHeaderSize + size + NeedleChecksumSize) % NeedlePaddingSize)
 	readSize := NeedleHeaderSize + size + NeedleChecksumSize + padding
-	block, isNew := getBytesForFileBlock(r, offset, int(readSize))
-	dataSlice = block.Bytes[0:int(readSize)]
-	if isNew {
-		_, err = r.ReadAt(dataSlice, offset)
-	}
-	return dataSlice, block, err
+	return getBytesForFileBlock(r, offset, int(readSize))
 }
 
 func (n *Needle) ReadData(r *os.File, offset int64, size uint32, version Version) (err error) {