Added fetch mode configuration

Author: Gusakov Nikita

src/Connection.php

@@ -51,8 +51,9 @@ class Connection extends \PDO
     public function __construct($dsn, $username = null, $password = null, array $attributes = [], Grammar $grammar = null)
     {
         parent::__construct($dsn, $username, $password, array_replace($attributes, [
-            \PDO::ATTR_ERRMODE         => \PDO::ERRMODE_EXCEPTION,
-            \PDO::ATTR_STATEMENT_CLASS => ['Flame\\Query', [&$this->grammar, &$this->placeholders, &$this->types]]
+            self::ATTR_ERRMODE            => self::ERRMODE_EXCEPTION,
+            self::ATTR_DEFAULT_FETCH_MODE => self::FETCH_ASSOC,
+            self::ATTR_STATEMENT_CLASS    => ['Flame\\Query', [&$this->grammar, &$this->placeholders, &$this->types]],
         ]));
 
         if ($grammar === null) {
@@ -113,6 +114,18 @@ public function commit()
         return $this;
     }
 
+    /**
+     * @param int $mode
+     *
+     * @return static
+     */
+    public function setDefaultFetchMode($mode)
+    {
+        $this->setAttribute(\PDO::ATTR_DEFAULT_FETCH_MODE, $mode);
+
+        return $this;
+    }
+
     /**
      * @param string $column,...
      *

src/Query.php

@@ -81,7 +81,7 @@ public function closeCursor()
      *
      * @return array
      */
-    public function fetchCallback(callable $callback, $mode = \PDO::FETCH_ASSOC)
+    public function fetchCallback(callable $callback, $mode = null)
     {
         $results = [];
         while (false !== $row = $this->fetch($mode)) {

tests/ConnectionTest.php

@@ -18,6 +18,16 @@ protected function setUp()
         $this->connection->query('CREATE TABLE users (id INT PRIMARY KEY, username CHAR(50), sex BOOL)');
     }
 
+    public function testSetDefaultFetchMode()
+    {
+        $this->assertSame(\PDO::FETCH_ASSOC, $this->connection->getAttribute(\PDO::ATTR_DEFAULT_FETCH_MODE));
+
+        $mode = \PDO::FETCH_OBJ;
+
+        $this->assertSame($this->connection, $this->connection->setDefaultFetchMode($mode));
+        $this->assertSame($mode, $this->connection->getAttribute(\PDO::ATTR_DEFAULT_FETCH_MODE));
+    }
+
     public function testAnotherGrammar()
     {
         $grammar = new MysqlGrammar();