triangle-utils 1.4.156 → 1.4.157

package/dist/src/UtilsRDS.d.ts

@@ -27,7 +27,8 @@ export declare class UtilsRDS {
         is_verbose?: boolean;
         is_forced?: boolean;
     }): Promise<Record<string, any> | undefined>;
-    batch_get(table_id: string, primary_key: Record<string, any[]>, options?: {
+    batch_get(table_id: string, primary_keys: Record<string, any>[], options?: {
+        batch_num_items?: number;
         attribute_names?: string[];
         is_verbose?: boolean;
         is_forced?: boolean;

package/dist/src/UtilsRDS.js

@@ -140,9 +140,15 @@ export class UtilsRDS {
         const item = items[0];
         return item;
     }
-    async batch_get(table_id, primary_key, options = {}) {
-        const sql = "SELECT " + (options.attribute_names === undefined ? "*" : options.attribute_names.join(", ")) + " FROM " + table_id + " WHERE " + Object.entries(primary_key).map(([key_name, values]) => key_name + " IN " + "(" + values.map(value => convert_rds_input(value)).join(", ") + ")").join(" AND ");
-        const items = await this.exec(sql, options);
+    async batch_get(table_id, primary_keys, options = {}) {
+        const batch_num_items = options.batch_num_items || 100;
+        const attribute_names = Array.from(new Set(primary_keys.map(primary_key => Object.keys(primary_key)).flat())).sort();
+        const items = [];
+        for (let i = 0; i < primary_keys.length; i += batch_num_items) {
+            const sql = "SELECT " + (options.attribute_names === undefined ? "*" : options.attribute_names.join(", ")) + " FROM " + table_id + " WHERE " + "(" + attribute_names.join(", ") + ")" + " IN " + "(" + primary_keys.map(primary_key => "(" + attribute_names.map(attribute_name => convert_rds_input(primary_key[attribute_name])).join(", ") + ")").join(", ") + ")";
+            const new_items = await this.exec(sql, options);
+            items.push(...new_items);
+        }
         return items;
     }
     async create(table_id, item, options = {}) {

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "triangle-utils",
-    "version": "1.4.156",
+    "version": "1.4.157",
     "main": "dist/src/index.js",
     "types": "dist/src/index.d.ts",
     "directories": {

package/src/UtilsRDS.ts

@@ -185,15 +185,22 @@ export class UtilsRDS {
 
     async batch_get(
         table_id : string,
-        primary_key : Record<string, any[]>,
+        primary_keys : Record<string, any>[],
         options : {
+            batch_num_items? : number,
             attribute_names? : string[],
             is_verbose? : boolean,
             is_forced? : boolean
         } = {}
     ) {
-        const sql = "SELECT " + (options.attribute_names === undefined ? "*" : options.attribute_names.join(", ")) + " FROM " + table_id + " WHERE " + Object.entries(primary_key).map(([key_name, values]) => key_name + " IN " + "(" + values.map(value => convert_rds_input(value)).join(", ") + ")").join(" AND ")
-        const items = await this.exec(sql, options)
+        const batch_num_items = options.batch_num_items || 100
+        const attribute_names = Array.from(new Set(primary_keys.map(primary_key => Object.keys(primary_key)).flat())).sort()
+        const items = []
+        for (let i = 0; i < primary_keys.length; i += batch_num_items) {
+            const sql = "SELECT " + (options.attribute_names === undefined ? "*" : options.attribute_names.join(", ")) + " FROM " + table_id + " WHERE " + "(" + attribute_names.join(", ") + ")" + " IN " + "(" + primary_keys.map(primary_key => "(" + attribute_names.map(attribute_name => convert_rds_input(primary_key[attribute_name])).join(", ") + ")" ).join(", ") + ")"
+            const new_items = await this.exec(sql, options)
+            items.push(...new_items)
+        }
         return items
     }