<?php
    2 /*
    3  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
    4  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
    5  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
    6  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
    7  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
    8  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
    9  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
   10  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
   11  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
   12  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
   13  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
   14  *
   15  * This software consists of voluntary contributions made by many individuals
   16  * and is licensed under the MIT license. For more information, see
   17  * <http://www.doctrine-project.org>.
   18  */

    namespace DoctrineDBALDriver;

    use PDO;

    class PDOConnection extends PDO implements Connection, ServerInfoAwareConnection
    {
        public function __construct($dsn, $user = null, $password = null, array $options = null)
        {
            try {
                parent::__construct($dsn, $user, $password, $options);
                $this->setAttribute(PDO::ATTR_STATEMENT_CLASS, array('DoctrineDBALDriverPDOStatement', array()));
               $this->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_EXCEPTION);
            } catch (PDOException $exception) {
                throw new PDOException($exception);
            }
        }

        public function exec($statement)
        {
            try {
                return parent::exec($statement);
            } catch (PDOException $exception) {
               throw new PDOException($exception);
            }
        }

        public function getServerVersion()
        {
            return PDO::getAttribute(PDO::ATTR_SERVER_VERSION);
        }

        public function prepare($prepareString, $driverOptions = array())
        {
           try {
                return parent::prepare($prepareString, $driverOptions);
            } catch (PDOException $exception) {
                throw new PDOException($exception);
            }
        }

        public function query()
       {
            $args = func_get_args();
            $argsCount = count($args);

            try {
                if ($argsCount == 4) {
                    return parent::query($args[0], $args[1], $args[2], $args[3]);
                }

               if ($argsCount == 3) {
                    return parent::query($args[0], $args[1], $args[2]);
                }

              if ($argsCount == 2) {
                   return parent::query($args[0], $args[1]);
               }

               return parent::query($args[0]);
           } catch (PDOException $exception) {
               throw new PDOException($exception);
           }
       }

       public function quote($input, $type = PDO::PARAM_STR)
       {
           return parent::quote($input, $type);
       }

      public function lastInsertId($name = null)
       {
           return parent::lastInsertId($name);
       }

       public function requiresQueryForServerVersion()
       {
           return false;
       }
    }