Handle named indices

Author: codeliner

composer.json

@@ -18,7 +18,7 @@
   "require": {
     "php": "^7.1",
     "ext-pdo": "*",
-    "event-engine/php-persistence": "^0.1"
+    "event-engine/php-persistence": "^0.2"
   },
   "require-dev": {
     "roave/security-advisories": "dev-master",

src/PostgresDocumentStore.php

@@ -166,6 +166,48 @@ public function dropCollection(string $collectionName): void
         });
     }
 
+    public function hasCollectionIndex(string $collectionName, string $indexName): bool
+    {
+        $query = <<<EOT
+SELECT INDEXNAME 
+FROM pg_indexes
+WHERE TABLENAME = '{$this->tableName($collectionName)}'
+AND INDEXNAME = '$indexName'
+EOT;
+
+        $stmt = $this->connection->prepare($query);
+
+        $stmt->execute();
+
+        $row = $stmt->fetchColumn();
+
+        return !!$row;
+    }
+
+    /**
+     * @param string $collectionName
+     * @param Index $index
+     * @throws \EventEngine\DocumentStore\Exception\RuntimeException if adding did not succeed
+     */
+    public function addCollectionIndex(string $collectionName, Index $index): void
+    {
+        $cmd = $this->indexToSqlCmd($index, $collectionName);
+
+        $this->connection->prepare($cmd)->execute();
+    }
+
+    /**
+     * @param string $collectionName
+     * @param string $indexName
+     * @throws \EventEngine\DocumentStore\Exception\RuntimeException if dropping did not succeed
+     */
+    public function dropCollectionIndex(string $collectionName, string $indexName): void
+    {
+        $cmd = "DROP INDEX $indexName";
+
+        $this->connection->prepare($cmd)->execute();
+    }
+
     /**
      * @param string $collectionName
      * @param string $docId
@@ -527,8 +569,10 @@ private function indexToSqlCmd(Index $index, string $collectionName): string
             throw new RuntimeException('Unsupported index type. Got ' . get_class($index));
         }
 
+        $name = $index->name() ?? '';
+
         $cmd = <<<EOT
-CREATE $type ON {$this->tableName($collectionName)}
+CREATE $type $name ON {$this->tableName($collectionName)}
 $fields;
 EOT;
 

tests/PostgresDocumentStoreTest.php

@@ -139,6 +139,32 @@ public function it_adds_collection_with_multi_field_index_unique(): void
         $this->assertStringStartsWith('CREATE UNIQUE INDEX', $indexes[1]['indexdef']);
     }
 
+    /**
+     * @test
+     */
+    public function it_handles_named_indices(): void
+    {
+        $collectionName = 'test_named_indices';
+
+        $this->documentStore->addCollection(
+            $collectionName,
+            FieldIndex::namedIndexForField('testidx_field_a', 'a'),
+            MultiFieldIndex::namedIndexForFields('multitestidx_fields_a_b', ['a', 'b'])
+        );
+
+        $this->assertTrue($this->documentStore->hasCollectionIndex($collectionName, 'testidx_field_a'));
+
+        $this->assertTrue($this->documentStore->hasCollectionIndex($collectionName, 'multitestidx_fields_a_b'));
+
+        $this->documentStore->dropCollectionIndex($collectionName, 'testidx_field_a');
+
+        $this->assertFalse($this->documentStore->hasCollectionIndex($collectionName, 'testidx_field_a'));
+
+        $this->documentStore->addCollectionIndex($collectionName, FieldIndex::namedIndexForField('testidx_field_b', 'b'));
+
+        $this->assertTrue($this->documentStore->hasCollectionIndex($collectionName, 'testidx_field_b'));
+    }
+
     /**
      * @test
      */