elasticsearch-drain 0.1.9 → 0.1.10

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 833e9ccb66eeabd7ec9e3ed2f193ec2dc389de6d
-  data.tar.gz: 42edac2e4d75dc0254b996dea328a8188a5c0d6e
+  metadata.gz: d6f8cdf47026cb2eb68de833e51f92ead6d2a0ad
+  data.tar.gz: db0f685a2028a42f597296ed89e6a5fb94a041da
 SHA512:
-  metadata.gz: 0a5e9b266c549472492101d76961441a46bd023868649ca640ba069bae82a36902c4cfb3d6b12b6a168f4f8a0c851ca8c1dbfdc1051d67c1bc408f3f550722d6
-  data.tar.gz: ba5815bfa7eba8f2b86af97072d9e4f89add797d39f4116ec22aaa1bfefe0067663a699f5dc542c544d9ab5d24dc592c65a57e57f52b0576d1915566ca7b10f8
+  metadata.gz: edd8562747b105ee3a7f2b6379224ed72aea324702828f92690a740265fd495d2fd18e8ba1fa474ea31ad071270438d55166f7aff0d42ce1f287bcb09d8ebf5e
+  data.tar.gz: 2e02dab0dea20881a1d87a2b28ae539aedf845110dc416a51f8f50fc395601a5a41799e841c1ee4ba3dd310219d652f3044593070004fa3f6c072f20864d78ec

data/lib/elasticsearch/drain/cli.rb

@@ -76,10 +76,12 @@ module Elasticsearch
           @active_nodes = nodes unless options[:continue]
 
           drain_nodes(nodes)
-          remove_nodes(nodes)
+          deleted_nodes = remove_nodes(nodes)
 
           # Remove the drained nodes from the list of active_nodes
-          @active_nodes -= nodes
+          deleted_nodes.each do |deleted_node|
+            active_nodes.delete_if { |n| n.id == deleted_node.id }
+	  end
 
           unless active_nodes.empty?
             say_status 'Drain Nodes', "#{active_nodes.length} nodes remaining", :green
@@ -142,6 +144,7 @@ module Elasticsearch
         end
 
         def remove_nodes(nodes) # rubocop:disable Metrics/MethodLength
+          deleted_nodes = []
           while nodes.length > 0
             sleep_time = wait_sleep_time
             nodes.each do |instance|
@@ -151,6 +154,7 @@ module Elasticsearch
                 sleep sleep_time
               else
                 next unless remove_node(instance)
+                deleted_nodes.push(nodes.find { |n| n.ipaddress == instance.ipaddress })
                 nodes.delete_if { |n| n.ipaddress == instance.ipaddress }
                 break if nodes.length < 1
                 say_status 'Waiting', 'Sleeping for 1 minute before removing the next node', :green
@@ -158,6 +162,7 @@ module Elasticsearch
               end
             end
           end
+          deleted_nodes
         end
 
         def remove_node(instance) # rubocop:disable Metrics/MethodLength

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: elasticsearch-drain
 version: !ruby/object:Gem::Version
-  version: 0.1.9
+  version: 0.1.10
 platform: ruby
 authors:
 - Andrew Thompson
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2020-08-20 00:00:00.000000000 Z
+date: 2020-08-21 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler