MS SQL Server fix for older versions than 2012 (alternative solution)

This alternative solution LIMIT, OFFSET problem SQL Server version older than 2012.
2025-11-13 12:49:04 +08:00 · 2014-05-08 13:04:36 +09:00
parent 3e351fb879
commit 3e45097126
1 changed files with 89 additions and 0 deletions
--- a/framework/db/mssql/QueryBuilder.php
+++ b/framework/db/mssql/QueryBuilder.php
@@ -139,4 +139,93 @@ class QueryBuilder extends \yii\db\QueryBuilder

        return "ALTER TABLE {$table} {$enable} CONSTRAINT ALL";
    }
+    
+    public function buildOrderBy($columns)
+    {
+        if (empty($columns)) {
+            return 'ORDER BY (SELECT NULL)'; // hack so limit will work if no order by is specified
+        } else {
+            return parent::buildOrderBy($columns);
+        }
+    }
+    
+    public function build($query, $params = [])
+    {
+        $query->prepareBuild($this);
+
+        $params = empty($params) ? $query->params : array_merge($params, $query->params);
+
+        $clauses = [
+            $this->buildSelect($query->select, $params, $query->distinct, $query->selectOption),
+            $this->buildFrom($query->from, $params),
+            $this->buildJoin($query->join, $params),
+            $this->buildWhere($query->where, $params),
+            $this->buildGroupBy($query->groupBy),
+            $this->buildHaving($query->having, $params),
+            $this->buildOrderBy($query->orderBy),
+            $this->olderMssql() ? '' : $this->buildLimit($query->limit, $query->offset),
+        ];
+
+        $sql = implode($this->separator, array_filter($clauses));
+        if ($this->olderMssql())
+            $sql = $this->applyLimit($sql, $query);
+        $union = $this->buildUnion($query->union, $params);
+        if ($union !== '') {
+            $sql = "($sql){$this->separator}$union";
+        }
+
+
+        return [$sql, $params];
+    }
+
+    public function applyLimit($sql, $query)
+    {
+        $limit = $query->limit !== null ? (int)$query->limit : -1;
+        $offset = $query->offset !== null ? (int)$query->offset : -1;
+        if ($limit > 0 || $offset >= 0)
+            $sql = $this->rewriteLimitOffsetSql($sql, $limit, $offset, $query);
+        return $sql;
+    }
+
+    protected function rewriteLimitOffsetSql($sql, $limit, $offset, $query)
+    {
+        $originalOrdering = $this->buildOrderBy($query->orderBy);
+        if ($query->select) {
+            $select = implode(', ', $query->select);
+        }
+        else {
+            $select = $query->select = '*';
+        }
+        if ($select === '*') {
+            $columns = $this->getAllColumnNames($query->from[0]);
+            if ($columns && is_array($columns))
+                $select = implode(', ', $columns);
+            else
+                $select = $columns;
+        }
+        $sql = str_replace($originalOrdering, '', $sql);
+        $sql = preg_replace('/^([\s(])*SELECT( DISTINCT)?(?!\s*TOP\s*\()/i', "\\1SELECT\\2 rowNum = ROW_NUMBER() over ({$originalOrdering}),", $sql);
+        $sql = "SELECT TOP {$limit} {$select} FROM ($sql) sub WHERE rowNum > {$offset}";
+        return $sql;
+    }
+
+    protected function getAllColumnNames($table = null)
+    {
+        if (!$table) {
+            return null;
+        }
+        $columns = (new \yii\db\Query())
+            ->select('name')
+            ->from('sys.columns')
+            ->where("object_id = OBJECT_ID('dbo.{$table}')")
+            ->column();
+        return $columns;
+    }
+
+    protected function olderMssql()
+    {
+        $this->db->open();
+        $version = preg_split("/\./", $this->db->pdo->getAttribute(\PDO::ATTR_SERVER_VERSION));
+        return $version[0] < 11;
+    }
 }