Add operator== and external operator<< to storage::spi::ResourceUsage.

3 files changed, 25 insertions, 0 deletions

diff --git a/persistence/src/vespa/persistence/spi/CMakeLists.txt b/persistence/src/vespa/persistence/spi/CMakeLists.txt
index aad99e12a69..ef871c2584f 100644
--- a/persistence/src/vespa/persistence/spi/CMakeLists.txt
+++ b/persistence/src/vespa/persistence/spi/CMakeLists.txt
@@ -10,6 +10,7 @@ vespa_add_library(persistence_spi OBJECT
     exceptions.cpp
     persistenceprovider.cpp
     read_consistency.cpp
+    resource_usage.cpp
     resource_usage_listener.cpp
     result.cpp
     selection.cpp
diff --git a/persistence/src/vespa/persistence/spi/resource_usage.cpp b/persistence/src/vespa/persistence/spi/resource_usage.cpp
new file mode 100644
index 00000000000..77ef19ea006
--- /dev/null
+++ b/persistence/src/vespa/persistence/spi/resource_usage.cpp
@@ -0,0 +1,15 @@
+// Copyright Verizon Media. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+
+#include "resource_usage.h"
+#include <iostream>
+
+namespace storage::spi {
+
+std::ostream& operator<<(std::ostream& out, const ResourceUsage& resource_usage)
+{
+    out << "{disk_usage=" << resource_usage.get_disk_usage() <<
+        ", memory_usage=" << resource_usage.get_memory_usage() << "}";
+    return out;
+}
+
+}
diff --git a/persistence/src/vespa/persistence/spi/resource_usage.h b/persistence/src/vespa/persistence/spi/resource_usage.h
index 1b2504b0f13..defdaab2fba 100644
--- a/persistence/src/vespa/persistence/spi/resource_usage.h
+++ b/persistence/src/vespa/persistence/spi/resource_usage.h
@@ -2,6 +2,8 @@
 
 #pragma once
 
+#include <iosfwd>
+
 namespace storage::spi {
 
 /*
@@ -27,7 +29,14 @@ public:
 
     double get_disk_usage() const noexcept { return _disk_usage; }
     double get_memory_usage() const noexcept { return _memory_usage; }
+
+    bool operator==(const ResourceUsage &rhs) const noexcept {
+        return (_disk_usage == rhs._disk_usage) &&
+            (_memory_usage == rhs._memory_usage);
+    }
 };
 
+std::ostream& operator<<(std::ostream& out, const ResourceUsage& resource_usage);
+
 }