From d3bdb50bd5f4c2b76c7bdd6e7dd4712013cb710a Mon Sep 17 00:00:00 2001
From: Geir Storli <geirst@verizonmedia.com>
Date: Thu, 3 Jun 2021 10:31:18 +0000
Subject: Dispatch get and visitor messages to correct distributor stripe.

---
 storage/src/vespa/storage/distributor/distributor.cpp | 7 +++++++
 1 file changed, 7 insertions(+)

(limited to 'storage/src')

diff --git a/storage/src/vespa/storage/distributor/distributor.cpp b/storage/src/vespa/storage/distributor/distributor.cpp
index 368a25315fb..65945b2c6ae 100644
--- a/storage/src/vespa/storage/distributor/distributor.cpp
+++ b/storage/src/vespa/storage/distributor/distributor.cpp
@@ -24,6 +24,7 @@
 #include <vespa/storage/config/distributorconfiguration.h>
 #include <vespa/storage/distributor/maintenance/simplebucketprioritydatabase.h>
 #include <vespa/storageapi/message/persistence.h>
+#include <vespa/storageapi/message/visitor.h>
 #include <vespa/storageframework/generic/status/xmlstatusreporter.h>
 #include <vespa/vdslib/distribution/distribution.h>
 #include <vespa/vespalib/util/memoryusage.h>
@@ -350,6 +351,12 @@ get_bucket_id_for_striping(const api::StorageMessage& msg, const DistributorNode
                     return reply.getBucketId();
                 }
             }
+            case api::MessageType::GET_ID:
+                return node_ctx.bucket_id_factory().getBucketId(dynamic_cast<const api::GetCommand&>(msg).getDocumentId());
+            case api::MessageType::VISITOR_CREATE_ID:
+                return dynamic_cast<const api::CreateVisitorCommand&>(msg).super_bucket_id();
+            case api::MessageType::VISITOR_CREATE_REPLY_ID:
+                return dynamic_cast<const api::CreateVisitorReply&>(msg).super_bucket_id();
             default:
                 return msg.getBucketId();
         }
-- 
cgit v1.2.3