From aa092f5474da3e9c3ff4f40b88849726b645f39f Mon Sep 17 00:00:00 2001
From: Jeremy Stribling <strib@alum.mit.edu>
Date: Fri, 22 Sep 2017 21:14:17 -0700
Subject: plumbing: add `HasEncodedObject` method to Storer

This allows the user to check whether an object exists, without
reading all the object data from storage.

Issue: KBFS-2445
---
 storage/filesystem/object.go | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)

(limited to 'storage/filesystem')

diff --git a/storage/filesystem/object.go b/storage/filesystem/object.go
index 13b8ddb..34501ec 100644
--- a/storage/filesystem/object.go
+++ b/storage/filesystem/object.go
@@ -126,6 +126,32 @@ func (s *ObjectStorage) SetEncodedObject(o plumbing.EncodedObject) (plumbing.Has
 	return o.Hash(), err
 }
 
+// HasEncodedObject returns nil if the object exists, without actually
+// reading the object data from storage.
+func (s *ObjectStorage) HasEncodedObject(h plumbing.Hash) (err error) {
+	// Check unpacked objects
+	f, err := s.dir.Object(h)
+	if err != nil {
+		if !os.IsNotExist(err) {
+			return err
+		}
+		// Fall through to check packed objects.
+	} else {
+		defer ioutil.CheckClose(f, &err)
+		return nil
+	}
+
+	// Check packed objects.
+	if err := s.requireIndex(); err != nil {
+		return err
+	}
+	_, _, offset := s.findObjectInPackfile(h)
+	if offset == -1 {
+		return plumbing.ErrObjectNotFound
+	}
+	return nil
+}
+
 // EncodedObject returns the object with the given hash, by searching for it in
 // the packfile and the git object directories.
 func (s *ObjectStorage) EncodedObject(t plumbing.ObjectType, h plumbing.Hash) (plumbing.EncodedObject, error) {
-- 
cgit