polish

alibaba · Sep 14, 2024 · 3d010df · 3d010df
1 parent 028cc81
commit 3d010df
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/core/monitor/LogtailMetric.h b/core/monitor/LogtailMetric.h
@@ -58,6 +58,8 @@ class MetricsRecord {
 
 class MetricsRecordRef {
     friend class WriteMetrics;
+    friend bool operator==(const MetricsRecordRef& lhs, std::nullptr_t rhs);
+    friend bool operator==(std::nullptr_t rhs, const MetricsRecordRef& lhs);
 
 private:
     MetricsRecord* mMetrics = nullptr;
@@ -83,6 +85,22 @@ class MetricsRecordRef {
 #endif
 };
 
+inline bool operator==(const MetricsRecordRef& lhs, std::nullptr_t rhs) {
+    return lhs.mMetrics == rhs;
+}
+
+inline bool operator==(std::nullptr_t lhs, const MetricsRecordRef& rhs) {
+    return lhs == rhs.mMetrics;
+}
+
+inline bool operator!=(const MetricsRecordRef& lhs, std::nullptr_t rhs) {
+    return !(lhs == rhs);
+}
+
+inline bool operator!=(std::nullptr_t lhs, const MetricsRecordRef& rhs) {
+    return !(lhs == rhs);
+}
+
 class ReentrantMetricsRecord {
 private:
     MetricsRecordRef mMetricsRecordRef;