Pass on bucket space from create visitor command to visitor.

2 files changed, 2 insertions, 0 deletions

diff --git a/storage/src/vespa/storage/visiting/visitor.h b/storage/src/vespa/storage/visiting/visitor.h
index 1c156ff187a..4436312032f 100644
--- a/storage/src/vespa/storage/visiting/visitor.h
+++ b/storage/src/vespa/storage/visiting/visitor.h
@@ -387,6 +387,7 @@ public:
     void setMemoryManager(framework::MemoryManagerInterface& mm)
         { _memoryManager = &mm; }
     void setOwnNodeIndex(uint16_t nodeIndex) { _ownNodeIndex = nodeIndex; }
+    void setBucketSpace(document::BucketSpace bucketSpace) { _bucketSpace = bucketSpace; }
 
     const documentapi::LoadType& getLoadType() const {
         return _initiatingCmd->getLoadType();
diff --git a/storage/src/vespa/storage/visiting/visitorthread.cpp b/storage/src/vespa/storage/visiting/visitorthread.cpp
index 8056d7e0ad2..d3fed86b741 100644
--- a/storage/src/vespa/storage/visiting/visitorthread.cpp
+++ b/storage/src/vespa/storage/visiting/visitorthread.cpp
@@ -497,6 +497,7 @@ VisitorThread::onCreateVisitor(
         visitor->setDocBlockTimeout(_defaultDocBlockTimeout);
         visitor->setVisitorInfoTimeout(_defaultVisitorInfoTimeout);
         visitor->setOwnNodeIndex(_component.getIndex());
+        visitor->setBucketSpace(cmd->getBucketSpace());
 
         // Parse document selection
         try{