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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
|
# encoding: UTF-8
require 'yaml'
require 'socket'
# require 'mauve/datamapper'
require 'mauve/proto'
require 'mauve/alert'
require 'mauve/mauve_thread'
require 'mauve/mauve_time'
require 'mauve/timer'
require 'mauve/udp_server'
require 'mauve/processor'
require 'mauve/http_server'
require 'log4r'
module Mauve
class Server
DEFAULT_CONFIGURATION = {
:ip => "127.0.0.1",
:port => 32741,
:database => "sqlite3:///./mauvealert.db",
:log_file => "stdout",
:log_level => 1,
:transmission_cache_expire_time => 600
}
#
# This is the order in which the threads should be started.
#
THREAD_CLASSES = [UDPServer, HTTPServer, Processor, Notifier, Timer]
attr_accessor :web_interface
attr_reader :stopped_at, :started_at, :initial_sleep
include Singleton
def initialize
# Set the logger up
@logger = Log4r::Logger.new(self.class.to_s)
# Sleep time between pooling the @buffer buffer.
@sleep = 1
@freeze = false
@stop = false
@stopped_at = MauveTime.now
@started_at = MauveTime.now
@initial_sleep = 300
@config = DEFAULT_CONFIGURATION
end
def configure(config_spec = nil)
#
# Update the configuration
#
if config_spec.nil?
# Do nothing
elsif config_spec.kind_of?(String) and File.exists?(config_spec)
@config.update(YAML.load_file(config_spec))
elsif config_spec.kind_of?(Hash)
@config.update(config_spec)
else
raise ArgumentError.new("Unknown configuration spec "+config_spec.inspect)
end
#
DataMapper.setup(:default, @config[:database])
DataObjects::Sqlite3.logger = Log4r::Logger.new("Mauve::DataMapper")
#
# Update any tables.
#
Alert.auto_upgrade!
AlertChanged.auto_upgrade!
Mauve::AlertEarliestDate.create_view!
#
# Work out when the server was last stopped
#
@stopped_at = self.last_heartbeat
end
def last_heartbeat
#
# Work out when the last update was
#
[ Alert.last(:order => :updated_at.asc),
AlertChanged.last(:order => :updated_at.asc) ].
reject{|a| a.nil?}.
collect{|a| a.updated_at.to_time}.
sort.
last
end
def freeze
@frozen = true
end
def thaw
@thaw = true
end
def stop
@stop = true
THREAD_CLASSES.reverse.each do |klass|
klass.instance.stop
end
@logger.info("All threads stopped")
end
def run
loop do
THREAD_CLASSES.each do |klass|
next if @frozen or @stop
unless klass.instance.alive?
# ugh something has died.
klass.instance.join
klass.instance.start unless @stop
end
end
break if @stop
sleep 1
end
logger.debug("Thread stopped")
end
alias start run
end
end
|
