PHPLIB-1394: Support "type" option for createSearchIndex(es) (#1375)

jmikola · web-flow · commit 99f3b77233d2 · 2024-09-06T14:41:21.000-04:00
Synced with mongodb/specifications@8be1189 * Update psalm baseline
diff --git a/psalm-baseline.xml b/psalm-baseline.xml
@@ -57,11 +57,6 @@
       <code><![CDATA[$mergedDriver['platform']]]></code>
     </MixedAssignment>
   </file>
-  <file src="src/Collection.php">
-    <MixedArgumentTypeCoercion>
-      <code><![CDATA[$options]]></code>
-    </MixedArgumentTypeCoercion>
-  </file>
   <file src="src/Command/ListCollections.php">
     <MixedAssignment>
       <code><![CDATA[$cmd[$option]]]></code>
diff --git a/src/Collection.php b/src/Collection.php
@@ -371,22 +371,22 @@ public function createIndexes(array $indexes, array $options = [])
      *
      * @see https://www.mongodb.com/docs/manual/reference/command/createSearchIndexes/
      * @see https://mongodb.com/docs/manual/reference/method/db.collection.createSearchIndex/
-     * @param array|object                          $definition Atlas Search index mapping definition
-     * @param array{name?: string, comment?: mixed} $options    Command options
+     * @param array|object                                         $definition Atlas Search index mapping definition
+     * @param array{comment?: mixed, name?: string, type?: string} $options    Index and command options
      * @return string The name of the created search index
      * @throws UnsupportedException if options are not supported by the selected server
      * @throws InvalidArgumentException for parameter/option parsing errors
      * @throws DriverRuntimeException for other driver errors (e.g. connection errors)
      */
     public function createSearchIndex($definition, array $options = []): string
     {
-        $index = ['definition' => $definition];
-        if (isset($options['name'])) {
-            $index['name'] = $options['name'];
-            unset($options['name']);
-        }
+        $indexOptionKeys = ['name' => 1, 'type' => 1];
+        /** @psalm-var array{name?: string, type?: string} */
+        $indexOptions = array_intersect_key($options, $indexOptionKeys);
+        /** @psalm-var array{comment?: mixed} */
+        $operationOptions = array_diff_key($options, $indexOptionKeys);
 
-        $names = $this->createSearchIndexes([$index], $options);
+        $names = $this->createSearchIndexes([['definition' => $definition] + $indexOptions], $operationOptions);
 
         return current($names);
     }
@@ -400,16 +400,16 @@ public function createSearchIndex($definition, array $options = []): string
      * For example:
      *
      *     $indexes = [
-     *         // Create a search index with the default name, on
+     *         // Create a search index with the default name on a single field
      *         ['definition' => ['mappings' => ['dynamic' => false, 'fields' => ['title' => ['type' => 'string']]]]],
      *         // Create a named search index on all fields
      *         ['name' => 'search_all', 'definition' => ['mappings' => ['dynamic' => true]]],
      *     ];
      *
      * @see https://www.mongodb.com/docs/manual/reference/command/createSearchIndexes/
      * @see https://mongodb.com/docs/manual/reference/method/db.collection.createSearchIndex/
-     * @param list<array{name?: string, definition: array|object}> $indexes List of search index specifications
-     * @param array{comment?: string}                              $options Command options
+     * @param list<array{definition: array|object, name?: string, type?: string}> $indexes List of search index specifications
+     * @param array{comment?: mixed}                                              $options Command options
      * @return string[] The names of the created search indexes
      * @throws UnsupportedException if options are not supported by the selected server
      * @throws InvalidArgumentException for parameter/option parsing errors
diff --git a/src/Model/SearchIndexInput.php b/src/Model/SearchIndexInput.php
@@ -39,7 +39,7 @@ class SearchIndexInput implements Serializable
     private array $index;
 
     /**
-     * @param array{name?: string, definition: array|object} $index Search index specification
+     * @param array{definition: array|object, name?: string, type?: string} $index Search index specification
      * @throws InvalidArgumentException
      */
     public function __construct(array $index)
@@ -57,6 +57,10 @@ public function __construct(array $index)
             throw InvalidArgumentException::invalidType('"name" option', $index['name'], 'string');
         }
 
+        if (isset($index['type']) && ! is_string($index['type'])) {
+            throw InvalidArgumentException::invalidType('"type" option', $index['type'], 'string');
+        }
+
         $this->index = $index;
     }
 
diff --git a/tests/Model/SearchIndexInputTest.php b/tests/Model/SearchIndexInputTest.php
@@ -32,6 +32,14 @@ public function testConstructorShouldRequireNameToBeString($name): void
         new SearchIndexInput(['definition' => ['mapping' => ['dynamic' => true]], 'name' => $name]);
     }
 
+    /** @dataProvider provideInvalidStringValues */
+    public function testConstructorShouldRequireTypeToBeString($type): void
+    {
+        $this->expectException(InvalidArgumentException::class);
+        $this->expectExceptionMessage('Expected "type" option to have type "string"');
+        new SearchIndexInput(['definition' => ['mapping' => ['dynamic' => true]], 'type' => $type]);
+    }
+
     public function testBsonSerialization(): void
     {
         $expected = (object) [
diff --git a/tests/Operation/CreateSearchIndexesTest.php b/tests/Operation/CreateSearchIndexesTest.php
@@ -30,6 +30,14 @@ public function testConstructorIndexNameMustBeAString($name): void
         new CreateSearchIndexes($this->getDatabaseName(), $this->getCollectionName(), [['name' => $name, 'definition' => ['mappings' => ['dynamic' => true]]]], []);
     }
 
+    /** @dataProvider provideInvalidStringValues */
+    public function testConstructorIndexTypeMustBeAString($type): void
+    {
+        $this->expectException(InvalidArgumentException::class);
+        $this->expectExceptionMessage('Expected "type" option to have type "string"');
+        new CreateSearchIndexes($this->getDatabaseName(), $this->getCollectionName(), [['type' => $type, 'definition' => ['mappings' => ['dynamic' => true]]]], []);
+    }
+
     public function testConstructorIndexDefinitionMustBeDefined(): void
     {
         $this->expectException(InvalidArgumentException::class);
diff --git a/tests/UnifiedSpecTests/Operation.php b/tests/UnifiedSpecTests/Operation.php
@@ -537,19 +537,25 @@ private function executeForCollection(Collection $collection)
                 );
 
             case 'createSearchIndex':
-                $options = [];
-                if (isset($args['model']->name)) {
-                    assertIsString($args['model']->name);
-                    $options['name'] = $args['model']->name;
-                }
-
+                assertArrayHasKey('model', $args);
+                assertIsObject($args['model']);
+                assertObjectHasAttribute('definition', $args['model']);
                 assertInstanceOf(stdClass::class, $args['model']->definition);
 
-                return $collection->createSearchIndex($args['model']->definition, $options);
+                /* Note: tests specify options within "model". A top-level
+                 * "options" key (CreateSearchIndexOptions) is not used. */
+                $definition = $args['model']->definition;
+                $options = array_diff_key((array) $args['model'], ['definition' => 1]);
+
+                return $collection->createSearchIndex($definition, $options);
 
             case 'createSearchIndexes':
+                assertArrayHasKey('models', $args);
+                assertIsArray($args['models']);
+
                 $indexes = array_map(function ($index) {
                     $index = (array) $index;
+                    assertArrayHasKey('definition', $index);
                     assertInstanceOf(stdClass::class, $index['definition']);
 
                     return $index;
diff --git a/tests/UnifiedSpecTests/index-management/createSearchIndex.json b/tests/UnifiedSpecTests/index-management/createSearchIndex.json
@@ -28,7 +28,17 @@
   ],
   "runOnRequirements": [
     {
-      "minServerVersion": "7.0.0",
+      "minServerVersion": "7.0.5",
+      "maxServerVersion": "7.0.99",
+      "topologies": [
+        "replicaset",
+        "load-balanced",
+        "sharded"
+      ],
+      "serverless": "forbid"
+    },
+    {
+      "minServerVersion": "7.2.0",
       "topologies": [
         "replicaset",
         "load-balanced",
@@ -50,7 +60,8 @@
                 "mappings": {
                   "dynamic": true
                 }
-              }
+              },
+              "type": "search"
             }
           },
           "expectError": {
@@ -73,7 +84,8 @@
                         "mappings": {
                           "dynamic": true
                         }
-                      }
+                      },
+                      "type": "search"
                     }
                   ],
                   "$db": "database0"
@@ -97,7 +109,8 @@
                   "dynamic": true
                 }
               },
-              "name": "test index"
+              "name": "test index",
+              "type": "search"
             }
           },
           "expectError": {
@@ -121,7 +134,68 @@
                           "dynamic": true
                         }
                       },
-                      "name": "test index"
+                      "name": "test index",
+                      "type": "search"
+                    }
+                  ],
+                  "$db": "database0"
+                }
+              }
+            }
+          ]
+        }
+      ]
+    },
+    {
+      "description": "create a vector search index",
+      "operations": [
+        {
+          "name": "createSearchIndex",
+          "object": "collection0",
+          "arguments": {
+            "model": {
+              "definition": {
+                "fields": [
+                  {
+                    "type": "vector",
+                    "path": "plot_embedding",
+                    "numDimensions": 1536,
+                    "similarity": "euclidean"
+                  }
+                ]
+              },
+              "name": "test index",
+              "type": "vectorSearch"
+            }
+          },
+          "expectError": {
+            "isError": true,
+            "errorContains": "Atlas"
+          }
+        }
+      ],
+      "expectEvents": [
+        {
+          "client": "client0",
+          "events": [
+            {
+              "commandStartedEvent": {
+                "command": {
+                  "createSearchIndexes": "collection0",
+                  "indexes": [
+                    {
+                      "definition": {
+                        "fields": [
+                          {
+                            "type": "vector",
+                            "path": "plot_embedding",
+                            "numDimensions": 1536,
+                            "similarity": "euclidean"
+                          }
+                        ]
+                      },
+                      "name": "test index",
+                      "type": "vectorSearch"
                     }
                   ],
                   "$db": "database0"
diff --git a/tests/UnifiedSpecTests/index-management/createSearchIndexes.json b/tests/UnifiedSpecTests/index-management/createSearchIndexes.json
@@ -28,7 +28,17 @@
   ],
   "runOnRequirements": [
     {
-      "minServerVersion": "7.0.0",
+      "minServerVersion": "7.0.5",
+      "maxServerVersion": "7.0.99",
+      "topologies": [
+        "replicaset",
+        "load-balanced",
+        "sharded"
+      ],
+      "serverless": "forbid"
+    },
+    {
+      "minServerVersion": "7.2.0",
       "topologies": [
         "replicaset",
         "load-balanced",
@@ -83,7 +93,8 @@
                   "mappings": {
                     "dynamic": true
                   }
-                }
+                },
+                "type": "search"
               }
             ]
           },
@@ -107,7 +118,8 @@
                         "mappings": {
                           "dynamic": true
                         }
-                      }
+                      },
+                      "type": "search"
                     }
                   ],
                   "$db": "database0"
@@ -132,7 +144,8 @@
                     "dynamic": true
                   }
                 },
-                "name": "test index"
+                "name": "test index",
+                "type": "search"
               }
             ]
           },
@@ -157,7 +170,70 @@
                           "dynamic": true
                         }
                       },
-                      "name": "test index"
+                      "name": "test index",
+                      "type": "search"
+                    }
+                  ],
+                  "$db": "database0"
+                }
+              }
+            }
+          ]
+        }
+      ]
+    },
+    {
+      "description": "create a vector search index",
+      "operations": [
+        {
+          "name": "createSearchIndexes",
+          "object": "collection0",
+          "arguments": {
+            "models": [
+              {
+                "definition": {
+                  "fields": [
+                    {
+                      "type": "vector",
+                      "path": "plot_embedding",
+                      "numDimensions": 1536,
+                      "similarity": "euclidean"
+                    }
+                  ]
+                },
+                "name": "test index",
+                "type": "vectorSearch"
+              }
+            ]
+          },
+          "expectError": {
+            "isError": true,
+            "errorContains": "Atlas"
+          }
+        }
+      ],
+      "expectEvents": [
+        {
+          "client": "client0",
+          "events": [
+            {
+              "commandStartedEvent": {
+                "command": {
+                  "createSearchIndexes": "collection0",
+                  "indexes": [
+                    {
+                      "definition": {
+                        "fields": [
+                          {
+                            "type": "vector",
+                            "path": "plot_embedding",
+                            "numDimensions": 1536,
+                            "similarity": "euclidean"
+                          }
+                        ]
+                      },
+                      "name": "test index",
+                      "type": "vectorSearch"
                     }
                   ],
                   "$db": "database0"
diff --git a/tests/UnifiedSpecTests/index-management/searchIndexIgnoresReadWriteConcern.json b/tests/UnifiedSpecTests/index-management/searchIndexIgnoresReadWriteConcern.json

Original file line number	Diff line number	Diff line change
`@@ -32,6 +32,14 @@ public function testConstructorShouldRequireNameToBeString($name): void`
`32`	`32`	`new SearchIndexInput(['definition' => ['mapping' => ['dynamic' => true]], 'name' => $name]);`
`33`	`33`	`}`
`34`	`34`
	`35`	`+ /** @dataProvider provideInvalidStringValues */`
	`36`	`+ public function testConstructorShouldRequireTypeToBeString($type): void`
	`37`	`+ {`
	`38`	`+ $this->expectException(InvalidArgumentException::class);`
	`39`	`+ $this->expectExceptionMessage('Expected "type" option to have type "string"');`
	`40`	`+ new SearchIndexInput(['definition' => ['mapping' => ['dynamic' => true]], 'type' => $type]);`
	`41`	`+ }`
	`42`	`+`
`35`	`43`	`public function testBsonSerialization(): void`
`36`	`44`	`{`
`37`	`45`	`$expected = (object) [`
Original file line number	Diff line number	Diff line change
`@@ -30,6 +30,14 @@ public function testConstructorIndexNameMustBeAString($name): void`
`30`	`30`	`new CreateSearchIndexes($this->getDatabaseName(), $this->getCollectionName(), [['name' => $name, 'definition' => ['mappings' => ['dynamic' => true]]]], []);`
`31`	`31`	`}`
`32`	`32`
	`33`	`+ /** @dataProvider provideInvalidStringValues */`
	`34`	`+ public function testConstructorIndexTypeMustBeAString($type): void`
	`35`	`+ {`
	`36`	`+ $this->expectException(InvalidArgumentException::class);`
	`37`	`+ $this->expectExceptionMessage('Expected "type" option to have type "string"');`
	`38`	`+ new CreateSearchIndexes($this->getDatabaseName(), $this->getCollectionName(), [['type' => $type, 'definition' => ['mappings' => ['dynamic' => true]]]], []);`
	`39`	`+ }`
	`40`	`+`
`33`	`41`	`public function testConstructorIndexDefinitionMustBeDefined(): void`
`34`	`42`	`{`
`35`	`43`	`$this->expectException(InvalidArgumentException::class);`