added replace(), currId and nextId() methods
This commit is contained in:
parent
d4df743a96
commit
5a5934db0b
@ -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);
|
||||
}
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user