SERVER-68487 add BSONElement::isNaN()

1 files changed, 20 insertions, 0 deletions

diff --git a/src/mongo/bson/bsonelement.h b/src/mongo/bson/bsonelement.h
index e0b3b8e4414..1d31c647901 100644
--- a/src/mongo/bson/bsonelement.h
+++ b/src/mongo/bson/bsonelement.h
@@ -373,6 +373,11 @@ public:
     bool isNumber() const;
 
     /**
+     * True if element is a NaN double or decimal.
+     */
+    bool isNaN() const;
+
+    /**
      * Return double value for this field. MUST be NumberDouble type.
      */
     double _numberDouble() const {
@@ -1005,6 +1010,21 @@ inline bool BSONElement::isNumber() const {
     }
 }
 
+inline bool BSONElement::isNaN() const {
+    switch (type()) {
+        case NumberDouble: {
+            double d = _numberDouble();
+            return std::isnan(d);
+        }
+        case NumberDecimal: {
+            Decimal128 d = _numberDecimal();
+            return d.isNaN();
+        }
+        default:
+            return false;
+    }
+}
+
 inline Decimal128 BSONElement::numberDecimal() const {
     switch (type()) {
         case NumberDouble: