mongar 0.0.4 → 0.0.5

data/README.rdoc

@@ -52,6 +52,10 @@ Run it with rails runner (rails 3.x) or script/runner (rails 2.x) script/mongar.
     end
   
     replicate Client do
+      db_time_selector do
+        current_time_on_db_server
+      end
+    
       column :id do
         primary_index
       end
@@ -84,7 +88,9 @@ Run it with rails runner (rails 3.x) or script/runner (rails 2.x) script/mongar.
   end
 
 == Contributing to mongar
- 
+
+Find this project on Pivotal Tracker here: https://www.pivotaltracker.com/projects/434475
+
 * Check out the latest master to make sure the feature hasn't been implemented or the bug hasn't been fixed yet
 * Check out the issue tracker to make sure someone already hasn't requested it and/or contributed it
 * Fork the project

data/VERSION

@@ -1 +1 @@
-0.0.4
+0.0.5

data/lib/mongar/replica.rb

@@ -4,7 +4,7 @@ class Mongar
     
     attr_accessor :source, :destination, :log_level
     attr_accessor :mongodb_name
-    attr_accessor :created_finder, :deleted_finder, :updated_finder
+    attr_accessor :created_finder, :deleted_finder, :updated_finder, :db_time_selector
     attr_accessor :columns
     
     def initialize(args = {})
@@ -24,11 +24,12 @@ class Mongar
       end
       self.updated_finder = Proc.new do |last_replicated_at|
         find(:all, :conditions => ["updated_at > ? AND deleted_at IS NULL", last_replicated_at])
-      end 
+      end
     end
     
     def run
-      time = Time.now
+      time = current_time_on_database_server
+      
       if do_full_refresh?
         info "Running full refresh on Replica #{source.to_s} to #{destination.name}"
         
@@ -149,5 +150,30 @@ class Mongar
         self.send("#{finder_type}_finder=".to_sym, nil)
       end
     end
+    
+    def db_time_selector &block
+      return @db_time_selector unless block_given?
+      
+      @db_time_selector = block
+    end
+    
+    def default_time_selector(object)
+      adapter = object.connection.class.to_s
+      adapter = $1 if adapter =~ /::([^:]+)$/
+      
+      time = if adapter == 'MysqlAdapter'
+        Time.parse(object.connection.execute("SELECT UTC_TIMESTAMP()").fetch_row.first)
+      elsif adapter == 'Mysql2Adapter'
+        object.connection.execute("SELECT UTC_TIMESTAMP()").first.first
+      elsif adapter == 'SQLServerAdapter'
+        object.connection.select_one("SELECT getutcdate() AS date")['date']
+      end
+      
+      time.is_a?(Time) ? time : nil
+    end
+    
+    def current_time_on_database_server
+      @db_time_selector.nil? ? default_time_selector(source) : source.instance_exec(&@db_time_selector)
+    end
   end
 end

data/mongar.gemspec

@@ -5,11 +5,11 @@
 
 Gem::Specification.new do |s|
   s.name = "mongar"
-  s.version = "0.0.4"
+  s.version = "0.0.5"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["Philippe Green"]
-  s.date = "2011-12-14"
+  s.date = "2011-12-15"
   s.description = "Replicates data from ActiveRecord (or other Ruby data mapping class) to MongoDB"
   s.email = "phil@greenviewdata.com"
   s.extra_rdoc_files = [

data/spec/fixtures/sources.rb

@@ -1,7 +1,17 @@
+class Mysql2Adapter
+  def execute(sql)
+    [[Time.now.utc]]
+  end
+end
+
 class Domain
   attr_accessor :name, :client_id
   attr_accessor :created_at, :updated_at, :deleted_at
   
+  def self.connection
+    return Mysql2Adapter.new
+  end
+  
   def initialize(args = {})
     args.each do |key, value|
       self.send(:"#{key}=", value)

data/spec/mongar/replica_spec.rb

@@ -26,6 +26,8 @@ describe "Mongar::Replica" do
       @mongo = Mongar::Mongo.new
       Mongar::Mongo.databases[:default] = @mongo
       
+      @replica.stub!(:default_time_selector).and_return(@time)
+      
       @last_replicated_time = Time.now - 86400
       @collection.stub!(:last_replicated_at).and_return(@last_replicated_time)
       
@@ -60,6 +62,8 @@ describe "Mongar::Replica" do
       before do
         @time = Time.parse("1/1/2011 00:00:00")
         Time.stub!(:now).and_return(@time)
+        @replica.stub!(:default_time_selector).and_return(@time)
+        
         @replica.stub!(:do_full_refresh?).and_return(false)
         
         @deleted_client1 = Client.new(:name => "Widget Co", :employee_count => 500)
@@ -370,4 +374,97 @@ describe "Mongar::Replica" do
       @replica.mongodb.should == @fake_db
     end
   end
+  
+  describe "#db_time_selector" do
+    before do
+      @block = lambda {}
+    end
+    
+    it "should default to nil" do
+      @replica.db_time_selector.should be_nil
+    end
+    
+    it "should set the time sql to something else given an argument" do
+      @replica.db_time_selector &@block
+      @replica.db_time_selector.should == @block
+    end
+  end
+  
+  describe "#current_time_on_database_server" do
+    before do
+      class Client; end
+      @replica.source = Client
+      @time = Time.now
+    end
+    context "default time selector" do
+      it "should call default_time_selector" do
+        @replica.should_receive(:default_time_selector).with(Client).and_return(@time)
+        @replica.current_time_on_database_server.should == @time
+      end
+    end
+    
+    context "custom time selector" do
+      before do
+        @replica.db_time_selector do
+          current_db_time
+        end
+      end
+      
+      it "should call the custom time selector function on the source class" do
+        Client.should_receive(:current_db_time).and_return(@time)
+        @replica.current_time_on_database_server.should == @time
+      end
+    end
+  end
+  
+  describe "#default_time_selector" do
+    before do
+      class MockAdapters
+        class MysqlAdapter
+        end
+      
+        class Mysql2Adapter
+        end
+      
+        class SQLServerAdapter
+        end
+      end
+      
+      @time = Time.now
+    end
+    
+    context "mysql server backend" do
+      before do
+        @mock_connection = MockAdapters::MysqlAdapter.new
+        @mock_source = mock(Object, :connection => @mock_connection)
+        @mock_result = mock(Object, :fetch_row => [@time.to_s])
+      end
+      it "should run SELECT UTC_TIMESTAMP()" do
+        @mock_connection.should_receive(:execute).with("SELECT UTC_TIMESTAMP()").and_return(@mock_result)
+        @replica.default_time_selector(@mock_source).to_s.should == @time.to_s
+      end
+    end
+    
+    context "mysql server backend with mysql2 adapter" do
+      before do
+        @mock_connection = MockAdapters::Mysql2Adapter.new
+        @mock_source = mock(Object, :connection => @mock_connection)
+      end
+      it "should run SELECT UTC_TIMESTAMP()" do
+        @mock_connection.should_receive(:execute).with("SELECT UTC_TIMESTAMP()").and_return([[@time]])
+        @replica.default_time_selector(@mock_source).should == @time
+      end
+    end
+  
+    context "sql server backend" do
+      before do
+        @mock_connection = MockAdapters::SQLServerAdapter.new
+        @mock_source = mock(Object, :connection => @mock_connection)
+      end
+      it "should run SELECT getutcdate() as date" do
+        @mock_connection.should_receive(:select_one).with("SELECT getutcdate() AS date").and_return({"date" => @time})
+        @replica.default_time_selector(@mock_source).should == @time      
+      end
+    end
+  end
 end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: mongar
 version: !ruby/object:Gem::Version 
-  hash: 23
+  hash: 21
   prerelease: 
   segments: 
   - 0
   - 0
-  - 4
-  version: 0.0.4
+  - 5
+  version: 0.0.5
 platform: ruby
 authors: 
 - Philippe Green
@@ -15,7 +15,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2011-12-14 00:00:00 Z
+date: 2011-12-15 00:00:00 Z
 dependencies: 
 - !ruby/object:Gem::Dependency 
   requirement: &id001 !ruby/object:Gem::Requirement