aubergine 0.0.6 → 0.0.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: b71615e6febfb431ba5dade675f228246bf019c0
-  data.tar.gz: 07ae43b9c3735d5f00450f5429e8454eb479e5af
+  metadata.gz: f8c748eb0ceea9e180717350d161488a6a2583cb
+  data.tar.gz: aae09a118f236dbe68b2fc58386b77c5b130bbda
 SHA512:
-  metadata.gz: 12a39c268d2195015d306978a81b5c9890474cef9ec672e0b9367194a2c9c08395028dbd214c3bc2e6345464d3e91a2426243fe0406e33e3cf2a6aa1f177b022
-  data.tar.gz: fd5942d4ba9c8de2a6a2ab9679474f7373d69dbfaf47dcbed25c0aaef0ea660a6fd361475116f0f5108b55120ba3bb6f626d7917a4aa07a490812de57cf1a612
+  metadata.gz: 08ed759c2782b093f9c6cd3da93adb0585ec68ba386973635eefb327b6de8ffc2233568f0fb0371faaecaa186b97ce60de4103ae34aefbe9f8510b5c84223562
+  data.tar.gz: 812ff25cc6802bc42c3c04c646e7856b893b40a8d286536d131e7f9affdae7190c726ae924e42c9bc9d2ff7161ee8290b19367ac30a52131afac26d3b0165655

data/lib/courgette/executor.rb

@@ -8,20 +8,24 @@ module Courgette
       @logger = logger
     end
 
-    def launch!
-      Parallel.each(client.devices, in_threads: @executors) do |device|
-        begin
-          status = client.update_device(device.ip, device.fetch)
-          logger.info "#{device.ip} #{status}"
-        rescue Timeout::Error
-          logger.info "#{device.ip} is unreachable (timeout)"
-        rescue Net::SSH::Disconnect
-          logger.info "#{device.ip} is unreachable (disconnect)"
-        rescue Net::SSH::AuthenticationFailed
-          logger.info "#{device.ip} credentials are incorrect"
-        rescue Errno::ECONNREFUSED
-          logger.info "#{device.ip} is unreachable (connection refused)"
-        end
+    def launch!(filter)
+      Parallel.each(client.devices.select { |device| filter.call(device) }, in_threads: @executors) do |device|
+        fetch_and_update(device)
+      end
+    end
+
+    def fetch_and_update(device)
+      begin
+        status = client.update_device(device.ip, device.fetch)
+        logger.info "#{device.ip} #{status}"
+      rescue Timeout::Error
+        logger.info "#{device.ip} is unreachable (timeout)"
+      rescue Net::SSH::Disconnect
+        logger.info "#{device.ip} is unreachable (disconnect)"
+      rescue Net::SSH::AuthenticationFailed
+        logger.info "#{device.ip} credentials are incorrect"
+      rescue Errno::ECONNREFUSED
+        logger.info "#{device.ip} is unreachable (connection refused)"
       end
     end
   end

data/lib/courgette.rb

@@ -11,7 +11,7 @@ require_relative './courgette/client.rb'
 require_relative './courgette/executor.rb'
 
 module Courgette
-  def self.run!(aubergine, token, executors = 4, logger = Logger.new(STDOUT))
-    Executor.new(Client.new(aubergine, token), executors, logger).launch!
+  def self.run!(aubergine, token, executors = 4, logger = Logger.new(STDOUT), filter = Proc.new { |device| true })
+    Executor.new(Client.new(aubergine, token), executors, logger).launch!(filter)
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: aubergine
 version: !ruby/object:Gem::Version
-  version: 0.0.6
+  version: 0.0.7
 platform: ruby
 authors:
 - Guillaume Rose