| 
									
										
										
										
											2015-07-20 18:59:26 +03:00
										 |  |  | require "providers/provider_factory" | 
					
						
							|  |  |  | require "commands/stack" | 
					
						
							|  |  |  | require "db/mongo/models/stack/stack_factory" | 
					
						
							| 
									
										
										
										
											2015-07-30 02:14:45 +03:00
										 |  |  | require "db/mongo/models/project" | 
					
						
							| 
									
										
										
										
											2015-07-20 18:59:26 +03:00
										 |  |  | require "db/mongo/models/report" | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | class StackSyncWorker < Worker | 
					
						
							|  |  |  |   include StackCommands | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2015-07-23 12:54:36 +03:00
										 |  |  |   # besides options came from JobStarter we need: | 
					
						
							|  |  |  |   #   :provider | 
					
						
							|  |  |  |   #   :stack_id | 
					
						
							|  |  |  |   def perform(options) | 
					
						
							| 
									
										
										
										
											2015-07-30 15:37:43 +03:00
										 |  |  |     call(options['config'], options['provider'], options['dir']) do |provider, out, file| | 
					
						
							|  |  |  |       mongo = Devops::Db.connector | 
					
						
							| 
									
										
										
										
											2015-07-23 12:54:36 +03:00
										 |  |  |       stack = mongo.stack(options['stack_id']) | 
					
						
							| 
									
										
										
										
											2015-07-30 02:14:45 +03:00
										 |  |  |       save_report(mongo, stack, file) | 
					
						
							| 
									
										
										
										
											2015-07-20 18:59:26 +03:00
										 |  |  | 
 | 
					
						
							|  |  |  |       status = sync_stack_till_not_in_progress_proc.call(out, stack, mongo) | 
					
						
							| 
									
										
										
										
											2015-07-30 02:14:45 +03:00
										 |  |  |       persist_stack_servers_in_mongo!(mongo, stack, provider) | 
					
						
							| 
									
										
										
										
											2015-07-20 18:59:26 +03:00
										 |  |  |       status | 
					
						
							|  |  |  |     end | 
					
						
							|  |  |  |   end | 
					
						
							| 
									
										
										
										
											2015-07-30 02:14:45 +03:00
										 |  |  | 
 | 
					
						
							|  |  |  |   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 | 
					
						
							| 
									
										
										
										
											2015-07-20 18:59:26 +03:00
										 |  |  | end | 
					
						
							| 
									
										
										
										
											2015-07-23 12:54:36 +03:00
										 |  |  | 
 | 
					
						
							|  |  |  | WorkersStorage.add_worker(sync_stack_till_not_in_progress: StackSyncWorker) |