From d79ef7ef223250fb78f4635413bf99acee6ccc4c Mon Sep 17 00:00:00 2001
From: SatelliteMind <stuff.szymek.655@outlook.com>
Date: Fri, 12 Jul 2024 11:26:52 +0200
Subject: storage: filesystem, Fix object cache not working due to
 uninitialised objects being put into cache; The hash (which is the cache key)
 returned by the object at the previous put-into-cache point is always zero,
 so subsequent loads of the same object were never resolved from the cache.

---
 storage/filesystem/object.go      |  4 +--
 storage/filesystem/object_test.go | 61 +++++++++++++++++++++++++++++++++++++++
 2 files changed, 63 insertions(+), 2 deletions(-)

diff --git a/storage/filesystem/object.go b/storage/filesystem/object.go
index e812fe9..91b4ace 100644
--- a/storage/filesystem/object.go
+++ b/storage/filesystem/object.go
@@ -431,13 +431,13 @@ func (s *ObjectStorage) getFromUnpacked(h plumbing.Hash) (obj plumbing.EncodedOb
 
 	defer ioutil.CheckClose(w, &err)
 
-	s.objectCache.Put(obj)
-
 	bufp := copyBufferPool.Get().(*[]byte)
 	buf := *bufp
 	_, err = io.CopyBuffer(w, r, buf)
 	copyBufferPool.Put(bufp)
 
+	s.objectCache.Put(obj)
+
 	return obj, err
 }
 
diff --git a/storage/filesystem/object_test.go b/storage/filesystem/object_test.go
index 251077a..4f98458 100644
--- a/storage/filesystem/object_test.go
+++ b/storage/filesystem/object_test.go
@@ -547,3 +547,64 @@ func BenchmarkGetObjectFromPackfile(b *testing.B) {
 		})
 	}
 }
+
+func (s *FsSuite) TestGetFromUnpackedCachesObjects(c *C) {
+	fs := fixtures.ByTag(".git").ByTag("unpacked").One().DotGit()
+	objectCache := cache.NewObjectLRUDefault()
+	objectStorage := NewObjectStorage(dotgit.New(fs), objectCache)
+	hash := plumbing.NewHash("f3dfe29d268303fc6e1bbce268605fc99573406e")
+
+	// Assert the cache is empty initially
+	_, ok := objectCache.Get(hash)
+	c.Assert(ok, Equals, false)
+
+	// Load the object
+	obj, err := objectStorage.EncodedObject(plumbing.AnyObject, hash)
+	c.Assert(err, IsNil)
+	c.Assert(obj.Hash(), Equals, hash)
+
+	// The object should've been cached during the load
+	cachedObj, ok := objectCache.Get(hash)
+	c.Assert(ok, Equals, true)
+	c.Assert(cachedObj, DeepEquals, obj)
+
+	// Assert that both objects can be read and that they both produce the same bytes
+
+	objReader, err := obj.Reader()
+	c.Assert(err, IsNil)
+	objBytes, err := io.ReadAll(objReader)
+	c.Assert(err, IsNil)
+	c.Assert(len(objBytes), Not(Equals), 0)
+	err = objReader.Close()
+	c.Assert(err, IsNil)
+
+	cachedObjReader, err := cachedObj.Reader()
+	c.Assert(err, IsNil)
+	cachedObjBytes, err := io.ReadAll(cachedObjReader)
+	c.Assert(len(cachedObjBytes), Not(Equals), 0)
+	c.Assert(err, IsNil)
+	err = cachedObjReader.Close()
+	c.Assert(err, IsNil)
+
+	c.Assert(cachedObjBytes, DeepEquals, objBytes)
+}
+
+func (s *FsSuite) TestGetFromUnpackedDoesNotCacheLargeObjects(c *C) {
+	fs := fixtures.ByTag(".git").ByTag("unpacked").One().DotGit()
+	objectCache := cache.NewObjectLRUDefault()
+	objectStorage := NewObjectStorageWithOptions(dotgit.New(fs), objectCache, Options{LargeObjectThreshold: 1})
+	hash := plumbing.NewHash("f3dfe29d268303fc6e1bbce268605fc99573406e")
+
+	// Assert the cache is empty initially
+	_, ok := objectCache.Get(hash)
+	c.Assert(ok, Equals, false)
+
+	// Load the object
+	obj, err := objectStorage.EncodedObject(plumbing.AnyObject, hash)
+	c.Assert(err, IsNil)
+	c.Assert(obj.Hash(), Equals, hash)
+
+	// The object should not have been cached during the load
+	_, ok = objectCache.Get(hash)
+	c.Assert(ok, Equals, false)
+}
-- 
cgit