From f9777c989ce04d33288655ea4d1c274bd5b90b5f Mon Sep 17 00:00:00 2001
From: Filipp Lepalaan <filipp@mac.com>
Date: Wed, 1 Sep 2010 18:28:32 +0300
Subject: return row count on delete

---
 MainDb.php | 17 ++++++++++++++++-
 1 file changed, 16 insertions(+), 1 deletion(-)

(limited to 'MainDb.php')

diff --git a/MainDb.php b/MainDb.php
index f26527d..6a18f9e 100644
--- a/MainDb.php
+++ b/MainDb.php
@@ -103,7 +103,12 @@ class MainDb
         return MainApp::error($error);
     }
     
-    // select statements need the query results
+    // DELETE statements should report number of rows
+    if (preg_match('/^DELETE/i', $sql)) {
+      return $result->rowCount();
+    }
+    
+    // SELECT statements need the query results
     if (preg_match('/^SELECT/i', $sql)) {
       return $stmt;
     }
@@ -147,6 +152,16 @@ class MainDb
     $stmt = self::query($sql, $args) or exit(MainApp::error('Error executing query '.$sql));
     return $stmt->fetchAll(PDO::FETCH_ASSOC);
   }
+  
+  ////
+  // count something
+  public static function total($table)
+  {
+    $sql = 'SELECT COUNT(*) AS the_count FROM `%s`';
+    $res = self::fetch(sprintf($sql, $table));
+    $res = current($res);
+    return $res['the_count'];
+  }
 	
 }
 
-- 
cgit v1.2.3