Added bit type for Doctrine [skip ci]

Author: ankane

src/doctrine/BitType.php

@@ -0,0 +1,38 @@
+<?php
+
+namespace Pgvector\Doctrine;
+
+use Doctrine\DBAL\Types\Type;
+use Doctrine\DBAL\Platforms\AbstractPlatform;
+
+class BitType extends Type
+{
+    public function getName(): string
+    {
+        return 'bit';
+    }
+
+    public function getSQLDeclaration(array $fieldDeclaration, AbstractPlatform $platform): string
+    {
+        $length = $fieldDeclaration['length'];
+        return is_null($length) ? 'bit' : sprintf('bit(%d)', $length);
+    }
+
+    public function convertToPHPValue(mixed $value, AbstractPlatform $platform): ?string
+    {
+        if (is_null($value)) {
+            return null;
+        }
+
+        return (string) $value;
+    }
+
+    public function convertToDatabaseValue(mixed $value, AbstractPlatform $platform): ?string
+    {
+        if (is_null($value)) {
+            return null;
+        }
+
+        return (string) $value;
+    }
+}

tests/DoctrineTest.php

@@ -32,6 +32,7 @@ public function testTypes()
 
         Type::addType('vector', 'Pgvector\Doctrine\VectorType');
         Type::addType('halfvec', 'Pgvector\Doctrine\HalfVectorType');
+        Type::addType('bit', 'Pgvector\Doctrine\BitType');
         Type::addType('sparsevec', 'Pgvector\Doctrine\SparseVectorType');
 
         $platform = $entityManager->getConnection()->getDatabasePlatform();
@@ -52,6 +53,7 @@ public function testTypes()
         $item = new DoctrineItem();
         $item->setEmbedding(new Vector([1, 2, 3]));
         $item->setHalfEmbedding(new HalfVector([4, 5, 6]));
+        $item->setBinaryEmbedding('101');
         $item->setSparseEmbedding(new SparseVector([7, 8, 9]));
         $entityManager->persist($item);
         $entityManager->flush();
@@ -60,6 +62,7 @@ public function testTypes()
         $item = $itemRepository->find(1);
         $this->assertEquals([1, 2, 3], $item->getEmbedding()->toArray());
         $this->assertEquals([4, 5, 6], $item->getHalfEmbedding()->toArray());
+        $this->assertEquals('101', $item->getBinaryEmbedding());
         $this->assertEquals([7, 8, 9], $item->getSparseEmbedding()->toArray());
     }
 }

tests/models/DoctrineItem.php

@@ -20,6 +20,9 @@ class DoctrineItem
     #[ORM\Column(type: 'halfvec', length: 3)]
     private HalfVector $halfEmbedding;
 
+    #[ORM\Column(type: 'bit', length: 3)]
+    private string $binaryEmbedding;
+
     #[ORM\Column(type: 'sparsevec', length: 3)]
     private SparseVector $sparseEmbedding;
 
@@ -43,6 +46,16 @@ public function setHalfEmbedding(HalfVector $embedding): void
         $this->halfEmbedding = $embedding;
     }
 
+    public function getBinaryEmbedding(): string
+    {
+        return $this->binaryEmbedding;
+    }
+
+    public function setBinaryEmbedding(string $embedding): void
+    {
+        $this->binaryEmbedding = $embedding;
+    }
+
     public function getSparseEmbedding(): SparseVector
     {
         return $this->sparseEmbedding;