Replaced remaining instances of connection

ankane · Oct 7, 2024 · 5db4b3f · 5db4b3f
1 parent 29189ab
commit 5db4b3f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/lib/neighbor/model.rb b/lib/neighbor/model.rb
@@ -154,7 +154,7 @@ def self.neighbor_attributes
               if operator == "BIT_COUNT"
                 "BIT_COUNT(#{quoted_attribute} ^ #{query})"
               else
-                "DISTANCE(#{quoted_attribute}, #{query}, #{connection.quote(operator)})"
+                "DISTANCE(#{quoted_attribute}, #{query}, #{connection_pool.with_connection { |c| c.quote(operator) }})"
               end
             else
               if operator == "#"

diff --git a/lib/neighbor/utils.rb b/lib/neighbor/utils.rb
@@ -50,7 +50,7 @@ def self.adapter(model)
       when /sqlite/i
         :sqlite
       when /mysql|trilogy/i
-        model.connection.try(:mariadb?) ? :mariadb : :mysql
+        model.connection_pool.with_connection { |c| c.try(:mariadb?) } ? :mariadb : :mysql
       else
         :postgresql
       end