transient_record 1.0.0 → 1.0.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 9567dc6a94b5693ba7522664172ec95257734f81820307ed705d1a8a55323aef
-  data.tar.gz: 82833fbb5361c27361347d89b38fb709a102fa5be820be30710a5b1debc061de
+  metadata.gz: 62717fcb04dc95460008a291475193ad256735122cef3502311ac02d62326baf
+  data.tar.gz: c0d9ac5cd578b60eb9e6e5774886d5e16737eb773766f999b9c201d5a923471c
 SHA512:
-  metadata.gz: 2a5ce0b5c9859a9e88ee78b119891cbc8f2daaa930db25a52fb14a6e93723d1d91e850a67f99fd70b8c1ccda80a52f206bb83a529f8cd52e1e11c9e9fe89c657
-  data.tar.gz: 3c64275220d6138377788c9e294b3f67f5285801ac977e03933489a65654604248a757feeca210cd7b642c01d08941e534d510f9814c252d1d5e8885b298f4d7
+  metadata.gz: fb93249d31a827da8ef2a652e9ebe96435d93468bd392d4a228b2200ae03fef6673c0b4995970927f5aa049b4f11d16d096b23fe8f8312749fb8562f45872083
+  data.tar.gz: 36c7bf8748bee4035084b33be89871bb1eeee83f7a75d02a16f52e769bc09587d145318df1279a2932d399ce76245b76f8d9995314d41ba8cbd2a8b9218827a2

data/lib/transient_record.rb

@@ -35,7 +35,10 @@
 #   end
 module TransientRecord
   # Transient Record version number.
-  VERSION = "1.0.0"
+  VERSION = "1.0.1"
+
+  # A class representing Transient Record errors.
+  class Error < RuntimeError; end
 
   # A module where all temporary models are defined.
   #
@@ -74,6 +77,10 @@ module TransientRecord
     TransientRecord.define_model(*args, &block)
   end
 
+  # Transient table names are stored in a module instance variable, so that
+  # only transient tables can be removed in .cleanup.
+  @transient_tables = []
+
   class << self
     # Create a transient table.
     #
@@ -101,6 +108,7 @@ module TransientRecord
     # @see https://api.rubyonrails.org/classes/ActiveRecord/ConnectionAdapters/SchemaStatements.html#method-i-create_table Documentation for #create_table in Ruby on Rails
     def create_table table_name, options = {}, &block
       table_name = table_name.to_sym
+      @transient_tables << table_name
 
       ::ActiveRecord::Migration.suppress_messages do
         ::ActiveRecord::Migration.create_table table_name, **options, &block
@@ -178,7 +186,7 @@ module TransientRecord
       Models.remove_all_consts
 
       connection = ::ActiveRecord::Base.connection
-      tables_to_remove = connection.tables
+      tables_to_remove = @transient_tables
       drop_attempts = tables_to_remove.count * (1 + tables_to_remove.count) / 2
 
       drop_attempts.times do
@@ -194,6 +202,12 @@ module TransientRecord
         end
       end
 
+      if !@transient_tables.empty?
+        raise Error.new(<<~ERROR)
+          The following transient tables could not be removed: #{@transient_tables.join(', ')}.
+        ERROR
+      end
+
       GC.start
 
       nil

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: transient_record
 version: !ruby/object:Gem::Version
-  version: 1.0.0
+  version: 1.0.1
 platform: ruby
 authors:
 - Greg Navis
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-01-24 00:00:00.000000000 Z
+date: 2023-01-25 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activerecord