CXX-685: Support Filtered (Partial) Indexes

Author: markbenvenuto

src/mongo/client/index_spec.cpp

@@ -124,6 +124,12 @@ IndexSpec& IndexSpec::version(int value) {
     return *this;
 }
 
+IndexSpec& IndexSpec::partialFilterExpression(const BSONObj& value) {
+    uassert(0, kDuplicateOption, !_options.asTempObj().hasField("partialFilterExpression"));
+    _options.append("partialFilterExpression", value);
+    return *this;
+}
+
 IndexSpec& IndexSpec::textWeights(const BSONObj& value) {
     uassert(0, kDuplicateOption, !_options.asTempObj().hasField("weights"));
     _options.append("weights", value);

src/mongo/client/index_spec.h

@@ -125,6 +125,8 @@ class MONGO_CLIENT_API IndexSpec {
      */
     IndexSpec& version(int value);
 
+    /** Sets the 'partialFilterExpression' document for a partial index. */
+    IndexSpec& partialFilterExpression(const BSONObj& value);
 
     //
     // Text options

src/mongo/client/index_spec_test.cpp

@@ -30,6 +30,8 @@ TEST(Options, RepeatedOptionsFail) {
     ASSERT_UASSERTS(IndexSpec().sparse().sparse());
     ASSERT_UASSERTS(IndexSpec().expireAfterSeconds(1).expireAfterSeconds(1));
     ASSERT_UASSERTS(IndexSpec().version(0).version(0));
+    ASSERT_UASSERTS(
+        IndexSpec().partialFilterExpression(BSONObj()).partialFilterExpression(BSONObj()));
     ASSERT_UASSERTS(IndexSpec().textWeights(BSONObj()).textWeights(BSONObj()));
     ASSERT_UASSERTS(IndexSpec().textDefaultLanguage("foo").textDefaultLanguage("foo"));
     ASSERT_UASSERTS(IndexSpec().textLanguageOverride("foo").textLanguageOverride("foo"));

src/mongo/integration/standalone/dbclient_test.cpp

@@ -1409,6 +1409,19 @@ TEST_F(DBClientTest, CreateHashedIndex) {
     c->createIndex(TEST_NS, IndexSpec().addKey("aField", IndexSpec::kIndexTypeHashed));
 }
 
+TEST_F(DBClientTest, CreatePartialIndex) {
+    if (serverGTE(c.get(), 3, 1)) {
+        c->createIndex(
+            TEST_NS,
+            IndexSpec().addKey("aField").partialFilterExpression(BSON("aField" << GT << 5)));
+
+        std::list<BSONObj> indexes = c->getIndexSpecs(TEST_NS);
+        ASSERT_EQUALS(2U, indexes.size());
+        indexes.pop_front();
+        ASSERT_EQUALS(BSON("aField" << GT << 5), indexes.front()["partialFilterExpression"].Obj());
+    }
+}
+
 TEST_F(DBClientTest, CreateUser) {
     createUser(c.get(), TEST_DB, "user1", "password1");
 }