appshot 0.0.3 → 0.0.4

data/lib/appshot.rb

@@ -52,6 +52,14 @@ class Appshot
 		@callables << Appshot::Mysql.new(args)
 	end
 
+	def mongodb(args={})
+		@callables << Appshot::MongoDB.new(args)
+	end
+
+	def redis(args={})
+		@callables << Appshot::Redis.new(args)
+	end
+
 	def ebs_prune(args={})
     @callables << Appshot::EBS_Prune.new(args)
 	end

data/lib/appshot/app/mongodb.rb

@@ -0,0 +1,25 @@
+class Appshot
+  class MongoDB
+
+    def initialize(opts={})
+      @port         = opts[:port]             || 27017
+      @mongo_binary = opts[:database_command] || "mongo"
+    end
+
+    def call(call_chain)
+      next_action = call_chain.shift
+      lock
+      next_action.call(call_chain) unless next_action.nil?
+    ensure
+      unlock
+    end
+
+    def lock
+      %x[#{@mongo_binary} --host localhost:#{@port} --eval "db.runCommand({ fsync:1, lock:1 })" admin ]
+    end
+
+    def unlock
+      %x[#{@mongo_binary} --host localhost:#{@port} --eval "db.$cmd.sys.unlock.findOne()" admin ]
+    end
+  end
+end

data/lib/appshot/app/mysql.rb

@@ -19,6 +19,7 @@ class Appshot
 
       lock
       next_action.call(call_chain) unless next_action.nil?
+    ensure
       unlock
     end
 

data/lib/appshot/app/redis.rb

@@ -0,0 +1,19 @@
+class Appshot
+  class Redis
+
+    def initialize(opts={})
+      @save_before_snapshot = opts[:save_before_snapshot] || false
+      @redis_binary         = opts[:database_command]     || "redis-cli"
+    end
+
+    def call(call_chain)
+      next_action = call_chain.shift
+      invoke_save if @save_before_snapshot
+      next_action.call(call_chain) unless next_action.nil?
+    end
+
+    def invoke_save
+      %x[ #{redis_binary} bgsave ]
+    end
+  end
+end

data/lib/appshot/version.rb

@@ -1,3 +1,3 @@
 class Appshot
-  VERSION = "0.0.3"
+  VERSION = "0.0.4"
 end

data/spec/lib/appshot/app/mongodb_spec.rb

@@ -0,0 +1,37 @@
+require 'appshot/app/mongodb'
+
+describe Appshot::MongoDB do
+  it { should be_a_kind_of Appshot::MongoDB }
+  it { should respond_to(:call) }
+
+  let(:callable1) { Appshot::MongoDB.new }
+  let(:callable2) { Appshot::MongoDB.new }
+  let(:items) { [ callable1, callable2 ] }
+  let(:mongo) { Appshot::MongoDB.new }
+
+  describe "#call" do
+    before do
+      Appshot::MongoDB.any_instance.stub(:lock).and_return(true)
+      Appshot::MongoDB.any_instance.stub(:unlock).and_return(true)
+    end
+    it "should require an argument to #call" do
+      lambda { mongo.call }.should raise_error(ArgumentError)
+    end
+    it "should invoke #call on the next object in the call_chain" do
+      callable1.should_receive(:call).with(items)
+      mongo.call(items)
+    end
+
+    it "should run #lock before call is invoked" do
+      callable1.should_receive(:call).with([callable2])
+      mongo.should_receive(:lock)
+      mongo.call(items)
+    end
+
+    it "should run #unlock after call is invoked" do
+      callable1.should_receive(:call).with([callable2])
+      mongo.should_receive(:unlock)
+      mongo.call(items)
+    end
+  end
+end

data/spec/lib/appshot/app/mysql_spec.rb

@@ -2,36 +2,39 @@ require 'appshot/app/mysql'
 require 'mysql2'
 
 describe Appshot::Mysql do
-  subject { Appshot::Mysql.new }
   it { should be_a_kind_of Appshot::Mysql }
-  it { subject.respond_to?(:call).should be_true }
+  it { should respond_to(:call) }
+
+  let(:mysql) { Appshot::Mysql.new }
 
   describe "#call" do
     before do
       Mysql2::Client.stub(:new).and_return(mysql)
+      Appshot::Mysql.any_instance.stub(:lock).and_return(true)
+      Appshot::Mysql.any_instance.stub(:unlock).and_return(true)
     end
-    let(:mysql) { double(:mysql2, query: true, close: true) }
+
+    let(:callable1) { Appshot::Mysql.new }
+    let(:callable2) { Appshot::Mysql.new }
+    let(:items) { [ callable1, callable2 ] }
 
     it "should require an argument to #call" do
-      lambda { subject.call }.should raise_error(ArgumentError)
+      lambda { mysql.call }.should raise_error(ArgumentError)
     end
 
     it "should invoke #call on the next object in the call_chain" do
-      items = [double(:appshot1), double(:appshot2)]
-      items.first.should_receive(:call).with(items).and_return(true)
-      subject.call(items)
+      mysql.should_receive(:call).with(items).and_return(true)
+      mysql.call(items)
     end
 
     it "should run #lock before call is invoked" do
-      items = [Appshot::Mysql.new, Appshot::Mysql.new]
-      items.first.should_receive(:lock)
-      subject.call(items)
+      mysql.should_receive(:lock)
+      mysql.call(items)
     end
 
     it "should run #unlock after call is invoked" do
-      items = [Appshot::Mysql.new, Appshot::Mysql.new]
-      items.first.should_receive(:unlock)
-      subject.call(items)
+      mysql.should_receive(:unlock)
+      mysql.call(items)
     end
   end
 end

data/spec/lib/appshot/app/redis_spec.rb

@@ -0,0 +1,43 @@
+require 'appshot/app/redis'
+
+describe Appshot::Redis do
+  it { should be_a_kind_of Appshot::Redis }
+  it { should respond_to(:call) }
+
+  let(:redis) { Appshot::Redis.new }
+
+  describe "#call" do
+    before do
+      Appshot::Redis.any_instance.stub(:invoke_save).and_return(true)
+    end
+    let(:callable1) { Appshot::Redis.new }
+    let(:callable2) { Appshot::Redis.new }
+    let(:items) { [ callable1, callable2 ] }
+
+
+    it "should require an argument to #call" do
+      lambda { redis.call }.should raise_error(ArgumentError)
+    end
+
+    it "should invoke #call on the next object in the call_chain" do
+      redis.should_receive(:call).with(items).and_return(true)
+      redis.call(items)
+    end
+
+    describe "with save_before_snapshot" do
+      context "set to false" do
+        it "should not run #invoke_save" do
+          redis.should_not_receive(:invoke_save)
+          redis.call(items)
+        end
+      end
+      context "set to true" do
+        let(:redis) { Appshot::Redis.new(save_before_snapshot: true) }
+        it "should run #invoke_save" do
+          redis.should_receive(:invoke_save)
+          redis.call(items)
+        end
+      end
+    end
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: appshot
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.4
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-08-08 00:00:00.000000000 Z
+date: 2012-08-17 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: fog
@@ -244,7 +244,9 @@ files:
 - features/support/env.rb
 - lib/appshot.rb
 - lib/appshot/app.rb
+- lib/appshot/app/mongodb.rb
 - lib/appshot/app/mysql.rb
+- lib/appshot/app/redis.rb
 - lib/appshot/filesystem.rb
 - lib/appshot/filesystem/dm.rb
 - lib/appshot/version.rb
@@ -253,7 +255,9 @@ files:
 - lib/appshot/volume/ebs_snapshot.rb
 - lib/appshot/volume/ebs_volume.rb
 - rvmrc.example
+- spec/lib/appshot/app/mongodb_spec.rb
 - spec/lib/appshot/app/mysql_spec.rb
+- spec/lib/appshot/app/redis_spec.rb
 - spec/lib/appshot/filesystem/dm_spec.rb
 - spec/lib/appshot/volume/ebs_prune_spec.rb
 - spec/lib/appshot/volume/ebs_snapshot_spec.rb
@@ -274,7 +278,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: 4374607639784639822
+      hash: -2737267038638303815
 required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
   requirements:
@@ -283,7 +287,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: 4374607639784639822
+      hash: -2737267038638303815
 requirements: []
 rubyforge_project: 
 rubygems_version: 1.8.24
@@ -294,7 +298,9 @@ test_files:
 - features/appshot.feature
 - features/step_definitions/appshot_steps.rb
 - features/support/env.rb
+- spec/lib/appshot/app/mongodb_spec.rb
 - spec/lib/appshot/app/mysql_spec.rb
+- spec/lib/appshot/app/redis_spec.rb
 - spec/lib/appshot/filesystem/dm_spec.rb
 - spec/lib/appshot/volume/ebs_prune_spec.rb
 - spec/lib/appshot/volume/ebs_snapshot_spec.rb