table_copy 0.0.8 → 0.0.9

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 3febd5ce84fd65678b0993f9b1eddea26b053b17
-  data.tar.gz: e0bd696e15c019f977597a32f6c8a5f50ec7aeb9
+  metadata.gz: 93757d1afc37f9c620e57c50504028503039419c
+  data.tar.gz: 61782d83b5cd3622adc3a49ad6f6a799bcec14a2
 SHA512:
-  metadata.gz: 6deff2bfac7c138bc3666e3812d51ccea182bb6fe900fd9d8eaae314c024e0f24de2963d23d34703d94883dea15bc8eac0048198874a0dcbce573b497b2564a8
-  data.tar.gz: 7245d1fe638e3effd048556dd93625754aa015fd253a40651a7107dd21e64e39afeafafe7f0f1dffe2f535b5073b5c60e9acfe91c8355192417c940c9c6eeceb
+  metadata.gz: 79613aeac02a20849882a9523a75a819b4afa2e2492ef16e229d3c51f427deda9c644cabd38deae0eafac11f6a2d0ded606100bc5c30b3d978efcb18add99686
+  data.tar.gz: 4d64aefcbaa43f0eaf07bc52844c3076b1e702191b7302d1e945b48de7abd80da3cd1d334d7d0ccdd0d57e5682997d4403e1c1c63fffc71b48e4267ff7a08a86

data/lib/table_copy/pg/destination.rb

@@ -1,15 +1,17 @@
 module TableCopy
   module PG
     class Destination
-      attr_reader :table_name, :conn_method, :indexes, :fields, :primary_key, :sequence_field
+      attr_reader :table_name, :conn_method, :indexes, :fields, :primary_key, :sequence_field, :after_create, :soft_delete_field
 
       def initialize(args)
-        @table_name     = args[:table_name]
-        @primary_key    = args[:primary_key]
-        @sequence_field = args[:sequence_field]
-        @conn_method    = args[:conn_method]
-        @indexes        = args[:indexes] || []
-        @fields         = args[:fields]
+        @table_name        = args[:table_name]
+        @primary_key       = args[:primary_key]
+        @sequence_field    = args[:sequence_field]
+        @conn_method       = args[:conn_method]
+        @indexes           = args[:indexes] || []
+        @fields            = args[:fields]
+        @after_create      = args[:after_create]
+        @soft_delete_field = args[:soft_delete_field]
       end
 
       def transaction
@@ -26,9 +28,11 @@ module TableCopy
       end
 
       def create(fields_ddl)
+        sd = ", #{soft_delete_field} bool default false" if soft_delete_field
         with_conn do |conn|
-          conn.exec("create table #{table_name} (#{fields_ddl})")
+          conn.exec("create table #{table_name} (#{fields_ddl}#{sd})")
         end
+        after_create.call(table_name) if after_create
       end
 
       def drop(opts={})
@@ -97,7 +101,11 @@ module TableCopy
 
       def delete_not_in_temp
         with_conn do |conn|
-          conn.exec("delete from #{table_name} where #{primary_key} in (select #{primary_key} from #{table_name} except select #{primary_key} from temp_#{table_name})")
+          if soft_delete_field
+            conn.exec("update #{table_name} set #{soft_delete_field}=true where #{not_in_temp} and (#{soft_delete_field} is null or #{soft_delete_field} != true)")
+          else
+            conn.exec("delete from #{table_name} where #{not_in_temp}")
+          end
         end
       end
 
@@ -123,6 +131,10 @@ module TableCopy
 
       private
 
+      def not_in_temp
+        "#{primary_key} in (select #{primary_key} from #{table_name} except select #{primary_key} from temp_#{table_name})"
+      end
+
       attr_reader :primary_key
 
       def fields_list

data/lib/table_copy/version.rb

@@ -1,3 +1,3 @@
 module TableCopy
-  VERSION = "0.0.8"
+  VERSION = "0.0.9"
 end

data/spec/db.rb

@@ -28,7 +28,11 @@ class DB
   end
 
   def create_table(name=table_name)
-    conn.exec("create table #{name} (column1 integer, column2 varchar(123), column3 varchar(256)[])")
+    conn.exec("create table #{name} (#{create_fields_ddl})")
+  end
+
+  def create_fields_ddl
+    "column1 integer, column2 varchar(123), column3 varchar(256)[]"
   end
 
   def create_view(name=view_name, t_name: table_name)

data/spec/lib/table_copy/pg/destination_spec.rb

@@ -1,4 +1,5 @@
 require 'table_copy/pg/destination'
+require 'table_copy/pg/source'
 require 'table_copy/pg/index'
 
 describe TableCopy::PG::Destination do
@@ -6,14 +7,16 @@ describe TableCopy::PG::Destination do
   let(:view_name)  { 'view_name' }
   let(:db) { DB.new(table_name: table_name, view_name: view_name) }
 
-  let(:dest) { TableCopy::PG::Destination.new(
+  let(:conf) { {
     table_name: table_name,
     conn_method: db.method(:with_conn),
     indexes: [ TableCopy::PG::Index.new(table_name, nil, ['column1']) ],
     fields: [ 'column1', 'column2', 'column3' ],
     primary_key: 'column1',
     sequence_field: 'column1'
-  )}
+  } }
+
+  let(:dest) { TableCopy::PG::Destination.new(conf) }
 
   after do
     db.drop_table
@@ -27,7 +30,7 @@ describe TableCopy::PG::Destination do
 
   context 'a table exists' do
     before do
-      db.create_table
+      dest.create(db.create_fields_ddl)
     end
 
     let(:expected_view) {
@@ -280,12 +283,28 @@ describe TableCopy::PG::Destination do
           db.insert_data
         end
 
-        it 'deletes row that are not in the temp table' do
-          expect {
-            dest.delete_not_in_temp
-          }.to change {
-            db.row_count
-          }.from(1).to(0)
+        context 'no soft delete' do
+          it 'deletes row that are not in the temp table' do
+            expect {
+              dest.delete_not_in_temp
+            }.to change {
+              db.row_count
+            }.from(1).to(0)
+          end
+        end
+
+        context 'with soft delete' do
+          let(:soft_delete_field) { 'is_soft_deleted' }
+          let(:conf2) { conf.merge(soft_delete_field: soft_delete_field) }
+          let(:dest)  { TableCopy::PG::Destination.new(conf2) }
+
+          it 'sets the field to true' do
+            expect {
+              dest.delete_not_in_temp
+            }.to change {
+              db.with_conn { |c| c.exec("select #{soft_delete_field} from #{table_name}").first[soft_delete_field]}
+            }.from('f').to('t')
+          end
         end
       end
     end
@@ -299,6 +318,17 @@ describe TableCopy::PG::Destination do
         db.table_exists?
       }.from(false).to(true)
     end
+
+    context 'an after_create proc is specified' do
+      let(:after_create) { double }
+      let(:conf2) { conf.merge(after_create: after_create) }
+      let(:dest) { TableCopy::PG::Destination.new(conf2) }
+
+      it 'calls the after_create proc, passing the table_name' do
+        expect(after_create).to receive(:call).with(dest.table_name)
+        dest.create('column1 integer')
+      end
+    end
   end
 
   describe '#create_temp' do

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: table_copy
 version: !ruby/object:Gem::Version
-  version: 0.0.8
+  version: 0.0.9
 platform: ruby
 authors:
 - TLH
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-10-07 00:00:00.000000000 Z
+date: 2014-10-22 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler