source: trunk/daemon/togad.py @ 16

Last change on this file since 16 was 16, checked in by bastiaans, 18 years ago

daemon/togad.py:

Storing and updating now working for ALL hosts and metrics

File size: 7.4 KB
Line 
1#!/usr/bin/env python
2
3from xml.sax import make_parser
4from xml.sax.handler import ContentHandler
5import socket
6import sys
7import rrdtool
8import string
9import os
10import os.path
11
12# Specify debugging level here;
13#
14# >10 = metric XML
15# >9  = host,cluster,grid,ganglia XML
16# >8  = RRD activity,gmetad config parsing
17#
18DEBUG_LEVEL = 8
19
20# Where is the gmetad.conf located
21#
22GMETAD_CONF = '/etc/gmetad.conf'
23
24# Where to store the archived rrd's
25#
26ARCHIVE_PATH = '/data/toga/rrds'
27
28# List of data_source names to archive for
29#
30ARCHIVE_SOURCES = [ "LISA Cluster" ]
31
32# Amount of hours to store in one single archived .rrd
33#
34ARCHIVE_HOURS_PER_RRD = 24
35
36
37#
38# Configuration ends
39#
40
41"""
42This is TOrque-GAnglia's data Daemon
43"""
44
45class GangliaXMLHandler( ContentHandler ):
46        "Parse Ganglia's XML"
47
48        metrics = [ ]
49
50        def startElement( self, name, attrs ):
51                "Store appropriate data from xml start tags"
52
53                if name == 'GANGLIA_XML':
54                        self.XMLSource = attrs.get('SOURCE',"")
55                        self.gangliaVersion = attrs.get('VERSION',"")
56                        debug_msg( 10, 'Found XML data: source %s version %s' %( self.XMLSource, self.gangliaVersion ) )
57
58                elif name == 'GRID':
59                        self.gridName = attrs.get('NAME',"")
60                        self.time = attrs.get('LOCALTIME',"")
61                        debug_msg( 10, '`-Grid found: %s' %( self.gridName ) )
62
63                elif name == 'CLUSTER':
64                        self.clusterName = attrs.get('NAME',"")
65                        self.time = attrs.get('LOCALTIME',"")
66                        self.rrd = RRDHandler( self.clusterName )
67                        debug_msg( 10, ' |-Cluster found: %s' %( self.clusterName ) )
68
69                elif name == 'HOST' and self.clusterName in ARCHIVE_SOURCES:     
70                        self.hostName = attrs.get('NAME',"")
71                        self.hostIp = attrs.get('IP',"")
72                        self.hostReported = attrs.get('REPORTED',"")
73                        # Reset the metrics list for each host
74                        self.metrics = [ ]
75                        debug_msg( 10, ' | |-Host found: %s - ip %s reported %s' %( self.hostName, self.hostIp, self.hostReported ) )
76
77                elif name == 'METRIC' and self.clusterName in ARCHIVE_SOURCES:
78                        myMetric = { }
79                        myMetric['name'] = attrs.get('NAME',"")
80                        myMetric['val'] = attrs.get('VAL',"")
81                        myMetric['time'] = self.time
82                        myMetric['type'] = attrs.get('TYPE',"")
83
84                        self.metrics.append( myMetric ) 
85                        debug_msg( 11, ' | | |-metric: %s:%s' %( myMetric['name'], myMetric['val'] ) )
86
87                return
88
89        def endElement( self, name ):
90                #if name == 'GANGLIA_XML':
91
92                #if name == 'GRID':
93
94                #if name == 'CLUSTER':
95
96                if name == 'HOST' and self.clusterName in ARCHIVE_SOURCES:     
97                        self.storeMetrics( self.hostName )
98
99                #if name == 'METRIC':
100
101        def storeMetrics( self, hostname ):
102
103                for metric in self.metrics:
104                        if metric['type'] not in [ 'string' ]:
105
106                                self.rrd.createCheck( hostname, metric )       
107                                self.rrd.update( hostname, metric )
108                                debug_msg( 9, 'stored metric %s for %s: %s' %( hostname, metric['name'], metric['val'] ) )
109                                #sys.exit(1)
110       
111
112class GangliaXMLGatherer:
113        "Setup a connection and file object to Ganglia's XML"
114
115        s = None
116
117        def __init__( self, host, port ):
118                "Store host and port for connection"
119
120                self.host = host
121                self.port = port
122
123        def __del__( self ):
124                "Kill the socket before we leave"
125
126                self.s.close()
127
128        def getFileObject( self ):
129                "Connect, and return a file object"
130
131                for res in socket.getaddrinfo( self.host, self.port, socket.AF_UNSPEC, socket.SOCK_STREAM ):
132                        af, socktype, proto, canonname, sa = res
133                        try:
134                                self.s = socket.socket( af, socktype, proto )
135                        except socket.error, msg:
136                                self.s = None
137                                continue
138                        try:
139                                self.s.connect( sa )
140                        except socket.error, msg:
141                                self.s.close()
142                                self.s = None
143                                continue
144                        break
145
146                if self.s is None:
147                        print 'Could not open socket'
148                        sys.exit(1)
149
150                return self.s.makefile( 'r' )
151
152class GangliaXMLProcessor:
153
154        def daemon( self ):
155                "Run as daemon forever"
156
157                self.DAEMON = 1
158
159                # Fork the first child
160                #
161                pid = os.fork()
162                if pid > 0:
163                        sys.exit(0)  # end parrent
164
165                # creates a session and sets the process group ID
166                #
167                os.setsid()
168
169                # Fork the second child
170                #
171                pid = os.fork()
172                if pid > 0:
173                        sys.exit(0)  # end parrent
174
175                # Go to the root directory and set the umask
176                #
177                os.chdir('/')
178                os.umask(0)
179
180                sys.stdin.close()
181                sys.stdout.close()
182                if (DEBUGLEVEL == 0):
183                        sys.stderr.close()
184
185                os.open('/dev/null', 0)
186                os.dup(0)
187                os.dup(0)
188
189                self.run()
190
191        def run( self ):
192                "Main thread"
193
194                while ( 1 ):
195                        self.processXML()
196                        time.sleep( 5 )
197
198        def processXML( self ):
199                "Process XML"
200
201                myXMLGatherer = GangliaXMLGatherer( 'localhost', 8651 ) 
202
203                myParser = make_parser()   
204                myHandler = GangliaXMLHandler()
205                myParser.setContentHandler( myHandler )
206
207                myParser.parse( myXMLGatherer.getFileObject() )
208
209class GangliaConfigParser:
210
211        sources = [ ]
212
213        def __init__( self, config ):
214                self.config = config
215                self.parseValues()
216
217        def parseValues(self):
218                "Parse certain values from gmetad.conf"
219
220                readcfg = open( self.config, 'r' )
221
222                for line in readcfg.readlines():
223
224                        if line.count( '"' ) > 1:
225
226                                if line.find( 'data_source' ) != -1 and line[0] != '#':
227
228                                        source = { }
229                                        source['name'] = line.split( '"' )[1]
230                                        source_words = line.split( '"' )[2].split( ' ' )
231
232                                        for word in source_words:
233
234                                                valid_interval = 1
235
236                                                for letter in word:
237                                                        if letter not in string.digits:
238                                                                valid_interval = 0
239
240                                                if valid_interval and len(word) > 0:
241                                                        source['interval'] = word
242                                                        debug_msg( 9, 'polling interval for %s = %s' %(source['name'], source['interval'] ) )
243               
244                # No interval found, use Ganglia's default     
245                if not source.has_key( 'interval' ):
246                        source['interval'] = 15
247                        debug_msg( 9, 'polling interval for %s defaulted to 15' %(source['name']) )
248
249                self.sources.append( source )
250
251        def getInterval( self, source_name ):
252                for source in self.sources:
253                        if source['name'] == source_name:
254                                return source['interval']
255                return None
256
257class RRDHandler:
258
259        def __init__( self, cluster ):
260                self.cluster = cluster
261
262                self.gmetad_conf = GangliaConfigParser( GMETAD_CONF )
263
264        def createCheck( self, host, metric ):
265                "Check if an .rrd allready exists for this metric, create if not"
266
267                rrd_dir = '%s/%s/%s' %( check_dir(ARCHIVE_PATH), self.cluster, host )
268                rrd_file = '%s/%s.rrd' %( rrd_dir, metric['name'] )
269
270                if not os.path.exists( rrd_dir ):
271                        os.makedirs( rrd_dir )
272                        debug_msg( 9, 'created dir %s' %( str(rrd_dir) ) )
273
274                if not os.path.exists( rrd_file ):
275
276                        interval = self.gmetad_conf.getInterval( self.cluster )
277                        heartbeat = 8 * int(interval)
278
279                        param_step1 = '--step'
280                        param_step2 = str( interval )
281
282                        param_start1 = '--start'
283                        param_start2 = str( int( metric['time'] ) - 1 )
284
285                        param_ds = 'DS:sum:GAUGE:%d:U:U' %heartbeat
286                        param_rra = 'RRA:AVERAGE:0.5:1:%s' %(ARCHIVE_HOURS_PER_RRD * 240)
287
288                        rrdtool.create( str(rrd_file), param_step1, param_step2, param_start1, param_start2, param_ds, param_rra )
289
290                        debug_msg( 9, 'created rrd %s' %( str(rrd_file) ) )
291
292        def update( self, host, metric ):
293
294                rrd_dir = '%s/%s/%s' %( check_dir(ARCHIVE_PATH), self.cluster, host )
295                rrd_file = '%s/%s.rrd' %( rrd_dir, metric['name'] )
296
297                timestamp = metric['time']
298                val = metric['val']
299
300                update_string = '%s:%s' %(timestamp, val)
301
302                rrdtool.update( str(rrd_file), str(update_string) )
303                debug_msg( 9, 'updated rrd %s with %s' %( str(rrd_file), update_string ) )
304
305def main():
306        "Program startup"
307
308        myProcessor = GangliaXMLProcessor()
309        myProcessor.processXML()
310
311def check_dir( directory ):
312        "Check if directory is a proper directory. I.e.: Does _not_ end with a '/'"
313
314        if directory[-1] == '/':
315                directory = directory[:-1]
316
317        return directory
318
319def debug_msg( level, msg ):
320
321        if (DEBUG_LEVEL >= level):
322                sys.stderr.write( msg + '\n' )
323
324# Let's go
325if __name__ == '__main__':
326        main()
Note: See TracBrowser for help on using the repository browser.