dbi::update() returns affected rows

2 files changed, 15 insertions, 0 deletions

diff --git a/it_dbi.class b/it_dbi.class
index 0457142..14899d2 100644
--- a/it_dbi.class
+++ b/it_dbi.class
@@ -651,6 +651,7 @@ function replace($tags = array())
  * Update current record or a number of records given by where condition
  * @param $tags key => value pairs (these have priority over changes in member vars)
  * @param $where condition to select records to be modified (if not current record)
+ * @return number of updated records (or false on error)
  * Does not destroy internal state of last select() call
  */
 function update($tags = array(), $where = null)
@@ -668,6 +669,8 @@ function update($tags = array(), $where = null)
 
 		if ($result = $this->query("UPDATE {$this->_p['table']} $set " . $this->_where($where, $this->_link)))
 		{
+			$result = mysql_affected_rows($this->_link);
+
 			if (array_key_exists($this->_p['keyfield'], $tags))	# Did we just update the key?
 				$this->_key = $tags[$this->_p['keyfield']];
 
diff --git a/tests/it_dbi.t b/tests/it_dbi.t
index 3428c0f..3ee2562 100755
--- a/tests/it_dbi.t
+++ b/tests/it_dbi.t
@@ -132,6 +132,18 @@ is(
 	"update"
 );
 
+is(
+	$record->update(array('x' => 18), array('x' => 17)),
+	1,
+	"return affected rows",
+);
+
+is(
+	$record->update(array('x' => 18), array('x' => 17)),
+	0,
+	"return zero affected rows",
+);
+
 $record->update(array('-x' => 'RAND() * 10'));
 isnt(
 	array($record->_key, $record->x, $record->foo),