Contents

require 'active_record/connection_adapters/abstract_mysql_adapter'
require 'active_record/connection_adapters/mysql/database_statements'

require 'active_record/connection_adapters/aws_data_service_mysql/connection.rb'

module ActiveRecord
  module ConnectionHandling
    def aws_data_service_mysql_connection(config)
      ConnectionAdapters::AwsDataServiceMysqlAdapter.new(ConnectionAdapters::AwsDataServiceMysql::Connection.new(config), logger, nil, config)
    end
  end

  module ConnectionAdapters
    class AwsDataServiceMysqlAdapter < AbstractMysqlAdapter
      ADAPTER_NAME = 'AwsDataServiceMysql'.freeze

      include MySQL::DatabaseStatements

      def initialize(*)
        super
        @prepared_statements = false
      end

      def each_hash(result)
        if block_given?
          result.each(as: :hash, symbolize_keys: true) do |row|
            yield row
          end
        else
          to_enum(:each_hash, result)
        end
      end

      def begin_db_transaction
        log('BEGIN') do
          ActiveSupport::Dependencies.interlock.permit_concurrent_loads do
            @connection.begin_db_transaction
          end
        end
      end

      def commit_db_transaction
        log('COMMIT') do
          ActiveSupport::Dependencies.interlock.permit_concurrent_loads do
            @connection.commit_db_transaction
          end
        end
      end

      def exec_rollback_db_transaction
        log('ROLLBACK') do
          ActiveSupport::Dependencies.interlock.permit_concurrent_loads do
            @connection.exec_rollback_db_transaction
          end
        end
      end

      private

      def error_number(exception)
        exception.error_number if exception.respond_to?(:error_number)
      end

      def full_version
        schema_cache.database_version.full_version_string
      end

      def get_full_version
        @connection.server_info[:version]
      end
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
activerecord-aws_data_service_mysql-adapter-0.1.0	lib/active_record/connection_adapters/aws_data_service_mysql_adapter.rb