lib/applb/client_wrapper.rb in applb-0.1.3 vs lib/applb/client_wrapper.rb in applb-0.1.4

@@ -1,7 +1,7 @@
 require 'forwardable'
-require 'aws-sdk'
+require 'aws-sdk-elasticloadbalancingv2'
 
 module Applb
   class ClientWrapper
     extend Forwardable
     def_delegators :@client, *%i/
@@ -38,29 +38,32 @@
       results = []
       next_marker = nil
       begin
         resp = @client.describe_target_groups(*argv)
         results.push(*resp.target_groups)
+        next_marker = resp.next_marker
       end while next_marker
       results
     end
 
     def listeners(*argv)
       results = []
       next_marker = nil
       begin
         resp = @client.describe_listeners(*argv)
         results.push(*resp.listeners)
+        next_marker = resp.next_marker
       end while next_marker
       results
     end
 
     def rules(*argv)
       results = []
       next_marker = nil
       begin
         resp = @client.describe_rules(*argv)
         results.push(*resp.rules)
+        next_marker = resp.next_marker
       end while next_marker
       results
     end
 
     def load_balancer_attributes(*argv)