Skip to content

HTTPS clone URL

Subversion checkout URL

You can clone with
or
.
Download ZIP
Tree: d29727235a
Fetching contributors…

Cannot retrieve contributors at this time

77 lines (64 sloc) 2.228 kB
class ActiveRecord::Tasks::MySQLDatabaseTasks
DEFAULT_CHARSET = ENV['CHARSET'] || 'utf8'
DEFAULT_COLLATION = ENV['COLLATION'] || 'utf8_unicode_ci'
ACCESS_DENIED_ERROR = 1045
delegate :connection, :establish_connection, :to => ActiveRecord::Base
def initialize(configuration)
@configuration = configuration
end
def create
establish_connection configuration_without_database
connection.create_database configuration['database'], creation_options
establish_connection configuration
rescue error_class => error
raise error unless error.errno == ACCESS_DENIED_ERROR
$stdout.print error.error
establish_connection root_configuration_without_database
connection.create_database configuration['database'], creation_options
connection.execute grant_statement.gsub(/\s+/, ' ').strip
establish_connection configuration
rescue error_class => error
$stderr.puts error.error
$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['charset']
end
private
attr_reader :configuration
def configuration_without_database
configuration.merge('database' => nil)
end
def creation_options
{
:charset => (configuration['charset'] || DEFAULT_CHARSET),
:collation => (configuration['collation'] || DEFAULT_COLLATION)
}
end
def error_class
case configuration['adapter']
when /jdbc/
require 'active_record/railties/jdbcmysql_error'
error_class = ArJdbcMySQL::Error
when /mysql2/
Mysql2::Error
else
Mysql::Error
end
end
def grant_statement
<<-SQL
GRANT ALL PRIVILEGES ON #{configuration['database']}.*
TO '#{configuration['username']}'@'localhost'
IDENTIFIED BY '#{configuration['password']}' WITH GRANT OPTION;
SQL
end
def root_configuration_without_database
configuration_without_database.merge(
'username' => 'root',
'password' => root_password
)
end
def root_password
$stdout.print "Please provide the root password for your mysql installation\n>"
$stdin.gets.strip
end
end
Jump to Line
Something went wrong with that request. Please try again.