Merge pull request #230 from stillalivx/master

Added: Truncate statement

lib/database.ts

@@ -36,6 +36,8 @@ export type DatabaseOptions =
 export type SyncOptions = {
   /** If tables should be dropped if they exist. */
   drop?: boolean;
+  /** If tables should be truncated. Will raise errors if the linked tables don't exist. */
+  truncate?: boolean;
 };
 
 /** Database client which interacts with an external database instance. */
@@ -187,11 +189,17 @@ export class Database {
    */
   async sync(options: SyncOptions = {}) {
     if (options.drop) {
-      for (let i = this._models.length - 1; i >= 0; i--){
+      for (let i = this._models.length - 1; i >= 0; i--) {
         await this._models[i].drop();
       }
     }
 
+    if (options.truncate) {
+      for (let i = this._models.length - 1; i >= 0; i--) {
+        await this._models[i].truncate();
+      }
+    }
+
     for (const model of this._models) {
       await model.createTable();
     }

lib/model.ts

@@ -139,6 +139,17 @@ export class Model {
     this._isCreatedInDatabase = false;
   }
 
+  /** Truncate a model in the database. */
+  static async truncate() {
+    const truncateQuery = this._options.queryBuilder
+      .queryForSchema(this)
+      .table(this.table)
+      .truncate()
+      .toDescription();
+
+    await this._options.database.query(truncateQuery);
+  }
+
   /** Create a model in the database. Should not be called from a child model. */
   static async createTable() {
     if (this._isCreatedInDatabase) {

lib/query-builder.ts

@@ -8,6 +8,7 @@ export type OrderDirection = "desc" | "asc";
 export type QueryType =
   | "create"
   | "drop"
+  | "truncate"
   | "select"
   | "insert"
   | "update"
@@ -122,6 +123,11 @@ export class QueryBuilder {
     return this;
   }
 
+  truncate() {
+    this._query.type = "truncate";
+    return this;
+  }
+
   select(...fields: (string | FieldAlias)[]) {
     this._query.select = fields;
     return this;

lib/translators/sql-translator.ts

@@ -126,6 +126,10 @@ export class SQLTranslator implements Translator {
         queryBuilder = queryBuilder.schema[dropTableHelper](query.table);
         break;
 
+      case "truncate":
+        queryBuilder = queryBuilder.truncate(query.table);
+        break;
+
       case "create":
         if (!query.fields) {
           throw new Error(