Class: Simple::SQL::Connection::Reflection

Inherits:
Object
  • Object
show all
Defined in:
lib/simple/sql/connection/reflection.rb

Constant Summary collapse

TIMESTAMP_COLUMN_NAMES = 
%w(inserted_at created_at updated_at)

Instance Method Summary collapse

Constructor Details

#initialize(connection) ⇒ Reflection

Returns a new instance of Reflection.



7
8
9
 
# File 'lib/simple/sql/connection/reflection.rb', line 7

def initialize(connection)
  @connection = connection
end

Instance Method Details

#column_info(table_name) ⇒ Object 



50
51
52
53
 
# File 'lib/simple/sql/connection/reflection.rb', line 50

def column_info(table_name)
  @column_info ||= {}
  @column_info[table_name] ||= _column_info(table_name)
end

#columns(table_name) ⇒ Object 



37
38
39
 
# File 'lib/simple/sql/connection/reflection.rb', line 37

def columns(table_name)
  column_info(table_name).keys
end

#looked_up_pg_classes ⇒ Object 



88
89
90
 
# File 'lib/simple/sql/connection/reflection.rb', line 88

def looked_up_pg_classes
  @looked_up_pg_classes ||= Hash.new { |hsh, key| hsh[key] = _lookup_pg_class(key) }
end

#lookup_pg_class(oid) ⇒ Object 



92
93
94
 
# File 'lib/simple/sql/connection/reflection.rb', line 92

def lookup_pg_class(oid)
  looked_up_pg_classes[oid]
end

#primary_key_columns(table_name) ⇒ Object 



15
16
17
18
19
20
21
22
23
 
# File 'lib/simple/sql/connection/reflection.rb', line 15

def primary_key_columns(table_name)
  @connection.all "    SELECT pg_attribute.attname\n    FROM   pg_index\n    JOIN   pg_attribute ON pg_attribute.attrelid = pg_index.indrelid AND pg_attribute.attnum = ANY(pg_index.indkey)\n    WHERE  pg_index.indrelid = $1::regclass\n    AND    pg_index.indisprimary;\n  SQL\nend\n", table_name

#table_info(schema: "public") ⇒ Object 



41
42
43
44
45
46
47
48
 
# File 'lib/simple/sql/connection/reflection.rb', line 41

def table_info(schema: "public")
  recs = @connection.all "    SELECT table_schema || '.' || table_name AS name, *\n    FROM information_schema.tables\n    WHERE table_schema=$1\n  SQL\n  records_by_attr(recs, :name)\nend\n", schema, into: Hash

#tables(schema: "public") ⇒ Object 



11
12
13
 
# File 'lib/simple/sql/connection/reflection.rb', line 11

def tables(schema: "public")
  table_info(schema: schema).keys
end

#timestamp_columns(table_name) ⇒ Object

timestamp_columns are columns that will be set automatically after inserting or updating a record. This includes:

  • inserted_at (for Ecto)

  • created_at (for ActiveRecord)

  • updated_at (for Ecto and ActiveRecord)



33
34
35
 
# File 'lib/simple/sql/connection/reflection.rb', line 33

def timestamp_columns(table_name)
  columns(table_name) & TIMESTAMP_COLUMN_NAMES
end