feat: query builder adapter now supports row values method

CHANGELOG.md

@@ -5,6 +5,7 @@
 * refactor: simplify `KeysetExpressionCalculator`
 * chore: cleanup
 * feat: native query adapter now supports row values method
+* feat: query builder adapter now supports row values method
 
 # 0.15.1
 

packages/rekapager-doctrine-orm-adapter/src/QueryBuilderAdapter.php

@@ -18,6 +18,7 @@
 use Doctrine\DBAL\ParameterType;
 use Doctrine\DBAL\Types\Type;
 use Doctrine\DBAL\Types\Types;
+use Doctrine\ORM\Query\Expr\Andx;
 use Doctrine\ORM\Query\Expr\From;
 use Doctrine\ORM\Query\Expr\OrderBy;
 use Doctrine\ORM\QueryBuilder;
@@ -27,6 +28,7 @@
 use Rekalogika\Rekapager\Adapter\Common\Field;
 use Rekalogika\Rekapager\Adapter\Common\IndexResolver;
 use Rekalogika\Rekapager\Adapter\Common\KeysetExpressionCalculator;
+use Rekalogika\Rekapager\Adapter\Common\SeekMethod;
 use Rekalogika\Rekapager\Doctrine\ORM\Exception\UnsupportedQueryBuilderException;
 use Rekalogika\Rekapager\Doctrine\ORM\Internal\KeysetQueryBuilderVisitor;
 use Rekalogika\Rekapager\Doctrine\ORM\Internal\QueryBuilderKeysetItem;
@@ -61,6 +63,7 @@ public function __construct(
         private readonly array $typeMapping = [],
         private readonly bool|null $useOutputWalkers = null,
         private readonly string|null $indexBy = null,
+        private readonly SeekMethod $seekMethod = SeekMethod::Approximated,
     ) {
         if ($queryBuilder->getFirstResult() !== 0 || $queryBuilder->getMaxResults() !== null) {
             throw new UnsupportedQueryBuilderException();
@@ -139,20 +142,14 @@ private function getQueryBuilder(
 
         // returns early if there are no boundary values
 
-        $fields = $this->createCalculatorFields($boundaryValues, $orderings);
-
-        if ($fields === []) {
-            return $queryBuilder;
-        }
+        [$where, $parameters] = $this->generateWhereExpression(
+            boundaryValues: $boundaryValues,
+            orderings: $orderings
+        );
 
-        // adds where expression to the querybuilder
-
-        $keysetExpression = KeysetExpressionCalculator::calculate($fields);
-
-        $visitor = new KeysetQueryBuilderVisitor();
-        $queryBuilder->andWhere($visitor->dispatch($keysetExpression));
+        $queryBuilder->andWhere($where);
 
-        foreach ($visitor->getParameters() as $template => $parameter) {
+        foreach ($parameters as $template => $parameter) {
             $queryBuilder->setParameter(
                 $template,
                 $parameter->getValue(),
@@ -163,6 +160,46 @@ private function getQueryBuilder(
         return $queryBuilder;
     }
 
+    /**
+     * @param array<string,QueryParameter> $boundaryValues Key is the property name, value is the bound value. Null if unbounded.
+     * @param non-empty-array<string,'ASC'|'DESC'> $orderings
+     * @return array{?Andx,array<string,QueryParameter>}
+     */
+    private function generateWhereExpression(
+        array $boundaryValues,
+        array $orderings,
+    ): array {
+        $fields = $this->createCalculatorFields($boundaryValues, $orderings);
+
+        if ($fields === []) {
+            return [null, []];
+        }
+
+        return match ($this->seekMethod) {
+            SeekMethod::Approximated => $this->generateApproximatedWhereExpression($fields),
+                // SeekMethod::RowValues => $this->generateRowValuesWhereExpression($fields),
+                // SeekMethod::Auto => $this->generateAutoWhereExpression($fields),
+            default => throw new LogicException('Unsupported seek method'),
+        };
+    }
+
+    /**
+     * @param non-empty-list<Field> $fields
+     * @return array{?Andx,array<string,QueryParameter>}
+     */
+    private function generateApproximatedWhereExpression(array $fields): array
+    {
+        $expression = KeysetExpressionCalculator::calculate($fields);
+
+        $visitor = new KeysetQueryBuilderVisitor();
+        $where = $visitor->dispatch($expression);
+        assert($where instanceof Andx);
+
+        $parameters = $visitor->getParameters();
+
+        return [$where, $parameters];
+    }
+
     /**
      * @param array<string,QueryParameter> $boundaryValues
      * @param non-empty-array<string, 'ASC'|'DESC'> $orderings

packages/rekapager-doctrine-orm-adapter/src/RowValuesFunction.php

@@ -0,0 +1,53 @@
+<?php
+
+declare(strict_types=1);
+
+/*
+ * This file is part of rekalogika/rekapager package.
+ *
+ * (c) Priyadi Iman Nurcahyo <https://rekalogika.dev>
+ *
+ * For the full copyright and license information, please view the LICENSE file
+ * that was distributed with this source code.
+ */
+
+namespace Rekalogika\Rekapager\Doctrine\ORM;
+
+use Doctrine\ORM\Query\AST\Functions\FunctionNode;
+use Doctrine\ORM\Query\AST\Node;
+use Doctrine\ORM\Query\Parser;
+use Doctrine\ORM\Query\SqlWalker;
+use Doctrine\ORM\Query\TokenType;
+
+class RowValuesFunction extends FunctionNode
+{
+    /** @var list<Node|string> */
+    public array $values = [];
+
+    public function getSql(SqlWalker $sqlWalker): string
+    {
+        $queryBuilder = [];
+
+        foreach ($this->values as $value) {
+            $queryBuilder[] = $sqlWalker->walkArithmeticPrimary($value);
+        }
+
+        return '(' . implode(', ', $queryBuilder) . ')';
+    }
+
+    public function parse(Parser $parser): void
+    {
+        $parser->match(TokenType::T_IDENTIFIER);
+        $parser->match(TokenType::T_OPEN_PARENTHESIS);
+        $this->values[] = $parser->ArithmeticPrimary();
+        $parser->match(TokenType::T_COMMA);
+        $this->values[] = $parser->ArithmeticPrimary();
+
+        while ($parser->getLexer()->isNextToken(TokenType::T_COMMA)) {
+            $parser->match(TokenType::T_COMMA);
+            $this->values[] = $parser->ArithmeticPrimary();
+        }
+
+        $parser->match(TokenType::T_CLOSE_PARENTHESIS);
+    }
+}

tests/config/packages/doctrine.yaml

@@ -17,3 +17,6 @@ doctrine:
                 dir: "%kernel.project_dir%/src/App/Entity"
                 prefix: 'Rekalogika\Rekapager\Tests\App\Entity'
                 alias: App
+        dql:
+            string_functions:
+                rekapager_row_values: Rekalogika\Rekapager\Doctrine\ORM\RowValuesFunction