feat(db): add truncateTable methodpulsejet/truncate-1

Signed-off-by: Varun Patil <varunpatil@ucla.edu>

3 files changed, 36 insertions, 0 deletions

diff --git a/lib/private/DB/Connection.php b/lib/private/DB/Connection.php
index 3fddfdcb094..eecf83ace95 100644
--- a/lib/private/DB/Connection.php
+++ b/lib/private/DB/Connection.php
@@ -699,6 +699,19 @@ class Connection extends PrimaryReadReplicaConnection {
 	}
 
 	/**
+	 * Truncate a table data if it exists
+	 *
+	 * @param string $table table name without the prefix
+	 * @param bool $cascade whether to truncate cascading
+	 *
+	 * @throws Exception
+	 */
+	public function truncateTable(string $table, bool $cascade) {
+		$this->executeStatement($this->getDatabasePlatform()
+			->getTruncateTableSQL($this->tablePrefix . trim($table), $cascade));
+	}
+
+	/**
 	 * Check if a table exists
 	 *
 	 * @param string $table table name without the prefix
diff --git a/lib/private/DB/ConnectionAdapter.php b/lib/private/DB/ConnectionAdapter.php
index 2baeda9cfb7..ba3bf90c2e8 100644
--- a/lib/private/DB/ConnectionAdapter.php
+++ b/lib/private/DB/ConnectionAdapter.php
@@ -189,6 +189,14 @@ class ConnectionAdapter implements IDBConnection {
 		}
 	}
 
+	public function truncateTable(string $table, bool $cascade): void {
+		try {
+			$this->inner->truncateTable($table, $cascade);
+		} catch (Exception $e) {
+			throw DbalException::wrap($e);
+		}
+	}
+
 	public function tableExists(string $table): bool {
 		try {
 			return $this->inner->tableExists($table);
diff --git a/lib/public/IDBConnection.php b/lib/public/IDBConnection.php
index 36369732b64..e0fe603ec57 100644
--- a/lib/public/IDBConnection.php
+++ b/lib/public/IDBConnection.php
@@ -296,6 +296,21 @@ interface IDBConnection {
 	public function dropTable(string $table): void;
 
 	/**
+	 * Truncate a table data if it exists
+	 *
+	 * Cascade is not supported on many platforms but would optionally cascade the truncate by
+	 * following the foreign keys.
+	 *
+	 * @param string $table table name without the prefix
+	 * @param bool $cascade whether to truncate cascading
+	 * @throws Exception
+	 * @since 32.0.0
+	 *
+	 * @psalm-taint-sink sql $table
+	 */
+	public function truncateTable(string $table, bool $cascade): void;
+
+	/**
 	 * Check if a table exists
 	 *
 	 * @param string $table table name without the prefix