More reliably determine database adapter

derrickreimer · Nov 20, 2015 · 0f308e0 · 0f308e0
1 parent a506f9c
commit 0f308e0
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/lib/sequenced/generator.rb b/lib/sequenced/generator.rb
@@ -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