Class: MAS

Inherits:
Object
  • Object
show all
Defined in:
lib/agens/mas.rb

Constant Summary collapse

@@counters = 
{}
@@recipes = 
[]

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize ⇒ MAS

Returns a new instance of MAS.



18
19
20
21
22
23
24
25
26
27
28
29
30
31
 
# File 'lib/agens/mas.rb', line 18

def initialize
	@config = Celluloid::Supervision::Configuration.new
	@names = []
	
	@@recipes.each do |agent_recipe|
		if !agent_recipe.args.nil? && agent_recipe.args.any?
			@config.define(type: agent_recipe.type, as: agent_recipe.name, args: agent_recipe.args)
		else
			@config.define(type: agent_recipe.type, as: agent_recipe.name)
		end
		
		@names << agent_recipe.name
	end
end

Class Method Details

.agent(agent, options = {}) ⇒ Object

region DSL agent MyAgent agent MyAgent, as: :james_bond agent MyAgent, as: :james_bond, args: [1, 2]



63
64
65
 
# File 'lib/agens/mas.rb', line 63

def self.agent(agent, options = {})
	@@recipes << self.generate_recipe(agent, options)
end

.agents(agent, options = {}) ⇒ Object

agents MyAgent agents MyAgent, as: :james_bonds agents MyAgent, count: 10 agents MyAgent, as: :james_bonds, count: 10 agents MyAgent, as: :james_bonds, count: 10, args: [1, 2]



72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
 
# File 'lib/agens/mas.rb', line 72

def self.agents(agent, options = {})
	mas_name = self.name.to_sym
	pool_name = "#{agent.class.name.to_sym}_pool"
	@@counters[pool_name] ||= 0
	@@counters[pool_name] += 1
	
	name = options.key?(:as) ? options[:as] : "#{mas_name}_#{pool_name}_#{@@counters[pool_name]}"
	count = options.key?(:count) ? options[:count] : nil
	
	count = 2 if count == nil
	
	count.times do
		self.agent(agent, args: options[:args])
	end
	
	# TODO add support for new celluloid supervision pools (when they will be documented, see https://github.com/celluloid/celluloid-supervision)
	# if count == nil
	# 	pool agent, as: name
	# else
	# 	pool agent, as: name, size: count
	# end
end

.generate_recipe(agent, options = {}) ⇒ Object

endregion DSL



96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
 
# File 'lib/agens/mas.rb', line 96

def self.generate_recipe(agent, options = {})
	name = nil
	if options.key?(:as)
		name = options[:as]
	else
		mas_name = self.name.to_sym
		agent_name = agent.name.to_sym
		@@counters[agent_name] ||= 0
		@@counters[agent_name] += 1
		
		name = "#{mas_name}_#{agent_name}_#{@@counters[agent_name]}"
	end
	
	AgentConfigRecipe.new(agent, name, options[:args])
end

Instance Method Details

#add_agent(agent, options = {}) ⇒ Object 



47
48
49
50
51
52
53
54
55
56
57
 
# File 'lib/agens/mas.rb', line 47

def add_agent(agent, options = {})
	recipe << self.generate_recipe(agent, options)
	
	if !recipe.args.nil? && recipe.args.any?
		@config.add(type: recipe.type, as: recipe.name, args: recipe.args)
	else
		@config.add(type: recipe.type, as: recipe.name)
	end
	
	self
end

#run ⇒ Object 



33
34
35
36
 
# File 'lib/agens/mas.rb', line 33

def run
	puts "Running MAS #{self.class.name}..."
	@config.deploy
end

#shutdown ⇒ Object 



38
39
40
41
 
# File 'lib/agens/mas.rb', line 38

def shutdown
	puts "Stoping MAS #{self.class.name}..."
	@config.shutdown
end

#world ⇒ Object 



43
44
45
 
# File 'lib/agens/mas.rb', line 43

def world
	World.new(@names)
end