feat: allow to give number and BigInt to executionParameters (#16)

Author: yamatatsu

lib/athena-query.ts

@@ -15,12 +15,26 @@ export class AthenaQuery {
 
   async *query(
     sql: string,
-    options?: { executionParameters?: string[]; maxResults?: number }
+    options?: {
+      executionParameters?: (string | number | BigInt)[];
+      maxResults?: number;
+    }
   ): AsyncGenerator<helpers.AtheneRecordData, void, undefined> {
     const QueryExecutionId = await helpers.startQueryExecution({
       athena: this.athena,
       sql,
-      executionParameters: options?.executionParameters,
+      executionParameters: options?.executionParameters?.map((param) => {
+        const typeOfParam = typeof param;
+        switch (typeOfParam) {
+          case "bigint":
+          case "number":
+            return param.toString();
+          case "string":
+            return `'${param}'`;
+          default:
+            throw new Error(`${typeOfParam} type is not allowed.`);
+        }
+      }),
       ...this.options,
     });
 

test/index.test.ts

@@ -251,7 +251,7 @@ test("pass args to sdk", async () => {
     catalog: "test-catalog",
   });
   const resultGen = athenaQuery.query("SELECT test FROM test;", {
-    executionParameters: ["'test'", "123"],
+    executionParameters: ["test", 123, 456n],
     maxResults: 100,
   });
 
@@ -261,7 +261,7 @@ test("pass args to sdk", async () => {
     athenaMock.commandCalls(StartQueryExecutionCommand)[0].args[0].input
   ).toEqual({
     QueryString: "SELECT test FROM test;",
-    ExecutionParameters: ["'test'", "123"],
+    ExecutionParameters: ["'test'", "123", "456"],
     WorkGroup: "test-workgroup",
     QueryExecutionContext: {
       Catalog: "test-catalog",