Add limit to delete queries (#60)

Author: G4brym

package-lock.json

@@ -1,6 +1,6 @@
 {
   "name": "workers-qb",
-  "version": "1.6.3",
+  "version": "1.6.4",
   "description": "Zero dependencies Query Builder for Cloudflare Workers",
   "main": "./dist/index.js",
   "module": "./dist/index.mjs",

package.json

@@ -382,6 +382,7 @@ export class QueryBuilder<GenericResultWrapper, IsAsync extends boolean = true>
       `DELETE
             FROM ${params.tableName}` +
       this._where(params.where) +
+      this._limit(params.limit) +
       this._returning(params.returning)
     )
   }

src/builder.ts

@@ -105,6 +105,7 @@ export type UpdateWithoutReturning = Omit<Update, 'returning'>
 export type Delete = {
   tableName: string
   where: Where // This field is optional, but is kept required in type to warn users of delete without where
+  limit?: number
   returning?: string | Array<string>
 }
 

src/interfaces.ts

@@ -26,6 +26,18 @@ describe('Delete Builder', () => {
     expect(result.fetchType).toEqual('ALL')
   })
 
+  test('delete with limit', async () => {
+    const result = new QuerybuilderTest().delete({
+      tableName: 'testTable',
+      where: 'field = false',
+      limit: 10,
+    })
+
+    expect(result.query).toEqual('DELETE FROM testTable WHERE field = false LIMIT 10')
+    expect(result.arguments).toEqual(undefined)
+    expect(result.fetchType).toEqual('ALL')
+  })
+
   test('delete with simplified where list', async () => {
     const result = new QuerybuilderTest().delete({
       tableName: 'testTable',
@@ -80,4 +92,20 @@ describe('Delete Builder', () => {
     expect(result.arguments).toEqual(['test', 123])
     expect(result.fetchType).toEqual('ALL')
   })
+
+  test('delete with multiple where with multiple returning and limit', async () => {
+    const result = new QuerybuilderTest().delete({
+      tableName: 'testTable',
+      where: {
+        conditions: ['field = ?1', 'id = ?2'],
+        params: ['test', 123],
+      },
+      returning: ['id', 'field'],
+      limit: 10000
+    })
+
+    expect(result.query).toEqual('DELETE FROM testTable WHERE field = ?1 AND id = ?2 LIMIT 10000 RETURNING id, field')
+    expect(result.arguments).toEqual(['test', 123])
+    expect(result.fetchType).toEqual('ALL')
+  })
 })