[D1] add rows read/written to wrangler d1 info output (#4621)

* [D1] add rows read/written to `wrangler d1 info` output * Update info.tsx
cloudflare · Dec 18, 2023 · 98dee93 · 98dee93
1 parent e2205e3
commit 98dee93
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 2 deletions.
diff --git a/.changeset/four-students-joke.md b/.changeset/four-students-joke.md
@@ -0,0 +1,5 @@
+---
+"wrangler": patch
+---
+
+feat: add rows written/read in the last 24 hours to `wrangler d1 info` output
diff --git a/packages/wrangler/src/d1/info.tsx b/packages/wrangler/src/d1/info.tsx
@@ -63,6 +63,8 @@ export const Handler = withConfig<HandlerOptions>(
 											sum {
 												readQueries
 												writeQueries
+												rowsRead
+												rowsWritten
 										}
 										dimensions {
 												datetimeHour
@@ -90,16 +92,25 @@ export const Handler = withConfig<HandlerOptions>(
 				},
 			});
 
-			const metrics = { readQueries: 0, writeQueries: 0 };
+			const metrics = {
+				readQueries: 0,
+				writeQueries: 0,
+				rowsRead: 0,
+				rowsWritten: 0,
+			};
 			if (graphqlResult) {
 				graphqlResult.data?.viewer?.accounts[0]?.d1AnalyticsAdaptiveGroups?.forEach(
 					(row) => {
 						metrics.readQueries += row?.sum?.readQueries ?? 0;
 						metrics.writeQueries += row?.sum?.writeQueries ?? 0;
+						metrics.rowsRead += row?.sum?.rowsRead ?? 0;
+						metrics.rowsWritten += row?.sum?.rowsWritten ?? 0;
 					}
 				);
 				output.read_queries_24h = metrics.readQueries;
 				output.write_queries_24h = metrics.writeQueries;
+				output.rows_read_24h = metrics.rowsRead;
+				output.rows_written_24h = metrics.rowsWritten;
 			}
 		}
 
@@ -113,7 +124,12 @@ export const Handler = withConfig<HandlerOptions>(
 				let value;
 				if (k === "database_size") {
 					value = prettyBytes(Number(v));
-				} else if (k === "read_queries_24h" || k === "write_queries_24h") {
+				} else if (
+					k === "read_queries_24h" ||
+					k === "write_queries_24h" ||
+					k === "rows_read_24h" ||
+					k === "rows_written_24h"
+				) {
 					value = v.toLocaleString();
 				} else {
 					value = v;

diff --git a/packages/wrangler/src/d1/types.ts b/packages/wrangler/src/d1/types.ts
@@ -44,6 +44,8 @@ export interface D1Metrics {
 	sum?: {
 		readQueries?: number;
 		writeQueries?: number;
+		rowsRead?: number;
+		rowsWritten?: number;
 		queryBatchResponseBytes?: number;
 	};
 	quantiles?: {