diff --git a/dibi/dibi.php b/dibi/dibi.php index cf84d64e..bd610bf8 100644 --- a/dibi/dibi.php +++ b/dibi/dibi.php @@ -63,7 +63,8 @@ class dibi DATETIME = 't', TIME = 't'; - const IDENTIFIER = 'n'; + const IDENTIFIER = 'n', + AFFECTED_ROWS = 'a'; /** @deprecated */ const FIELD_TEXT = dibi::TEXT, diff --git a/dibi/libs/DibiFluent.php b/dibi/libs/DibiFluent.php index 1d63a9e3..9ba0af46 100644 --- a/dibi/libs/DibiFluent.php +++ b/dibi/libs/DibiFluent.php @@ -309,7 +309,14 @@ class DibiFluent extends DibiObject implements IDataSource public function execute($return = NULL) { $res = $this->query($this->_export()); - return $return === dibi::IDENTIFIER ? $this->connection->getInsertId() : $res; + switch ($return) { + case dibi::IDENTIFIER: + return $this->connection->getInsertId(); + case dibi::AFFECTED_ROWS: + return $this->connection->getAffectedRows(); + default: + return $res; + } }