Class: ActiveRecord::Tasks::PostgreSQLDatabaseTasks

Inherits:
Object
  • Object
show all
Defined in:
lib/active_record/tasks/postgresql_database_tasks.rb

Overview

:nodoc:

Constant Summary collapse

DEFAULT_ENCODING = 
ENV["CHARSET"] || "utf8"
ON_ERROR_STOP_1 = 
"ON_ERROR_STOP=1"
SQL_COMMENT_BEGIN = 
"--"

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(db_config) ⇒ PostgreSQLDatabaseTasks

Returns a new instance of PostgreSQLDatabaseTasks.

[View source]

19
20
21
22
 
# File 'lib/active_record/tasks/postgresql_database_tasks.rb', line 19

def initialize(db_config)
  @db_config = db_config
  @configuration_hash = db_config.configuration_hash
end

Class Method Details

.using_database_configurations? ⇒ Boolean

Returns:

  • (Boolean)
[View source]

15
16
17
 
# File 'lib/active_record/tasks/postgresql_database_tasks.rb', line 15

def self.using_database_configurations?
  true
end

Instance Method Details

#charset ⇒ Object

[View source]

35
36
37
 
# File 'lib/active_record/tasks/postgresql_database_tasks.rb', line 35

def charset
  connection.encoding
end

#collation ⇒ Object

[View source]

39
40
41
 
# File 'lib/active_record/tasks/postgresql_database_tasks.rb', line 39

def collation
  connection.collation
end

#create(master_established = false) ⇒ Object

[View source]

24
25
26
27
28
 
# File 'lib/active_record/tasks/postgresql_database_tasks.rb', line 24

def create(master_established = false)
  establish_master_connection unless master_established
  connection.create_database(db_config.database, configuration_hash.merge(encoding: encoding))
  establish_connection(db_config)
end

#drop ⇒ Object

[View source]

30
31
32
33
 
# File 'lib/active_record/tasks/postgresql_database_tasks.rb', line 30

def drop
  establish_master_connection
  connection.drop_database(db_config.database)
end

#purge ⇒ Object

[View source]

43
44
45
46
47
 
# File 'lib/active_record/tasks/postgresql_database_tasks.rb', line 43

def purge
  clear_active_connections!
  drop
  create true
end

#structure_dump(filename, extra_flags) ⇒ Object

[View source]

49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
 
# File 'lib/active_record/tasks/postgresql_database_tasks.rb', line 49

def structure_dump(filename, extra_flags)
  search_path = \
    case ActiveRecord.dump_schemas
    when :schema_search_path
      configuration_hash[:schema_search_path]
    when :all
      nil
    when String
      ActiveRecord.dump_schemas
    end

  args = ["--schema-only", "--no-privileges", "--no-owner"]
  args.concat(["--file", filename])

  args.concat(Array(extra_flags)) if extra_flags

  unless search_path.blank?
    args += search_path.split(",").map do |part|
      "--schema=#{part.strip}"
    end
  end

  ignore_tables = ActiveRecord::SchemaDumper.ignore_tables
  if ignore_tables.any?
    args += ignore_tables.flat_map { |table| ["-T", table] }
  end

  args << db_config.database
  run_cmd("pg_dump", args, "dumping")
  remove_sql_header_comments(filename)
  File.open(filename, "a") { |f| f << "SET search_path TO #{connection.schema_search_path};\n\n" }
end

#structure_load(filename, extra_flags) ⇒ Object

[View source]

82
83
84
85
86
87
 
# File 'lib/active_record/tasks/postgresql_database_tasks.rb', line 82

def structure_load(filename, extra_flags)
  args = ["--set", ON_ERROR_STOP_1, "--quiet", "--no-psqlrc", "--output", File::NULL, "--file", filename]
  args.concat(Array(extra_flags)) if extra_flags
  args << db_config.database
  run_cmd("psql", args, "loading")
end