aws_advanced_ruby_driver_wrapper 1.0.0

data/lib/aws_advanced_ruby_driver_wrapper/db_dialects/multi_az_cluster_mysql_dialect.rb

@@ -0,0 +1,95 @@
+# frozen_string_literal: true
+
+#  Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+#  Licensed under the Apache License, Version 2.0 (the "License").
+#  You may not use this file except in compliance with the License.
+#  You may obtain a copy of the License at
+#
+#  http://www.apache.org/licenses/LICENSE-2.0
+#
+#  Unless required by applicable law or agreed to in writing, software
+#  distributed under the License is distributed on an "AS IS" BASIS,
+#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#  See the License for the specific language governing permissions and
+#  limitations under the License.
+
+require_relative 'mysql_dialect'
+require_relative 'utils/dialect_utils'
+
+module AwsAdvancedRubyDriverWrapper
+  module DbDialects
+    class MultiAzClusterMysqlDialect < MysqlDialect
+      REPORT_HOST_EXISTS_QUERY = <<~SQL
+        SHOW VARIABLES LIKE 'report_host'
+      SQL
+
+      TOPOLOGY_TABLE_EXISTS_QUERY = <<~SQL
+        SELECT 1 AS tmp FROM information_schema.tables WHERE
+        table_schema = 'mysql' AND table_name = 'rds_topology'
+      SQL
+
+      TOPOLOGY_QUERY = <<~SQL
+        SELECT id AS instance_id, endpoint AS endpoint FROM mysql.rds_topology
+      SQL
+
+      INSTANCE_IDENTITY_QUERY = <<~SQL
+        SELECT id AS instance_id,
+        SUBSTRING_INDEX(endpoint, '.', 1) AS instance_name
+        FROM mysql.rds_topology
+        WHERE id = @@server_id
+      SQL
+
+      WRITER_ID_QUERY = <<~SQL
+        SHOW REPLICA STATUS
+      SQL
+
+      WRITER_ID_QUERY_COLUMN_NAME = 'Source_Server_Id'
+
+      def dialect?(connection)
+        return false unless
+          check_existence_queries(@driver_dialect, connection, TOPOLOGY_TABLE_EXISTS_QUERY, TOPOLOGY_QUERY)
+
+        begin
+          result = @driver_dialect.execute(connection, REPORT_HOST_EXISTS_QUERY)
+          return false if result.none?
+
+          row = result.first
+          report_host = row.values[1]
+          report_host.is_a?(String) && !report_host.empty?
+        rescue StandardError
+          false
+        end
+      end
+
+      def dialect_update_candidates
+        []
+      end
+
+      def topology_query
+        TOPOLOGY_QUERY
+      end
+
+      def instance_identity(connection)
+        query_instance_identity(@driver_dialect, connection, INSTANCE_IDENTITY_QUERY)
+      end
+
+      def writer_id_query
+        WRITER_ID_QUERY
+      end
+
+      def writer_id_column_name
+        WRITER_ID_QUERY_COLUMN_NAME
+      end
+
+      # @param service_container [Services::ServiceContainer]
+      # @return [Host::RdsHostListProvider] the host list provider
+      def create_host_list_provider(service_container)
+        require_relative '../utils/multi_az_topology_utils'
+        require_relative '../host/rds_host_list_provider'
+        topology_utils = Utils::MultiAzTopologyUtils.new(dialect: self)
+        Host::RdsHostListProvider.new(service_container:, topology_utils:)
+      end
+    end
+  end
+end

data/lib/aws_advanced_ruby_driver_wrapper/db_dialects/multi_az_cluster_pg_dialect.rb

@@ -0,0 +1,86 @@
+# frozen_string_literal: true
+
+#  Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+#  Licensed under the Apache License, Version 2.0 (the "License").
+#  You may not use this file except in compliance with the License.
+#  You may obtain a copy of the License at
+#
+#  http://www.apache.org/licenses/LICENSE-2.0
+#
+#  Unless required by applicable law or agreed to in writing, software
+#  distributed under the License is distributed on an "AS IS" BASIS,
+#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#  See the License for the specific language governing permissions and
+#  limitations under the License.
+
+require_relative 'pg_dialect'
+require_relative 'utils/dialect_utils'
+
+module AwsAdvancedRubyDriverWrapper
+  module DbDialects
+    class MultiAzClusterPgDialect < PgDialect
+      IS_RDS_CLUSTER_QUERY = <<~SQL
+        SELECT multi_az_db_cluster_source_dbi_resource_id FROM rds_tools.multi_az_db_cluster_source_dbi_resource_id()
+      SQL
+
+      TOPOLOGY_QUERY = <<~SQL.freeze
+        SELECT id AS instance_id, endpoint AS endpoint FROM rds_tools.show_topology('aws_advanced_ruby_driver_wrapper-#{AwsAdvancedRubyDriverWrapper::VERSION}')
+      SQL
+
+      INSTANCE_IDENTITY_QUERY = <<~SQL
+        SELECT id AS instance_id,
+        SUBSTRING(endpoint FROM 0 FOR POSITION('.' IN endpoint)) AS instance_name
+        FROM rds_tools.show_topology()
+        WHERE id OPERATOR(pg_catalog.=) rds_tools.dbi_resource_id()
+      SQL
+
+      WRITER_ID_QUERY = <<~SQL
+        SELECT multi_az_db_cluster_source_dbi_resource_id FROM rds_tools.multi_az_db_cluster_source_dbi_resource_id()
+        WHERE multi_az_db_cluster_source_dbi_resource_id OPERATOR(pg_catalog.!=)
+        (SELECT dbi_resource_id FROM rds_tools.dbi_resource_id())
+      SQL
+
+      WRITER_ID_QUERY_COLUMN_NAME = 'multi_az_db_cluster_source_dbi_resource_id'
+
+      def dialect?(connection)
+        result = @driver_dialect.execute(connection, IS_RDS_CLUSTER_QUERY)
+        return false unless result.any?
+
+        row = result.first
+        !row['multi_az_db_cluster_source_dbi_resource_id'].nil?
+      rescue StandardError
+        false
+      end
+
+      def dialect_update_candidates
+        []
+      end
+
+      def topology_query
+        TOPOLOGY_QUERY
+      end
+
+      def instance_identity(connection)
+        query_instance_identity(@driver_dialect, connection, INSTANCE_IDENTITY_QUERY)
+      end
+
+      def writer_id_query
+        WRITER_ID_QUERY
+      end
+
+      def writer_id_column_name
+        WRITER_ID_QUERY_COLUMN_NAME
+      end
+
+      # @param service_container [Services::ServiceContainer]
+      # @return [Host::RdsHostListProvider] the host list provider
+      def create_host_list_provider(service_container)
+        require_relative '../utils/multi_az_topology_utils'
+        require_relative '../host/rds_host_list_provider'
+        topology_utils = Utils::MultiAzTopologyUtils.new(dialect: self)
+        Host::RdsHostListProvider.new(service_container:, topology_utils:)
+      end
+    end
+  end
+end

data/lib/aws_advanced_ruby_driver_wrapper/db_dialects/mysql_dialect.rb

@@ -0,0 +1,98 @@
+# frozen_string_literal: true
+
+#  Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+#  Licensed under the Apache License, Version 2.0 (the "License").
+#  You may not use this file except in compliance with the License.
+#  You may obtain a copy of the License at
+#
+#  http://www.apache.org/licenses/LICENSE-2.0
+#
+#  Unless required by applicable law or agreed to in writing, software
+#  distributed under the License is distributed on an "AS IS" BASIS,
+#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#  See the License for the specific language governing permissions and
+#  limitations under the License.
+
+require_relative 'dialect_codes'
+require_relative 'utils/dialect_utils'
+
+module AwsAdvancedRubyDriverWrapper
+  module DbDialects
+    class MysqlDialect
+      include AwsAdvancedRubyDriverWrapper::DbDialects::DialectUtils
+
+      VERSION_QUERY = <<~SQL
+        SHOW VARIABLES LIKE 'version_comment'
+      SQL
+
+      INSTANCE_IDENTITY_QUERY = <<~SQL
+        SELECT @@hostname AS instance_id, CONCAT(@@hostname, ':', @@port) AS instance_name
+      SQL
+
+      IS_READER_QUERY = <<~SQL
+        SELECT @@read_only
+      SQL
+
+      DIALECT_UPDATE_CANDIDATES = [
+        AwsAdvancedRubyDriverWrapper::DialectCodes::GLOBAL_AURORA_MYSQL,
+        AwsAdvancedRubyDriverWrapper::DialectCodes::AURORA_MYSQL,
+        AwsAdvancedRubyDriverWrapper::DialectCodes::MULTI_AZ_MYSQL_CLUSTER,
+        AwsAdvancedRubyDriverWrapper::DialectCodes::RDS_MYSQL
+      ].freeze
+
+      def initialize(driver_dialect)
+        @driver_dialect = driver_dialect
+      end
+
+      def execute(connection, sql)
+        @driver_dialect.execute(connection, sql)
+      end
+
+      def execute_with_params(connection, sql, params)
+        @driver_dialect.execute_with_params(connection, sql, params)
+      end
+
+      def dialect?(connection)
+        result = @driver_dialect.execute(connection, VERSION_QUERY)
+        result.any? do |row|
+          row.values[1]&.downcase&.include?('mysql')
+        end
+      rescue StandardError
+        false
+      end
+
+      def default_port
+        @default_port ||= 3306
+      end
+
+      # @return [Boolean] whether this dialect represents an Aurora Global Database.
+      #   Overridden to true by the global dialects.
+      def global?
+        false
+      end
+
+      def dialect_update_candidates
+        DIALECT_UPDATE_CANDIDATES
+      end
+
+      def exception_handler(driver_dialect)
+        AwsAdvancedRubyDriverWrapper::Errors::MysqlErrorHandler.new(driver_dialect)
+      end
+
+      def host_role(connection)
+        query_host_role(@driver_dialect, connection, self.class::IS_READER_QUERY)
+      end
+
+      def instance_identity(connection)
+        query_instance_identity(@driver_dialect, connection, INSTANCE_IDENTITY_QUERY)
+      end
+
+      # @param service_container [Services::ServiceContainer]
+      # @return [Host::ConnectionStringHostListProvider] the host list provider
+      def create_host_list_provider(service_container)
+        Host::ConnectionStringHostListProvider.new(service_container:)
+      end
+    end
+  end
+end

data/lib/aws_advanced_ruby_driver_wrapper/db_dialects/pg_dialect.rb

@@ -0,0 +1,95 @@
+# frozen_string_literal: true
+
+#  Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+#  Licensed under the Apache License, Version 2.0 (the "License").
+#  You may not use this file except in compliance with the License.
+#  You may obtain a copy of the License at
+#
+#  http://www.apache.org/licenses/LICENSE-2.0
+#
+#  Unless required by applicable law or agreed to in writing, software
+#  distributed under the License is distributed on an "AS IS" BASIS,
+#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#  See the License for the specific language governing permissions and
+#  limitations under the License.
+
+require_relative 'dialect_codes'
+require_relative 'utils/dialect_utils'
+require_relative '../host/connection_string_host_list_provider'
+
+module AwsAdvancedRubyDriverWrapper
+  module DbDialects
+    class PgDialect
+      include AwsAdvancedRubyDriverWrapper::DbDialects::DialectUtils
+
+      PG_PROC_EXISTS_QUERY = <<~SQL
+        SELECT 1 FROM pg_catalog.pg_proc LIMIT 1
+      SQL
+
+      INSTANCE_IDENTITY_QUERY = <<~SQL
+        SELECT pg_catalog.inet_server_addr() AS instance_id,
+        pg_catalog.CONCAT(pg_catalog.inet_server_addr(), ':', pg_catalog.inet_server_port()) AS instance_name
+      SQL
+
+      IS_READER_QUERY = <<~SQL
+        SELECT pg_catalog.pg_is_in_recovery()
+      SQL
+
+      DIALECT_UPDATE_CANDIDATES = [
+        AwsAdvancedRubyDriverWrapper::DialectCodes::GLOBAL_AURORA_PG,
+        AwsAdvancedRubyDriverWrapper::DialectCodes::AURORA_PG,
+        AwsAdvancedRubyDriverWrapper::DialectCodes::MULTI_AZ_PG_CLUSTER,
+        AwsAdvancedRubyDriverWrapper::DialectCodes::RDS_PG
+      ].freeze
+
+      def initialize(driver_dialect)
+        @driver_dialect = driver_dialect
+      end
+
+      def execute(connection, sql)
+        @driver_dialect.execute(connection, sql)
+      end
+
+      def execute_with_params(connection, sql, params)
+        @driver_dialect.execute_with_params(connection, sql, params)
+      end
+
+      def dialect?(connection)
+        check_existence_queries(@driver_dialect, connection, PG_PROC_EXISTS_QUERY)
+      end
+
+      def default_port
+        @default_port ||= 5432
+      end
+
+      # @return [Boolean] whether this dialect represents an Aurora Global Database.
+      #   Overridden to true by the global dialects.
+      def global?
+        false
+      end
+
+      def dialect_update_candidates
+        DIALECT_UPDATE_CANDIDATES
+      end
+
+      def exception_handler(driver_dialect)
+        AwsAdvancedRubyDriverWrapper::Errors::PgErrorHandler.new(driver_dialect)
+      end
+
+      def host_role(connection)
+        query_host_role(@driver_dialect, connection, IS_READER_QUERY)
+      end
+
+      def instance_identity(connection)
+        query_instance_identity(@driver_dialect, connection, INSTANCE_IDENTITY_QUERY)
+      end
+
+      # @param service_container [Services::ServiceContainer]
+      # @return [Host::ConnectionStringHostListProvider] the host list provider
+      def create_host_list_provider(service_container)
+        Host::ConnectionStringHostListProvider.new(service_container:)
+      end
+    end
+  end
+end

data/lib/aws_advanced_ruby_driver_wrapper/db_dialects/rds_mysql_dialect.rb

@@ -0,0 +1,88 @@
+# frozen_string_literal: true
+
+#  Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+#  Licensed under the Apache License, Version 2.0 (the "License").
+#  You may not use this file except in compliance with the License.
+#  You may obtain a copy of the License at
+#
+#  http://www.apache.org/licenses/LICENSE-2.0
+#
+#  Unless required by applicable law or agreed to in writing, software
+#  distributed under the License is distributed on an "AS IS" BASIS,
+#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#  See the License for the specific language governing permissions and
+#  limitations under the License.
+
+require_relative 'mysql_dialect'
+require_relative 'utils/dialect_utils'
+
+module AwsAdvancedRubyDriverWrapper
+  module DbDialects
+    class RdsMysqlDialect < MysqlDialect
+      REPORT_HOST_EXISTS_QUERY = <<~SQL
+        SHOW VARIABLES LIKE 'report_host'
+      SQL
+
+      TOPOLOGY_TABLE_EXISTS_QUERY = <<~SQL
+        SELECT 1 AS tmp FROM information_schema.tables WHERE
+        table_schema = 'mysql' AND table_name = 'rds_topology'
+      SQL
+
+      INSTANCE_IDENTITY_QUERY = <<~SQL
+        SELECT id AS instance_id,
+        SUBSTRING_INDEX(endpoint, '.', 1) AS instance_name
+        FROM mysql.rds_topology
+        WHERE id = @@server_id
+      SQL
+
+      BG_STATUS_QUERY = <<~SQL
+        SELECT * FROM mysql.rds_topology
+      SQL
+
+      DIALECT_UPDATE_CANDIDATES = [
+        AwsAdvancedRubyDriverWrapper::DialectCodes::GLOBAL_AURORA_MYSQL,
+        AwsAdvancedRubyDriverWrapper::DialectCodes::AURORA_MYSQL,
+        AwsAdvancedRubyDriverWrapper::DialectCodes::MULTI_AZ_MYSQL_CLUSTER
+      ].freeze
+
+      def dialect?(connection)
+        return false if super
+
+        begin
+          result = @driver_dialect.execute(connection, VERSION_QUERY)
+          return false unless result.any?
+
+          row = result.first
+          column_value = row.values[1]
+          return false unless 'Source distribution'.casecmp?(column_value)
+
+          result = @driver_dialect.execute(connection, REPORT_HOST_EXISTS_QUERY)
+          return false unless result.any?
+
+          row = result.first
+          report_host = row.values[1]
+          report_host.nil? || report_host.empty?
+        rescue StandardError
+          false
+        end
+      end
+
+      def dialect_update_candidates
+        DIALECT_UPDATE_CANDIDATES
+      end
+
+      def blue_green_status_available?(connection)
+        check_existence_queries(@driver_dialect, connection, TOPOLOGY_TABLE_EXISTS_QUERY)
+      end
+
+      def instance_identity(connection)
+        query_instance_identity(@driver_dialect, connection, INSTANCE_IDENTITY_QUERY)
+      end
+
+      def blue_green_status_query
+        BG_STATUS_QUERY
+      end
+    end
+  end
+end

data/lib/aws_advanced_ruby_driver_wrapper/db_dialects/rds_pg_dialect.rb

@@ -0,0 +1,86 @@
+# frozen_string_literal: true
+
+#  Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+#  Licensed under the Apache License, Version 2.0 (the "License").
+#  You may not use this file except in compliance with the License.
+#  You may obtain a copy of the License at
+#
+#  http://www.apache.org/licenses/LICENSE-2.0
+#
+#  Unless required by applicable law or agreed to in writing, software
+#  distributed under the License is distributed on an "AS IS" BASIS,
+#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#  See the License for the specific language governing permissions and
+#  limitations under the License.
+
+require_relative 'pg_dialect'
+require_relative 'utils/dialect_utils'
+
+module AwsAdvancedRubyDriverWrapper
+  module DbDialects
+    class RdsPgDialect < PgDialect
+      EXTENSIONS_EXIST_SQL = <<~SQL
+        SELECT (setting OPERATOR(pg_catalog.~~) '%rds_tools%') AS rds_tools,
+        (setting OPERATOR(pg_catalog.~~) '%aurora_stat_utils%') AS aurora_stat_utils
+        FROM pg_catalog.pg_settings
+        WHERE name OPERATOR(pg_catalog.=) 'rds.extensions'
+      SQL
+
+      TOPOLOGY_TABLE_EXISTS_QUERY = <<~SQL
+        SELECT 'rds_tools.show_topology'::pg_catalog.regproc
+      SQL
+
+      INSTANCE_IDENTITY_QUERY = <<~SQL
+        SELECT id AS instance_id,
+        SUBSTRING(endpoint FROM 0 FOR POSITION('.' IN endpoint)) AS instance_name
+        FROM rds_tools.show_topology()
+        WHERE id OPERATOR(pg_catalog.=) rds_tools.dbi_resource_id()
+      SQL
+
+      BG_STATUS_QUERY = <<~SQL.freeze
+        SELECT * FROM rds_tools.show_topology('aws_advanced_ruby_driver_wrapper-#{AwsAdvancedRubyDriverWrapper::VERSION}')
+      SQL
+
+      DIALECT_UPDATE_CANDIDATES = [
+        AwsAdvancedRubyDriverWrapper::DialectCodes::MULTI_AZ_PG_CLUSTER,
+        AwsAdvancedRubyDriverWrapper::DialectCodes::GLOBAL_AURORA_PG,
+        AwsAdvancedRubyDriverWrapper::DialectCodes::AURORA_PG
+      ].freeze
+
+      def dialect?(connection)
+        return false unless super
+
+        begin
+          result = @driver_dialect.execute(connection, EXTENSIONS_EXIST_SQL)
+          result.each do |row|
+            rds_tools_enabled = row['rds_tools'] == 't'
+            aurora_utils_enabled = row['aurora_stat_utils'] == 't'
+
+            return true if rds_tools_enabled && !aurora_utils_enabled
+          end
+        rescue StandardError
+          return false
+        end
+
+        false
+      end
+
+      def dialect_update_candidates
+        DIALECT_UPDATE_CANDIDATES
+      end
+
+      def blue_green_status_available?(connection)
+        check_existence_queries(@driver_dialect, connection, TOPOLOGY_TABLE_EXISTS_QUERY)
+      end
+
+      def instance_identity(connection)
+        query_instance_identity(@driver_dialect, connection, INSTANCE_IDENTITY_QUERY)
+      end
+
+      def blue_green_status_query
+        BG_STATUS_QUERY
+      end
+    end
+  end
+end

data/lib/aws_advanced_ruby_driver_wrapper/db_dialects/unknown_dialect.rb

@@ -0,0 +1,72 @@
+# frozen_string_literal: true
+
+#  Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+#  Licensed under the Apache License, Version 2.0 (the "License").
+#  You may not use this file except in compliance with the License.
+#  You may obtain a copy of the License at
+#
+#  http://www.apache.org/licenses/LICENSE-2.0
+#
+#  Unless required by applicable law or agreed to in writing, software
+#  distributed under the License is distributed on an "AS IS" BASIS,
+#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#  See the License for the specific language governing permissions and
+#  limitations under the License.
+
+require_relative 'dialect_codes'
+
+module AwsAdvancedRubyDriverWrapper
+  module DbDialects
+    class UnknownDialect
+      DIALECT_UPDATE_CANDIDATES = [
+        AwsAdvancedRubyDriverWrapper::DialectCodes::GLOBAL_AURORA_PG,
+        AwsAdvancedRubyDriverWrapper::DialectCodes::GLOBAL_AURORA_MYSQL,
+        AwsAdvancedRubyDriverWrapper::DialectCodes::AURORA_PG,
+        AwsAdvancedRubyDriverWrapper::DialectCodes::AURORA_MYSQL,
+        AwsAdvancedRubyDriverWrapper::DialectCodes::MULTI_AZ_PG_CLUSTER,
+        AwsAdvancedRubyDriverWrapper::DialectCodes::MULTI_AZ_MYSQL_CLUSTER,
+        AwsAdvancedRubyDriverWrapper::DialectCodes::RDS_PG,
+        AwsAdvancedRubyDriverWrapper::DialectCodes::RDS_MYSQL,
+        AwsAdvancedRubyDriverWrapper::DialectCodes::PG,
+        AwsAdvancedRubyDriverWrapper::DialectCodes::MYSQL
+      ].freeze
+
+      def initialize(_driver_dialect = nil); end
+
+      def default_port
+        @default_port ||= -1
+      end
+
+      def host_role(connection)
+        raise NotImplementedError, 'Unable to gather host role, connected to unknown DB type.'
+      end
+
+      def host_id(connection)
+        raise NotImplementedError, 'Unable to gather host id, connected to unknown DB type.'
+      end
+
+      def instance_identity(_connection)
+        nil
+      end
+
+      def dialect?(_connection)
+        false
+      end
+
+      def dialect_update_candidates
+        DIALECT_UPDATE_CANDIDATES
+      end
+
+      def execute(connection, sql)
+        raise NotImplementedError, 'Unable to execute query, connected to unknown DB type.'
+      end
+
+      # @param service_container [Services::ServiceContainer]
+      # @return [Host::ConnectionStringHostListProvider] the host list provider
+      def create_host_list_provider(service_container)
+        Host::ConnectionStringHostListProvider.new(service_container:)
+      end
+    end
+  end
+end

data/lib/aws_advanced_ruby_driver_wrapper/db_dialects/utils/dialect_utils.rb

@@ -0,0 +1,71 @@
+# frozen_string_literal: true
+
+#  Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+#  Licensed under the Apache License, Version 2.0 (the "License").
+#  You may not use this file except in compliance with the License.
+#  You may obtain a copy of the License at
+#
+#  http://www.apache.org/licenses/LICENSE-2.0
+#
+#  Unless required by applicable law or agreed to in writing, software
+#  distributed under the License is distributed on an "AS IS" BASIS,
+#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#  See the License for the specific language governing permissions and
+#  limitations under the License.module DialectUtils
+#
+require_relative '../../host/host_role'
+
+module AwsAdvancedRubyDriverWrapper
+  module DbDialects
+    module DialectUtils
+      # Returns the instance identity for the given connection.
+      #
+      # @param driver_dialect [Object] the driver dialect used to execute queries
+      # @param connection [Object] the connection to analyze
+      # @param host_id_query [String] the SQL query to retrieve the instance identity
+      # @return [Array(String, String), nil] the instance_id and instance_name, or nil if they cannot be determined
+      def query_instance_identity(driver_dialect, connection, host_id_query)
+        result = driver_dialect.execute(connection, host_id_query)
+        return nil if result.nil? || result.none?
+
+        row = result.first
+        values = row.values
+        [values[0], values[1]]
+      rescue StandardError
+        nil
+      end
+
+      # Determines the role of the host behind the given connection.
+      #
+      # @param driver_dialect [Object] the driver dialect used to execute queries
+      # @param connection [Object] the connection to analyze
+      # @param reader_check_query [String] the SQL query to check if the host is a reader
+      # @return [HostRole] :writer or :reader
+      def query_host_role(driver_dialect, connection, reader_check_query)
+        result = driver_dialect.execute(connection, reader_check_query)
+        row = result.first
+        value = row.is_a?(Hash) ? row.values.first : row
+        %w[t 1].include?(value.to_s) ? Host::HostRole::READER : Host::HostRole::WRITER
+      end
+
+      # Checks that all given queries return non-empty results.
+      #
+      # @param driver_dialect [Object] the driver dialect used to execute queries
+      # @param connection [Object] the connection to execute queries against
+      # @param queries [Array<String>] one or more SQL queries to verify
+      # @return [Boolean] true if all queries return results, false otherwise
+      def check_existence_queries(driver_dialect, connection, *queries)
+        return false unless driver_dialect.respond_to?(:execute)
+
+        queries.each do |existence_query|
+          result = driver_dialect.execute(connection, existence_query)
+          return false if result.nil? || result.none?
+        end
+        true
+      rescue StandardError
+        false
+      end
+    end
+  end
+end