activerecord-aurora-serverless-adapter 1.0.0

data/lib/active_record/connection_adapters/aurora_serverless/mysql2/client.rb

@@ -0,0 +1,71 @@
+# Keep `AuroraServerless::Client` abstract and mix these methods
+# in specifically to mimic the Mysql2 gem's interfaces.
+#
+module ActiveRecord
+  module ConnectionAdapters
+    module AuroraServerless
+      module Mysql2
+        module Client
+
+          ESCAPE_MAP = {
+            "\x00" => "0",
+            "\n"   => "n",
+            "\r"   => "r",
+            "\\"   => "\\",
+            "'"    => "'",
+            '"'    => '"'
+          }.freeze
+
+          ESCAPE_PATTERN = Regexp.union(*ESCAPE_MAP.keys)
+
+          def self.default_query_options
+            {}
+          end
+
+          def query_options
+            {}
+          end
+
+          def query(sql)
+            raise ActiveRecord::StatementInvalid if @closed
+            result = execute_statement(sql)
+            AuroraServerless::Mysql2::Result.new(result)
+          end
+
+          def server_info
+            @server_info || begin
+              r = query 'SHOW VARIABLES LIKE "version"'
+              version = r.to_a.detect{ |r| r.detect { |v| v == 'version' } }.last
+              { version: version }
+            end
+          end
+
+          def close
+            @closed = true
+          end
+
+          def automatic_close=(*)
+            nil
+          end
+
+          def escape(string)
+            string.gsub(ESCAPE_PATTERN) { |x| "\\#{ESCAPE_MAP[x]}" }
+          end
+
+          def ping
+            return false if @closed
+            query('SELECT 1').to_a.first.first == 1
+          rescue
+            false
+          end
+
+          def abandon_results!
+            nil
+          end
+
+        end
+      end
+      AuroraServerless::Client.include AuroraServerless::Mysql2::Client
+    end
+  end
+end

data/lib/active_record/connection_adapters/aurora_serverless/mysql2/connection_handling.rb

@@ -0,0 +1,20 @@
+module ActiveRecord
+  module ConnectionHandling
+
+    def aurora_serverless_connection(config)
+      client = aurora_serverless_connection_from_config(config)
+      ConnectionAdapters::AuroraServerlessAdapter.new(client, logger, nil, config)
+    end
+
+    def aurora_serverless_connection_from_config(config)
+      ConnectionAdapters::AuroraServerless::Client.new(
+        config[:database],
+        config[:resource_arn],
+        config[:secret_arn],
+        config
+      )
+    end
+    module_function :aurora_serverless_connection_from_config
+
+  end
+end

data/lib/active_record/connection_adapters/aurora_serverless/mysql2/result.rb

@@ -0,0 +1,116 @@
+# Wraps a `Aws::RDSDataService::Types::ExecuteStatementResponse` response object.
+# https://docs.aws.amazon.com/sdk-for-ruby/v3/api/Aws/RDSDataService/Types/ExecuteStatementResponse.html
+#
+module ActiveRecord
+  module ConnectionAdapters
+    module AuroraServerless
+      module Mysql2
+        class Result
+          include Enumerable
+
+          attr_reader :response
+
+          def initialize(response)
+            @response = response
+          end
+
+          def fields
+            @fields ||= begin
+              md = @response.column_metadata
+              md ? md.map(&:label) : []
+            end
+          end
+
+          def to_a
+            as_array
+          end
+
+          def each(**kwargs)
+            eobj = each_object(kwargs)
+            block_given? ? eobj.each { |r| yield(r) } : eobj
+          end
+
+          private
+
+          def each_object(**kwargs)
+            symbolize_keys = kwargs[:symbolize_keys]
+            kwargs[:as] == :hash ? as_hash(symbolize_keys) : as_array
+          end
+
+          def as_array
+            @as_array ||= (@response.records || []).map do |fields|
+              fields.each_with_index.map do |field, index|
+                type = @response.column_metadata[index].type_name
+                type_cast(field, type)
+              end
+            end
+          end
+
+          def as_hash(symbolize_keys = true)
+            @as_hash ||= begin
+              h = ActiveRecord::Result.new(fields, as_array).to_a
+              symbolize_keys ? h.map { |r| r.symbolize_keys! } : h
+            end
+          end
+
+          def type_cast(field, type)
+            return if field.is_null
+            vmeth = VALUE_METHODS[type] || :string_value
+            value = field.public_send(vmeth)
+            case type
+            when 'DECIMAL'   then type_cast_decimal(value)
+            when 'DATE'      then type_cast_date(value)
+            when 'YEAR'      then type_cast_year(value)
+            else
+              value
+            end
+          end
+
+          def type_cast_decimal(v)
+            BigDecimal(v)
+          end
+
+          def type_cast_date(v)
+            Date.parse(v)
+          end
+
+          def type_cast_year(v)
+            v.to_i
+          end
+
+          VALUE_METHODS = {
+            'BIT'        => :boolean_value,
+            'TINYINT'    => :long_value,
+            'SMALLINT'   => :long_value,
+            'MEDIUMINT'  => :long_value,
+            'INT'        => :long_value,
+            'BIGINT'     => :long_value,
+            'FLOAT'      => :double_value,
+            'DOUBLE'     => :double_value,
+            'DECIMAL'    => :string_value,
+            'DATE'       => :string_value,
+            'DATETIME'   => :string_value,
+            'TIMESTAMP'  => :string_value,
+            'TIME'       => :string_value,
+            'YEAR'       => :string_value,
+            'CHAR'       => :string_value,
+            'VARCHAR'    => :string_value,
+            'BINARY'     => :blob_value,
+            'VARBINARY'  => :blob_value,
+            'TINYBLOB'   => :blob_value,
+            'TINYTEXT'   => :string_value,
+            'BLOB'       => :blob_value,
+            'TEXT'       => :string_value,
+            'MEDIUMBLOB' => :blob_value,
+            'MEDIUMTEXT' => :string_value,
+            'LONGBLOB'   => :blob_value,
+            'LONGTEXT'   => :string_value,
+          }.freeze
+          private_constant :VALUE_METHODS
+
+        end
+      end
+    end
+  end
+end
+

data/lib/active_record/connection_adapters/aurora_serverless/version.rb

@@ -0,0 +1,7 @@
+module ActiveRecord
+  module ConnectionAdapters
+    module AuroraServerless
+      VERSION = "1.0.0"
+    end
+  end
+end

data/lib/active_record/connection_adapters/aurora_serverless_adapter.rb

@@ -0,0 +1,7 @@
+require 'active_record'
+require 'aws-sdk-rdsdataservice'
+require 'active_record/connection_adapters/aurora_serverless/version'
+require 'active_record/connection_adapters/aurora_serverless/client'
+require 'active_record/connection_adapters/aurora_serverless/abstract'
+# TODO: [PG] Make this a conditional require? Railtie config? Just do both?
+require 'active_record/connection_adapters/aurora_serverless/mysql2'

data/lib/activerecord-aurora-serverless-adapter.rb

@@ -0,0 +1 @@
+require 'active_record/connection_adapters/aurora_serverless_adapter'

data/lib/mysql2.rb

@@ -0,0 +1,4 @@
+# This is here so the Mysql2 gem require will no-op.
+#
+module Mysql2
+end

metadata

@@ -0,0 +1,217 @@
+--- !ruby/object:Gem::Specification
+name: activerecord-aurora-serverless-adapter
+version: !ruby/object:Gem::Version
+  version: 1.0.0
+platform: ruby
+authors:
+- Ken Collins
+autorequire: 
+bindir: exe
+cert_chain: []
+date: 2019-12-29 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: activerecord
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '6.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '6.0'
+- !ruby/object:Gem::Dependency
+  name: aws-sdk-rdsdataservice
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: appraisal
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: dotenv
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: minitest
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: minitest-reporters
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: minitest-retry
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: pry
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: sqlite3
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Amazon Aurora Serverless is an on-demand, auto-scaling configuration
+  for Amazon Aurora (MySQL-compatible and PostgreSQL-compatible editions). Perfect
+  for small Rails on AWS Lambda.
+email:
+- kcollins@customink.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".github/workflows/ci.yaml"
+- ".gitignore"
+- ".ruby-version"
+- CODE_OF_CONDUCT.md
+- Dockerfile-cdk
+- Dockerfile-ci
+- Gemfile
+- Gemfile.lock
+- LICENSE.txt
+- README.md
+- Rakefile
+- activerecord-aurora-serverless-adapter.gemspec
+- bin/_setup
+- bin/_test
+- bin/bootstrap
+- bin/run
+- bin/setup
+- bin/test
+- bin/test-ci-setup
+- docker-compose.yml
+- lib/active_record/connection_adapters/aurora_serverless/abstract.rb
+- lib/active_record/connection_adapters/aurora_serverless/client.rb
+- lib/active_record/connection_adapters/aurora_serverless/gem_hack.rb
+- lib/active_record/connection_adapters/aurora_serverless/mysql2.rb
+- lib/active_record/connection_adapters/aurora_serverless/mysql2/client.rb
+- lib/active_record/connection_adapters/aurora_serverless/mysql2/connection_handling.rb
+- lib/active_record/connection_adapters/aurora_serverless/mysql2/result.rb
+- lib/active_record/connection_adapters/aurora_serverless/version.rb
+- lib/active_record/connection_adapters/aurora_serverless_adapter.rb
+- lib/activerecord-aurora-serverless-adapter.rb
+- lib/mysql2.rb
+homepage: https://github.com/customink/activerecord-aurora-serverless-adapter
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.7.6.2
+signing_key: 
+specification_version: 4
+summary: ActiveRecord Adapter for Amazon Aurora Serverless
+test_files: []