Implement hashing of BucketSpace and Bucket

2 files changed, 16 insertions, 0 deletions

diff --git a/document/src/vespa/document/bucket/bucket.h b/document/src/vespa/document/bucket/bucket.h
index af60027d7e3..cc7aceb4ea7 100644
--- a/document/src/vespa/document/bucket/bucket.h
+++ b/document/src/vespa/document/bucket/bucket.h
@@ -32,6 +32,15 @@ public:
     BucketId getBucketId() const noexcept { return _bucketId; }
     void print(std::ostream& os) const;
     vespalib::string toString() const;
+
+    struct hash {
+        size_t operator () (const Bucket& b) const {
+            size_t hash1 = BucketId::hash()(b.getBucketId());
+            size_t hash2 = BucketSpace::hash()(b.getBucketSpace());
+            // Formula taken from std::hash_combine proposal
+            return hash1 ^ (hash2 + 0x9e3779b9 + (hash1<<6) + (hash1>>2));
+        }
+    };
 private:
     BucketSpace _bucketSpace;
     BucketId _bucketId;
diff --git a/document/src/vespa/document/bucket/bucketspace.h b/document/src/vespa/document/bucket/bucketspace.h
index 22dc6dc98a8..a78004b271b 100644
--- a/document/src/vespa/document/bucket/bucketspace.h
+++ b/document/src/vespa/document/bucket/bucketspace.h
@@ -5,6 +5,7 @@
 #include <vespa/vespalib/stllike/string.h>
 #include <vespa/vespalib/stllike/asciistream.h>
 #include <cstdint>
+#include <functional>
 #include <iostream>
 
 namespace document {
@@ -24,6 +25,12 @@ public:
     Type getId() const noexcept { return _id; }
     void print(std::ostream& out) const;
     vespalib::string toString() const;
+
+    struct hash {
+        size_t operator () (const BucketSpace& bs) const {
+            return std::hash<Type>()(bs.getId());
+        }
+    };
 private:
     Type _id;
 };