forked from gnufied/backgroundrb
/
backgroundrb_tasks.rake
99 lines (83 loc) · 3.68 KB
/
backgroundrb_tasks.rake
1
2
3
4
5
6
7
8
9
10
11
12
13
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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
namespace :backgroundrb do
require 'yaml'
desc 'Setup backgroundrb in your rails application'
task :setup do
script_dest = "#{RAILS_ROOT}/script/backgroundrb"
script_src = File.dirname(__FILE__) + "/../script/backgroundrb"
FileUtils.chmod 0774, script_src
defaults = {:backgroundrb => {:ip => '0.0.0.0',:port => 11006 } }
config_dest = "#{RAILS_ROOT}/config/backgroundrb.yml"
unless File.exists?(config_dest)
puts "Copying backgroundrb.yml config file to #{config_dest}"
File.open(config_dest, 'w') { |f| f.write(YAML.dump(defaults)) }
end
unless File.exists?(script_dest)
puts "Copying backgroundrb script to #{script_dest}"
FileUtils.cp_r(script_src, script_dest)
end
workers_dest = "#{RAILS_ROOT}/lib/workers"
unless File.exists?(workers_dest)
puts "Creating #{workers_dest}"
FileUtils.mkdir(workers_dest)
end
test_helper_dest = "#{RAILS_ROOT}/test/bdrb_test_helper.rb"
test_helper_src = File.dirname(__FILE__) + "/../script/bdrb_test_helper.rb"
unless File.exists?(test_helper_dest)
puts "Copying Worker Test helper file #{test_helper_dest}"
FileUtils.cp_r(test_helper_src,test_helper_dest)
end
worker_env_loader_dest = "#{RAILS_ROOT}/script/load_worker_env.rb"
worker_env_loader_src = File.join(File.dirname(__FILE__),"..","script","load_worker_env.rb")
unless File.exists? worker_env_loader_dest
puts "Copying Worker envionment loader file #{worker_env_loader_dest}"
FileUtils.cp_r(worker_env_loader_src,worker_env_loader_dest)
end
# Generate the migration
Rake::Task['backgroundrb:queue_migration'].invoke
end
desc "Drops and recreate backgroundrb queue table"
task :redo_queue => :queue_migration do
end
desc 'update backgroundrb config files from your rails application'
task :update do
temp_scripts = ["backgroundrb","load_worker_env.rb"].map {|x| "#{RAILS_ROOT}/script/#{x}"}
temp_scripts.each do |file_name|
if File.exists?(file_name)
puts "Removing #{file_name} ..."
FileUtils.rm(file_name,:force => true)
end
end
new_temp_scripts = ["backgroundrb","load_worker_env.rb"].map {|x| File.dirname(__FILE__) + "/../script/#{x}" }
new_temp_scripts.each do |file_name|
puts "Updating file #{File.expand_path(file_name)} ..."
FileUtils.cp_r(file_name,"#{RAILS_ROOT}/script/")
end
end
desc 'Generate a migration for the backgroundrb queue table. The migration name can be ' +
'specified with the MIGRATION environment variable.'
task :queue_migration => :environment do
raise "Task unavailable to this database (no migration support)" unless ActiveRecord::Base.connection.supports_migrations?
require 'rails_generator'
require 'rails_generator/scripts/generate'
Rails::Generator::Scripts::Generate.new.run(['bdrb_migration', ENV['MIGRATION'] || 'CreateBackgroundrbQueueTable'])
end
desc 'Remove backgroundrb from your rails application'
task :remove do
script_src = "#{RAILS_ROOT}/script/backgroundrb"
temp_scripts = ["backgroundrb","load_worker_env.rb"].map {|x| "#{RAILS_ROOT}/script/#{x}"}
if File.exists?(script_src)
puts "Removing #{script_src} ..."
FileUtils.rm(script_src, :force => true)
end
test_helper_src = "#{RAILS_ROOT}/test/bdrb_test_helper.rb"
if File.exists?(test_helper_src)
puts "Removing backgroundrb test helper.."
FileUtils.rm(test_helper_src,:force => true)
end
workers_dest = "#{RAILS_ROOT}/lib/workers"
if File.exists?(workers_dest) && Dir.entries("#{workers_dest}").size == 2
puts "#{workers_dest} is empty...deleting!"
FileUtils.rmdir(workers_dest)
end
end
end