Class: Baza::Driver::Pg::Database

Inherits:
Baza::Database show all
Defined in:
lib/baza/driver/pg/database.rb

Constant Summary collapse

CREATE_ALLOWED_KEYS = 
[:columns, :indexes, :temp, :return_sql].freeze

Instance Attribute Summary

Attributes inherited from Baza::Database

#db, #driver, #name

Instance Method Summary collapse

Methods inherited from Baza::Database

#import_file!, #initialize, #name_changed?, #name_was, #table_exists?, #to_param

Methods included from Baza::DatabaseModelFunctionality

#model_name, #to_model

Constructor Details

This class inherits a constructor from Baza::Database

Instance Method Details

#create_table(table_name, data, args = nil) ⇒ Object

Creates a new table by the given name and data.



58
59
60
61
62
 
# File 'lib/baza/driver/pg/database.rb', line 58

def create_table(table_name, data, args = nil)
  use do
    db.tables.create(table_name, data, args)
  end
end

#drop ⇒ Object 



7
8
9
10
11
12
13
14
 
# File 'lib/baza/driver/pg/database.rb', line 7

def drop
  with_cloned_conn_and_terminated_connections do |cloned_conn|
    # Drop the database
    cloned_conn.query("DROP DATABASE #{db.sep_database}#{db.escape_database(name)}#{db.sep_database}")
  end

  self
end

#save! ⇒ Object 



2
3
4
5
 
# File 'lib/baza/driver/pg/database.rb', line 2

def save!
  rename(name) unless name.to_s == name_was.to_s
  self
end

#table(table_name) ⇒ Object

Raises:



16
17
18
19
20
 
# File 'lib/baza/driver/pg/database.rb', line 16

def table(table_name)
  table = tables(name: table_name).first
  raise Baza::Errors::TableNotFound unless table
  table
end

#tables(args = {}) ⇒ Object 



22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
 
# File 'lib/baza/driver/pg/database.rb', line 22

def tables(args = {})
  tables_list = [] unless block_given?

  where_args = {
    table_catalog: name,
    table_schema: "public"
  }
  where_args[:table_name] = args.fetch(:name) if args[:name]

  use do
    db.select([:information_schema, :tables], where_args, orderby: :table_name) do |table_data|
      table = Baza::Driver::Pg::Table.new(
        driver: db.driver,
        data: table_data
      )

      next if table.native?

      if tables_list
        tables_list << table
      else
        yield table
      end
    end
  end

  tables_list
end

#use(&blk) ⇒ Object 



51
52
53
54
 
# File 'lib/baza/driver/pg/database.rb', line 51

def use(&blk)
  db.with_database(name, &blk)
  self
end