Added support for deleting tables from Ceen.Database

kenkendk · May 15, 2020 · 291d573 · 291d573
1 parent a3eb59c
commit 291d573
Show file tree

Hide file tree

Showing 4 changed files with 56 additions and 0 deletions.
diff --git a/Ceen.Database/DatabaseDialectBase.cs b/Ceen.Database/DatabaseDialectBase.cs
@@ -117,6 +117,13 @@ public virtual TableMapping GetTypeMap(Type type)
         /// <param name="ifNotExists">Only create table if it does not exist</param>
         public abstract string CreateTableSql(Type recordtype, bool ifNotExists = true);
 
+        /// <summary>
+        /// Returns a delete-table sql statement
+        /// </summary>
+        /// <param name="recordtype">The datatype to delete from the table.</param>
+        /// <param name="ifExists">Only delete table if it exists</param>
+        public abstract string DeleteTableSql(Type recordtype, bool ifExists = true);
+
         /// <summary>
         /// Creates a command that checks if a table exists
         /// </summary>

diff --git a/Ceen.Database/DatabaseDialectSQLite.cs b/Ceen.Database/DatabaseDialectSQLite.cs
@@ -154,6 +154,23 @@ public override string CreateTableSql(Type recordtype, bool ifNotExists = true)
             return sql;
         }
 
+        /// <summary>
+        /// Returns a delete-table sql statement
+        /// </summary>
+        /// <param name="recordtype">The datatype to delete from the table.</param>
+        /// <param name="ifExists">Only delete table if it exists</param>
+        public override string DeleteTableSql(Type recordtype, bool ifExists = true)
+        {
+            var mapping = GetTypeMap(recordtype);
+
+            return string.Format(
+                @"DROP TABLE{0} {1}",
+                ifExists ? " IF EXISTS" : "",
+                QuoteName(mapping.Name)
+            );
+        }
+
+
         /// <summary>
         /// Creates a command that checks if a table exists
         /// </summary>

diff --git a/Ceen.Database/DatabaseHelper.cs b/Ceen.Database/DatabaseHelper.cs
@@ -174,6 +174,31 @@ public static void CreateTable(this IDbConnection connection, Type type, bool if
             }
         }
 
+        /// <summary>
+        /// Deletes the table for the given type
+        /// </summary>
+        /// <typeparam name="T">The type of the table to delete.</typeparam>
+        /// <param name="connection">The connection to use</param>
+        public static void DeleteTable<T>(this IDbConnection connection)
+        {
+            DeleteTable(connection, typeof(T));
+        }
+
+        /// <summary>
+        /// Deletes a table for the given type
+        /// </summary>
+        /// <param name="type">The type of the table to delete.</param>
+        /// <param name="connection">The connection to use</param>
+        /// <param name="ifExists">Only delete table if it exists</param>
+        public static void DeleteTable(this IDbConnection connection, Type type, bool ifExists = true)
+        {
+            var dialect = GetDialect(connection);
+            var mapping = dialect.GetTypeMap(type);
+
+            var sql = dialect.DeleteTableSql(type);
+            using (var cmd = connection.CreateCommand(sql))
+                cmd.ExecuteNonQuery();
+        }
 
         /// <summary>
         /// Checks if the table for the given type exists

diff --git a/Ceen.Database/IDatabaseDialect.cs b/Ceen.Database/IDatabaseDialect.cs
@@ -71,6 +71,13 @@ public interface IDatabaseDialect
         /// <param name="ifNotExists">Only create table if it does not exist</param>
         string CreateTableSql(Type recordtype, bool ifNotExists = true);
 
+        /// <summary>
+        /// Returns a delete-table sql statement
+        /// </summary>
+        /// <param name="recordtype">The datatype to delete from the table.</param>
+        /// <param name="ifExists">Only delete table if it exists</param>
+        string DeleteTableSql(Type recordtype, bool ifExists = true);
+
         /// <summary>
         /// Creates a command that checks if a table exists
         /// </summary>