Module: Sequel::StatementTimeout

Defined in:
lib/sequel/extensions/statement_timeout.rb

Instance Method Summary collapse

Instance Method Details

#with_statement_timeout(timeout_seconds = 20) ⇒ Object 



3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
 
# File 'lib/sequel/extensions/statement_timeout.rb', line 3

def with_statement_timeout(timeout_seconds = 20)
  # Might not have postgres class loaded, use class name
  if self.class.name == "Sequel::Postgres::Database"
      # Don't want to use a transaction because this will often be a read and a transaction is unnecessary.
      # Also, when using it for clean, want to control the transactions outside this.
      current_statement_timeout = execute("show statement_timeout") { |r| r.first.values.first }
      run("SET statement_timeout = '#{timeout_seconds}s'")
      begin
        yield
      ensure
        run("SET statement_timeout = '#{current_statement_timeout}'")
      end
  else
    yield
  end
end