Allow to parse heroku-generated database.yml config file.

Cyril Rohr
1 parent 11d3a00c
Showing 1 changed file with 1 additions and 1 deletions Show diff stats
lib/tasks/gitlab/setup.rake
@@ -16,7 +16,7 @@ namespace :gitlab do
     Rake::Task["db:setup"].invoke
-    config = YAML.load_file(File.join(Rails.root,'config','database.yml'))[Rails.env]
+    config = YAML.load(ERB.new(File.read(File.join(Rails.root, "config","database.yml"))).result)
     success = case config["adapter"]
               when /^mysql/ then
                 Rake::Task["add_limits_mysql"].invoke
	@@ -16,7 +16,7 @@ namespace :gitlab do		@@ -16,7 +16,7 @@ namespace :gitlab do
16		16
17	Rake::Task["db:setup"].invoke	17	Rake::Task["db:setup"].invoke
18		18
19	- config = YAML.load_file(File.join(Rails.root,'config','database.yml'))[Rails.env]	19	+ config = YAML.load(ERB.new(File.read(File.join(Rails.root, "config","database.yml"))).result)
20	success = case config["adapter"]	20	success = case config["adapter"]
21	when /^mysql/ then	21	when /^mysql/ then
22	Rake::Task["add_limits_mysql"].invoke	22	Rake::Task["add_limits_mysql"].invoke