Class: BigBang::Universe

Inherits:
Object
  • Object
show all
Includes:
ExplodeCmd, KillCmd, TestCmd
Defined in:
lib/bigbang/universe.rb

Instance Method Summary collapse

Methods included from ExplodeCmd

#allocate_addresses, #assign_addresses, #create_dns_entries, #create_lb, #create_load_balancers, #dns_entry_for, #elb_dns_entries, #explode, #instance_dns_entries, #run_instance, #run_instances, #tag_instance, #wait_for_eips, #wait_for_running

Methods included from TestCmd

#test, #test_amis, #test_availability_zones, #test_dns

Methods included from KillCmd

#kill, #kill_dns_entry, #kill_eip, #kill_elb, #kill_instance

Constructor Details

#initialize(dsl) ⇒ Universe

Returns a new instance of Universe.



15
16
17
18
19
 
# File 'lib/bigbang/universe.rb', line 15

def initialize(dsl)
	@instances = dsl.instances
	@runs = dsl.runs
	@config = dsl.conf
end

Instance Method Details

#configured_elbs ⇒ Object 



55
56
57
58
59
60
61
62
63
 
# File 'lib/bigbang/universe.rb', line 55

def configured_elbs
	lbs = []
	@runs.each do |r|
		next unless r.is_a?(ClusterRun)
		next if r.lb.nil?
		lbs << r.lb
	end
	lbs
end

#confirm(msg) ⇒ Object 



107
108
109
110
111
112
113
114
115
116
 
# File 'lib/bigbang/universe.rb', line 107

def confirm(msg)
	print "#{msg}? [y/N] "
	if STDIN.gets.strip == "y"
		yield
		return true
	else
		puts "skipping"
		return false
	end
end

#get_addresses ⇒ Object 



65
66
67
68
69
 
# File 'lib/bigbang/universe.rb', line 65

def get_addresses
	aset = provider.ec2.describe_addresses.addressesSet
	return [] if aset.nil?
	aset.item
end

#get_availability_zones ⇒ Object 



26
27
28
29
30
 
# File 'lib/bigbang/universe.rb', line 26

def get_availability_zones
	item = provider.ec2.describe_availability_zones.availabilityZoneInfo.item
	item = [] if item.nil?
	item
end

#get_instances ⇒ Object 



32
33
34
35
36
37
38
39
40
41
42
43
44
45
 
# File 'lib/bigbang/universe.rb', line 32

def get_instances
	rset = provider.ec2.describe_instances.reservationSet
	if rset.nil?
		return []
	else
		instances = []
		rset.item.each do |reservation|
			reservation.instancesSet.item.each do |i|
				instances << i
			end
		end
		return instances
	end
end

#get_instances_map ⇒ Object 



47
48
49
50
51
52
53
 
# File 'lib/bigbang/universe.rb', line 47

def get_instances_map
	map = {}
	get_instances.each do |i|
		map[i.instanceId] = i
	end
	map
end

#get_tags ⇒ Object 



79
80
81
82
83
84
85
86
87
 
# File 'lib/bigbang/universe.rb', line 79

def get_tags
	tag_set = provider.ec2.describe_tags(:filter => 
			[{'key' => 'bb_universe'}]).tagSet
	if tag_set.nil?
		return []
	end

	tag_set.item
end

#instance_tags(tags = nil) ⇒ Object 



131
132
133
134
 
# File 'lib/bigbang/universe.rb', line 131

def instance_tags(tags = nil)
	tags = get_tags if tags.nil?
	tags.find_all { |t| t.resourceType == 'instance' }
end

#list ⇒ Object 



89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
 
# File 'lib/bigbang/universe.rb', line 89

def list
	universes = Set.new
	get_tags.each do |tag|
		universes << tag.value
	end

	running = running_instances

	universes.each do |u|
		instances = universe_running_instances(running, universe_tags(u))
		if instances.empty?
			puts "#{u} (defunct)"
		else
			puts "#{u} (#{instances.size} running instances)"
		end
	end
end

#notify(msg) ⇒ Object 



118
119
120
121
122
123
124
125
126
127
128
129
 
# File 'lib/bigbang/universe.rb', line 118

def notify(msg)
	print "#{msg}: "
	STDOUT.flush
	begin
		r = yield
		puts "\033[01;32mOK\033[00m"
		return r
	rescue => e
		puts "\033[01;31mERROR"
		puts "#{e.to_s}\033[00m"
	end
end

#provider ⇒ Object 



21
22
23
24
 
# File 'lib/bigbang/universe.rb', line 21

def provider
	return @provider unless @provider.nil?
	@provider = Provider.new(@config)
end

#running_instances ⇒ Object 



71
72
73
 
# File 'lib/bigbang/universe.rb', line 71

def running_instances
	get_instances.find_all { |i| i.instanceState.name == "running" }
end

#running_instances_count(ids) ⇒ Object 



75
76
77
 
# File 'lib/bigbang/universe.rb', line 75

def running_instances_count(ids)
	running_instances.count { |i| ids.include?(i.instanceId) }
end

#universe_running_instances(running, universe_tags) ⇒ Object 



136
137
138
139
140
141
142
143
144
 
# File 'lib/bigbang/universe.rb', line 136

def universe_running_instances(running, universe_tags)
	res = []
	instance_tags(universe_tags).each do |t|
		i = running.find { |i| i.instanceId == t.resourceId }
		res << i unless i.nil?
	end

	res
end

#universe_tags(name) ⇒ Object 



146
147
148
149
150
 
# File 'lib/bigbang/universe.rb', line 146

def universe_tags(name)
	get_tags.find_all do |tag|
		tag.value == name
	end
end