db:structure:load for Rails 3.1

db.rake

namespace :db do

  namespace :test do |schema|
    schema[:clone_structure].abandon
    desc "OVERWRITTEN - load the development_structure file using psql shell"
    task :clone_structure => ["db:structure:dump", "db:test:purge"] do
      config = ActiveRecord::Base.configurations['test']
      cmd = "psql #{psql_options} < db/development_structure.sql"
      system cmd
    end
  end

  namespace :structure do |schema|
    desc "load the development_structure file using psql shell"
    task :load => :environment do
      config = ActiveRecord::Base.configurations[Rails.env]
      cmd = "psql #{psql_options} < db/development_structure.sql"
      system cmd
    end
  end

  def psql_options
    config = ActiveRecord::Base.configurations[Rails.env]
    pwd = "-W #{config['password']} " if config['password'].to_s.length > 0
    user = "-U #{config['username'] || config['user']}"
    [pwd, user, config['database']].join(" ")
  end
end