module ActiveRecord::ConnectionAdapters::Materialize::DatabaseStatements

Public Instance Methods

begin_db_transaction() click to toggle source

Begins a transaction.

# File lib/active_record/connection_adapters/materialize/database_statements.rb, line 134
def begin_db_transaction
  execute "BEGIN"
end
begin_isolated_db_transaction(isolation) click to toggle source
# File lib/active_record/connection_adapters/materialize/database_statements.rb, line 138
def begin_isolated_db_transaction(isolation)
  begin_db_transaction
  execute "SET TRANSACTION ISOLATION LEVEL #{transaction_isolation_levels.fetch(isolation)}"
end
commit_db_transaction() click to toggle source

Commits a transaction.

# File lib/active_record/connection_adapters/materialize/database_statements.rb, line 144
def commit_db_transaction
  execute "COMMIT"
end
exec_delete(sql, name = nil, binds = []) click to toggle source
# File lib/active_record/connection_adapters/materialize/database_statements.rb, line 117
def exec_delete(sql, name = nil, binds = [])
  execute_and_clear(sql, name, binds) { |result| result.cmd_tuples }
end
Also aliased as: exec_update
exec_query(sql, name = "SQL", binds = [], prepare: false) click to toggle source
# File lib/active_record/connection_adapters/materialize/database_statements.rb, line 103
def exec_query(sql, name = "SQL", binds = [], prepare: false)
  execute_and_clear(sql, name, binds, prepare: prepare) do |result|
    types = {}
    fields = result.fields
    fields.each_with_index do |fname, i|
      ftype = result.ftype i
      fmod  = result.fmod i

      types[fname] = get_oid_type(ftype, fmod, fname)
    end
    ActiveRecord::Result.new(fields, result.values, types)
  end
end
exec_rollback_db_transaction() click to toggle source

Aborts a transaction.

# File lib/active_record/connection_adapters/materialize/database_statements.rb, line 149
def exec_rollback_db_transaction
  execute "ROLLBACK"
end
exec_update(sql, name = nil, binds = [])
Alias for: exec_delete
execute(sql, name = nil) click to toggle source

Executes an SQL statement, returning a PG::Result object on success or raising a PG::Error exception otherwise. Note: the PG::Result object is manually memory managed; if you don't need it specifically, you may want consider the exec_query wrapper.

# File lib/active_record/connection_adapters/materialize/database_statements.rb, line 87
def execute(sql, name = nil)
  if preventing_writes? && write_query?(sql)
    raise ActiveRecord::ReadOnlyError, "Write query attempted while in readonly mode: #{sql}"
  end

  materialize_transactions

  result = nil
  log(sql, name) do
    ActiveSupport::Dependencies.interlock.permit_concurrent_loads do
      result = execute_async_and_raise(sql)
    end
  end
  result
end
explain(arel, binds = []) click to toggle source
# File lib/active_record/connection_adapters/materialize/database_statements.rb, line 9
def explain(arel, binds = [])
  sql = "EXPLAIN #{to_sql(arel, binds)}"
  Materialize::ExplainPrettyPrinter.new.pp(exec_query(sql, "EXPLAIN", binds))
end

Private Instance Methods

build_truncate_statements(table_names) click to toggle source
# File lib/active_record/connection_adapters/materialize/database_statements.rb, line 170
def build_truncate_statements(table_names)
  ["TRUNCATE TABLE #{table_names.map(&method(:quote_table_name)).join(", ")}"]
end
execute_async_and_raise(sql) click to toggle source

Known issue: PG::InternalError: ERROR: At least one input has no complete timestamps yet github.com/MaterializeInc/materialize/issues/2917

# File lib/active_record/connection_adapters/materialize/database_statements.rb, line 156
def execute_async_and_raise(sql)
  @connection.async_exec(sql)
rescue PG::InternalError => error
  if error.message.include? "At least one input has no complete timestamps yet"
    raise ::Materialize::Errors::IncompleteInput, error.message
  else
    raise
  end
end
execute_batch(statements, name = nil) click to toggle source
# File lib/active_record/connection_adapters/materialize/database_statements.rb, line 166
def execute_batch(statements, name = nil)
  execute(combine_multi_statements(statements))
end
suppress_composite_primary_key(pk) click to toggle source
# File lib/active_record/connection_adapters/materialize/database_statements.rb, line 174
def suppress_composite_primary_key(pk)
  pk unless pk.is_a?(Array)
end