Allow passing in of PDO startup options.

miccheng · miccheng · commit ef77493a25c0 · 2013-01-05T09:48:34.000+08:00
diff --git a/src/PHPQueue/Backend/PDO.php b/src/PHPQueue/Backend/PDO.php
@@ -9,6 +9,7 @@ class PDO extends Base
     private $db_user;
     private $db_password;
     private $db_table;
+    private $pdo_options = array();
 
     public function __construct($options=array())
     {
@@ -25,6 +26,9 @@ public function __construct($options=array())
         if (!empty($options['db_table'])) {
             $this->db_table = $options['db_table'];
         }
+        if (!empty($options['pdo_options']) && is_array($options['pdo_options'])) {
+            $this->pdo_options = array_merge($this->pdo_options, $options['pdo_options']);
+        }
     }
 
     public function setTable($table_name=null)
@@ -39,7 +43,7 @@ public function setTable($table_name=null)
 
     public function connect()
     {
-        $this->connection = new \PDO($this->connection_string, $this->db_user, $this->db_password);
+        $this->connection = new \PDO($this->connection_string, $this->db_user, $this->db_password, $this->pdo_options);
     }
 
     public function add($data = null)
diff --git a/test/PHPQueue/Backend/PDOTest.php b/test/PHPQueue/Backend/PDOTest.php
@@ -18,6 +18,9 @@ public function setUp()
                 , 'db_user'           => 'root'
                 , 'db_password'       => 'media1'
                 , 'db_table'          => 'pdotest'
+                , 'pdo_options'       => array(
+                    \PDO::ATTR_PERSISTENT => true
+                )
             );
             $this->object = new PDO($options);
         }

Original file line number	Diff line number	Diff line change
`@@ -9,6 +9,7 @@ class PDO extends Base`
`9`	`9`	`private $db_user;`
`10`	`10`	`private $db_password;`
`11`	`11`	`private $db_table;`
	`12`	`+ private $pdo_options = array();`
`12`	`13`
`13`	`14`	`public function __construct($options=array())`
`14`	`15`	`{`
`@@ -25,6 +26,9 @@ public function __construct($options=array())`
`25`	`26`	`if (!empty($options['db_table'])) {`
`26`	`27`	`$this->db_table = $options['db_table'];`
`27`	`28`	`}`
	`29`	`+ if (!empty($options['pdo_options']) && is_array($options['pdo_options'])) {`
	`30`	`+ $this->pdo_options = array_merge($this->pdo_options, $options['pdo_options']);`
	`31`	`+ }`
`28`	`32`	`}`
`29`	`33`
`30`	`34`	`public function setTable($table_name=null)`
`@@ -39,7 +43,7 @@ public function setTable($table_name=null)`
`39`	`43`
`40`	`44`	`public function connect()`
`41`	`45`	`{`
`42`		`- $this->connection = new \PDO($this->connection_string, $this->db_user, $this->db_password);`
	`46`	`+ $this->connection = new \PDO($this->connection_string, $this->db_user, $this->db_password, $this->pdo_options);`
`43`	`47`	`}`
`44`	`48`
`45`	`49`	`public function add($data = null)`