fluke/devops-service/workers/stack_sync_worker.rb
2015-07-30 01:14:45 +02:00

69 lines
2.0 KiB
Ruby

require "providers/provider_factory"
require "commands/stack"
require "db/mongo/models/stack/stack_factory"
require "db/mongo/models/project"
require "db/mongo/models/report"
require 'workers/workers_storage'
class StackSyncWorker < Worker
include StackCommands
# besides options came from JobStarter we need:
# :provider
# :stack_id
def perform(options)
call(options['config'], options['provider'], options['dir']) do |mongo, provider, out, file|
stack = mongo.stack(options['stack_id'])
save_report(mongo, stack, file)
status = sync_stack_till_not_in_progress_proc.call(out, stack, mongo)
persist_stack_servers_in_mongo!(mongo, stack, provider)
status
end
end
private
def save_report(mongo, stack, file)
report = ::Devops::Model::Report.new(
"file" => file,
"_id" => jid,
"created_by" => stack.owner,
"project" => stack.project,
"deploy_env" => stack.deploy_env,
"type" => ::Devops::Model::Report::STACK_TYPE
)
mongo.save_report(report)
end
def persist_stack_servers_in_mongo!(mongo, stack, provider)
project = mongo.project(stack.project)
deploy_env = project.deploy_env(stack.deploy_env)
stack_servers = []
stack.resources.each do |resource|
logical_name = resource.resource_name
extended_info = stack.resource(logical_name)
body = {
'name' => logical_name,
'key' => extended_info.key_name
}
servers = extract_servers(provider, project, deploy_env, body, stack.owner, mongo)
servers.each do |server|
# TODO: improve IP assigning logic
server.private_ip = extended_info.addresses.values.first.first['addr']
end
stack_servers += servers
end
stack_servers.each_with_index do |server, i|
server.id = "stack-#{stack.id}-server-#{i+1}"
server.stack = stack.id
mongo.server_insert server
end
end
end
WorkersStorage.add_worker(sync_stack_till_not_in_progress: StackSyncWorker)