mongolly 0.0.3 → 0.0.4

data/bin/mongolly

@@ -23,7 +23,7 @@ module Mongolly
       db = Mongo::Connection.new( @db_config["host"], @db_config["port"] )
       db['admin'].authenticate( @db_config["user"], @db_config["pass"] )
 
-      SnapshotManager.take_snapshots( db, @config["aws_access_key_id"], @config["aws_secret_asccess_key"], @db_config["volumes"] )
+      SnapshotManager.take_snapshots( db, @config["aws_access_key_id"], @config["aws_secret_access_key"], @db_config["region"], @db_config["volumes"] )
     end
 
     desc "clean", "removes snapshots older than the given data"
@@ -34,7 +34,7 @@ module Mongolly
 
       puts " ** Cleaning snapshots older than #{max_age}"
 
-      ec2 = AWS::EC2.new(access_key_id: @config["aws_access_key_id"], secret_access_key: @config["aws_secret_asccess_key"])
+      ec2 = AWS::EC2.new(access_key_id: @config["aws_access_key_id"], secret_access_key: @config["aws_secret_access_key"]).regions[@db_config["region"]]
       ec2.snapshots.with_owner(:self).each do |snapshot|
         unless snapshot.tags[:created_at].nil? || snapshot.tags[:backup_key].nil?
           if Time.parse(snapshot.tags[:created_at]) < max_age
@@ -56,11 +56,12 @@ module Mongolly
             port: nil,
             user: nil,
             pass: nil,
+            region: nil,
             volumes: [nil]
           }
         },
         aws_access_key_id: nil,
-        aws_secret_asccess_key: nil,
+        aws_secret_access_key: nil,
       }
 
       File.open( CONFIG_PATH, "w" ) do |f|
@@ -108,7 +109,7 @@ module Mongolly
     end
 
     def valid_config?
-      %w(aws_access_key_id aws_secret_asccess_key).each do |arg|
+      %w(aws_access_key_id aws_secret_access_key).each do |arg|
         raise ArgumentError.new("#{arg} cannot be empty") if @config[arg].to_s.strip.empty?
       end
       return true

data/lib/mongolly.rb

@@ -2,5 +2,4 @@ require "mongolly/version"
 require "mongolly/snapshot_manager"
 
 module Mongolly
-  # Your code goes here...
 end

data/lib/mongolly/snapshot_manager.rb

@@ -3,14 +3,23 @@ require 'aws-sdk'
 module Mongolly
   class SnapshotManager
 
-    def self.take_snapshots(db, aws_key_id, aws_secret_key, volume_ids)
+    def self.take_snapshots(db, aws_key_id, aws_secret_key, region, volume_ids)
+      profile_levels = {}
+
       unless db.locked?
         puts " ** Locking Database"
+        db.database_names.each do |db_name|
+          level = db[db_name].profiling_level
+          if level != :off
+            profile_levels[db_name] = level
+            db[db_name].profiling_level = :off
+          end
+        end
         db.lock!
       end
 
       begin
-        ec2 = AWS::EC2.new(access_key_id: aws_key_id, secret_access_key: aws_secret_key)
+        ec2 = AWS::EC2.new(access_key_id: aws_key_id, secret_access_key: aws_secret_key).regions[region]
         backup_key = (0...8).map{65.+(rand(25)).chr}.join
 
         puts " ** Starting Snapshot with key #{backup_key}"
@@ -28,6 +37,13 @@ module Mongolly
         if db.locked?
           puts " ** Unlocking Database"
           db.unlock!
+          db.database_names.each do |db_name|
+            level = profile_levels[db_name]
+            unless level.nil? || level == :off
+              puts " ** Setting #{db_name} profile level to #{level.to_s}"
+              db[db_name].profiling_level = level
+            end
+          end
         end
       end
     end

data/lib/mongolly/version.rb

@@ -1,3 +1,3 @@
 module Mongolly
-  VERSION = "0.0.3"
+  VERSION = "0.0.4"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: mongolly
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.4
   prerelease: 
 platform: ruby
 authors:
@@ -107,7 +107,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: -2926264446981327269
+      hash: 3409233611429320048
 required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
   requirements:
@@ -116,7 +116,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: -2926264446981327269
+      hash: 3409233611429320048
 requirements: []
 rubyforge_project: 
 rubygems_version: 1.8.24