Module: Sequel::Postgres::AdapterMethods

Included in:
Adapter
Defined in:
lib/sequel_core/adapters/shared/postgres.rb

Overview

Methods shared by adapter/connection instances.

Constant Summary collapse

SELECT_CURRVAL = 
"SELECT currval('%s')".freeze
SELECT_CUSTOM_SEQUENCE = 
<<-end_sql
  SELECT '"' || name.nspname || '"."' || CASE  
      WHEN split_part(def.adsrc, '''', 2) ~ '.' THEN  
        substr(split_part(def.adsrc, '''', 2),  
               strpos(split_part(def.adsrc, '''', 2), '.')+1) 
      ELSE split_part(def.adsrc, '''', 2)  
    END || '"'
  FROM pg_class t
  JOIN pg_namespace  name ON (t.relnamespace = name.oid)
  JOIN pg_attribute  attr ON (t.oid = attrelid)
  JOIN pg_attrdef    def  ON (adrelid = attrelid AND adnum = attnum)
  JOIN pg_constraint cons ON (conrelid = adrelid AND adnum = conkey[1])
  WHERE cons.contype = 'p'
    AND def.adsrc ~* 'nextval'
    AND name.nspname = '%s'
    AND t.relname = '%s'
end_sql
SELECT_PK = 
<<-end_sql
  SELECT pg_attribute.attname
  FROM pg_class, pg_attribute, pg_index, pg_namespace
  WHERE pg_class.oid = pg_attribute.attrelid
    AND pg_class.relnamespace  = pg_namespace.oid
    AND pg_class.oid = pg_index.indrelid
    AND pg_index.indkey[0] = pg_attribute.attnum
    AND pg_index.indisprimary = 't'
    AND pg_namespace.nspname = '%s'
    AND pg_class.relname = '%s'
end_sql
SELECT_SERIAL_SEQUENCE = 
<<-end_sql
  SELECT  '"' || name.nspname || '"."' || seq.relname || '"'
  FROM pg_class seq, pg_attribute attr, pg_depend dep,
    pg_namespace name, pg_constraint cons
  WHERE seq.oid = dep.objid
    AND seq.relnamespace  = name.oid
    AND seq.relkind = 'S'
    AND attr.attrelid = dep.refobjid
    AND attr.attnum = dep.refobjsubid
    AND attr.attrelid = cons.conrelid
    AND attr.attnum = cons.conkey[1]
    AND cons.contype = 'p'
    AND name.nspname = '%s'
    AND seq.relname = '%s'
end_sql

Instance Attribute Summary collapse

Instance Method Summary collapse

Instance Attribute Details

#db=(value) ⇒ Object (writeonly)

Sets the attribute db

Parameters:

  • value —

    the value to set the attribute db to.



46
47
48
 
# File 'lib/sequel_core/adapters/shared/postgres.rb', line 46

def db=(value)
  @db = value
end

#transaction_depth ⇒ Object

Depth of the current transaction on this connection, used to implement multi-level transactions with savepoints.



95
96
97
 
# File 'lib/sequel_core/adapters/shared/postgres.rb', line 95

def transaction_depth
  @transaction_depth
end

Instance Method Details

#apply_connection_settings ⇒ Object

Apply connection settings for this connection. Currently, turns standard_conforming_strings ON if Postgres.force_standard_strings is true.



100
101
102
103
104
105
106
107
108
109
 
# File 'lib/sequel_core/adapters/shared/postgres.rb', line 100

def apply_connection_settings
  if Postgres.force_standard_strings
    sql = "SET standard_conforming_strings = ON"
    @db.log_info(sql)
    # This setting will only work on PostgreSQL 8.2 or greater
    # and we don't know the server version at this point, so
    # try it unconditionally and rescue any errors.
    execute(sql) rescue nil
  end
end

#last_insert_id(sequence) ⇒ Object

Get the last inserted value for the given sequence.



112
113
114
115
116
117
118
119
 
# File 'lib/sequel_core/adapters/shared/postgres.rb', line 112

def last_insert_id(sequence)
  sql = SELECT_CURRVAL % sequence
  @db.log_info(sql)
  execute(sql) do |r|
    val = single_value(r)
    return val.to_i if val
  end
end

#primary_key(schema, table) ⇒ Object

Get the primary key for the given table.



122
123
124
125
126
127
128
 
# File 'lib/sequel_core/adapters/shared/postgres.rb', line 122

def primary_key(schema, table)
  sql = SELECT_PK % [schema, table]
  @db.log_info(sql)
  execute(sql) do |r|
    return single_value(r)
  end
end

#sequence(schema, table) ⇒ Object

Get the primary key and sequence for the given table.



131
132
133
134
135
136
137
138
139
140
141
142
143
144
 
# File 'lib/sequel_core/adapters/shared/postgres.rb', line 131

def sequence(schema, table)
  sql = SELECT_SERIAL_SEQUENCE % [schema, table]
  @db.log_info(sql)
  execute(sql) do |r|
    seq = single_value(r)
    return seq if seq
  end
  
  sql = SELECT_CUSTOM_SEQUENCE % [schema, table]
  @db.log_info(sql)
  execute(sql) do |r|
    return single_value(r)
  end
end