Contents

# frozen_string_literal: true

module PgEasyReplicate
  class Query
    extend Helper

    class << self
      def run(
        query:,
        connection_url:,
        user: internal_user_name,
        schema: nil,
        transaction: true
      )
        conn =
          connect(connection_url: connection_url, schema: schema, user: user)
        if transaction
          r =
            conn.transaction do
              conn.run("SET search_path to #{quote_ident(schema)}") if schema
              conn.run("SET statement_timeout to '5s'")
              conn.fetch(query).to_a
            end
        else
          conn.run("SET search_path to #{quote_ident(schema)}") if schema
          conn.run("SET statement_timeout to '5s'")
          r = conn.fetch(query).to_a
        end
        conn.disconnect
        r
      ensure
        conn&.fetch("RESET statement_timeout")
        conn&.disconnect
      end

      def connect(connection_url:, user: internal_user_name, schema: nil)
        c =
          Sequel.connect(
            connection_url,
            user: user,
            logger: ENV.fetch("DEBUG", nil) ? logger : nil,
            search_path: schema,
          )
        logger.debug("Connection established")
        c
      end
    end
  end
end

Version data entries

7 entries across 7 versions & 1 rubygems

Version	Path
pg_easy_replicate-0.2.2	lib/pg_easy_replicate/query.rb
pg_easy_replicate-0.2.1	lib/pg_easy_replicate/query.rb
pg_easy_replicate-0.2.0	lib/pg_easy_replicate/query.rb
pg_easy_replicate-0.1.12	lib/pg_easy_replicate/query.rb
pg_easy_replicate-0.1.11	lib/pg_easy_replicate/query.rb
pg_easy_replicate-0.1.10	lib/pg_easy_replicate/query.rb
pg_easy_replicate-0.1.9	lib/pg_easy_replicate/query.rb