Add FilteredBAckingStore::co_getBlobAuxData

Summary: Add coroutine (now_task) version of getBlobAuxData.

Reviewed By: SBones

Differential Revision: D103303864

fbshipit-source-id: 5d8951c84f4cfb7c677e8ba9ba57b61193cd2751

Author: janezhang10

eden/fs/store/FilteredBackingStore.cpp

@@ -545,6 +545,19 @@ FilteredBackingStore::getBlobAuxData(
   return backingStore_->getBlobAuxData(filteredId.object(), context);
 }
 
+folly::coro::now_task<BackingStore::GetBlobAuxResult>
+FilteredBackingStore::co_getBlobAuxData(
+    const ObjectId& id,
+    const ObjectFetchContextPtr& context) {
+  if (isSlOid(id)) {
+    co_return co_await backingStore_->getBlobAuxData(id, context);
+  }
+
+  auto filteredId = FilteredObjectId::fromObjectId(id);
+  co_return co_await backingStore_->getBlobAuxData(
+      filteredId.object(), context);
+}
+
 folly::SemiFuture<BackingStore::GetBlobResult> FilteredBackingStore::getBlob(
     const ObjectId& id,
     const ObjectFetchContextPtr& context) {

eden/fs/store/FilteredBackingStore.h

@@ -145,6 +145,7 @@ class FilteredBackingStore
   FRIEND_TEST(SaplingFilteredBackingStoreTest, testMercurialFFIInvalidFOID);
   FRIEND_TEST(FakeSubstringFilteredBackingStoreTest, getNonExistent);
   FRIEND_TEST(FakeSubstringFilteredBackingStoreTest, getBlob);
+  FRIEND_TEST(FakeSubstringFilteredBackingStoreTest, co_getBlobAuxData);
   FRIEND_TEST(FakeSubstringFilteredBackingStoreTest, getTree);
   FRIEND_TEST(FakeSubstringFilteredBackingStoreTest, getRootTree);
   FRIEND_TEST(FakeSubstringFilteredBackingStoreTest, getGlobFiles);
@@ -187,6 +188,10 @@ class FilteredBackingStore
       const ObjectId& id,
       const ObjectFetchContextPtr& context) override;
 
+  folly::coro::now_task<GetBlobAuxResult> co_getBlobAuxData(
+      const ObjectId& id,
+      const ObjectFetchContextPtr& context);
+
   [[nodiscard]] folly::SemiFuture<folly::Unit> prefetchBlobs(
       ObjectIdRange ids,
       const ObjectFetchContextPtr& context) override;

eden/fs/store/test/FilteredBackingStoreTest.cpp

@@ -286,6 +286,23 @@ TEST_F(FakeSubstringFilteredBackingStoreTest, getBlob) {
   EXPECT_EQ("foobar", blobContents(*std::move(future6).get(0ms).blob));
 }
 
+CO_TEST_F(FakeSubstringFilteredBackingStoreTest, co_getBlobAuxData) {
+  auto id = makeTestId("1");
+  auto filteredId = ObjectId{
+      FilteredObjectId{id, FilteredObjectIdType::OBJECT_TYPE_BLOB}.getValue()};
+  auto* storedBlob = wrappedStore_->putBlob(id, "foobar");
+  storedBlob->setReady();
+
+  auto result = co_await filteredStore_->co_getBlobAuxData(
+      filteredId, ObjectFetchContext::getNullContext());
+
+  EXPECT_NE(result.blobAux, nullptr);
+  EXPECT_EQ(result.blobAux->size, 6);
+  EXPECT_EQ(
+      result.blobAux->sha1,
+      Hash20::sha1(folly::ByteRange{folly::StringPiece{"foobar"}}));
+}
+
 TEST_F(FakeSubstringFilteredBackingStoreTest, getTree) {
   // Populate some files in the store
   auto [runme, runme_id] =