From 7ec187ddc16d5e90605ab935ade6296daceebfc7 Mon Sep 17 00:00:00 2001
From: Chiri Vulpes <chiri@chiri.works>
Date: Tue, 3 Dec 2024 22:44:05 +1300
Subject: [PATCH] Return row count for update when not returning any columns

---
 src/statements/Update.ts | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/statements/Update.ts b/src/statements/Update.ts
index 4427daf..00cbd0e 100644
--- a/src/statements/Update.ts
+++ b/src/statements/Update.ts
@@ -4,7 +4,7 @@ import Schema, { TableSchema } from "../Schema";
 import Expression, { ExpressionInitialiser } from "../expressions/Expression";
 import Statement from "./Statement";
 
-export default class UpdateTable<SCHEMA extends TableSchema, RESULT = Record<string, never>[], VARS = {}> extends Statement<RESULT> {
+export default class UpdateTable<SCHEMA extends TableSchema, RESULT = number, VARS = {}> extends Statement<RESULT> {
 
 	private vars: any[];
 	public constructor (public readonly tableName: string | undefined, public readonly schema: SCHEMA, vars?: any[]) {
@@ -43,6 +43,7 @@ export default class UpdateTable<SCHEMA extends TableSchema, RESULT = Record<str
 	}
 
 	private returningColumns?: (Schema.Column<SCHEMA> | "*")[];
+	public returning (): UpdateTable<SCHEMA, number, VARS>;
 	public returning<RETURNING_COLUMNS extends Schema.Column<SCHEMA>[]> (...columns: RETURNING_COLUMNS): UpdateTable<SCHEMA, { [KEY in RETURNING_COLUMNS[number]]: OutputTypeFromString<SCHEMA[KEY]> }[], VARS>;
 	public returning<RETURNING_COLUMN extends Schema.Column<SCHEMA> | "*"> (columns: RETURNING_COLUMN): UpdateTable<SCHEMA, { [KEY in RETURNING_COLUMN extends "*" ? Schema.Column<SCHEMA> : RETURNING_COLUMN]: OutputTypeFromString<SCHEMA[KEY]> }[], VARS>;
 	public returning (...columns: (Schema.Column<SCHEMA> | "*")[]) {
@@ -58,6 +59,6 @@ export default class UpdateTable<SCHEMA extends TableSchema, RESULT = Record<str
 	}
 
 	protected override resolveQueryOutput (output: QueryResult<any>) {
-		return output.rows as RESULT;
+		return (!this.returningColumns?.length ? output.rowCount : output.rows) as RESULT;
 	}
 }
\ No newline at end of file