Pass StorageConfig and Storage to Serve

1 files changed, 11 insertions, 1 deletions

diff --git a/src/buildtool/serve_api/serve_service/target.hpp b/src/buildtool/serve_api/serve_service/target.hpp
index 5301b82c..9488e19c 100644
--- a/src/buildtool/serve_api/serve_service/target.hpp
+++ b/src/buildtool/serve_api/serve_service/target.hpp
@@ -34,6 +34,8 @@
 #include "src/buildtool/logging/logger.hpp"
 #include "src/buildtool/serve_api/remote/config.hpp"
 #include "src/buildtool/serve_api/remote/serve_api.hpp"
+#include "src/buildtool/storage/config.hpp"
+#include "src/buildtool/storage/storage.hpp"
 #include "src/utils/cpp/expected.hpp"
 
 // The target-level cache service.
@@ -41,9 +43,15 @@ class TargetService final : public justbuild::just_serve::Target::Service {
   public:
     explicit TargetService(
         gsl::not_null<RemoteServeConfig const*> const& serve_config,
+        gsl::not_null<StorageConfig const*> const& storage_config,
+        gsl::not_null<Storage const*> const& storage,
         gsl::not_null<ApiBundle const*> const& apis,
         ServeApi const* serve = nullptr) noexcept
-        : serve_config_{*serve_config}, apis_{*apis}, serve_{serve} {}
+        : serve_config_{*serve_config},
+          storage_config_{*storage_config},
+          storage_{*storage},
+          apis_{*apis},
+          serve_{serve} {}
 
     // Given a target-level caching key, returns the computed value. In doing
     // so, it can build on the associated endpoint passing the
@@ -122,6 +130,8 @@ class TargetService final : public justbuild::just_serve::Target::Service {
 
   private:
     RemoteServeConfig const& serve_config_;
+    StorageConfig const& storage_config_;
+    Storage const& storage_;
     ApiBundle const& apis_;
     ServeApi const* const serve_ = nullptr;
     std::shared_ptr<Logger> logger_{std::make_shared<Logger>("target-service")};