From 0a8dbe086a23bbb905514007d21e2e04cdaae1e8 Mon Sep 17 00:00:00 2001
From: Maksim Denisov <denisov.maksim@huawei.com>
Date: Tue, 4 Feb 2025 18:26:39 +0100
Subject: IExecutionApi: Use unordered_set in IsAvailable

---
 src/buildtool/execution_api/common/common_api.hpp    | 7 ++++---
 src/buildtool/execution_api/common/execution_api.hpp | 5 +++--
 2 files changed, 7 insertions(+), 5 deletions(-)

(limited to 'src/buildtool/execution_api/common')

diff --git a/src/buildtool/execution_api/common/common_api.hpp b/src/buildtool/execution_api/common/common_api.hpp
index 991b9875..d5daa3dd 100644
--- a/src/buildtool/execution_api/common/common_api.hpp
+++ b/src/buildtool/execution_api/common/common_api.hpp
@@ -21,6 +21,7 @@
 #include <iterator>
 #include <optional>
 #include <unordered_map>
+#include <unordered_set>
 #include <vector>
 
 #include "gsl/gsl"
@@ -39,7 +40,7 @@
 /// to some given original type.
 template <typename T>
 struct MissingArtifactsInfo {
-    std::vector<ArtifactDigest> digests;
+    std::unordered_set<ArtifactDigest> digests;
     std::unordered_map<ArtifactDigest, T> back_map;
 };
 
@@ -69,7 +70,7 @@ template <typename TValue, typename TIterator>
     TIterator const& end,
     typename std::function<ArtifactDigest(TValue const&)> const&
         converter) noexcept -> std::optional<MissingArtifactsInfo<TValue>> {
-    std::vector<ArtifactDigest> digests;
+    std::unordered_set<ArtifactDigest> digests;
     digests.reserve(std::distance(begin, end));
     MissingArtifactsInfo<TValue> res{};
     for (auto it = begin; it != end; ++it) {
@@ -77,7 +78,7 @@ template <typename TValue, typename TIterator>
             auto const inserted =
                 res.back_map.insert({std::invoke(converter, *it), *it});
             if (inserted.second) {
-                digests.emplace_back(inserted.first->first);
+                digests.emplace(inserted.first->first);
             }
         } catch (...) {
             return std::nullopt;
diff --git a/src/buildtool/execution_api/common/execution_api.hpp b/src/buildtool/execution_api/common/execution_api.hpp
index 02c78bcc..b868b57d 100644
--- a/src/buildtool/execution_api/common/execution_api.hpp
+++ b/src/buildtool/execution_api/common/execution_api.hpp
@@ -21,6 +21,7 @@
 #include <memory>
 #include <optional>
 #include <string>
+#include <unordered_set>
 #include <vector>
 
 #include "src/buildtool/common/artifact.hpp"  // Artifact::ObjectInfo
@@ -129,8 +130,8 @@ class IExecutionApi {
         ArtifactDigest const& digest) const noexcept -> bool = 0;
 
     [[nodiscard]] virtual auto IsAvailable(
-        std::vector<ArtifactDigest> const& digests) const noexcept
-        -> std::vector<ArtifactDigest> = 0;
+        std::unordered_set<ArtifactDigest> const& digests) const noexcept
+        -> std::unordered_set<ArtifactDigest> = 0;
 
     [[nodiscard]] virtual auto SplitBlob(ArtifactDigest const& /*blob_digest*/)
         const noexcept -> std::optional<std::vector<ArtifactDigest>> {
-- 
cgit v1.2.3