-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathmain.rb
More file actions
45 lines (40 loc) · 1.19 KB
/
Copy pathmain.rb
File metadata and controls
45 lines (40 loc) · 1.19 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
# Requires for load classes
require 'json'
require_relative 'source/classes/agent'
require_relative 'source/classes/job'
require_relative 'source/classes/ruby_queue'
require_relative 'source/utils/hash'
# Reading from STDIN
stdin = ARGF.read
parsed_input = JSON.parse(stdin)
stdout = []
# Instanciating queue
queue = RubyQueue.new
# Reading inputs
parsed_input.each do |individual_command|
individual_command.each do |command, params|
# Symbolizing params keys to normalize key names
params.clean_spaces_keys!
params.deep_symbolize_keys!
# Selecting each option
case command.to_sym
when :new_agent
queue.new_agent Agent.new(params[:id],
params[:name],
params[:primary_skillset],
params[:secondary_skillset])
when :new_job
queue.new_job Job.new(params[:id], params[:type], params[:urgent])
when :job_request
job = queue.dequeue(params[:agent_id])
next if job.nil?
stdout << {
job_assigned: {
job_id: job.id,
agent_id: params[:agent_id]
}
}
end
end
end
open('output.json', 'w') { |f| f.puts stdout.to_json }