Class: ActiveRecord::Tasks::MySQLDatabaseTasks

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

Overview

:nodoc:

Constant Summary collapse

ACCESS_DENIED_ERROR = 
1045

Instance Method Summary collapse

Constructor Details

#initialize(configuration) ⇒ MySQLDatabaseTasks

Returns a new instance of MySQLDatabaseTasks.



10
11
12
 
# File 'activerecord/lib/active_record/tasks/mysql_database_tasks.rb', line 10

def initialize(configuration)
  @configuration = configuration
end

Instance Method Details

#charset ⇒ Object 



50
51
52
 
# File 'activerecord/lib/active_record/tasks/mysql_database_tasks.rb', line 50

def charset
  connection.charset
end

#collation ⇒ Object 



54
55
56
 
# File 'activerecord/lib/active_record/tasks/mysql_database_tasks.rb', line 54

def collation
  connection.collation
end

#create ⇒ Object 



14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
 
# File 'activerecord/lib/active_record/tasks/mysql_database_tasks.rb', line 14

def create
  establish_connection configuration_without_database
  connection.create_database configuration["database"], creation_options
  establish_connection configuration
rescue ActiveRecord::StatementInvalid => error
  if error.message.include?("database exists")
    raise DatabaseAlreadyExists
  else
    raise
  end
rescue error_class => error
  if error.respond_to?(:errno) && error.errno == ACCESS_DENIED_ERROR
    $stdout.print error.message
    establish_connection root_configuration_without_database
    connection.create_database configuration["database"], creation_options
    if configuration["username"] != "root"
      connection.execute grant_statement.gsub(/\s+/, " ").strip
    end
    establish_connection configuration
  else
    $stderr.puts error.inspect
    $stderr.puts "Couldn't create database for #{configuration.inspect}, #{creation_options.inspect}"
    $stderr.puts "(If you set the charset manually, make sure you have a matching collation)" if configuration["encoding"]
  end
end

#drop ⇒ Object 



40
41
42
43
 
# File 'activerecord/lib/active_record/tasks/mysql_database_tasks.rb', line 40

def drop
  establish_connection configuration
  connection.drop_database configuration["database"]
end

#purge ⇒ Object 



45
46
47
48
 
# File 'activerecord/lib/active_record/tasks/mysql_database_tasks.rb', line 45

def purge
  establish_connection configuration
  connection.recreate_database configuration["database"], creation_options
end

#structure_dump(filename, extra_flags) ⇒ Object 



58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
 
# File 'activerecord/lib/active_record/tasks/mysql_database_tasks.rb', line 58

def structure_dump(filename, extra_flags)
  args = prepare_command_options
  args.concat(["--result-file", "#{filename}"])
  args.concat(["--no-data"])
  args.concat(["--routines"])
  args.concat(["--skip-comments"])

  ignore_tables = ActiveRecord::SchemaDumper.ignore_tables
  if ignore_tables.any?
    args += ignore_tables.map { |table| "--ignore-table=#{configuration['database']}.#{table}" }
  end

  args.concat(["#{configuration['database']}"])
  args.unshift(*extra_flags) if extra_flags

  run_cmd("mysqldump", args, "dumping")
end

#structure_load(filename, extra_flags) ⇒ Object 



76
77
78
79
80
81
82
83
 
# File 'activerecord/lib/active_record/tasks/mysql_database_tasks.rb', line 76

def structure_load(filename, extra_flags)
  args = prepare_command_options
  args.concat(["--execute", %{SET FOREIGN_KEY_CHECKS = 0; SOURCE #{filename}; SET FOREIGN_KEY_CHECKS = 1}])
  args.concat(["--database", "#{configuration['database']}"])
  args.unshift(*extra_flags) if extra_flags

  run_cmd("mysql", args, "loading")
end