Contents

require "pg"

class Simple::SQL::Connection::RawConnection < Simple::SQL::Connection
  SELF = self

  attr_reader :raw_connection

  def initialize(database_url)
    @database_url = database_url
    @raw_connection = PG::Connection.new(@database_url)
    ObjectSpace.define_finalizer(self, SELF.finalizer(@raw_connection))
  end

  def self.finalizer(raw_connection)
    proc do
      raw_connection.finish unless raw_connection.finished?
    end
  end

  def disconnect!
    return unless @raw_connection

    @raw_connection.finish unless @raw_connection.finished?
    @raw_connection = nil
  end

  def transaction(&_block)
    @tx_nesting_level ||= 0

    # Notes: by using "ensure" (as opposed to rescue) we are rolling back
    # both when an exception was raised and when a value was thrown. This
    # also means we have to track whether or not to rollback. i.e. do roll
    # back when we yielded to &block but not otherwise.
    #
    # Also the transaction support is a bit limited: you cannot rollback.
    # Rolling back from inside a nested transaction would require SAVEPOINT
    # support; without the code is simpler at least :)

    if @tx_nesting_level == 0
      exec "BEGIN"
      tx_started = true
    end

    @tx_nesting_level += 1

    return_value = yield

    # Only commit if we started a transaction here.
    if tx_started
      exec "COMMIT"
      tx_committed = true
    end

    return_value
  ensure
    @tx_nesting_level -= 1
    if tx_started && !tx_committed
      exec "ROLLBACK"
    end
  end
end

Version data entries

21 entries across 21 versions & 1 rubygems

Version	Path
simple-sql-0.5.14	lib/simple/sql/connection/raw_connection.rb
simple-sql-0.5.13	lib/simple/sql/connection/raw_connection.rb
simple-sql-0.5.12	lib/simple/sql/connection/raw_connection.rb
simple-sql-0.5.11	lib/simple/sql/connection/raw_connection.rb
simple-sql-0.5.10	lib/simple/sql/connection/raw_connection.rb
simple-sql-0.5.9	lib/simple/sql/connection/raw_connection.rb
simple-sql-0.5.8	lib/simple/sql/connection/raw_connection.rb
simple-sql-0.5.7	lib/simple/sql/connection/raw_connection.rb
simple-sql-0.5.6	lib/simple/sql/connection/raw_connection.rb
simple-sql-0.5.5	lib/simple/sql/connection/raw_connection.rb
simple-sql-0.5.4	lib/simple/sql/connection/raw_connection.rb
simple-sql-0.5.3	lib/simple/sql/connection/raw_connection.rb
simple-sql-0.4.41	lib/simple/sql/connection/raw_connection.rb
simple-sql-0.5.2	lib/simple/sql/connection/raw_connection.rb
simple-sql-0.4.40	lib/simple/sql/connection/raw_connection.rb
simple-sql-0.4.39	lib/simple/sql/connection/raw_connection.rb
simple-sql-0.4.38	lib/simple/sql/connection/raw_connection.rb
simple-sql-0.4.37	lib/simple/sql/connection/raw_connection.rb
simple-sql-0.4.36	lib/simple/sql/connection/raw_connection.rb
simple-sql-0.4.35	lib/simple/sql/connection/raw_connection.rb