activerecord_archive 1.0.3 → 1.0.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ec6caa129b25d19c6f0f399c91509018130b38f4bbed884de7efebf7df1d5137
-  data.tar.gz: f02746ec310b8ad11af8a08e2c04a4b5ee023b442e194c7d236be8ae4134bec4
+  metadata.gz: 2d3c010163dccd7143c45dbaf0680598ea4cc5dc688c389bd174d75ebaa9ae36
+  data.tar.gz: 2a495578d9ec68a760283ee2f40660037b433d112024818c5e6dc03292a961af
 SHA512:
-  metadata.gz: dcb76d080936e53506cadf4e2d2353d10072ab8472c6d2b8d861fc5829e1a690a1e3e2fdee0b0d67c34cdd9eeab8c01d7ef87b03f98033fdcf7054ba52301a87
-  data.tar.gz: 914075c3c87ae28ede48f1500b226cf7d213a355fd836c59a0c9e6962fe50b6ff46d123daa1b4e2934157f27a536012a0939d5deef5507fa0e1243acbf1350d1
+  metadata.gz: fb595069da61e90ff49e74a29cfa3d04eb18460d7be9d1826253e4c1cfef29120c942dd28e6a02a9212e6f6f110fef985fab7ec5a3bff30d0d1a5412582a3e12
+  data.tar.gz: 77f0556795226e4204c40e844a1d788962d4f02468e07627bc0d85c118e984808ec121492309eaf03c538ac802dd96ddeee869b0e7c14ecb82826fd3270918e4

data/lib/activerecord_archive.rb

@@ -30,6 +30,18 @@ module ArchiveMethods # :nodoc:
         SELECT * FROM #{tabname} WHERE #{conditions}
     ")
 
+    if options[:recursive_foreign_key] && !options[:recursive_foreign_key].blank?
+      # for self-referencing foreign keys, delete child records first
+      # delete only records in parent table where ids match those in archive table
+      ActiveRecord::Base.connection.execute("
+        DELETE FROM #{tabname}
+        WHERE #{tabname}.#{options[:recursive_foreign_key]} IS NOT NULL AND EXISTS(
+          SELECT #{options[:prefix]}#{tabname}.id
+          FROM #{options[:prefix]}#{tabname}
+          WHERE #{options[:prefix]}#{tabname}.id = #{tabname}.id)
+      ")
+    end
+
     # delete only records in parent table where ids match those in archive table
     ActiveRecord::Base.connection.execute("
       DELETE FROM #{tabname}
@@ -60,11 +72,25 @@ module ArchiveMethods # :nodoc:
       WHERE #{conditions}
     ")
 
-    # use replace into in case of duplicate inserts
-    ActiveRecord::Base.connection.execute("
+    if options[:recursive_foreign_key] && !options[:recursive_foreign_key].blank?
+      # use replace into in case of duplicate inserts
+      # for self-referencing foreign keys, insert parent records first
+      ActiveRecord::Base.connection.execute("
+      REPLACE INTO #{tabname}
+        SELECT * FROM #{options[:prefix]}#{tabname} WHERE (#{conditions}) AND #{options[:prefix]}#{tabname}.#{options[:recursive_foreign_key]} IS NULL
+      ")
+      # now insert child records
+      ActiveRecord::Base.connection.execute("
+      REPLACE INTO #{tabname}
+        SELECT * FROM #{options[:prefix]}#{tabname} WHERE (#{conditions}) AND #{options[:prefix]}#{tabname}.#{options[:recursive_foreign_key]} IS NOT NULL
+      ")
+    else
+      # use replace into in case of duplicate inserts
+      ActiveRecord::Base.connection.execute("
       REPLACE INTO #{tabname}
         SELECT * FROM #{options[:prefix]}#{tabname} WHERE #{conditions}
-    ")
+      ")
+    end
 
     # delete only records in archive table where ids match those in parent table
     ActiveRecord::Base.connection.execute("

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: activerecord_archive
 version: !ruby/object:Gem::Version
-  version: 1.0.3
+  version: 1.0.4
 platform: ruby
 authors:
 - Vic Spanner