- Patch #396578 by Damien Tournoud: added db_truncate_table() to the database layer.

1 files changed, 44 insertions, 0 deletions

diff --git a/includes/database/database.inc b/includes/database/database.inc
index f07e07373..60d4b415f 100644
--- a/includes/database/database.inc
+++ b/includes/database/database.inc
@@ -255,6 +255,13 @@ abstract class DatabaseConnection extends PDO {
   protected $deleteClass = NULL;
 
   /**
+   * The name of the Truncate class for this connection.
+   *
+   * @var string
+   */
+  protected $truncateClass = NULL;
+
+  /**
    * The name of the Insert class for this connection.
    *
    * @var string
@@ -763,6 +770,26 @@ abstract class DatabaseConnection extends PDO {
   }
 
   /**
+   * Prepare and return a TRUNCATE query object.
+   *
+   * @see TruncateQuery
+   * @param $options
+   *   An array of options on the query.
+   * @return
+   *   A new DeleteQuery object.
+   */
+  public function truncate($table, array $options = array()) {
+    if (empty($this->truncateClass)) {
+      $this->truncateClass = 'TruncateQuery_' . $this->driver();
+      if (!class_exists($this->truncateClass)) {
+        $this->truncateClass = 'TruncateQuery';
+      }
+    }
+    $class = $this->truncateClass;
+    return new $class($this, $table, $options);
+  }
+
+  /**
    * Returns a DatabaseSchema object for manipulating the schema of this database.
    *
    * This method will lazy-load the appropriate schema library file.
@@ -1881,6 +1908,23 @@ function db_delete($table, array $options = array()) {
 }
 
 /**
+ * Returns a new TruncateQuery object for the active database.
+ *
+ * @param $table
+ *   The table from which to delete.
+ * @param $options
+ *   An array of options to control how the query operates.
+ * @return
+ *   A new TruncateQuery object for this connection.
+ */
+function db_truncate($table, array $options = array()) {
+  if (empty($options['target']) || $options['target'] == 'slave') {
+    $options['target'] = 'default';
+  }
+  return Database::getConnection($options['target'])->truncate($table, $options);
+}
+
+/**
  * Returns a new SelectQuery object for the active database.
  *
  * @param $table