swa 0.8.3 → 0.8.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: '0863918be3963c4a4fefbf9f7d4c7f049617cf5e6b9fe53ca1d3743eace7c364'
-  data.tar.gz: d1d2972dbcfd1c90379202aabc80dbc4968c231bf27054842270d98aae5a6e00
+  metadata.gz: f97ea85d4f432b6ea61fb652ce14f5ea50f1827aa3056437869f9eb6fcd28f83
+  data.tar.gz: 00e6d4f165f754f6dc4d03d4c45c09075b7235961527d1792eec649f8475b67e
 SHA512:
-  metadata.gz: dcb11477f1210b8a6e766a0ed9b3623d60d4a7cc4b95b643cf2b634e95cb4230c024e5853e7cf4b71bc789cd006885d5ab7313275739bc7d31daf706739d2d6f
-  data.tar.gz: e7c298215ee0cf718992d8afde6b78ab1c8e00b25a82d3f8cd45615112bc544db33791350abb2698991f5b0d9e3d711fefaf8c38dcc4341c5664c3eaab326814
+  metadata.gz: da5a54d7e1ec1da3b3860cf27ae60e5548eb007679ff7158eb6c8e0ae7f82b49592c553b6920b230773302a7318d5f2d651ba2bfc1d5c7c37d9256bc51fc1aa6
+  data.tar.gz: 10134f56ec3a62439470d5f0148a8e3df4804338ff86a18c824da2795866fd1512cfab71c75f8bd9de45b1f3f5b1ab7b3c092825e4d88b6c89c47c3640bf084c

data/lib/swa/cli/athena_command.rb

@@ -148,8 +148,7 @@ module Swa
       end
 
       def query_for(query_method, response_key, model, **query_args)
-        records = athena_client.public_send(query_method, **query_args).public_send(response_key)
-        model.list(records)
+        model.list_from_query(athena_client, query_method, response_key, **query_args)
       end
 
       def output_results_as_csv(result_set)

data/lib/swa/cli/glue_command.rb

@@ -211,8 +211,7 @@ module Swa
       end
 
       def query_for(query_method, response_key, model, **query_args)
-        records = glue_client.public_send(query_method, **query_args).public_send(response_key)
-        model.list(records)
+        model.list_from_query(glue_client, query_method, response_key, **query_args)
       end
 
     end

data/lib/swa/cli/kms_command.rb

@@ -54,9 +54,8 @@ module Swa
         ::Aws::KMS::Client.new(aws_config)
       end
 
-      def query_for(query_method, response_key, model)
-        records = kms_client.public_send(query_method).public_send(response_key)
-        model.list(records)
+      def query_for(query_method, response_key, model, **query_args)
+        model.list_from_query(kms_client, query_method, response_key, **query_args)
       end
 
     end

data/lib/swa/cli/lake_formation_command.rb

@@ -59,8 +59,7 @@ module Swa
       end
 
       def query_for(query_method, response_key, model, **query_args)
-        records = lf_client.public_send(query_method, **query_args).public_send(response_key)
-        model.list(records)
+        model.list_from_query(lf_client, query_method, response_key, **query_args)
       end
 
     end

data/lib/swa/record.rb

@@ -9,6 +9,13 @@ module Swa
       records.lazy.map(&method(:new))
     end
 
+    def self.list_from_query(client, query_method, response_key, **query_args)
+      records = client.public_send(query_method, **query_args).each.lazy.flat_map { |page|
+        page.public_send(response_key)
+      }
+      list(records)
+    end
+
     def initialize(aws_record)
       @aws_record = aws_record
     end

data/lib/swa/version.rb

@@ -1,3 +1,3 @@
 module Swa
-  VERSION = "0.8.3"
+  VERSION = "0.8.4"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: swa
 version: !ruby/object:Gem::Version
-  version: 0.8.3
+  version: 0.8.4
 platform: ruby
 authors:
 - Mike Williams
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-03-21 00:00:00.000000000 Z
+date: 2024-05-09 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler