db-rotator 0.0.3 → 0.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 950b76e1099d543a0d7e7ea90ef9f5821dd8b11d
-  data.tar.gz: b51297f8b370983ec0445621d1e2f8f39c2b3334
+  metadata.gz: 548ebd4b7086d44bccfe3f6640b4dc4ac90c9b77
+  data.tar.gz: 8e64b24dcfb012bd525512314d69cdf716c5bfff
 SHA512:
-  metadata.gz: 7d6ab0d69db0a1c288b82b8ae5b1468a6aee4e75ebe58f7418ac7b186534e6b81ef8307f3ae83cfb5234d7df14aa959969a0d84ceb2e8334ee37eb6ff8858e2a
-  data.tar.gz: e449a83884b4a2441816db1388e8493afbaab46c612d592ea9c30b9c5208f6f01468dcdade65d4172ab59bfdba8eb9d1bfa1ef43c6a782137e19db12fb4485ca
+  metadata.gz: 6a98adb179de802fcabd20a9310a11ea513a99c2ea9b66c3e31cf73e2ae98d2d1a719374949c180457f323e0a4216c1a3bb6cc4ef13ca66e64785cfdf48a87df
+  data.tar.gz: 88132fb3cf019e39d93bedeeaadb96a1b21001cd702edd2b6908d98c9e21832fe6441a85487f59631e1927659a2d280fdf1ae5dffd568b7b5f00eb389954b5af

data/README.md

@@ -60,3 +60,5 @@ Example: `scp hostname:/path/to/mysql/backups/backup_filename.sql.bz2`
 - **reasonable_diskspace** (-s). Rough estimate of temporary disk space required to import a typical dump, in GB. Ensures this amount of space is free before importing. Default: nil
 - **rails_db_yaml_path** (-y). Updates database name in your YAML file. Example: `/path/to/railsroot/config/database.yml` Default: nil
 - **rails_environments** (-e). In conjunction with -y, which rails envs to update DB name for. If passing multiple via command line, use a comma to separate, like `-e "development,staging"`. Default: ["development"]
+- **on_success** (-S). Executes COMMAND on successful completion
+- **on_failure** (-F). Executes COMMAND if there is an error

data/db-rotator.gemspec

@@ -1,6 +1,6 @@
 Gem::Specification.new do |s|
   s.name        = "db-rotator"
-  s.version     = "0.0.3"
+  s.version     = "0.1.0"
   s.summary     = "Easy MySQL database rotation and pruning"
   s.description = "Easy MySQL database rotation and pruning"
   s.authors     = ["Michael Nelson"]

data/lib/db_rotator.rb

@@ -6,12 +6,19 @@ class DBRotator
   def initialize(config)
     @config = config.config
     @schemas = []
+
     populate_schemas
+  rescue
+    on_failure $!
   end
 
   def rotate
     refresh
     update_db_yaml
+    on_success
+
+  rescue
+    on_failure $!
   end
 
   def refresh
@@ -28,6 +35,18 @@ class DBRotator
 
   private
 
+  def on_success
+    bash_exec "#{@config[:on_success]}" if @config[:on_success]
+  end
+
+  def on_failure(err)
+    if @config[:on_failure]
+      bash_exec "#{@config[:on_failure]} #{err.class.name} '#{err.message}'"
+    else
+      raise err
+    end
+  end
+
   def download_dump
     verbose_message "Downloading dump..."
     bash_exec "rm -f #{local_dump_path}"

data/lib/db_rotator_config.rb

@@ -19,6 +19,9 @@ class DBRotatorConfig
     reasonable_diskspace:   [['-s', "--minimum-diskspace [GB]", "Rough estimate of temporary disk space required to import a typical dump, in GB."], nil],
     rails_db_yaml_path:     [['-y', "--rails-db-yaml [PATH]", "Updates database name in your YAML file when given.", "Default: nil"], nil],
     rails_environments:     [['-e', "--rails-db-environments [ENV1,ENV2]", Array, "In conjunction with -y, which rails envs to update DB name for. Default: development"], ["development"]],
+    on_success:             [['-S', "--on_success [COMMAND]", "Executes command on successful completion."], nil],
+    on_failure:             [['-F', "--on_failure [COMMAND]", "Executes command if there is an error."], nil],
+
 
     config_file:            [['-f', "--config-file PATH", "Runs rotator with configuration from this .yml file."], nil],
   }

data/test/helper.rb

@@ -3,6 +3,7 @@ require_relative '../lib/db_rotator'
 require_relative '../lib/db_rotator_config'
 
 TEST_SCHEMA_PREFIX = "__minitest_"
+TEST_MYSQL_COMMAND = "mysql -u root"
 
 def dummy_config(config={})
   DBRotatorConfig.new.tap do |cfg|
@@ -10,7 +11,8 @@ def dummy_config(config={})
       db_prefix:   TEST_SCHEMA_PREFIX,
       scp_command: "cp test/fixtures/basic_dump.sql.bz2",
 
-      local_dump_destination: "/tmp"
+      local_dump_destination: "/tmp",
+      mysql_command: TEST_MYSQL_COMMAND,
     }.merge(config))
 
     cfg.add_default_values

data/test/test_db_rotator.rb

@@ -2,9 +2,9 @@ require_relative '../test/helper'
 
 describe DBRotator do
   after do
-    `mysql -B -e 'SHOW SCHEMAS;'`.split.each do |schema|
+    `#{TEST_MYSQL_COMMAND} -B -e 'SHOW SCHEMAS;'`.split.each do |schema|
       if /^#{TEST_SCHEMA_PREFIX}/.match(schema)
-        `mysql -e "DROP SCHEMA #{schema}"`
+        `#{TEST_MYSQL_COMMAND} -e "DROP SCHEMA #{schema}"`
       end
     end
   end
@@ -23,22 +23,49 @@ describe DBRotator do
 
       Time.stub(:now, t1) do
         dbr.rotate
-        `mysql -B -e 'SHOW SCHEMAS;'`.must_include(dbname_with(t1))
-        `mysql -B -e 'SHOW TABLES FROM #{dbname_with(t1)};'`.must_include("test")
+        `#{TEST_MYSQL_COMMAND} -B -e 'SHOW SCHEMAS;'`.must_include(dbname_with(t1))
+        `#{TEST_MYSQL_COMMAND} -B -e 'SHOW TABLES FROM #{dbname_with(t1)};'`.must_include("test")
       end
 
       Time.stub(:now, t2) do
         dbr.rotate
-        `mysql -B -e 'SHOW SCHEMAS;'`.must_include(dbname_with(t2))
-        `mysql -B -e 'SHOW SCHEMAS;'`.must_include(dbname_with(t1))
+        `#{TEST_MYSQL_COMMAND} -B -e 'SHOW SCHEMAS;'`.must_include(dbname_with(t2))
+        `#{TEST_MYSQL_COMMAND} -B -e 'SHOW SCHEMAS;'`.must_include(dbname_with(t1))
       end
 
       Time.stub(:now, t3) do
         dbr.rotate
-        `mysql -B -e 'SHOW SCHEMAS;'`.must_include(dbname_with(t3))
-        `mysql -B -e 'SHOW SCHEMAS;'`.must_include(dbname_with(t2))
+        `#{TEST_MYSQL_COMMAND} -B -e 'SHOW SCHEMAS;'`.must_include(dbname_with(t3))
+        `#{TEST_MYSQL_COMMAND} -B -e 'SHOW SCHEMAS;'`.must_include(dbname_with(t2))
 
-        `mysql -B -e 'SHOW SCHEMAS;'`.wont_include(dbname_with(t1))
+        `#{TEST_MYSQL_COMMAND} -B -e 'SHOW SCHEMAS;'`.wont_include(dbname_with(t1))
+      end
+    end
+  end
+
+  describe "callbacks" do
+    before do
+      `test -e /tmp/success && rm /tmp/success`
+      `test -e /tmp/failure && rm /tmp/failure`
+    end
+
+    describe "on_success" do
+      it "calls on_success when specified" do
+        dbr = DBRotator.new(dummy_config(on_success: 'touch /tmp/success'))
+        dbr.rotate
+        FileTest.exists?('/tmp/success').must_equal true
+        FileTest.exists?('/tmp/failure').must_equal false
+      end
+    end
+
+    describe "on_failure" do
+      it "calls on_failure" do
+        dbr = DBRotator.new(dummy_config(
+                                   on_failure: 'touch /tmp/failure && true',
+                                   mysql_command: 'crap command 2> /dev/null'))
+        dbr.rotate
+        FileTest.exists?('/tmp/success').must_equal false
+        FileTest.exists?('/tmp/failure').must_equal true
       end
     end
   end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: db-rotator
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.1.0
 platform: ruby
 authors:
 - Michael Nelson
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-12-05 00:00:00.000000000 Z
+date: 2014-10-08 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: minitest
@@ -65,7 +65,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.0.14
+rubygems_version: 2.1.11
 signing_key: 
 specification_version: 4
 summary: Easy MySQL database rotation and pruning