From de21426cf1a4f46a3ffecffee9262dad664fe519 Mon Sep 17 00:00:00 2001
From: Brian Celenza <bri@celenza.dev>
Date: Thu, 13 Jul 2023 07:37:02 -0700
Subject: [PATCH] feat(logs): Add support for specifying the result limit

---
 src/index.ts | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/src/index.ts b/src/index.ts
index aa2446f..4112334 100755
--- a/src/index.ts
+++ b/src/index.ts
@@ -62,6 +62,12 @@ process.on("SIGINT", async () => {
       default: false,
       description: "Return just the message in plain-text format (no CSV)",
     })
+    .option("limit", {
+      alias: "i",
+      type: "number",
+      default: 1000,
+      description: "The limit to the number of results returned",
+    })
     .demandOption("log-group", "At least one log group is required.")
     .help()
     .alias("help", "h").argv;
@@ -76,6 +82,7 @@ process.on("SIGINT", async () => {
 
   const startTime = time.parseTimeOrDuration(args.start);
   const endTime = time.parseTimeOrDuration(args.end);
+  const limit = args.limit;
 
   const query = args._[0];
   if (!query) {
@@ -107,6 +114,7 @@ process.on("SIGINT", async () => {
       // CloudWatch uses standard unix timestamp (seconds since epoch)
       startTime: Math.ceil(startTime / 1000),
       endTime: Math.ceil(endTime / 1000),
+      limit,
     })
   );
   runningQueryId = startOutput.queryId;