Merge pull request #73 from sarahkemp/master

Add offset_compatibility_mode option for newer versions of DB2
cooperl22 · Dec 10, 2021 · f5cad53 · f5cad53
2 parents fed6ccf + 1ee2072
commit f5cad53
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 2 deletions.
diff --git a/src/Database/DB2Connection.php b/src/Database/DB2Connection.php
@@ -108,6 +108,10 @@ protected function getDefaultQueryGrammar()
             $defaultGrammar->setDateFormat($this->config['date_format']);
         }
 
+        if (array_key_exists('offset_compatibility_mode', $this->config)) {
+            $defaultGrammar->setOffsetCompatibilityMode($this->config['offset_compatibility_mode']);
+        }
+
         return $this->withTablePrefix($defaultGrammar);
     }
 

diff --git a/src/Database/Query/Grammars/DB2Grammar.php b/src/Database/Query/Grammars/DB2Grammar.php
@@ -19,6 +19,12 @@ class DB2Grammar extends Grammar
      */
     protected $dateFormat;
 
+    /**
+     * Offset compatibility mode true triggers FETCH FIRST X ROWS and ROW_NUM behavior for older versions of DB2
+     * @var bool
+     */
+    protected $offsetCompatibilityMode = true;
+
     /**
      * Wrap a single string in keyword identifiers.
      *
@@ -45,7 +51,10 @@ protected function wrapValue($value)
      */
     protected function compileLimit(Builder $query, $limit)
     {
-        return "FETCH FIRST $limit ROWS ONLY";
+        if($this->offsetCompatibilityMode){
+            return "FETCH FIRST $limit ROWS ONLY";
+        }
+        return parent::compileLimit($query, $limit);
     }
 
     /**
@@ -57,6 +66,10 @@ protected function compileLimit(Builder $query, $limit)
      */
     public function compileSelect(Builder $query)
     {
+        if(!$this->offsetCompatibilityMode){
+            return parent::compileSelect($query);
+        }
+
         if (is_null($query->columns)) {
             $query->columns = ['*'];
         }
@@ -183,7 +196,10 @@ protected function compileTableExpression($sql, $constraint)
      */
     protected function compileOffset(Builder $query, $offset)
     {
-        return '';
+        if($this->offsetCompatibilityMode){
+            return '';
+        }
+        return parent::compileOffset($query, $offset);
     }
 
     /**
@@ -221,6 +237,16 @@ public function setDateFormat($dateFormat)
         $this->dateFormat = $dateFormat;
     }
 
+    /**
+     * Set offset compatibility mode to trigger FETCH FIRST X ROWS and ROW_NUM behavior for older versions of DB2
+     *
+     * @param $bool
+     */
+    public function setOffsetCompatibilityMode($bool)
+    {
+        $this->offsetCompatibilityMode = $bool;
+    }
+
     /**
      * Compile the SQL statement to define a savepoint.
      *