Merge pull request #6 from fac/fix/connection-flags-passthrough

Fix: Ensure connection flags are supported

Author: scottclk

Gemfile.lock

@@ -1,68 +1,78 @@
 PATH
   remote: .
   specs:
-    mysql2-aurora (0.5.2.1)
-      mysql2 (~> 0.5.2)
+    mysql2-aurora (0.5.6)
+      mysql2 (~> 0.5.6)
 
 GEM
   remote: https://rubygems.org/
   specs:
-    ast (2.4.0)
-    coderay (1.1.2)
+    ast (2.4.2)
+    coderay (1.1.3)
     coveralls (0.8.23)
       json (>= 1.8, < 3)
       simplecov (~> 0.16.1)
       term-ansicolor (~> 1.3)
       thor (>= 0.19.4, < 2.0)
       tins (~> 1.6)
-    diff-lcs (1.3)
-    docile (1.3.1)
-    jaro_winkler (1.5.2)
-    json (2.6.2)
-    method_source (0.9.2)
-    mysql2 (0.5.2)
-    parallel (1.17.0)
-    parser (2.6.3.0)
-      ast (~> 2.4.0)
-    pry (0.12.2)
-      coderay (~> 1.1.0)
-      method_source (~> 0.9.0)
-    rainbow (3.0.0)
-    rake (13.0.6)
-    rspec (3.8.0)
-      rspec-core (~> 3.8.0)
-      rspec-expectations (~> 3.8.0)
-      rspec-mocks (~> 3.8.0)
-    rspec-core (3.8.0)
-      rspec-support (~> 3.8.0)
-    rspec-expectations (3.8.4)
+    diff-lcs (1.5.1)
+    docile (1.4.0)
+    json (2.7.1)
+    method_source (1.0.0)
+    mysql2 (0.5.6)
+    parallel (1.24.0)
+    parser (3.3.0.5)
+      ast (~> 2.4.1)
+      racc
+    pry (0.14.2)
+      coderay (~> 1.1)
+      method_source (~> 1.0)
+    racc (1.7.3)
+    rainbow (3.1.1)
+    rake (13.2.0)
+    regexp_parser (2.9.0)
+    rexml (3.2.6)
+    rspec (3.13.0)
+      rspec-core (~> 3.13.0)
+      rspec-expectations (~> 3.13.0)
+      rspec-mocks (~> 3.13.0)
+    rspec-core (3.13.0)
+      rspec-support (~> 3.13.0)
+    rspec-expectations (3.13.0)
       diff-lcs (>= 1.2.0, < 2.0)
-      rspec-support (~> 3.8.0)
-    rspec-mocks (3.8.0)
+      rspec-support (~> 3.13.0)
+    rspec-mocks (3.13.0)
       diff-lcs (>= 1.2.0, < 2.0)
-      rspec-support (~> 3.8.0)
-    rspec-support (3.8.2)
-    rubocop (0.71.0)
-      jaro_winkler (~> 1.5.1)
+      rspec-support (~> 3.13.0)
+    rspec-support (3.13.1)
+    rubocop (0.93.1)
       parallel (~> 1.10)
-      parser (>= 2.6)
+      parser (>= 2.7.1.5)
       rainbow (>= 2.2.2, < 4.0)
+      regexp_parser (>= 1.8)
+      rexml
+      rubocop-ast (>= 0.6.0)
       ruby-progressbar (~> 1.7)
-      unicode-display_width (>= 1.4.0, < 1.7)
-    ruby-progressbar (1.10.1)
+      unicode-display_width (>= 1.4.0, < 2.0)
+    rubocop-ast (1.31.2)
+      parser (>= 3.3.0.4)
+    ruby-progressbar (1.13.0)
     simplecov (0.16.1)
       docile (~> 1.1)
       json (>= 1.8, < 3)
       simplecov-html (~> 0.10.0)
     simplecov-html (0.10.2)
-    term-ansicolor (1.7.1)
+    sync (0.5.0)
+    term-ansicolor (1.7.2)
       tins (~> 1.0)
-    thor (0.20.3)
-    tins (1.20.3)
-    unicode-display_width (1.6.0)
+    thor (1.3.1)
+    tins (1.32.1)
+      sync
+    unicode-display_width (1.8.0)
     yard (0.9.36)
 
 PLATFORMS
+  arm64-darwin-23
   ruby
 
 DEPENDENCIES
@@ -77,4 +87,4 @@ DEPENDENCIES
   yard (~> 0.9)
 
 BUNDLED WITH
-   1.17.2
+   2.5.6

lib/mysql2/aurora.rb

@@ -88,6 +88,12 @@ def self.method_missing(name, *args, &block) # rubocop:disable Style/MethodMissi
       def self.const_missing(name)
         Mysql2::Aurora::ORIGINAL_CLIENT_CLASS.const_get(name)
       end
+
+      # Delegate const definition to class.
+      # @param [Symbol] name Const name
+      def self.const_defined?(name)
+        Mysql2::Aurora::ORIGINAL_CLIENT_CLASS.const_defined?(name)
+      end
     end
 
     # Swap Mysql2::Client

lib/mysql2/aurora/version.rb

@@ -3,6 +3,6 @@ module Aurora
     # Major Version: Support `mysql2` major version.
     # Minor Version: Support `mysql2` minor version.
     # Tiny Version:  Mysql2::Aurora version.
-    VERSION = '0.5.2.1'.freeze
+    VERSION = '0.5.6'.freeze
   end
 end

mysql2-aurora.gemspec

@@ -20,7 +20,7 @@ Gem::Specification.new do |spec|
 
   spec.required_ruby_version = '>= 2.4.6'
 
-  spec.add_dependency 'mysql2', '~> 0.5.2'
+  spec.add_dependency 'mysql2', '~> 0.5.6'
 
   spec.add_development_dependency 'bundler',   '>= 1.16'
   spec.add_development_dependency 'coveralls', '~> 0.8'

spec/mysql2/aurora_spec.rb

@@ -186,6 +186,27 @@
     end
   end
 
+  describe 'connection flags' do
+    let :multi_client do
+      Mysql2::Client.new(
+        host:                          ENV['TEST_DB_HOST'],
+        username:                      ENV['TEST_DB_USER'],
+        password:                      ENV['TEST_DB_PASS'],
+        aurora_max_retry:              10,
+        aurora_disconnect_on_readonly: aurora_disconnect_on_readonly,
+        flags: ["MULTI_STATEMENTS"]
+      )
+    end
+
+    subject do
+      multi_client.query('SELECT CURRENT_USER() AS user; SELECT CURRENT_USER() AS user;')
+    end
+
+    it 'supports multi statements after reconnect' do
+      expect { subject }.to_not raise_error
+    end
+  end
+
   describe '#method_missing' do
     subject do
       client.ping