Add limit method:

Add limit method to limit output records.

Author: Mohammadreza-73

src/Database/PDOQueryBuilder.php

@@ -12,6 +12,7 @@ class PDOQueryBuilder
     private array $values;
     private string $operator;
     private array $order;
+    private int $limit;
 
     private static $pdo;
     private static $table;
@@ -119,7 +120,12 @@ public function get(array $columns = ['*']): array
             $order = " ORDER BY {$orderColumn} {$sort}";
         }
 
-        $sql = "SELECT {$columns} FROM " . self::$table . "{$conditions}{$order}";
+        $limit = '';
+        if (isset($this->limit)) {
+            $limit = " LIMIT {$this->limit}";
+        }
+
+        $sql = "SELECT {$columns} FROM " . self::$table . "{$conditions}{$order}{$limit}";
         $stmt = self::$pdo->prepare($sql);
         $stmt->execute($this->values ?? []);
 
@@ -153,6 +159,13 @@ public function orderBy(string $column, string $sort = 'ASC'): self
         return $this;
     }
 
+    public function limit(int $limit): self
+    {
+        $this->limit = $limit;
+
+        return $this;
+    }
+
     public static function truncateAllTable(): void
     {
         $stmt = self::$pdo->prepare("SHOW TABLES");

tests/Unit/PDOQueryBuilderTest.php

@@ -152,6 +152,17 @@ public function testItCanGetOrderdData(): void
         $this->assertEquals('Zend', $result[0]->skill); // First row
     }
 
+    public function testItCanLimitData(): void
+    {
+        $this->multipleInsertIntoDb(10);
+
+        $result = PDOQueryBuilder::table('users')
+            ->limit(5)
+            ->get();
+        
+        $this->assertCount(5, $result);
+    }
+
     private function insertIntoDb(array $data = []): int
     {
         $data = array_merge([