Sha256: 62220993faa85de28bfc7d5c5f213865d0ca92fa3b4c5034256c4b7a7c7d7a78
Contents?: true
Size: 1.64 KB
Versions: 42
Compression:
Stored size: 1.64 KB
Contents
require_relative '../celluloid/rake_worker'
require_relative './input_stream'
require_relative './output_stream'
module CapistranoMulticonfigParallel
# class used to handle the rake worker and sets all the hooks before and after running the worker
class RakeTaskHooks
attr_accessor :task, :env
def initialize(env, task)
@env = env
@task = task
end
def automatic_hooks(&block)
if job_id.present?
actor_start_working
actor.wait_execution until actor.task_approved
actor_execute_block(&block)
else
block.call
end
end
private
def output_stream
CapistranoMulticonfigParallel::OutputStream
end
def input_stream
CapistranoMulticonfigParallel::InputStream
end
def before_hooks
stringio = StringIO.new
output_stream.hook(stringio)
input_stream.hook(actor, stringio)
end
def after_hooks
input_stream.unhook
output_stream.unhook
end
def actor_execute_block(&block)
before_hooks
block.call
after_hooks
end
def actor_start_working
if actor.blank?
supervise_actor
actor.work(@env, actor_id: rake_actor_id, job_id: job_id, task: @task)
else
actor.publish_new_work(@env, task: @task)
end
end
def supervise_actor
return unless actor.blank?
CapistranoMulticonfigParallel::RakeWorker.supervise_as(rake_actor_id)
end
def actor
Celluloid::Actor[rake_actor_id]
end
def job_id
@env[CapistranoMulticonfigParallel::ENV_KEY_JOB_ID]
end
def rake_actor_id
"rake_worker_#{job_id}"
end
end
end
Version data entries
42 entries across 42 versions & 1 rubygems