feat: run method that returns affected row count (#598)

Because we throw away all other details of the result, this adds a new
method that will be able to retain more than just the result. For now
we can return the row count.

Author: amerryma

packages/runtime/src/tag.ts

@@ -8,7 +8,10 @@ export interface ICursor<T> {
 }
 
 export interface IDatabaseConnection {
-  query: (query: string, bindings: any[]) => Promise<{ rows: any[] }>;
+  query: (
+    query: string,
+    bindings: any[],
+  ) => Promise<{ rows: any[]; rowCount: number }>;
   stream?: (query: string, bindings: any[]) => ICursor<any[]>;
 }
 
@@ -95,6 +98,11 @@ export class PreparedQuery<TParamType, TResultType> {
     dbConnection: IDatabaseConnection,
   ) => Promise<Array<TResultType>>;
 
+  public runWithCounts: (
+    params: TParamType,
+    dbConnection: IDatabaseConnection,
+  ) => Promise<{ result: Array<TResultType>; rowCount: number }>;
+
   public stream: (
     params: TParamType,
     dbConnection: IDatabaseConnection,
@@ -112,6 +120,17 @@ export class PreparedQuery<TParamType, TResultType> {
       const result = await connection.query(processedQuery, bindings);
       return mapQueryResultRows(result.rows);
     };
+    this.runWithCounts = async (params, connection) => {
+      const { query: processedQuery, bindings } = processSQLQueryIR(
+        this.queryIR,
+        params as any,
+      );
+      const result = await connection.query(processedQuery, bindings);
+      return {
+        result: mapQueryResultRows(result.rows),
+        rowCount: result.rowCount,
+      };
+    };
     this.stream = (params, connection) => {
       const { query: processedQuery, bindings } = processSQLQueryIR(
         this.queryIR,