SERVER-49686 Translate $type match expression to SBE

2 files changed, 24 insertions, 0 deletions

diff --git a/src/mongo/db/matcher/matcher_type_set.cpp b/src/mongo/db/matcher/matcher_type_set.cpp
index 17681045243..1c48d32ec25 100644
--- a/src/mongo/db/matcher/matcher_type_set.cpp
+++ b/src/mongo/db/matcher/matcher_type_set.cpp
@@ -33,6 +33,7 @@
 
 #include "mongo/db/matcher/expression_parser.h"
 #include "mongo/db/matcher/schema/json_schema_parser.h"
+#include "mongo/db/query/bson_typemask.h"
 
 namespace mongo {
 namespace {
@@ -173,6 +174,22 @@ void MatcherTypeSet::toBSONArray(BSONArrayBuilder* builder) const {
     }
 }
 
+uint32_t MatcherTypeSet::getBSONTypeMask() const {
+    uint32_t mask = 0;
+    if (allNumbers) {
+        mask |= (mongo::getBSONTypeMask(BSONType::NumberInt) |
+                 mongo::getBSONTypeMask(BSONType::NumberLong) |
+                 mongo::getBSONTypeMask(BSONType::NumberDouble) |
+                 mongo::getBSONTypeMask(BSONType::NumberDecimal));
+    }
+
+    for (auto t : bsonTypes) {
+        mask |= mongo::getBSONTypeMask(t);
+    }
+
+    return mask;
+}
+
 BSONTypeSet BSONTypeSet::parseFromBSON(const BSONElement& element) {
     // BSON type can be specified with a type alias, other values will be rejected.
     auto typeSet = uassertStatusOK(JSONSchemaParser::parseTypeSet(element, findBSONTypeAlias));
diff --git a/src/mongo/db/matcher/matcher_type_set.h b/src/mongo/db/matcher/matcher_type_set.h
index a217b96129b..7903d03f441 100644
--- a/src/mongo/db/matcher/matcher_type_set.h
+++ b/src/mongo/db/matcher/matcher_type_set.h
@@ -104,6 +104,13 @@ struct MatcherTypeSet {
         return !allNumbers && bsonTypes.empty();
     }
 
+    /**
+     * Returns a bitmask representing the set of BSONTypes that this MatcherTypeSet contains.
+     *
+     * For details on how these bitmasks are encoded, see mongo::getBSONTypeMask().
+     */
+    uint32_t getBSONTypeMask() const;
+
     void toBSONArray(BSONArrayBuilder* builder) const;
 
     BSONArray toBSONArray() const {