blackhoundjob 1.0.2
Sign up to get free protection for your applications and to get access to all the features.
- data/HISTORY +64 -0
- data/README +308 -0
- data/TODO +40 -0
- data/bin/bj +679 -0
- data/bj-1.0.2.gem +0 -0
- data/gemspec.rb +30 -0
- data/install.rb +210 -0
- data/lib/bj/api.rb +161 -0
- data/lib/bj/attributes.rb +120 -0
- data/lib/bj/bj.rb +72 -0
- data/lib/bj/errors.rb +4 -0
- data/lib/bj/joblist.rb +112 -0
- data/lib/bj/logger.rb +50 -0
- data/lib/bj/runner.rb +357 -0
- data/lib/bj/stdext.rb +86 -0
- data/lib/bj/table.rb +384 -0
- data/lib/bj/util.rb +111 -0
- data/lib/bj.rb +87 -0
- data/rakefile +3 -0
- metadata +113 -0
data/install.rb
ADDED
@@ -0,0 +1,210 @@
|
|
1
|
+
#!/usr/bin/env ruby
|
2
|
+
require 'rbconfig'
|
3
|
+
require 'find'
|
4
|
+
require 'ftools'
|
5
|
+
require 'tempfile'
|
6
|
+
include Config
|
7
|
+
|
8
|
+
LIBDIR = "lib"
|
9
|
+
LIBDIR_MODE = 0644
|
10
|
+
|
11
|
+
BINDIR = "bin"
|
12
|
+
BINDIR_MODE = 0755
|
13
|
+
|
14
|
+
|
15
|
+
$srcdir = CONFIG["srcdir"]
|
16
|
+
$version = CONFIG["MAJOR"]+"."+CONFIG["MINOR"]
|
17
|
+
$libdir = File.join(CONFIG["libdir"], "ruby", $version)
|
18
|
+
$archdir = File.join($libdir, CONFIG["arch"])
|
19
|
+
$site_libdir = $:.find {|x| x =~ /site_ruby$/}
|
20
|
+
$bindir = CONFIG["bindir"] || CONFIG['BINDIR']
|
21
|
+
$ruby_install_name = CONFIG['ruby_install_name'] || CONFIG['RUBY_INSTALL_NAME'] || 'ruby'
|
22
|
+
$ruby_ext = CONFIG['EXEEXT'] || ''
|
23
|
+
$ruby = File.join($bindir, ($ruby_install_name + $ruby_ext))
|
24
|
+
|
25
|
+
if !$site_libdir
|
26
|
+
$site_libdir = File.join($libdir, "site_ruby")
|
27
|
+
elsif $site_libdir !~ %r/#{Regexp.quote($version)}/
|
28
|
+
$site_libdir = File.join($site_libdir, $version)
|
29
|
+
end
|
30
|
+
|
31
|
+
def install_rb(srcdir=nil, destdir=nil, mode=nil, bin=nil)
|
32
|
+
#{{{
|
33
|
+
path = []
|
34
|
+
dir = []
|
35
|
+
Find.find(srcdir) do |f|
|
36
|
+
next unless FileTest.file?(f)
|
37
|
+
next if (f = f[srcdir.length+1..-1]) == nil
|
38
|
+
next if (/CVS$/ =~ File.dirname(f))
|
39
|
+
next if f =~ %r/\.lnk/
|
40
|
+
path.push f
|
41
|
+
dir |= [File.dirname(f)]
|
42
|
+
end
|
43
|
+
for f in dir
|
44
|
+
next if f == "."
|
45
|
+
next if f == "CVS"
|
46
|
+
File::makedirs(File.join(destdir, f))
|
47
|
+
end
|
48
|
+
for f in path
|
49
|
+
next if (/\~$/ =~ f)
|
50
|
+
next if (/^\./ =~ File.basename(f))
|
51
|
+
unless bin
|
52
|
+
File::install(File.join(srcdir, f), File.join(destdir, f), mode, true)
|
53
|
+
else
|
54
|
+
from = File.join(srcdir, f)
|
55
|
+
to = File.join(destdir, f)
|
56
|
+
shebangify(from) do |sf|
|
57
|
+
$deferr.print from, " -> ", File::catname(from, to), "\n"
|
58
|
+
$deferr.printf "chmod %04o %s\n", mode, to
|
59
|
+
File::install(sf, to, mode, false)
|
60
|
+
end
|
61
|
+
end
|
62
|
+
end
|
63
|
+
#}}}
|
64
|
+
end
|
65
|
+
def shebangify f
|
66
|
+
#{{{
|
67
|
+
open(f) do |fd|
|
68
|
+
buf = fd.read 42
|
69
|
+
if buf =~ %r/^\s*#\s*!.*ruby/o
|
70
|
+
ftmp = Tempfile::new("#{ $$ }_#{ File::basename(f) }")
|
71
|
+
begin
|
72
|
+
fd.rewind
|
73
|
+
ftmp.puts "#!#{ $ruby }"
|
74
|
+
while((buf = fd.read(8192)))
|
75
|
+
ftmp.write buf
|
76
|
+
end
|
77
|
+
ftmp.close
|
78
|
+
yield ftmp.path
|
79
|
+
ensure
|
80
|
+
ftmp.close!
|
81
|
+
end
|
82
|
+
else
|
83
|
+
yield f
|
84
|
+
end
|
85
|
+
end
|
86
|
+
#}}}
|
87
|
+
end
|
88
|
+
def ARGV.switch
|
89
|
+
#{{{
|
90
|
+
return nil if self.empty?
|
91
|
+
arg = self.shift
|
92
|
+
return nil if arg == '--'
|
93
|
+
if arg =~ /^-(.)(.*)/
|
94
|
+
return arg if $1 == '-'
|
95
|
+
raise 'unknown switch "-"' if $2.index('-')
|
96
|
+
self.unshift "-#{$2}" if $2.size > 0
|
97
|
+
"-#{$1}"
|
98
|
+
else
|
99
|
+
self.unshift arg
|
100
|
+
nil
|
101
|
+
end
|
102
|
+
#}}}
|
103
|
+
end
|
104
|
+
def ARGV.req_arg
|
105
|
+
#{{{
|
106
|
+
self.shift || raise('missing argument')
|
107
|
+
#}}}
|
108
|
+
end
|
109
|
+
def linkify d, linked = []
|
110
|
+
#--{{{
|
111
|
+
if test ?d, d
|
112
|
+
versioned = Dir[ File::join(d, "*-[0-9].[0-9].[0-9].rb") ]
|
113
|
+
versioned.each do |v|
|
114
|
+
src, dst = v, v.gsub(%r/\-[\d\.]+\.rb$/, '.rb')
|
115
|
+
lnk = nil
|
116
|
+
begin
|
117
|
+
if test ?l, dst
|
118
|
+
lnk = "#{ dst }.lnk"
|
119
|
+
puts "#{ dst } -> #{ lnk }"
|
120
|
+
File::rename dst, lnk
|
121
|
+
end
|
122
|
+
unless test ?e, dst
|
123
|
+
puts "#{ src } -> #{ dst }"
|
124
|
+
File::copy src, dst
|
125
|
+
linked << dst
|
126
|
+
end
|
127
|
+
ensure
|
128
|
+
if lnk
|
129
|
+
at_exit do
|
130
|
+
puts "#{ lnk } -> #{ dst }"
|
131
|
+
File::rename lnk, dst
|
132
|
+
end
|
133
|
+
end
|
134
|
+
end
|
135
|
+
end
|
136
|
+
end
|
137
|
+
linked
|
138
|
+
#--}}}
|
139
|
+
end
|
140
|
+
|
141
|
+
|
142
|
+
#
|
143
|
+
# main program
|
144
|
+
#
|
145
|
+
|
146
|
+
libdir = $site_libdir
|
147
|
+
bindir = $bindir
|
148
|
+
no_linkify = false
|
149
|
+
linked = nil
|
150
|
+
help = false
|
151
|
+
|
152
|
+
usage = <<-usage
|
153
|
+
#{ File::basename $0 }
|
154
|
+
-d, --destdir <destdir>
|
155
|
+
-l, --libdir <libdir>
|
156
|
+
-b, --bindir <bindir>
|
157
|
+
-r, --ruby <ruby>
|
158
|
+
-n, --no_linkify
|
159
|
+
-s, --sudo
|
160
|
+
-h, --help
|
161
|
+
usage
|
162
|
+
|
163
|
+
begin
|
164
|
+
while switch = ARGV.switch
|
165
|
+
case switch
|
166
|
+
when '-d', '--destdir'
|
167
|
+
libdir = ARGV.req_arg
|
168
|
+
when '-l', '--libdir'
|
169
|
+
libdir = ARGV.req_arg
|
170
|
+
when '-b', '--bindir'
|
171
|
+
bindir = ARGV.req_arg
|
172
|
+
when '-r', '--ruby'
|
173
|
+
$ruby = ARGV.req_arg
|
174
|
+
when '-n', '--no_linkify'
|
175
|
+
no_linkify = true
|
176
|
+
when '-s', '--sudo'
|
177
|
+
sudo = 'sudo'
|
178
|
+
when '-h', '--help'
|
179
|
+
help = true
|
180
|
+
else
|
181
|
+
raise "unknown switch #{switch.dump}"
|
182
|
+
end
|
183
|
+
end
|
184
|
+
rescue
|
185
|
+
STDERR.puts $!.to_s
|
186
|
+
STDERR.puts usage
|
187
|
+
exit 1
|
188
|
+
end
|
189
|
+
|
190
|
+
if help
|
191
|
+
STDOUT.puts usage
|
192
|
+
exit
|
193
|
+
end
|
194
|
+
|
195
|
+
system "#{ sudo } #{ $ruby } pre-install.rb" if test(?s, 'pre-install.rb')
|
196
|
+
|
197
|
+
unless no_linkify
|
198
|
+
linked = linkify('lib') + linkify('bin')
|
199
|
+
end
|
200
|
+
|
201
|
+
system "#{ $ruby } extconf.rb && make && #{ sudo } make install" if test(?s, 'extconf.rb')
|
202
|
+
|
203
|
+
install_rb(LIBDIR, libdir, LIBDIR_MODE)
|
204
|
+
install_rb(BINDIR, bindir, BINDIR_MODE, bin=true)
|
205
|
+
|
206
|
+
if linked
|
207
|
+
linked.each{|path| File::rm_f path}
|
208
|
+
end
|
209
|
+
|
210
|
+
system "#{ sudo } #{ $ruby } post-install.rb" if test(?s, 'post-install.rb')
|
data/lib/bj/api.rb
ADDED
@@ -0,0 +1,161 @@
|
|
1
|
+
class Bj
|
2
|
+
#
|
3
|
+
# the api exposes nearly all the bj code you'll likely need, with the
|
4
|
+
# exception of accessing the job table for searching, which is done using
|
5
|
+
#
|
6
|
+
# eg.
|
7
|
+
#
|
8
|
+
# Bj.table.job.find :all
|
9
|
+
#
|
10
|
+
module API
|
11
|
+
#
|
12
|
+
# submit jobs for background processing. 'jobs' can be a string or array of
|
13
|
+
# strings. options are applied to each job in the 'jobs', and the list of
|
14
|
+
# submitted jobs is always returned. options (string or symbol) can be
|
15
|
+
#
|
16
|
+
# :rails_env => production|development|key_in_database_yml
|
17
|
+
# when given this keyword causes bj to submit jobs to the
|
18
|
+
# specified database. default is RAILS_ENV.
|
19
|
+
#
|
20
|
+
# :priority => any number, including negative ones. default is zero.
|
21
|
+
#
|
22
|
+
# :tag => a tag added to the job. simply makes searching easier.
|
23
|
+
#
|
24
|
+
# :env => a hash specifying any additional environment vars the background
|
25
|
+
# process should have.
|
26
|
+
#
|
27
|
+
# :stdin => any stdin the background process should have.
|
28
|
+
#
|
29
|
+
# eg:
|
30
|
+
#
|
31
|
+
# jobs = Bj.submit 'echo foobar', :tag => 'simple job'
|
32
|
+
#
|
33
|
+
# jobs = Bj.submit '/bin/cat', :stdin => 'in the hat', :priority => 42
|
34
|
+
#
|
35
|
+
# jobs = Bj.submit './script/runner ./scripts/a.rb', :rails_env => 'production'
|
36
|
+
#
|
37
|
+
# jobs = Bj.submit './script/runner /dev/stdin', :stdin => 'p RAILS_ENV', :tag => 'dynamic ruby code'
|
38
|
+
#
|
39
|
+
# jobs = Bj.submit array_of_commands, :priority => 451
|
40
|
+
#
|
41
|
+
# when jobs are run, they are run in RAILS_ROOT. various attributes are
|
42
|
+
# available *only* once the job has finished. you can check whether or not
|
43
|
+
# a job is finished by using the #finished method, which simple does a
|
44
|
+
# reload and checks to see if the exit_status is non-nil.
|
45
|
+
#
|
46
|
+
# eg:
|
47
|
+
#
|
48
|
+
# jobs = Bj.submit list_of_jobs, :tag => 'important'
|
49
|
+
# ...
|
50
|
+
#
|
51
|
+
# jobs.each do |job|
|
52
|
+
# if job.finished?
|
53
|
+
# p job.exit_status
|
54
|
+
# p job.stdout
|
55
|
+
# p job.stderr
|
56
|
+
# end
|
57
|
+
# end
|
58
|
+
#
|
59
|
+
#
|
60
|
+
def submit jobs, options = {}, &block
|
61
|
+
options.to_options!
|
62
|
+
Bj.transaction(options) do
|
63
|
+
table.job.submit jobs, options, &block
|
64
|
+
end
|
65
|
+
ensure
|
66
|
+
Bj.runner.tickle unless options[:no_tickle]
|
67
|
+
end
|
68
|
+
#
|
69
|
+
# this method changes the context under which bj is operating. a context is
|
70
|
+
# a RAILS_ENV. the method accepts a block and it used to alter the
|
71
|
+
# behaviour of the bj lib on a global scale such that all operations,
|
72
|
+
# spawning of background runnner processes, etc, occur in that context.
|
73
|
+
#
|
74
|
+
# eg:
|
75
|
+
#
|
76
|
+
# Bj.in :production do
|
77
|
+
# Bj.submit './script/runner ./scripts/facebook_notification.rb'
|
78
|
+
# end
|
79
|
+
#
|
80
|
+
# Bj.in :development do
|
81
|
+
# Bj.submit 'does_this_eat_memory.exe'
|
82
|
+
# end
|
83
|
+
#
|
84
|
+
def in rails_env = Bj.rails_env, &block
|
85
|
+
transaction(:rails_env => rails_env.to_s, &block)
|
86
|
+
end
|
87
|
+
#
|
88
|
+
# list simply returns a list of all jobs in the job table
|
89
|
+
#
|
90
|
+
def list options = {}, &block
|
91
|
+
options.to_options!
|
92
|
+
Bj.transaction(options) do
|
93
|
+
options.delete :rails_env
|
94
|
+
table.job.find(:all, options)
|
95
|
+
end
|
96
|
+
end
|
97
|
+
#
|
98
|
+
#
|
99
|
+
#
|
100
|
+
def run options = {}
|
101
|
+
runner.run options
|
102
|
+
end
|
103
|
+
#
|
104
|
+
# generate a migration and migrate a database (production/development/etc)
|
105
|
+
#
|
106
|
+
def setup options = {}
|
107
|
+
options.to_options!
|
108
|
+
chroot do
|
109
|
+
generate_migration options
|
110
|
+
migrate options
|
111
|
+
end
|
112
|
+
end
|
113
|
+
#
|
114
|
+
# generate_migration, suprisingly, generates the single migration needed for
|
115
|
+
# bj. you'll notice the the migration is very short as the migration
|
116
|
+
# classes themselves are inner classes of the respective bj table class.
|
117
|
+
# see lib/bj/table.rb for details.
|
118
|
+
#
|
119
|
+
def generate_migration options = {}
|
120
|
+
options.to_options!
|
121
|
+
chroot do
|
122
|
+
before = Dir.glob "./db/migrate/*"
|
123
|
+
n = Dir.glob("./db/migrate/*_bj_*").size
|
124
|
+
classname = "BjMigration#{ n }"
|
125
|
+
util.spawn "#{ Bj.ruby } ./script/generate migration #{ classname }", options rescue nil
|
126
|
+
after = Dir.glob "./db/migrate/*"
|
127
|
+
candidates = after - before
|
128
|
+
case candidates.size
|
129
|
+
when 0
|
130
|
+
false
|
131
|
+
when 1
|
132
|
+
generated = candidates.first
|
133
|
+
open(generated, "w"){|fd| fd.puts Bj.table.migration_code(classname)}
|
134
|
+
Bj.logger.info{ "generated <#{ generated }>" }
|
135
|
+
generated
|
136
|
+
else
|
137
|
+
raise "ambiguous migration <#{ candidates.inspect }>"
|
138
|
+
end
|
139
|
+
end
|
140
|
+
end
|
141
|
+
#
|
142
|
+
# migrate a database (production|development|etc)
|
143
|
+
#
|
144
|
+
def migrate options = {}
|
145
|
+
options.to_options!
|
146
|
+
chroot do
|
147
|
+
util.spawn "rake RAILS_ENV=#{ Bj.rails_env } db:migrate", options
|
148
|
+
end
|
149
|
+
end
|
150
|
+
#
|
151
|
+
# install plugin into this rails app
|
152
|
+
#
|
153
|
+
def plugin options = {}
|
154
|
+
options.to_options!
|
155
|
+
chroot do
|
156
|
+
util.spawn "#{ Bj.ruby } ./script/plugin install http://codeforpeople.rubyforge.org/svn/rails/plugins/bj --force", options
|
157
|
+
end
|
158
|
+
end
|
159
|
+
end
|
160
|
+
send :extend, API
|
161
|
+
end
|
@@ -0,0 +1,120 @@
|
|
1
|
+
module Attributes
|
2
|
+
Attributes::VERSION = '5.0.0' unless defined? Attributes::VERSION
|
3
|
+
def self.version() Attributes::VERSION end
|
4
|
+
|
5
|
+
class List < ::Array
|
6
|
+
def << element
|
7
|
+
super
|
8
|
+
self
|
9
|
+
ensure
|
10
|
+
uniq!
|
11
|
+
index!
|
12
|
+
end
|
13
|
+
def index!
|
14
|
+
@index ||= Hash.new
|
15
|
+
each{|element| @index[element] = true}
|
16
|
+
end
|
17
|
+
def include? element
|
18
|
+
@index ||= Hash.new
|
19
|
+
@index[element] ? true : false
|
20
|
+
end
|
21
|
+
def initializers
|
22
|
+
@initializers ||= Hash.new
|
23
|
+
end
|
24
|
+
end
|
25
|
+
|
26
|
+
def attributes *a, &b
|
27
|
+
unless a.empty?
|
28
|
+
returned = Hash.new
|
29
|
+
|
30
|
+
hashes, names = a.partition{|x| Hash === x}
|
31
|
+
names_and_defaults = {}
|
32
|
+
hashes.each{|h| names_and_defaults.update h}
|
33
|
+
names.flatten.compact.each{|name| names_and_defaults.update name => nil}
|
34
|
+
|
35
|
+
initializers = __attributes__.initializers
|
36
|
+
|
37
|
+
names_and_defaults.each do |name, default|
|
38
|
+
raise NameError, "bad instance variable name '@#{ name }'" if "@#{ name }" =~ %r/[!?=]$/o
|
39
|
+
name = name.to_s
|
40
|
+
|
41
|
+
initialize = b || lambda { default }
|
42
|
+
initializer = lambda do |this|
|
43
|
+
Object.instance_method('instance_eval').bind(this).call &initialize
|
44
|
+
end
|
45
|
+
initializer_id = initializer.object_id
|
46
|
+
__attributes__.initializers[name] = initializer
|
47
|
+
|
48
|
+
module_eval <<-code
|
49
|
+
def #{ name }=(*value, &block)
|
50
|
+
value.unshift block if block
|
51
|
+
@#{ name } = value.first
|
52
|
+
end
|
53
|
+
code
|
54
|
+
|
55
|
+
module_eval <<-code
|
56
|
+
def #{ name }(*value, &block)
|
57
|
+
value.unshift block if block
|
58
|
+
return self.send('#{ name }=', value.first) unless value.empty?
|
59
|
+
#{ name }! unless defined? @#{ name }
|
60
|
+
@#{ name }
|
61
|
+
end
|
62
|
+
code
|
63
|
+
|
64
|
+
module_eval <<-code
|
65
|
+
def #{ name }!
|
66
|
+
initializer = ObjectSpace._id2ref #{ initializer_id }
|
67
|
+
self.#{ name } = initializer.call(self)
|
68
|
+
@#{ name }
|
69
|
+
end
|
70
|
+
code
|
71
|
+
|
72
|
+
module_eval <<-code
|
73
|
+
def #{ name }?
|
74
|
+
#{ name }
|
75
|
+
end
|
76
|
+
code
|
77
|
+
|
78
|
+
attributes << name
|
79
|
+
returned[name] = initializer
|
80
|
+
end
|
81
|
+
|
82
|
+
returned
|
83
|
+
else
|
84
|
+
begin
|
85
|
+
__attribute_list__
|
86
|
+
rescue NameError
|
87
|
+
singleton_class =
|
88
|
+
class << self
|
89
|
+
self
|
90
|
+
end
|
91
|
+
klass = self
|
92
|
+
singleton_class.module_eval do
|
93
|
+
attribute_list = List.new
|
94
|
+
define_method('attribute_list'){ klass == self ? attribute_list : raise(NameError) }
|
95
|
+
alias_method '__attribute_list__', 'attribute_list'
|
96
|
+
end
|
97
|
+
__attribute_list__
|
98
|
+
end
|
99
|
+
end
|
100
|
+
end
|
101
|
+
|
102
|
+
%w( __attributes__ __attribute__ attribute ).each{|dst| alias_method dst, 'attributes'}
|
103
|
+
end
|
104
|
+
|
105
|
+
=begin
|
106
|
+
class Object
|
107
|
+
def attributes *a, &b
|
108
|
+
sc =
|
109
|
+
class << self
|
110
|
+
self
|
111
|
+
end
|
112
|
+
sc.attributes *a, &b
|
113
|
+
end
|
114
|
+
%w( __attributes__ __attribute__ attribute ).each{|dst| alias_method dst, 'attributes'}
|
115
|
+
end
|
116
|
+
=end
|
117
|
+
|
118
|
+
class Module
|
119
|
+
include Attributes
|
120
|
+
end
|
data/lib/bj/bj.rb
ADDED
@@ -0,0 +1,72 @@
|
|
1
|
+
class Bj
|
2
|
+
module ClassMethods
|
3
|
+
attribute("rails_root"){ Util.const_or_env("RAILS_ROOT"){ "." } }
|
4
|
+
attribute("rails_env"){ Util.const_or_env("RAILS_ENV"){ "development" } }
|
5
|
+
attribute("database_yml"){ File.join rails_root, "config", "database.yml" }
|
6
|
+
attribute("configurations"){ YAML::load(ERB.new(IO.read(database_yml)).result) }
|
7
|
+
attribute("tables"){ Table.list }
|
8
|
+
attribute("hostname"){ Socket.gethostname }
|
9
|
+
attribute("logger"){ Bj::Logger.off STDERR }
|
10
|
+
attribute("ruby"){ Util.which_ruby }
|
11
|
+
attribute("rake"){ Util.which_rake }
|
12
|
+
attribute("script"){ Util.find_script "bj" }
|
13
|
+
attribute("ttl"){ Integer(Bj::Table::Config["ttl"] || (twenty_four_hours = 24 * 60 * 60)) }
|
14
|
+
attribute("table"){ Table }
|
15
|
+
attribute("config"){ table.config }
|
16
|
+
attribute("util"){ Util }
|
17
|
+
attribute("runner"){ Runner }
|
18
|
+
attribute("joblist"){ Joblist }
|
19
|
+
attribute("default_path"){ %w'/bin /usr/bin /usr/local/bin /opt/local/bin'.join(File::PATH_SEPARATOR) }
|
20
|
+
|
21
|
+
def transaction options = {}, &block
|
22
|
+
options.to_options!
|
23
|
+
|
24
|
+
cur_rails_env = Bj.rails_env.to_s
|
25
|
+
new_rails_env = options[:rails_env].to_s
|
26
|
+
|
27
|
+
cur_spec = configurations[cur_rails_env]
|
28
|
+
table.establish_connection(cur_spec) unless table.connected?
|
29
|
+
|
30
|
+
if(new_rails_env.empty? or cur_rails_env == new_rails_env)
|
31
|
+
table.transaction{ block.call(table.connection) }
|
32
|
+
else
|
33
|
+
new_spec = configurations[new_rails_env]
|
34
|
+
table.establish_connection(new_spec)
|
35
|
+
Bj.rails_env = new_rails_env
|
36
|
+
begin
|
37
|
+
table.transaction{ block.call(table.connection) }
|
38
|
+
ensure
|
39
|
+
table.establish_connection(cur_spec)
|
40
|
+
Bj.rails_env = cur_rails_env
|
41
|
+
end
|
42
|
+
end
|
43
|
+
end
|
44
|
+
|
45
|
+
def chroot options = {}, &block
|
46
|
+
if defined? @chrooted and @chrooted
|
47
|
+
return(block ? block.call(@chrooted) : @chrooted)
|
48
|
+
end
|
49
|
+
if block
|
50
|
+
begin
|
51
|
+
chrooted = @chrooted
|
52
|
+
Dir.chdir(@chrooted = rails_root) do
|
53
|
+
raise RailsRoot, "<#{ Dir.pwd }> is not a rails root" unless Util.valid_rails_root?(Dir.pwd)
|
54
|
+
block.call(@chrooted)
|
55
|
+
end
|
56
|
+
ensure
|
57
|
+
@chrooted = chrooted
|
58
|
+
end
|
59
|
+
else
|
60
|
+
Dir.chdir(@chrooted = rails_root)
|
61
|
+
raise RailsRoot, "<#{ Dir.pwd }> is not a rails root" unless Util.valid_rails_root?(Dir.pwd)
|
62
|
+
@chrooted
|
63
|
+
end
|
64
|
+
end
|
65
|
+
|
66
|
+
def boot
|
67
|
+
load File.join(rails_root, "config", "boot.rb")
|
68
|
+
load File.join(rails_root, "config", "environment.rb")
|
69
|
+
end
|
70
|
+
end
|
71
|
+
send :extend, ClassMethods
|
72
|
+
end
|
data/lib/bj/errors.rb
ADDED
data/lib/bj/joblist.rb
ADDED
@@ -0,0 +1,112 @@
|
|
1
|
+
class Bj
|
2
|
+
class JobList < ::Array
|
3
|
+
module ClassMethods
|
4
|
+
def for jobs, options = {}
|
5
|
+
if Joblist === jobs
|
6
|
+
jobs.update options
|
7
|
+
return jobs
|
8
|
+
end
|
9
|
+
options.to_options!
|
10
|
+
jobs = [jobs].flatten.compact
|
11
|
+
list = []
|
12
|
+
jobs.each do |arg|
|
13
|
+
list.push(
|
14
|
+
case arg
|
15
|
+
when String
|
16
|
+
case arg
|
17
|
+
when %r/^\d+$/
|
18
|
+
job_from_id arg
|
19
|
+
else
|
20
|
+
job_from_string arg
|
21
|
+
end
|
22
|
+
when Hash
|
23
|
+
job_from_hash arg
|
24
|
+
when Io
|
25
|
+
jobs_from_io arg
|
26
|
+
when Fixnum, Bignum
|
27
|
+
job_from_number arg
|
28
|
+
else
|
29
|
+
job_from_string arg
|
30
|
+
end
|
31
|
+
)
|
32
|
+
end
|
33
|
+
list.flatten!
|
34
|
+
list.compact!
|
35
|
+
list.map!{|job| job.reverse_merge! options}
|
36
|
+
list
|
37
|
+
end
|
38
|
+
|
39
|
+
def job_from_hash arg
|
40
|
+
arg.to_hash.to_options!
|
41
|
+
end
|
42
|
+
|
43
|
+
def job_from_string arg
|
44
|
+
unless arg.strip.empty?
|
45
|
+
{ :command => arg.to_s }
|
46
|
+
else
|
47
|
+
nil
|
48
|
+
end
|
49
|
+
end
|
50
|
+
|
51
|
+
def job_from_number arg
|
52
|
+
id = arg.to_i
|
53
|
+
Table::Job.find(id).to_hash
|
54
|
+
end
|
55
|
+
|
56
|
+
def jobs_from_io arg
|
57
|
+
if arg == "-"
|
58
|
+
load_from_io STDIN
|
59
|
+
else
|
60
|
+
if arg.respond_to? :read
|
61
|
+
load_from_io arg
|
62
|
+
else
|
63
|
+
open(arg, "r"){|fd| load_from_io fd}
|
64
|
+
end
|
65
|
+
end
|
66
|
+
end
|
67
|
+
|
68
|
+
def load_from_io io
|
69
|
+
list = []
|
70
|
+
io.each do |line|
|
71
|
+
line.strip!
|
72
|
+
next if line.empty?
|
73
|
+
list << job_from_string(line)
|
74
|
+
end
|
75
|
+
list
|
76
|
+
end
|
77
|
+
|
78
|
+
def jobs_from_yaml arg
|
79
|
+
object =
|
80
|
+
if arg == "-"
|
81
|
+
YAML.load STDIN
|
82
|
+
else
|
83
|
+
if arg.respond_to? :read
|
84
|
+
YAML.load arg
|
85
|
+
else
|
86
|
+
open(arg, "r"){|fd| YAML.load fd}
|
87
|
+
end
|
88
|
+
end
|
89
|
+
Joblist.for object
|
90
|
+
end
|
91
|
+
end
|
92
|
+
send :extend, ClassMethods
|
93
|
+
|
94
|
+
module InstanceMethods
|
95
|
+
def update options = {}
|
96
|
+
options.to_options!
|
97
|
+
each{|job| job.update options}
|
98
|
+
end
|
99
|
+
|
100
|
+
def push other
|
101
|
+
Joblist.for(other).each do |job|
|
102
|
+
super job
|
103
|
+
end
|
104
|
+
self
|
105
|
+
end
|
106
|
+
alias_method "<<", "push"
|
107
|
+
end
|
108
|
+
send :include, InstanceMethods
|
109
|
+
end
|
110
|
+
|
111
|
+
Joblist = JobList
|
112
|
+
end
|