iron_worker 2.4.3 → 3.0.0
Sign up to get free protection for your applications and to get access to all the features.
- checksums.yaml +4 -4
- data/.gitignore +1 -0
- data/README.md +352 -0
- data/iron_worker.gemspec +3 -8
- data/lib/iron_worker.rb +13 -51
- data/lib/iron_worker/api_client.rb +152 -0
- data/lib/iron_worker/version.rb +9 -1
- data/lib/iron_worker/worker_helper.rb +72 -0
- metadata +18 -183
- data/README.markdown +0 -73
- data/lib/generators/iron_worker/iron_worker_generator.rb +0 -13
- data/lib/generators/iron_worker/templates/template_worker.erb +0 -7
- data/lib/iron_worker/api.rb +0 -244
- data/lib/iron_worker/base.rb +0 -449
- data/lib/iron_worker/config.rb +0 -287
- data/lib/iron_worker/rails2_init.rb +0 -8
- data/lib/iron_worker/railtie.rb +0 -16
- data/lib/iron_worker/server/overrides.rb +0 -198
- data/lib/iron_worker/server/runner.rb +0 -82
- data/lib/iron_worker/service.rb +0 -644
- data/lib/iron_worker/used_in_worker.rb +0 -11
- data/lib/iron_worker/utils.rb +0 -11
- data/rails/init.rb +0 -0
- data/test/Gemfile +0 -36
- data/test/Gemfile.lock +0 -169
- data/test/README.md +0 -5
- data/test/batch_run.rb +0 -121
- data/test/config_sample.yml +0 -13
- data/test/models/cool_model.rb +0 -7
- data/test/models/db_model.rb +0 -6
- data/test/models/model_1.rb +0 -16
- data/test/models/model_2.rb +0 -16
- data/test/quick_run.rb +0 -36
- data/test/quick_schedule.rb +0 -18
- data/test/resources/something.yml +0 -4
- data/test/simple_script.rb +0 -11
- data/test/test_base.rb +0 -79
- data/test/test_database.rb +0 -31
- data/test/test_gems.rb +0 -31
- data/test/test_inheritance.rb +0 -25
- data/test/test_iron_worker.rb +0 -194
- data/test/test_merging.rb +0 -51
- data/test/test_non_worker_script.rb +0 -14
- data/test/test_rails.rb +0 -35
- data/test/test_scheduling.rb +0 -74
- data/test/test_scheduling_no_active_support.rb +0 -37
- data/test/test_service.rb +0 -28
- data/test/test_uploads.rb +0 -30
- data/test/test_webhooks.rb +0 -45
- data/test/v2quick.rb +0 -29
- data/test/workers/awesome_job.rb +0 -26
- data/test/workers/aws_s3_worker.rb +0 -10
- data/test/workers/big_data_worker.rb +0 -15
- data/test/workers/big_gems_worker.rb +0 -35
- data/test/workers/cool_worker.rb +0 -14
- data/test/workers/db_worker.rb +0 -27
- data/test/workers/fail_worker.rb +0 -10
- data/test/workers/gem_dependency_worker.rb +0 -20
- data/test/workers/hacker_worker.rb +0 -65
- data/test/workers/iw_test_worker.rb +0 -22
- data/test/workers/iw_test_worker_2.rb +0 -53
- data/test/workers/iw_test_worker_3.rb +0 -17
- data/test/workers/local_vs_remote_worker.rb +0 -9
- data/test/workers/merging_worker.rb +0 -40
- data/test/workers/mq_worker.rb +0 -21
- data/test/workers/one_line_worker.rb +0 -10
- data/test/workers/prawn_worker.rb +0 -32
- data/test/workers/progress_worker.rb +0 -26
- data/test/workers/qb_worker.rb +0 -17
- data/test/workers/rails_worker.rb +0 -24
- data/test/workers/rails_worker2.rb +0 -12
- data/test/workers/requiring_worker.rb +0 -4
- data/test/workers/running_back_worker.rb +0 -7
- data/test/workers/scheduled_worker.rb +0 -15
- data/test/workers/second_worker.rb +0 -16
- data/test/workers/third_worker.rb +0 -6
- data/test/workers/webhook_worker.rb +0 -7
- data/test/zip_vs_gzip.rb +0 -26
@@ -1,82 +0,0 @@
|
|
1
|
-
# This is the file that gets executed on the server.
|
2
|
-
|
3
|
-
def init_database_connection(sw_config)
|
4
|
-
if sw_config
|
5
|
-
db_config = sw_config['database']
|
6
|
-
if db_config
|
7
|
-
#@logger.info "Connecting to database using ActiveRecord..."
|
8
|
-
require 'active_record'
|
9
|
-
ActiveRecord::Base.establish_connection(db_config)
|
10
|
-
end
|
11
|
-
end
|
12
|
-
end
|
13
|
-
|
14
|
-
def init_mailer(sw_config)
|
15
|
-
if sw_config
|
16
|
-
mailer_config = sw_config['mailer']
|
17
|
-
if mailer_config
|
18
|
-
require 'action_mailer'
|
19
|
-
ActionMailer::Base.raise_delivery_errors = true
|
20
|
-
ActionMailer::Base.smtp_settings = mailer_config
|
21
|
-
ActionMailer::Base.delivery_method = :smtp
|
22
|
-
end
|
23
|
-
end
|
24
|
-
end
|
25
|
-
|
26
|
-
def get_class_to_run(class_name)
|
27
|
-
runner_class = constantize(class_name)
|
28
|
-
return runner_class
|
29
|
-
end
|
30
|
-
|
31
|
-
# File activesupport/lib/active_support/inflector/methods.rb, line 107
|
32
|
-
# Shoutout to the MIT License
|
33
|
-
def constantize(camel_cased_word)
|
34
|
-
names = camel_cased_word.split('::')
|
35
|
-
names.shift if names.empty? || names.first.empty?
|
36
|
-
|
37
|
-
constant = Object
|
38
|
-
names.each do |name|
|
39
|
-
constant = constant.const_defined?(name) ? constant.const_get(name) : constant.const_missing(name)
|
40
|
-
end
|
41
|
-
constant
|
42
|
-
end
|
43
|
-
|
44
|
-
def init_runner(runner_class, job_data, user_dir, task_id)
|
45
|
-
# ensure initialize takes no arguments
|
46
|
-
init_arity = runner_class.instance_method(:initialize).arity
|
47
|
-
if init_arity == 0 || init_arity == -1
|
48
|
-
# good. -1 can be if it's not defined at all
|
49
|
-
else
|
50
|
-
raise IronWorker::InvalidWorkerError, "Worker initialize method must accept zero arguments."
|
51
|
-
end
|
52
|
-
runner = runner_class.new
|
53
|
-
runner.instance_variable_set(:@task_id, task_id)
|
54
|
-
if job_data
|
55
|
-
runner.instance_variable_set(:@job_data, job_data)
|
56
|
-
runner.instance_variable_set(:@sw_config, job_data['sw_config'])
|
57
|
-
runner.sw_set_data(job_data)
|
58
|
-
end
|
59
|
-
runner.instance_variable_set(:@user_dir, user_dir)
|
60
|
-
runner
|
61
|
-
end
|
62
|
-
|
63
|
-
def init_worker_service_for_runner(job_data)
|
64
|
-
IronWorker.configure do |config|
|
65
|
-
sw_config = job_data['sw_config']
|
66
|
-
config.token = sw_config['token']
|
67
|
-
config.project_id = sw_config['project_id']
|
68
|
-
config.scheme = sw_config['scheme'] if sw_config['scheme']
|
69
|
-
config.host = sw_config['host'] if sw_config['host']
|
70
|
-
config.port = sw_config['port'] if sw_config['port']
|
71
|
-
db_config = sw_config['database']
|
72
|
-
if db_config
|
73
|
-
config.database = db_config
|
74
|
-
end
|
75
|
-
mailer_config = sw_config['mailer']
|
76
|
-
if mailer_config && config.respond_to?(:mailer)
|
77
|
-
config.mailer = mailer_config
|
78
|
-
end
|
79
|
-
config.global_attributes = sw_config['global_attributes'] if sw_config['global_attributes']
|
80
|
-
end
|
81
|
-
end
|
82
|
-
|
data/lib/iron_worker/service.rb
DELETED
@@ -1,644 +0,0 @@
|
|
1
|
-
require 'base64'
|
2
|
-
require 'logger'
|
3
|
-
begin
|
4
|
-
# Little hack until everyone upgrades to rubyzip >= 1
|
5
|
-
require 'zip'
|
6
|
-
ZIPCLZ = Zip::File
|
7
|
-
rescue Exception => ex
|
8
|
-
require 'zip/zip'
|
9
|
-
ZIPCLZ = Zip::ZipFile
|
10
|
-
end
|
11
|
-
require 'bundler'
|
12
|
-
require 'rest_client'
|
13
|
-
require 'json'
|
14
|
-
|
15
|
-
require_relative 'api'
|
16
|
-
|
17
|
-
module IronWorker
|
18
|
-
|
19
|
-
class Service < IronWorker::Api::Client
|
20
|
-
|
21
|
-
attr_accessor :config
|
22
|
-
|
23
|
-
def initialize(token, options={})
|
24
|
-
if options[:config]
|
25
|
-
self.config = options[:config]
|
26
|
-
else
|
27
|
-
c = IronWorker::Config.new unless self.config
|
28
|
-
c.token = token
|
29
|
-
self.config = c
|
30
|
-
end
|
31
|
-
options[:version] = IronWorker.api_version
|
32
|
-
options[:logger] = IronWorker.logger
|
33
|
-
super("worker-aws-us-east-1.iron.io", token, options)
|
34
|
-
self.host = self.config.host if self.config && self.config.host
|
35
|
-
# automerge simple worker gem and dependencies
|
36
|
-
self.config.merge_gem('rubyzip', :require => 'zip')
|
37
|
-
self.config.merge_gem('rest-client')
|
38
|
-
self.config.merge_gem('rest')
|
39
|
-
self.config.merge_gem('iron_worker')
|
40
|
-
IronWorker.logger.info 'IronWorker initialized.'
|
41
|
-
IronWorker.logger.debug ' host = ' + self.host.inspect
|
42
|
-
end
|
43
|
-
|
44
|
-
# Options:
|
45
|
-
# - :callback_url
|
46
|
-
# - :merge => array of files to merge in with this file
|
47
|
-
def upload(filename, class_name, options={})
|
48
|
-
if self.config.no_upload
|
49
|
-
IronWorker.logger.debug 'NO UPLOAD!'
|
50
|
-
return
|
51
|
-
end
|
52
|
-
name = options[:name] || class_name
|
53
|
-
project_id = get_project_id(options)
|
54
|
-
tmp = Dir.tmpdir()
|
55
|
-
md5file = "iron_worker_#{class_name.gsub("::", ".")}_#{token[0, 8]}.md5"
|
56
|
-
existing_md5 = nil
|
57
|
-
md5_f = File.join(tmp, md5file)
|
58
|
-
if File.exists?(md5_f)
|
59
|
-
existing_md5 = IO.read(md5_f)
|
60
|
-
end
|
61
|
-
|
62
|
-
|
63
|
-
begin
|
64
|
-
|
65
|
-
zip_filename = build_merged_file(filename, options[:merge], options[:unmerge], options[:merged_gems], options[:unmerged_gems], options[:merged_mailers], options[:merged_folders], class_name)
|
66
|
-
|
67
|
-
# Check for code changes.
|
68
|
-
zipfile = ZIPCLZ.open(zip_filename, ZIPCLZ::CREATE)
|
69
|
-
crc = zipfile.entries.collect { |x| x.crc }.inject(:+)
|
70
|
-
new_code = false
|
71
|
-
if self.config.force_upload || crc.to_s != existing_md5
|
72
|
-
IronWorker.logger.info "Uploading #{class_name}, code modified."
|
73
|
-
File.open(md5_f, 'w') { |f| f.write(crc) }
|
74
|
-
new_code = true
|
75
|
-
# todo: delete md5 file if error occurs during upload process
|
76
|
-
else
|
77
|
-
# puts "#{class_name}: same code, not uploading"
|
78
|
-
return
|
79
|
-
end
|
80
|
-
|
81
|
-
if new_code
|
82
|
-
upload_code(name, zip_filename, 'runner.rb', :runtime => 'ruby')
|
83
|
-
end
|
84
|
-
|
85
|
-
rescue Exception => ex
|
86
|
-
# if it errors, let's delete md5 since it wouldn't have uploaded.
|
87
|
-
File.delete(md5_f) if File.exists?(md5_f)
|
88
|
-
raise ex
|
89
|
-
end
|
90
|
-
end
|
91
|
-
|
92
|
-
def get_server_gems
|
93
|
-
hash = get("gems/list")
|
94
|
-
JSON.parse(hash["gems"])
|
95
|
-
end
|
96
|
-
|
97
|
-
def logger
|
98
|
-
IronWorker.logger
|
99
|
-
end
|
100
|
-
|
101
|
-
|
102
|
-
def self.heroku?
|
103
|
-
!Gem::GemPathSearcher.private_instance_methods.include?(:_deprecated_initialize) rescue false
|
104
|
-
end
|
105
|
-
|
106
|
-
def webhook
|
107
|
-
|
108
|
-
end
|
109
|
-
|
110
|
-
def self.get_gem_path(gem_info)
|
111
|
-
# gem_name =(gem_info[:require] || gem_info[:name].match(/^[a-zA-Z0-9\-_\.]+/)[0])
|
112
|
-
gem_name =(gem_info[:name].match(/^[a-zA-Z0-9\-_\.]+/)[0])
|
113
|
-
IronWorker.logger.debug "Searching for gem #{gem_name}..."
|
114
|
-
gems= Gem::Specification.respond_to?(:each) ? Gem::Specification.find_all_by_name(gem_name) : Gem::GemPathSearcher.new.find_all(gem_name)
|
115
|
-
if (Service.heroku?) && (!gems || gems.empty?)
|
116
|
-
gems = Gem::GemPathSearcher.new.init_gemspecs.select { |gem| gem.name==gem_name }
|
117
|
-
end
|
118
|
-
IronWorker.logger.debug 'gems found=' + gems.inspect
|
119
|
-
gems = gems.select { |g| g.version.version==gem_info[:version] } if gem_info[:version]
|
120
|
-
if !gems.empty?
|
121
|
-
gem = gems.last
|
122
|
-
return gem, gem.full_gem_path
|
123
|
-
else
|
124
|
-
return nil, nil
|
125
|
-
end
|
126
|
-
end
|
127
|
-
|
128
|
-
def gem_dependencies(list_of_gems)
|
129
|
-
if Service.heroku?
|
130
|
-
gem_specs = Gem::GemPathSearcher.new.init_gemspecs
|
131
|
-
else
|
132
|
-
gem_specs = Gem::Specification.all
|
133
|
-
end
|
134
|
-
|
135
|
-
IronWorker.logger.debug "Getting gem_dependencies.."
|
136
|
-
deps = []
|
137
|
-
dependendent_gems ={}
|
138
|
-
list_of_gems.each_value do |v|
|
139
|
-
IronWorker.logger.debug "Dependencies for #{v}"
|
140
|
-
@deps = v[:gemspec].dependencies
|
141
|
-
@deps.each do |d|
|
142
|
-
IronWorker.logger.debug "dep: #{d}"
|
143
|
-
deps << Bundler::DepProxy.new(d, 'ruby')
|
144
|
-
end
|
145
|
-
end
|
146
|
-
filtered_deps = deps.select { |d| d.type != :development }
|
147
|
-
IronWorker.logger.debug "filtered_deps=#{filtered_deps}"
|
148
|
-
index = Bundler::Index.new
|
149
|
-
gem_specs.collect { |s| index<<s }
|
150
|
-
list = Bundler::Resolver.resolve(filtered_deps, index)
|
151
|
-
list.each do |gemspec|
|
152
|
-
next if list_of_gems.keys.include?(gemspec.name) || Config.system_gems.include?(gemspec.name)
|
153
|
-
IronWorker.logger.debug "gemspec from bundler " + gemspec.inspect
|
154
|
-
gi = IronWorker::MergeHelper.create_gem_info(gemspec.name, gemspec.version.version)
|
155
|
-
dependendent_gems[gemspec.name] = gi
|
156
|
-
end
|
157
|
-
dependendent_gems
|
158
|
-
end
|
159
|
-
|
160
|
-
def build_merged_file(filename, merged, unmerge, merged_gems, unmerged_gems, merged_mailers, merged_folders, class_name)
|
161
|
-
|
162
|
-
unmerged_gems ||= {}
|
163
|
-
|
164
|
-
merge = IronWorker.config.merged.dup
|
165
|
-
merge.merge!(merged) if merged
|
166
|
-
if unmerge
|
167
|
-
unmerge.each_pair do |x, y|
|
168
|
-
deleted = merge.delete x
|
169
|
-
IronWorker.logger.debug "Unmerging #{x}. Success? #{deleted}"
|
170
|
-
end
|
171
|
-
end
|
172
|
-
merged = merge
|
173
|
-
IronWorker.logger.debug 'merged=' + merged.inspect
|
174
|
-
merged_gems = IronWorker.config.merged_gems.merge(merged_gems)
|
175
|
-
IronWorker.logger.debug 'merged_gems=' + merged_gems.inspect
|
176
|
-
|
177
|
-
if config.skip_auto_dependencies
|
178
|
-
gems_dependencies = {}
|
179
|
-
else
|
180
|
-
gems_dependencies = gem_dependencies(merged_gems)
|
181
|
-
IronWorker.logger.debug 'gem_dependencies=' + gems_dependencies.inspect
|
182
|
-
# add dependencies to merged_gems
|
183
|
-
#require gems dependencies
|
184
|
-
gems_dependencies.each_pair do |k, gem|
|
185
|
-
IronWorker.logger.debug "Bundling dependent gem #{gem[:name]}..."
|
186
|
-
gem[:bypass_require] = true
|
187
|
-
# merged_gems[k] ||= gem
|
188
|
-
end
|
189
|
-
end
|
190
|
-
merged_gems = gems_dependencies.merge merged_gems
|
191
|
-
# Now remove unmerged gems
|
192
|
-
unmerged_gems = unmerged_gems.merge(IronWorker.config.unmerged_gems)
|
193
|
-
unmerged_gems.each_pair do |k, v|
|
194
|
-
IronWorker.logger.debug 'unmerging gem=' + k.inspect
|
195
|
-
merged_gems.delete(k)
|
196
|
-
end
|
197
|
-
IronWorker.logger.debug 'merged_gems_after=' + merged_gems.inspect
|
198
|
-
|
199
|
-
merged_mailers ||= {}
|
200
|
-
merged_mailers = merged_mailers.merge(IronWorker.config.mailers) if IronWorker.config.mailers
|
201
|
-
|
202
|
-
#tmp_file = File.join(Dir.tmpdir(), File.basename(filename))
|
203
|
-
tmp_file = File.join(Dir.tmpdir(), 'runner.rb')
|
204
|
-
File.open(tmp_file, "w") do |f|
|
205
|
-
|
206
|
-
f.write("
|
207
|
-
# Find environment (-e)
|
208
|
-
dirname = ''
|
209
|
-
i = 0
|
210
|
-
task_data_file = nil
|
211
|
-
task_id = nil
|
212
|
-
#puts \"args for single file=\" + ARGV.inspect
|
213
|
-
ARGV.each do |arg|
|
214
|
-
if arg == \"-d\"
|
215
|
-
# the user's writable directory
|
216
|
-
dirname = ARGV[i+1]
|
217
|
-
end
|
218
|
-
if arg == \"-id\"
|
219
|
-
# task_id
|
220
|
-
task_id = ARGV[i+1]
|
221
|
-
end
|
222
|
-
if arg == \"-payload\"
|
223
|
-
# path to job data
|
224
|
-
task_data_file = ARGV[i+1]
|
225
|
-
end
|
226
|
-
i+=1
|
227
|
-
end
|
228
|
-
require 'json'
|
229
|
-
")
|
230
|
-
|
231
|
-
# require merged gems
|
232
|
-
merged_gems.each_pair do |k, gem|
|
233
|
-
f.write "$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '/gems/#{gem[:name]}/lib'))\n"
|
234
|
-
end
|
235
|
-
|
236
|
-
merged_gems.each_pair do |k, gem|
|
237
|
-
IronWorker.logger.debug "Bundling gem #{gem[:name]}..."
|
238
|
-
#f.write "$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '/gems/#{gem[:name]}/lib'))\n"
|
239
|
-
next if gem[:bypass_require]
|
240
|
-
IronWorker.logger.debug 'writing requires: ' + gem[:require].inspect
|
241
|
-
if gem[:require].nil?
|
242
|
-
gem[:require] = []
|
243
|
-
elsif gem[:require].is_a?(String)
|
244
|
-
gem[:require] = [gem[:require]]
|
245
|
-
end
|
246
|
-
IronWorker.logger.debug "gem[:require]: " + gem[:require].inspect
|
247
|
-
gem[:require].each do |r|
|
248
|
-
IronWorker.logger.debug 'adding require to file ' + r.to_s
|
249
|
-
f.write "require '#{r}'\n"
|
250
|
-
end
|
251
|
-
end
|
252
|
-
|
253
|
-
File.open(File.join(File.dirname(__FILE__), 'server', 'overrides.rb'), 'r') do |fr|
|
254
|
-
while line = fr.gets
|
255
|
-
f.write line
|
256
|
-
end
|
257
|
-
end
|
258
|
-
|
259
|
-
# Now we must disable queuing while loading up classes. This is from the overrides.rb file
|
260
|
-
f.write("
|
261
|
-
IronWorker.disable_queueing()
|
262
|
-
")
|
263
|
-
|
264
|
-
|
265
|
-
File.open(File.join(File.dirname(__FILE__), "server", 'runner.rb'), 'r') do |fr|
|
266
|
-
while line = fr.gets
|
267
|
-
f.write line
|
268
|
-
end
|
269
|
-
end
|
270
|
-
|
271
|
-
# load job data
|
272
|
-
f.write("
|
273
|
-
# Change to user directory
|
274
|
-
#puts 'dirname=' + dirname.inspect
|
275
|
-
Dir.chdir(dirname)
|
276
|
-
# Load in job data
|
277
|
-
payload = IO.read(task_data_file)
|
278
|
-
IronWorker.payload = payload
|
279
|
-
begin
|
280
|
-
job_data = JSON.parse(payload)
|
281
|
-
#puts 'payload=' + job_data.inspect
|
282
|
-
sw_config = job_data['sw_config']
|
283
|
-
IronWorker.task_data = job_data
|
284
|
-
|
285
|
-
if IronWorker.task_data['rails']
|
286
|
-
module ::Rails
|
287
|
-
def self.version
|
288
|
-
IronWorker.task_data['rails']['version']
|
289
|
-
end
|
290
|
-
def self.env
|
291
|
-
IronWorker.task_data['rails']['env']
|
292
|
-
end
|
293
|
-
end
|
294
|
-
end
|
295
|
-
rescue JSON::ParserError => ex
|
296
|
-
puts 'Payload is not json, raw payload can be accessed with IronWorker.payload.'
|
297
|
-
end
|
298
|
-
")
|
299
|
-
|
300
|
-
if IronWorker.config.extra_requires
|
301
|
-
IronWorker.config.extra_requires.each do |r|
|
302
|
-
f.write "require '#{r}'\n"
|
303
|
-
end
|
304
|
-
end
|
305
|
-
if merged_mailers && !merged_mailers.empty?
|
306
|
-
# todo: isn't 'action_mailer already required in railtie?
|
307
|
-
f.write "require 'action_mailer'\n"
|
308
|
-
f.write "init_mailer(sw_config)\n"
|
309
|
-
f.write "ActionMailer::Base.prepend_view_path('templates')\n"
|
310
|
-
end
|
311
|
-
f.write "init_database_connection(sw_config)\n"
|
312
|
-
|
313
|
-
merged.each_pair do |k, v|
|
314
|
-
if v[:extname] == ".rb"
|
315
|
-
f.write "require_relative '#{File.basename(v[:path])}'\n"
|
316
|
-
end
|
317
|
-
end
|
318
|
-
merged_mailers.each_pair do |k, mailer|
|
319
|
-
f.write "require_relative '#{mailer[:name]}'\n"
|
320
|
-
end
|
321
|
-
#end
|
322
|
-
#f.write File.open(filename, 'r') { |mo| mo.read }
|
323
|
-
f.write("require_relative '#{File.basename(filename)}'\n")
|
324
|
-
|
325
|
-
f.write("
|
326
|
-
clz_name = '#{class_name}' # job_data['class_name']
|
327
|
-
runner_class = get_class_to_run(clz_name)
|
328
|
-
IronWorker.running_class = runner_class
|
329
|
-
runner = init_runner(runner_class, job_data, dirname, task_id)
|
330
|
-
if IronWorker.task_data
|
331
|
-
init_worker_service_for_runner(job_data)
|
332
|
-
end
|
333
|
-
# Now reenable after loading
|
334
|
-
IronWorker.enable_queueing()
|
335
|
-
|
336
|
-
# Let's run it!
|
337
|
-
runner_return_data = runner.run
|
338
|
-
")
|
339
|
-
|
340
|
-
end
|
341
|
-
#puts 'funner.rb=' + tmp_file
|
342
|
-
merge['runner.rb'] = {:path => tmp_file}
|
343
|
-
#puts 'filename=' + filename
|
344
|
-
merge[File.basename(filename)] = {:path => filename}
|
345
|
-
#puts "merge before uniq! " + merge.inspect
|
346
|
-
# puts "merge after uniq! " + merge.inspect
|
347
|
-
|
348
|
-
fname2 = tmp_file + ".zip"
|
349
|
-
# puts 'fname2=' + fname2
|
350
|
-
# puts 'merged_file_array=' + merge.inspect
|
351
|
-
#File.open(fname2, "w") do |f|
|
352
|
-
File.delete(fname2) if File.exist?(fname2)
|
353
|
-
#merging all gems and deps
|
354
|
-
merged_gems.merge!(gems_dependencies)
|
355
|
-
ZIPCLZ.open(fname2, 'w') do |f|
|
356
|
-
if merged_gems && merged_gems.size > 0
|
357
|
-
merged_gems.each_pair do |k, gem|
|
358
|
-
next unless gem[:merge]
|
359
|
-
# puts 'gem=' + gem.inspect
|
360
|
-
path = gem[:path]
|
361
|
-
if path
|
362
|
-
#IronWorker.logger.debug "Collecting gem #{path}"
|
363
|
-
paths_to_use = ["#{path}/*", "#{path}/lib/**/**"]
|
364
|
-
if gem[:include_dirs]
|
365
|
-
#IronWorker.logger.debug "including extra dirs: " + gem[:include_dirs].inspect
|
366
|
-
gem[:include_dirs].each do |dir|
|
367
|
-
paths_to_use << "#{path}/#{dir}/**/**"
|
368
|
-
end
|
369
|
-
end
|
370
|
-
IronWorker.logger.debug 'paths_to_use: ' + paths_to_use.inspect
|
371
|
-
Dir.glob(paths_to_use).each do |file|
|
372
|
-
# todo: could check if directory and it not lib, skip it
|
373
|
-
#IronWorker.logger.debug 'file for gem=' + file.inspect
|
374
|
-
# puts 'gem2=' + gem.inspect
|
375
|
-
zdest = "gems/#{gem[:name]}/#{file.sub(path+'/', '')}"
|
376
|
-
# puts 'gem file=' + file.to_s
|
377
|
-
#IronWorker.logger.debug 'zip dest=' + zdest
|
378
|
-
f.add(zdest, file)
|
379
|
-
end
|
380
|
-
else
|
381
|
-
if gem[:auto_merged]
|
382
|
-
# todo: should only continue if the gem was auto merged.
|
383
|
-
IronWorker.logger.warn "Gem #{gem[:name]} #{gem[:version]} was not found, continuing anyways."
|
384
|
-
else
|
385
|
-
raise "Gem #{gem[:name]} #{gem[:version]} was not found. This will occour when gem_name.gemspec is not the same as the gems primary require."
|
386
|
-
end
|
387
|
-
|
388
|
-
end
|
389
|
-
end
|
390
|
-
end
|
391
|
-
if merged_folders && merged_folders.size > 0
|
392
|
-
merged_folders.each do |folder, files|
|
393
|
-
IronWorker.logger.debug "Collecting folder #{folder}"
|
394
|
-
if files and files.size>0
|
395
|
-
files.each do |file|
|
396
|
-
zdest = "#{Digest::MD5.hexdigest(folder)}/#{file.sub(':', '_').sub('/', '_')}"
|
397
|
-
#IronWorker.logger.debug 'put file to=' + zdest
|
398
|
-
f.add(zdest, file)
|
399
|
-
end
|
400
|
-
end
|
401
|
-
end
|
402
|
-
end
|
403
|
-
|
404
|
-
IronWorker.logger.debug "merge=" + merge.inspect
|
405
|
-
merge.each_pair do |k, v|
|
406
|
-
IronWorker.logger.debug "merging k=#{k.inspect} v=#{v.inspect} into #{filename}"
|
407
|
-
f.add(File.basename(v[:path]), v[:path])
|
408
|
-
end
|
409
|
-
if merged_mailers && merged_mailers.size > 0
|
410
|
-
# puts " MERGED MAILERS" + merged_mailers.inspect
|
411
|
-
merged_mailers.each_pair do |k, mailer|
|
412
|
-
IronWorker.logger.debug "Collecting mailer #{mailer[:name]}"
|
413
|
-
f.add(File.basename(mailer[:filename]), mailer[:filename])
|
414
|
-
path = mailer[:path_to_templates]
|
415
|
-
Dir["#{path}/**/**"].each do |file|
|
416
|
-
zdest = "templates/#{mailer[:name]}/#{file.sub(path+'/', '')}"
|
417
|
-
f.add(zdest, file)
|
418
|
-
end
|
419
|
-
path = mailer[:path_to_layouts]
|
420
|
-
Dir["#{path}/**/**"].each do |file|
|
421
|
-
zdest = "templates/layouts/#{file.sub(path+'/', '')}"
|
422
|
-
f.add(zdest, file)
|
423
|
-
end
|
424
|
-
end
|
425
|
-
end
|
426
|
-
end
|
427
|
-
fname2
|
428
|
-
end
|
429
|
-
|
430
|
-
# This will package up files into a zip file ready for uploading.
|
431
|
-
def package_code(files)
|
432
|
-
fname2 = "package.zip"
|
433
|
-
File.delete(fname2) if File.exist?(fname2)
|
434
|
-
ZIPCLZ.open(fname2, 'w') do |f|
|
435
|
-
files.each do |file|
|
436
|
-
f.add(file, file)
|
437
|
-
end
|
438
|
-
end
|
439
|
-
fname2
|
440
|
-
end
|
441
|
-
|
442
|
-
# options:
|
443
|
-
# :runtime => 'ruby', 'python', 'node', 'java', 'go'
|
444
|
-
def upload_code(name, package_file, exec_file, options={})
|
445
|
-
IronWorker.logger.info 'file size to upload: ' + File.size(package_file).to_s
|
446
|
-
options = {
|
447
|
-
"name" => name,
|
448
|
-
"standalone" => true,
|
449
|
-
"runtime" => options[:runtime] || "ruby",
|
450
|
-
"file_name" => exec_file # File.basename(filename)
|
451
|
-
}
|
452
|
-
#puts 'options for upload=' + options.inspect
|
453
|
-
IronWorker.logger.info "Uploading now..."
|
454
|
-
ret = post_file("#{project_url_prefix(get_project_id(options))}codes", File.new(package_file), options)
|
455
|
-
IronWorker.logger.info "Done uploading."
|
456
|
-
return ret
|
457
|
-
end
|
458
|
-
|
459
|
-
def project_url_prefix(project_id = 0)
|
460
|
-
# IronWorker.logger.info "project_url_prefix, project_id = " + project_id.inspect
|
461
|
-
if project_id == 0
|
462
|
-
return false
|
463
|
-
project_id = config.project_id
|
464
|
-
end
|
465
|
-
"projects/#{project_id}/"
|
466
|
-
end
|
467
|
-
|
468
|
-
def wait_until_complete(task_id, options={})
|
469
|
-
tries = 0
|
470
|
-
status = nil
|
471
|
-
sleep 1
|
472
|
-
while tries < 100
|
473
|
-
status = options[:schedule] ? schedule_status(task_id) : status(task_id)
|
474
|
-
puts "Waiting... status=" + status["status"]
|
475
|
-
if options[:schedule]
|
476
|
-
if status["status"] == "complete"
|
477
|
-
break
|
478
|
-
end
|
479
|
-
else
|
480
|
-
if status["status"] != "queued" && status["status"] != "running"
|
481
|
-
break
|
482
|
-
end
|
483
|
-
end
|
484
|
-
sleep 2
|
485
|
-
end
|
486
|
-
status
|
487
|
-
end
|
488
|
-
|
489
|
-
def add_sw_params(hash_to_send)
|
490
|
-
# todo: remove secret key?? Can use worker service from within a worker without it now
|
491
|
-
hash_to_send["oauth"] = self.token
|
492
|
-
hash_to_send["api_version"] = IronWorker.api_version
|
493
|
-
end
|
494
|
-
|
495
|
-
def check_config
|
496
|
-
if self.config.nil? || self.config.token.nil? || self.config.project_id.nil?
|
497
|
-
raise "Invalid IronWorker configuration, token and project_id required."
|
498
|
-
end
|
499
|
-
end
|
500
|
-
|
501
|
-
def enqueue(name, data={}, options={})
|
502
|
-
queue(name, data, options)
|
503
|
-
end
|
504
|
-
|
505
|
-
# name: The name of previously upload worker code, eg: MySuperWorker
|
506
|
-
# data: Arbitrary hash of your own data that your task will need to run.
|
507
|
-
def queue(name, data={}, options={})
|
508
|
-
puts "Queuing #{name}..."
|
509
|
-
check_config
|
510
|
-
unless data.is_a?(Array)
|
511
|
-
data = [data]
|
512
|
-
end
|
513
|
-
# Now we need to add class_name to the payload
|
514
|
-
tasks = []
|
515
|
-
data.each do |d|
|
516
|
-
d['class_name'] = name unless d['class_name']
|
517
|
-
task = {}
|
518
|
-
task["payload"] = d.to_json
|
519
|
-
task["code_name"] = name
|
520
|
-
task.merge!(options)
|
521
|
-
tasks << task
|
522
|
-
end
|
523
|
-
name = options[:name] || name
|
524
|
-
hash_to_send = {}
|
525
|
-
hash_to_send["options"] = options
|
526
|
-
hash_to_send["tasks"] = tasks
|
527
|
-
add_sw_params(hash_to_send)
|
528
|
-
if defined?(RAILS_ENV)
|
529
|
-
# todo: REMOVE THIS
|
530
|
-
hash_to_send["rails_env"] = RAILS_ENV
|
531
|
-
end
|
532
|
-
queue_raw(name, hash_to_send, options)
|
533
|
-
end
|
534
|
-
|
535
|
-
def queue_raw(name, data={}, options={})
|
536
|
-
params = nil
|
537
|
-
hash_to_send = data
|
538
|
-
#hash_to_send["class_name"] = name unless hash_to_send["class_name"]
|
539
|
-
hash_to_send["name"] = name unless hash_to_send["name"]
|
540
|
-
uri = project_url_prefix(get_project_id(options)) + "tasks"
|
541
|
-
IronWorker.logger.debug 'queue_raw , uri = ' + uri
|
542
|
-
ret = post(uri, hash_to_send)
|
543
|
-
ret
|
544
|
-
end
|
545
|
-
|
546
|
-
#
|
547
|
-
# schedule: hash of scheduling options that can include:
|
548
|
-
# Required:
|
549
|
-
# - start_at: Time of first run - DateTime or Time object.
|
550
|
-
# Optional:
|
551
|
-
# - run_every: Time in seconds between runs. If ommitted, task will only run once.
|
552
|
-
# - delay_type: Fixed Rate or Fixed Delay. Default is fixed_delay.
|
553
|
-
# - end_at: Scheduled task will stop running after this date (optional, if ommitted, runs forever or until cancelled)
|
554
|
-
# - run_times: Task will run exactly :run_times. For instance if :run_times is 5, then the task will run 5 times.
|
555
|
-
#
|
556
|
-
def schedule(name, data, schedule)
|
557
|
-
puts "Scheduling #{name}..."
|
558
|
-
raise "Schedule must be a hash." if !schedule.is_a? Hash
|
559
|
-
hash_to_send = {}
|
560
|
-
schedules = []
|
561
|
-
schedule["payload"] = data.to_json
|
562
|
-
schedule["name"] = name unless schedule["name"]
|
563
|
-
schedule["code_name"] = name unless schedule["code_name"]
|
564
|
-
schedules << schedule
|
565
|
-
hash_to_send["schedules"] = schedules
|
566
|
-
add_sw_params(hash_to_send)
|
567
|
-
# puts ' about to send ' + hash_to_send.inspect
|
568
|
-
uri = project_url_prefix(get_project_id(data)) + "schedules"
|
569
|
-
ret = post(uri, hash_to_send)
|
570
|
-
ret
|
571
|
-
end
|
572
|
-
|
573
|
-
def cancel_schedule(scheduled_task_id, options={})
|
574
|
-
raise "Must include a schedule id." if scheduled_task_id.blank?
|
575
|
-
hash_to_send = {}
|
576
|
-
hash_to_send["schedule_id"] = scheduled_task_id
|
577
|
-
uri = "#{project_url_prefix(get_project_id(options))}schedules/#{scheduled_task_id}/cancel"
|
578
|
-
post(uri, hash_to_send)
|
579
|
-
end
|
580
|
-
|
581
|
-
def get_project_id(options={})
|
582
|
-
options[:project_id] || config.project_id
|
583
|
-
end
|
584
|
-
|
585
|
-
def codes(options={})
|
586
|
-
hash_to_send = options
|
587
|
-
uri = "projects/" + get_project_id(options) + "/codes"
|
588
|
-
ret = get(uri, hash_to_send)
|
589
|
-
ret
|
590
|
-
end
|
591
|
-
|
592
|
-
def schedules(options={})
|
593
|
-
hash_to_send = options
|
594
|
-
uri = "projects/" + get_project_id(options) + "/schedules"
|
595
|
-
ret = get(uri, hash_to_send)
|
596
|
-
ret
|
597
|
-
end
|
598
|
-
|
599
|
-
def jobs(options={})
|
600
|
-
puts 'Service.jobs is deprecated, use Service.tasks instead'
|
601
|
-
tasks(options)
|
602
|
-
end
|
603
|
-
|
604
|
-
def tasks(options={})
|
605
|
-
hash_to_send = options
|
606
|
-
uri = "projects/" + get_project_id(options) + "/tasks"
|
607
|
-
ret = get(uri, hash_to_send)
|
608
|
-
ret
|
609
|
-
end
|
610
|
-
|
611
|
-
def get_log(job_id, options={})
|
612
|
-
puts 'Service.get_log is deprecated, use Service.log instead'
|
613
|
-
log(job_id, options)
|
614
|
-
end
|
615
|
-
|
616
|
-
def log(task_id, options={})
|
617
|
-
data = options
|
618
|
-
ret = get("#{project_url_prefix(get_project_id(options))}tasks/#{task_id}/log", data, :parse => false)
|
619
|
-
ret
|
620
|
-
end
|
621
|
-
|
622
|
-
|
623
|
-
def status(task_id, options={})
|
624
|
-
data = {"task_id" => task_id}
|
625
|
-
ret = get("#{project_url_prefix(get_project_id(options))}tasks/#{task_id}", data)
|
626
|
-
ret
|
627
|
-
end
|
628
|
-
|
629
|
-
def schedule_status(schedule_id, options={})
|
630
|
-
data = {"schedule_id" => schedule_id}
|
631
|
-
ret = get("#{project_url_prefix(get_project_id(options))}schedules/#{schedule_id}", data)
|
632
|
-
ret
|
633
|
-
end
|
634
|
-
|
635
|
-
# data is a hash, should include 'percent' and 'msg'
|
636
|
-
def set_progress(task_id, options={})
|
637
|
-
#data={"data"=>data, "task_id"=>task_id}
|
638
|
-
post("#{project_url_prefix(get_project_id(options))}tasks/#{task_id}/progress", options)
|
639
|
-
end
|
640
|
-
|
641
|
-
|
642
|
-
end
|
643
|
-
|
644
|
-
end
|