1 files changed, 7 insertions, 2 deletions
diff --git a/searchcore/src/apps/vespa-feed-bm/storage_api_chain_bm_feed_handler.h b/searchcore/src/apps/vespa-feed-bm/storage_api_chain_bm_feed_handler.h
index 79c04c98de9..521deddd19e 100644
--- a/searchcore/src/apps/vespa-feed-bm/storage_api_chain_bm_feed_handler.h
+++ b/searchcore/src/apps/vespa-feed-bm/storage_api_chain_bm_feed_handler.h
@@ -15,15 +15,20 @@ namespace feedbm {
  */
 class StorageApiChainBmFeedHandler : public IBmFeedHandler
 {
+public:
+    struct Context;
+private:
+    std::shared_ptr<Context> _context;
     void send_msg(std::shared_ptr<storage::api::StorageCommand> cmd, PendingTracker& tracker);
 public:
-    StorageApiChainBmFeedHandler();
+    StorageApiChainBmFeedHandler(std::shared_ptr<Context> context);
     ~StorageApiChainBmFeedHandler();
     void put(const document::Bucket& bucket, std::unique_ptr<document::Document> document, uint64_t timestamp, PendingTracker& tracker) override;
     void update(const document::Bucket& bucket, std::unique_ptr<document::DocumentUpdate> document_update, uint64_t timestamp, PendingTracker& tracker) override;
     void remove(const document::Bucket& bucket, const document::DocumentId& document_id,  uint64_t timestamp, PendingTracker& tracker) override;
 
-    static std::unique_ptr<storage::IStorageChainBuilder> get_storage_chain_builder();
+    static std::shared_ptr<Context> get_context();
+    static std::unique_ptr<storage::IStorageChainBuilder> get_storage_chain_builder(std::shared_ptr<Context> context);
 };
 
 }