Improved code

ankane · ankane · commit 7e12adc6bcb2 · 2025-02-15T15:18:37.000-08:00
diff --git a/src/doctrine/DistanceNode.php b/src/doctrine/DistanceNode.php
@@ -2,15 +2,16 @@
 
 namespace Pgvector\Doctrine;
 
+use Doctrine\ORM\Query\AST;
 use Doctrine\ORM\Query\AST\Functions\FunctionNode;
 use Doctrine\ORM\Query\Parser;
 use Doctrine\ORM\Query\SqlWalker;
 use Doctrine\ORM\Query\TokenType;
 
 abstract class DistanceNode extends FunctionNode
 {
-    public $left;
-    public $right;
+    public AST\Node|string $left;
+    public AST\Node|string $right;
 
     abstract protected function getOp(): string;
 
@@ -28,9 +29,9 @@ public function getSql(SqlWalker $sqlWalker): string
     {
         return sprintf(
             '(%s %s %s)',
-            $this->left->dispatch($sqlWalker),
+            $sqlWalker->walkArithmeticPrimary($this->left),
             $this->getOp(),
-            $this->right->dispatch($sqlWalker),
+            $sqlWalker->walkArithmeticPrimary($this->right)
         );
     }
 }
