/**
* Sets the number of rows affected by the query
*
* @param fResult $result The result object for the query
* @param mixed $resource Only applicable for `ibm_db2`, `pdo`, `oci8` and `sqlsrv` extentions or `mysqli` prepared statements - this is either the `PDOStatement` object, `mysqli_stmt` object or the `oci8` or `sqlsrv` resource
* @return void
*/
private function setAffectedRows($result, $resource = NULL)
{
if ($this->extension == 'ibm_db2') {
$insert_update_delete = preg_match('#^\\s*(INSERT|UPDATE|DELETE)\\b#i', $result->getSQL());
$result->setAffectedRows(!$insert_update_delete ? 0 : db2_num_rows($resource));
} elseif ($this->extension == 'mssql') {
$affected_rows_result = mssql_query('SELECT @@ROWCOUNT AS rows', $this->connection);
$result->setAffectedRows((int) mssql_result($affected_rows_result, 0, 'rows'));
} elseif ($this->extension == 'mysql') {
$result->setAffectedRows(mysql_affected_rows($this->connection));
} elseif ($this->extension == 'mysqli') {
if (is_object($resource)) {
$result->setAffectedRows($resource->affected_rows);
} else {
$result->setAffectedRows(mysqli_affected_rows($this->connection));
}
} elseif ($this->extension == 'oci8') {
$result->setAffectedRows(oci_num_rows($resource));
} elseif ($this->extension == 'pgsql') {
$result->setAffectedRows(pg_affected_rows($result->getResult()));
} elseif ($this->extension == 'sqlite') {
$result->setAffectedRows(sqlite_changes($this->connection));
} elseif ($this->extension == 'sqlsrv') {
$result->setAffectedRows(sqlsrv_rows_affected($resource));
} elseif ($this->extension == 'pdo') {
// This fixes the fact that rowCount is not reset for non INSERT/UPDATE/DELETE statements
try {
if (!$resource || !$resource->fetch()) {
throw new PDOException();
}
$result->setAffectedRows(0);
} catch (PDOException $e) {
// The SQLite PDO driver seems to return 1 when no rows are returned from a SELECT statement
if ($this->type == 'sqlite' && $this->extension == 'pdo' && preg_match('#^\\s*SELECT#i', $result->getSQL())) {
$result->setAffectedRows(0);
} elseif (!$resource) {
$result->setAffectedRows(0);
} else {
$result->setAffectedRows($resource->rowCount());
}
}
}
}
