AWS Data Api $count fix

Author: Sukairo-02

.github/workflows/release-feature-branch.yaml

@@ -14,7 +14,7 @@ jobs:
       matrix:
         shard:
           - gel
-          - planetscale
+          # - planetscale
           - singlestore-core
           - singlestore-proxy
           - singlestore-prefixed

drizzle-orm/src/aws-data-api/pg/session.ts

@@ -289,6 +289,17 @@ export class AwsDataApiSession<
 		);
 	}
 
+	override async count(sql: SQL): Promise<number> {
+		const query = this.dialect.sqlToQuery(sql);
+		const prepared = this.prepareQuery(query, undefined, undefined, true);
+
+		const { rows } = await prepared.values();
+		const count = rows[0]?.[0] ?? 0;
+
+		if (typeof count === 'number') return count;
+		return Number(count);
+	}
+
 	override execute<T>(query: SQL): Promise<T> {
 		return this.prepareQuery<PreparedQueryConfig & { execute: T; values: AwsDataApiPgQueryResult<unknown[]> }>(
 			this.dialect.sqlToQuery(query),