Make sure ArResolver works for other adapter with proxy

tsukasaoishi · Feb 24, 2021 · 66bc23f · 66bc23f
1 parent cb446dc
commit 66bc23f
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 1 deletion.
diff --git a/lib/fresh_connection/extend/adapters/other_adapter_proxy.rb b/lib/fresh_connection/extend/adapters/other_adapter_proxy.rb
@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+module FreshConnection
+  module Extend
+    module OtherAdapterProxy
+      private
+
+      def __change_connection
+        yield
+      end
+    end
+  end
+end
diff --git a/lib/fresh_connection/extend/ar_resolver.rb b/lib/fresh_connection/extend/ar_resolver.rb
@@ -14,7 +14,12 @@ def spec(*args)
           require 'fresh_connection/extend/adapters/pg_adapter'
           __extend_adapter_by_fc(::ActiveRecord::ConnectionAdapters::PostgreSQLAdapter, PgAdapter)
         else
-          raise NotImplementedError, "This adapter('#{specification.config[:adapter]}') is not supported. If you specified the mysql or postgres adapter, it's probably a bug in FreshConnection. Please teach me (https://github.com/tsukasaoishi/fresh_connection/issues/new)"
+          require 'fresh_connection/extend/adapters/other_adapter_proxy'
+          base_adapter_name = "#{specification.config[:adapter].to_s}_adapter".camelize
+          __extend_adapter_by_fc(
+            "::ActiveRecord::ConnectionAdapters::#{base_adapter_name}".constantize,
+            OtherAdapterProxy
+          )
         end
 
         specification