More reliably determine database adapter

derrickreimer · Nov 20, 2015 · b98beaa · b98beaa
1 parent a506f9c
commit b98beaa
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/lib/sequenced/generator.rb b/lib/sequenced/generator.rb
@@ -12,7 +12,7 @@ def initialize(record, options = {})
 
     def set
       return if id_set? || skip?
-      lock_table(record.class)
+      lock_table
       record.send(:"#{column}=", next_id)
     end
 
@@ -46,14 +46,14 @@ def unique?(id)
 
   private
 
-    def lock_table(klass)
+    def lock_table
       if postgresql?
-        klass.connection.execute("LOCK TABLE #{klass.table_name} IN EXCLUSIVE MODE")
+        record.class.connection.execute("LOCK TABLE #{record.class.table_name} IN EXCLUSIVE MODE")
       end
     end
 
     def postgresql?
-      ActiveRecord::Base.configurations[Rails.env]['adapter'] == 'postgresql'
+      record.class.connection.instance_of?(ActiveRecord::ConnectionAdapters::PostgreSQLAdapter)
     end
 
     def base_relation