| 
									
										
										
										
											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" | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2015-07-30 13:18:55 +03:00
										 |  |  | class StackBootstrapWorker < Worker | 
					
						
							| 
									
										
										
										
											2015-07-20 18:59:26 +03:00
										 |  |  |   include StackCommands | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2015-07-23 12:54:36 +03:00
										 |  |  |   def perform(options) | 
					
						
							| 
									
										
										
										
											2015-09-01 16:31:31 +03:00
										 |  |  |     provider_name = options.fetch('provider_name') | 
					
						
							|  |  |  |     stack_attrs = options.fetch('stack_attributes') | 
					
						
							| 
									
										
										
										
											2015-08-14 17:48:07 +03:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2015-09-01 16:31:31 +03:00
										 |  |  |     call(provider_name) do |provider, out, file| | 
					
						
							|  |  |  |       without_bootstrap = stack_attrs.delete('without_bootstrap') | 
					
						
							| 
									
										
										
										
											2015-09-17 11:50:35 +03:00
										 |  |  |       report = save_report(file, stack_attrs) | 
					
						
							| 
									
										
										
										
											2015-09-01 16:31:31 +03:00
										 |  |  | 
 | 
					
						
							|  |  |  |       stack = Devops::Model::StackFactory.create(provider_name, stack_attrs, out) | 
					
						
							| 
									
										
										
										
											2015-08-12 13:32:26 +03:00
										 |  |  |       mongo.stack_insert(stack) | 
					
						
							| 
									
										
										
										
											2015-07-20 18:59:26 +03:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2015-09-01 16:31:31 +03:00
										 |  |  |       operation_result = sync_stack_proc.call(out, stack, mongo) | 
					
						
							|  |  |  |       if operation_result == 0
 | 
					
						
							| 
									
										
										
										
											2015-08-20 21:55:44 +03:00
										 |  |  |         out << "\nStack '#{stack.name}' has been created\n" | 
					
						
							|  |  |  |         out.flush | 
					
						
							|  |  |  |         servers = persist_stack_servers!(stack, provider) | 
					
						
							| 
									
										
										
										
											2015-09-17 11:50:35 +03:00
										 |  |  |         bootstrap_servers(servers, out, report) unless without_bootstrap | 
					
						
							| 
									
										
										
										
											2015-08-12 12:24:48 +03:00
										 |  |  |       end | 
					
						
							| 
									
										
										
										
											2015-09-01 16:31:31 +03:00
										 |  |  |       operation_result | 
					
						
							| 
									
										
										
										
											2015-07-20 18:59:26 +03:00
										 |  |  |     end | 
					
						
							|  |  |  |   end | 
					
						
							| 
									
										
										
										
											2015-07-30 02:14:45 +03:00
										 |  |  | 
 | 
					
						
							|  |  |  |   private | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2015-09-17 11:50:35 +03:00
										 |  |  |   def bootstrap_servers(servers, out, report) | 
					
						
							| 
									
										
										
										
											2015-09-01 16:31:31 +03:00
										 |  |  |     out << "\n" | 
					
						
							|  |  |  |     servers.map do |server| | 
					
						
							| 
									
										
										
										
											2015-09-17 11:50:35 +03:00
										 |  |  |       executor = Devops::Executor::ServerExecutor.new(server, out) | 
					
						
							|  |  |  |       executor.report = report | 
					
						
							|  |  |  |       executor.two_phase_bootstrap({}) | 
					
						
							| 
									
										
										
										
											2015-09-01 16:31:31 +03:00
										 |  |  |     end | 
					
						
							|  |  |  |   end | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |   def save_report(file, stack_attrs) | 
					
						
							|  |  |  |     report = ::Devops::Model::Report.new( | 
					
						
							|  |  |  |       "file" => file, | 
					
						
							|  |  |  |       "_id" => jid, | 
					
						
							|  |  |  |       "created_by" => stack_attrs['owner'], | 
					
						
							|  |  |  |       "project" => stack_attrs["project"], | 
					
						
							|  |  |  |       "deploy_env" => stack_attrs["deploy_env"], | 
					
						
							|  |  |  |       "type" => ::Devops::Model::Report::STACK_TYPE | 
					
						
							|  |  |  |     ) | 
					
						
							|  |  |  |     mongo.save_report(report) | 
					
						
							| 
									
										
										
										
											2015-09-17 11:50:35 +03:00
										 |  |  |     report | 
					
						
							| 
									
										
										
										
											2015-09-01 16:31:31 +03:00
										 |  |  |   end | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2015-08-05 18:26:21 +03:00
										 |  |  |   def persist_stack_servers!(stack, provider) | 
					
						
							| 
									
										
										
										
											2015-07-30 02:14:45 +03:00
										 |  |  |     project = mongo.project(stack.project) | 
					
						
							|  |  |  |     deploy_env = project.deploy_env(stack.deploy_env) | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2015-08-31 14:26:16 +03:00
										 |  |  |     provider.stack_servers(stack).map do |extended_info| | 
					
						
							| 
									
										
										
										
											2015-08-05 18:26:21 +03:00
										 |  |  |       server_attrs = { | 
					
						
							|  |  |  |         'provider' => provider.name, | 
					
						
							|  |  |  |         'project' => project.id, | 
					
						
							|  |  |  |         'deploy_env' => deploy_env.identifier, | 
					
						
							|  |  |  |         'remote_user' => mongo.image(deploy_env.image).remote_user, | 
					
						
							| 
									
										
										
										
											2015-08-18 20:01:17 +03:00
										 |  |  |         'key' => extended_info["key_name"] || provider.ssh_key, | 
					
						
							|  |  |  |         '_id' => extended_info["id"], | 
					
						
							|  |  |  |         'chef_node_name' => extended_info["name"], | 
					
						
							|  |  |  |         'private_ip' => extended_info["private_ip"], | 
					
						
							|  |  |  |         'public_ip' => extended_info["public_ip"], | 
					
						
							| 
									
										
										
										
											2015-08-05 18:26:21 +03:00
										 |  |  |         'created_by' => stack.owner, | 
					
						
							|  |  |  |         'run_list' => stack.run_list || [], | 
					
						
							| 
									
										
										
										
											2015-08-20 22:42:21 +03:00
										 |  |  |         'stack' => stack.name | 
					
						
							| 
									
										
										
										
											2015-07-30 02:14:45 +03:00
										 |  |  |       } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2015-08-05 18:26:21 +03:00
										 |  |  |       server = ::Devops::Model::Server.new(server_attrs) | 
					
						
							|  |  |  |       mongo.server_insert(server) | 
					
						
							| 
									
										
										
										
											2015-08-31 14:26:16 +03:00
										 |  |  |       # server.chef_node_name =  provider.create_default_chef_node_name(server) | 
					
						
							| 
									
										
										
										
											2015-08-12 12:24:48 +03:00
										 |  |  |       server | 
					
						
							| 
									
										
										
										
											2015-07-30 02:14:45 +03:00
										 |  |  |     end | 
					
						
							|  |  |  |   end | 
					
						
							| 
									
										
										
										
											2015-08-05 18:26:21 +03:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2015-07-20 18:59:26 +03:00
										 |  |  | end |