ebs_snapper 0.0.7 → 0.0.8

data/config/sample.config.yaml

@@ -5,3 +5,4 @@ aws:
   volume_tag : 'Snapper' # tag to use in AWS for the EBS backup flags
   retain : '10.days'
 log_to: '/var/log/ebs_snapper.log'
+pause_time: 1

data/lib/ebs_snapper/cli.rb

@@ -54,6 +54,7 @@ class EbsSnapper::CLI
     options[:verbose] = false
     options[:out] = ''
     options[:config] = nil
+    options[:pause_time] = 1
     
     opts = OptionParser.new do |opts|
       opts.banner = "Usage: ebs_snapper [options]"

data/lib/ebs_snapper/ebs.rb

@@ -17,6 +17,7 @@ require 'aws-sdk'
 class EbsSnapper::Ebs
   
   DEFAULT_TAG_NAME = 'Snapper'
+  DEFAULT_PAUSE_TIME = 0
   
   def initialize(opts = {})
     if !opts[:secret_access_key].nil? && !opts[:access_key_id].nil?
@@ -27,6 +28,7 @@ class EbsSnapper::Ebs
     @logger = opts[:logger] || Logger.new(STDOUT)
     @retain = opts[:retain]
     @tag_name = opts[:volume_tag] || DEFAULT_TAG_NAME # default
+    PausingEnumerable.pause_time = opts[:pause_time] || DEFAULT_PAUSE_TIME
   end
   
   def snapshot_and_purge
@@ -41,7 +43,8 @@ class EbsSnapper::Ebs
     volumes = []
     
     each_region do |r|
-      r.tags.filter('resource-type', 'volume').filter('key', @tag_name).each do |tag|
+      tags = r.tags.filter('resource-type', 'volume').filter('key', @tag_name)
+      PausingEnumerable.wrap(tags).each do |tag|
         # if the tag exists, it's using the default retention (TTL)
         ttl_value = @retain
         if tag.value != nil && !tag.value.strip.empty?
@@ -72,14 +75,14 @@ class EbsSnapper::Ebs
   end
   
   def purge_old_snapshots(ttl, region, vol_id)
-    region.snapshots.filter('volume-id', vol_id).filter('tag-key', @tag_name).each do |snapshot|
+    snapshots = region.snapshots.filter('volume-id', vol_id).filter('tag-key', @tag_name)
+    PausingEnumerable.wrap(snapshots).each do |snapshot|
       unless snapshot.status == :pending
         ts = snapshot.tags[@tag_name]
         if ttl.purge?(ts)
           @logger.info {"Purging #{vol_id} snapshot: #{snapshot.id}"}
           begin
             snapshot.delete
-            sleep 1 # we need to slow down our processing..
           rescue => e
             @logger.error "Exception: #{e}\n" + e.backtrace().join("\n")
           end
@@ -87,8 +90,37 @@ class EbsSnapper::Ebs
       end
     end
   end
-  
-  
+
+  module PausingEnumerable
+    
+    def self.pause_time=(time)
+       @pause_time = time
+    end
+    
+    def self.pause_time
+       @pause_time || EbsSnapper::Ebs::DEFAULT_PAUSE_TIME
+    end
+    
+    def self.included(base)
+      base.class_eval do
+        if !method_defined?(:orig_each)
+          alias_method :orig_each, :each
+          def each(&block)
+            orig_each(&block)
+            sleep PausingEnumerable.pause_time # we need to slow down our processing..
+          end
+        end
+      end
+    end
+    
+    def self.wrap(enumerable)
+      enumerable.class.module_eval do
+        include PausingEnumerable
+      end
+      enumerable
+    end
+  end
+
   def each_region
     ec2.regions.each do |region|
       yield (region)

data/lib/ebs_snapper/version.rb

@@ -13,5 +13,5 @@
 # limitations under the License.
 
 module EbsSnapper
-  VERSION = "0.0.7"
+  VERSION = "0.0.8"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ebs_snapper
 version: !ruby/object:Gem::Version
-  version: 0.0.7
+  version: 0.0.8
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-01-03 00:00:00.000000000 Z
+date: 2014-01-09 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: aws-sdk
@@ -111,7 +111,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: 1005220110206046927
+      hash: 529299969700256006
 required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
   requirements:
@@ -120,7 +120,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: 1005220110206046927
+      hash: 529299969700256006
 requirements: []
 rubyforge_project: 
 rubygems_version: 1.8.24