MikeSofaer-saxual-replication 0.0.3 → 0.0.4

data/lib/saxual-replication.rb

@@ -39,6 +39,10 @@ module SAXualReplication
       @tag = value
     end
 
+    def key_column(value)
+      @key_column = value
+    end
+
     def datamapper_class
       klass = self.dup
       klass.send(:include, DataMapper::Resource)
@@ -60,8 +64,10 @@ module SAXualReplication
     end
 
     def sql(rows)
-      "INSERT INTO "+ @table_name + "(" + column_names.join(', ') + ", created_at, updated_at) VALUES " +
+      _sql = "INSERT INTO "+ @table_name + "(" + column_names.join(', ') + ", created_at, updated_at) VALUES " +
         ([("(" + (["?"] * (column_names.size + 2)).join(',') + ")")] * rows.size).join(',')
+      _sql << duplicate_key_clause if @key_column
+      _sql
     end
     def bind_values(rows)
       names = column_names
@@ -70,17 +76,20 @@ module SAXualReplication
       rows.each{|row| row.add_bind_values!(names, array, datetime)}
       array
     end
+    def duplicate_key_clause
+      " ON DUPLICATE KEY UPDATE " + (column_names - [:created_at, @key_column]).map {|c| c.to_s + "=VALUES(" + c.to_s + ")"}.join(', ')
+    end
 
     def save(rows)
       connection.execute sql(rows), *bind_values(rows)
     end
   end
 
-
- #+ " ON DUPLICATE KEY UPDATE " + update_keys.join(', ')
-
   def add_bind_values!(column_names, bind_array, datetime)
-    column_names.each{|c| bind_array << self.send(c)}
+    column_names.each do |c|
+      val = self.send(c)
+      bind_array << ((self.class.data_class(c) == DateTime && val) ? (DateTime.parse(val)) : val)
+    end
     bind_array << datetime << datetime
   end
 

data/spec/saxual-replication/saxual_replication_spec.rb

@@ -71,6 +71,19 @@ describe "SAXualReplication" do
           @klass.datamapper_class.all[1].title.should == "Someone's Cat"
         end
       end
+      describe "replication" do
+        it "should update fields on rows with a repeated primary key" do
+          @klass.key_column :written_on
+          @adapter.execute "create unique index key_column on documents(written_on)"
+          t = DateTime.now.to_s
+          xml1 = "<document><title>Hello, Everyone!</title><written_on>#{t}</written_on></document>"
+          xml2 = "<document><title>Someone's Cat</title><written_on>#{t}</written_on></document>"
+          @klass.save [@klass.parse(xml1)]
+          @klass.save [@klass.parse(xml2)]
+          @klass.datamapper_class.all.size.should == 1
+          @klass.datamapper_class.all[0].title.should == "Someone's Cat"
+        end
+      end
     end
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: MikeSofaer-saxual-replication
 version: !ruby/object:Gem::Version 
-  version: 0.0.3
+  version: 0.0.4
 platform: ruby
 authors: 
 - Michael Sofaer