ActiveRecord - 创建没有Rails的数据库

时间:2012-05-07 03:03:14

标签: postgresql activerecord

我无法找到如何从ActiveRecord创建数据库(如果尚不存在)。我看过这个:http://api.rubyonrails.org/classes/ActiveRecord/ConnectionAdapters/PostgreSQLAdapter.html#method-i-create_database,但它不想玩得很好,据我所知......我不确定我在这里缺少什么。这甚至可能吗?

谢谢!

1 个答案:

答案 0 :(得分:4)

我设法凑合了这个适用于heroku和我的本地机器的东西,并且如果不存在则会创建一个数据库。

# ./lib/db_connect.rb

require 'erb'
require 'uri'
require 'em-synchrony/activerecord'

class DbConnect
  attr_accessor :config
  def initialize
    @db = URI.parse(ENV['DATABASE_URL'] || 'http://localhost')
    if @db.scheme == 'postgres' # This section makes Heroku work
      ActiveRecord::Base.establish_connection(
        :adapter  => @db.scheme == 'postgres' ? 'postgresql' : @db.scheme,
        :host     => @db.host,
        :username => @db.user,
        :password => @db.password,
        :database => @db.path[1..-1],
        :encoding => 'utf8'
      )
    else # And this is for my local environment
      environment = ENV['DATABASE_URL'] ? 'production' : 'development'
      @db = YAML.load(ERB.new(File.read('config/database.yml')).result)[environment]
      ActiveRecord::Base.establish_connection(@db)
      @config = ActiveRecord::Base.connection.pool.spec.config
    end
  end
end

# connect to the database
DbConnect.new

这是一个使用它的Rakefile。

# ./Rakefile

$: << './lib'
require 'db_connect'

db = DbConnect.new
config = db.config

desc 'create the database'
task :create_db do
  unless ENV['DATABASE_URL'].present?
    ActiveRecord::Base.establish_connection adapter:'postgresql',username:config[:username],password:config[:password], database: 'postgres'
    ActiveRecord::Base.connection.create_database config[:database]
  else
    raise StandardError, "You cannot do this on Heroku!"
  end
end

desc 'create the users table'
task :create_users_table do
  ActiveRecord::Schema.define(:version => 0) do
    create_table "users", :force => true do |t|
      t.string   "first_name"
      t.string   "last_name"
      t.datetime "created_at", :null => false
      t.datetime "updated_at", :null => false
    end
  end
end

问题解决了。