blob: 9378ee2cdfe7d6312dd6a8a5abaa4a2bd90db027 (
plain)
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
46
47
48
49
50
51
|
require "msgpack/rpc/transport/unix"
require_relative "event_queue"
require_relative "node_connection"
require_relative "node_manager"
require_relative "control_server"
require_relative "notification_queue"
module Collector
module Daemon
class << self
attr_reader :start_time
def start
@start_time = Time.now
set_loggers
EM.run do
sock_path = File.join(Rails.root, "tmp", "sockets", "collector.sock")
File.delete(sock_path) if File.exist?(sock_path)
control = MessagePack::RPC::Server.new
control.listen(MessagePack::RPC::UNIXServerTransport.new(sock_path), Collector::ControlServer.new)
EM.defer { control.run }
event_queue = Collector::EventQueue.new
EM.add_periodic_timer(Settings.collector.flush_interval) do
event_queue.flush
end
NotificationQueue.start
nodes = EM.start_server("0.0.0.0", Settings.collector.server_port, Collector::NodeConnection, event_queue)
stop = -> _ do
control.stop
EM.stop_server(nodes)
EM.stop
end
Signal.trap("INT", &stop)
Signal.trap("TERM", &stop)
end
end
private
def set_loggers
_logger = Logger.new(STDOUT)
_logger.level = Rails.env.production? ? Logger::INFO : Logger::DEBUG
ActiveRecord::Base.logger = Rails.logger = _logger
end
end
end
end
|