added replace(), currId and nextId() methods

lib/Doctrine/Connection/Mysql.php

@@ -40,7 +40,7 @@ class Doctrine_Connection_Mysql extends Doctrine_Connection_Common {
      * the constructor
      *
      * @param Doctrine_Manager $manager
-     * @param PDO $pdo                          database handle
+     * @param PDO|Doctrine_Adapter $adapter     database handler
      */
     public function __construct(Doctrine_Manager $manager, $adapter) {
         $adapter->setAttribute(PDO::ATTR_EMULATE_PREPARES, true);
@@ -68,6 +68,145 @@ class Doctrine_Connection_Mysql extends Doctrine_Connection_Common {
 
         parent::__construct($manager, $adapter);
     }    
+    /**
+     * Returns the next free id of a sequence
+     *
+     * @param string $seq_name name of the sequence
+     * @param boolean $ondemand when true the sequence is
+     *                          automatic created, if it
+     *                          not exists
+     *
+     * TODO: on demand creation of sequence table
+     *
+     * @return integer
+     */
+    public function nextID($seqName, $ondemand = true) {
+        $sequenceName = $this->quoteIdentifier($this->getSequenceName($seqName), true);
+        $seqcolName   = $this->quoteIdentifier($this->getAttribute(Doctrine::ATTR_SEQCOL_NAME), true);
+        $query        = 'INSERT INTO ' . $sequenceName . ' (' . $seqcolName . ') VALUES (NULL)';
+
+        $value = $this->dbh->lastInsertId();
+
+        if(is_numeric($value)) {
+            $query  = 'DELETE FROM ' . $sequenceName . ' WHERE ' . $seqcolName . ' < ' . $value;
+            $result = $this->dbh->query($query);
+        }
+        return $value;
+    }
+    /**
+     * Returns the current id of a sequence
+     *
+     * @param string $seq_name name of the sequence
+     * @return integer
+     */
+    public function currID($seqName) {
+        $sequenceName = $this->quoteIdentifier($this->getSequenceName($seqName), true);
+        $seqcolName = $this->quoteIdentifier($this->options['seqcol_name'], true);
+        $query = 'SELECT MAX(' . $seqcolName . ') FROM ' . $sequenceName;
+        return $this->queryOne($query, 'integer');
+    }
+    /**
+     * Execute a SQL REPLACE query. A REPLACE query is identical to a INSERT
+     * query, except that if there is already a row in the table with the same
+     * key field values, the REPLACE query just updates its values instead of
+     * inserting a new row.
+     *
+     * The REPLACE type of query does not make part of the SQL standards. Since
+     * practically only MySQL implements it natively, this type of query is
+     * emulated through this method for other DBMS using standard types of
+     * queries inside a transaction to assure the atomicity of the operation.
+     *
+     * @access public
+     *
+     * @param string $table name of the table on which the REPLACE query will
+     *  be executed.
+     * @param array $fields associative array that describes the fields and the
+     *  values that will be inserted or updated in the specified table. The
+     *  indexes of the array are the names of all the fields of the table. The
+     *  values of the array are also associative arrays that describe the
+     *  values and other properties of the table fields.
+     *
+     *  Here follows a list of field properties that need to be specified:
+     *
+     *    value:
+     *          Value to be assigned to the specified field. This value may be
+     *          of specified in database independent type format as this
+     *          function can perform the necessary datatype conversions.
+     *
+     *    Default:
+     *          this property is required unless the Null property
+     *          is set to 1.
+     *
+     *    type
+     *          Name of the type of the field. Currently, all types Metabase
+     *          are supported except for clob and blob.
+     *
+     *    Default: no type conversion
+     *
+     *    null
+     *          Boolean property that indicates that the value for this field
+     *          should be set to null.
+     *
+     *          The default value for fields missing in INSERT queries may be
+     *          specified the definition of a table. Often, the default value
+     *          is already null, but since the REPLACE may be emulated using
+     *          an UPDATE query, make sure that all fields of the table are
+     *          listed in this function argument array.
+     *
+     *    Default: 0
+     *
+     *    key
+     *          Boolean property that indicates that this field should be
+     *          handled as a primary key or at least as part of the compound
+     *          unique index of the table that will determine the row that will
+     *          updated if it exists or inserted a new row otherwise.
+     *
+     *          This function will fail if no key field is specified or if the
+     *          value of a key field is set to null because fields that are
+     *          part of unique index they may not be null.
+     *
+     *    Default: 0
+     *
+     * @return integer      the number of affected rows
+     */
+    public function replace($table, $fields) {
+        $count = count($fields);
+        $query = $values = '';
+        $keys = $colnum = 0;
+
+        for(reset($fields); $colnum < $count; next($fields), $colnum++) {
+            $name = key($fields);
+
+            if ($colnum > 0) {
+                $query .= ',';
+                $values.= ',';
+            }
 
+            $query .= $name;
 
+            if (isset($fields[$name]['null']) && $fields[$name]['null']) {
+                $value = 'NULL';
+            } else {
+                $type = isset($fields[$name]['type']) ? $fields[$name]['type'] : null;
+                $value = $this->quote($fields[$name]['value'], $type);
+            }
+
+            $values .= $value;
+
+            if (isset($fields[$name]['key']) && $fields[$name]['key']) {
+                if ($value === 'NULL')
+                    throw new Doctrine_Connection_Mysql_Exception('key value '.$name.' may not be NULL');
+
+                $keys++;
+            }
+        }
+
+        if ($keys == 0)
+            throw new Doctrine_Connection_Mysql_Exception('not specified which fields are keys');
+
+
+        $query = 'REPLACE INTO ' . $table . ' (' . $query . ') VALUES (' . $values . ')';
+
+        return $this->dbh->exec($query);
+    }
 }