source: trunk/daemon/togad.py @ 31

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

daemon/togad.py:

Bug was caused by incorrect timestamping

  • Now host reporting time is used

New bug: when a ganglia monitor has a big interval
it will report the same value for a second xmlthread run.

  • One cannot update an rrd twice with the same timestamp
  • Will have to check timestamp of metrics each run

This implicates that for every metric for every host the
timestamp of the last xml run must be kept in memory

  • Might as well rewrite rrd code with in-memory metric storage
  • And do scheduled/queued rrd updates, in stead of every xml run: big performance increase
File size: 11.1 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
11import time
12import re
13
14# Specify debugging level here;
15#
16# <=11 = metric XML
17# <=10 = host,cluster,grid,ganglia XML
18# <=9  = RRD activity,gmetad config parsing
19# <=8  = host processing
20# <=7  = daemon threading - NOTE: Daemon will 'halt on all errors' from this level
21#
22DEBUG_LEVEL = 8
23
24# Where is the gmetad.conf located
25#
26GMETAD_CONF = '/etc/gmetad.conf'
27
28# Where to store the archived rrd's
29#
30ARCHIVE_PATH = '/data/toga/rrds'
31
32# List of data_source names to archive for
33#
34ARCHIVE_SOURCES = [ "LISA Cluster" ]
35
36# Amount of hours to store in one single archived .rrd
37#
38ARCHIVE_HOURS_PER_RRD = 12
39
40# Interval at which to grab&store XML
41#
42GRAB_INTERVAL = 15
43
44# Wether or not to run as a daemon in background
45#
46DAEMONIZE = 0
47
48######################
49#                    #
50# Configuration ends #
51#                    #
52######################
53
54# What XML data types not to store
55#
56UNSUPPORTED_ARCHIVE_TYPES = [ 'string' ]
57
58"""
59This is TOrque-GAnglia's data Daemon
60"""
61
62class GangliaXMLHandler( ContentHandler ):
63        "Parse Ganglia's XML"
64
65        metrics = [ ]
66
67        def startElement( self, name, attrs ):
68                "Store appropriate data from xml start tags"
69
70                if name == 'GANGLIA_XML':
71                        self.XMLSource = attrs.get('SOURCE',"")
72                        self.gangliaVersion = attrs.get('VERSION',"")
73                        debug_msg( 10, 'Found XML data: source %s version %s' %( self.XMLSource, self.gangliaVersion ) )
74
75                elif name == 'GRID':
76                        self.gridName = attrs.get('NAME',"")
77                        self.time = attrs.get('LOCALTIME',"")
78                        debug_msg( 10, '`-Grid found: %s' %( self.gridName ) )
79
80                elif name == 'CLUSTER':
81                        self.clusterName = attrs.get('NAME',"")
82                        self.time = attrs.get('LOCALTIME',"")
83                        self.rrd = RRDHandler( self.clusterName )
84                        debug_msg( 10, ' |-Cluster found: %s' %( self.clusterName ) )
85
86                elif name == 'HOST' and self.clusterName in ARCHIVE_SOURCES:     
87                        self.hostName = attrs.get('NAME',"")
88                        self.hostIp = attrs.get('IP',"")
89                        self.hostReported = attrs.get('REPORTED',"")
90                        # Reset the metrics list for each host
91                        self.metrics = [ ]
92                        debug_msg( 10, ' | |-Host found: %s - ip %s reported %s' %( self.hostName, self.hostIp, self.hostReported ) )
93
94                elif name == 'METRIC' and self.clusterName in ARCHIVE_SOURCES:
95                        myMetric = { }
96                        myMetric['name'] = attrs.get('NAME',"")
97                        myMetric['val'] = attrs.get('VAL',"")
98                        myMetric['time'] = self.hostReported
99                        myMetric['type'] = attrs.get('TYPE',"")
100
101                        self.metrics.append( myMetric ) 
102                        debug_msg( 11, ' | | |-metric: %s:%s' %( myMetric['name'], myMetric['val'] ) )
103
104                return
105
106        def endElement( self, name ):
107                #if name == 'GANGLIA_XML':
108
109                #if name == 'GRID':
110
111                #if name == 'CLUSTER':
112
113                if name == 'HOST' and self.clusterName in ARCHIVE_SOURCES:     
114
115                        # Determine time here, so all use same time in this run
116                        mytime = self.rrd.makeTimeSerial()
117                        correct_serial = self.rrd.checkNewRrdPeriod( self.hostName, mytime )
118
119                        #debug_msg( 8, 'time %s: Storing metrics for %s' %(mytime, self.hostName) )
120                        print 'Storing metrics for %s:' %(self.hostName),
121                        self.storeMetrics( self.hostName, correct_serial )
122
123                #if name == 'METRIC':
124
125        def storeMetrics( self, hostname, timeserial ):
126
127                for metric in self.metrics:
128                        if metric['type'] not in UNSUPPORTED_ARCHIVE_TYPES:
129
130                                self.rrd.createCheck( hostname, metric, timeserial )   
131                                print ' [%s.%s]' %(metric['name'], metric['time']),
132                                self.rrd.update( hostname, metric, timeserial )
133                                debug_msg( 9, 'stored metric %s for %s: %s' %( hostname, metric['name'], metric['val'] ) )
134                                #sys.exit(1)
135                print
136       
137
138class GangliaXMLGatherer:
139        "Setup a connection and file object to Ganglia's XML"
140
141        s = None
142
143        def __init__( self, host, port ):
144                "Store host and port for connection"
145
146                self.host = host
147                self.port = port
148
149        def __del__( self ):
150                "Kill the socket before we leave"
151
152                self.s.close()
153
154        def getFileObject( self ):
155                "Connect, and return a file object"
156
157                for res in socket.getaddrinfo( self.host, self.port, socket.AF_UNSPEC, socket.SOCK_STREAM ):
158                        af, socktype, proto, canonname, sa = res
159                        try:
160                                self.s = socket.socket( af, socktype, proto )
161                        except socket.error, msg:
162                                self.s = None
163                                continue
164                        try:
165                                self.s.connect( sa )
166                        except socket.error, msg:
167                                self.s.close()
168                                self.s = None
169                                continue
170                        break
171
172                if self.s is None:
173                        print 'Could not open socket'
174                        sys.exit(1)
175
176                return self.s.makefile( 'r' )
177
178class GangliaXMLProcessor:
179
180        def daemon( self ):
181                "Run as daemon forever"
182
183                self.DAEMON = 1
184
185                # Fork the first child
186                #
187                pid = os.fork()
188                if pid > 0:
189                        sys.exit(0)  # end parrent
190
191                # creates a session and sets the process group ID
192                #
193                os.setsid()
194
195                # Fork the second child
196                #
197                pid = os.fork()
198                if pid > 0:
199                        sys.exit(0)  # end parrent
200
201                # Go to the root directory and set the umask
202                #
203                os.chdir('/')
204                os.umask(0)
205
206                sys.stdin.close()
207                sys.stdout.close()
208                sys.stderr.close()
209
210                os.open('/dev/null', 0)
211                os.dup(0)
212                os.dup(0)
213
214                self.run()
215
216        def printTime( self ):
217
218                return time.strftime("%a, %d %b %Y %H:%M:%S")
219
220        def run( self ):
221                "Main thread"
222
223                while ( 1 ):
224
225                        debug_msg( 7, self.printTime() + ' - mainthread() - xmlthread() started' )
226                        pid = os.fork()
227
228                        if pid == 0:
229                                # Child - XML Thread
230                                #
231                                # Process XML and exit
232
233                                debug_msg( 7, self.printTime() + ' - xmlthread()  - Start XML processing..' )
234                                self.processXML()
235                                debug_msg( 7, self.printTime() + ' - xmlthread()  - Done processing; exiting.' )
236                                sys.exit( 0 )
237
238                        elif pid > 0:
239                                # Parent - Daemon Thread
240
241                                debug_msg( 7, self.printTime() + ' - mainthread() - Sleep '+ str(GRAB_INTERVAL) +'s: zzzzz..' )
242                                time.sleep( GRAB_INTERVAL )
243                                debug_msg( 7, self.printTime() + ' - mainthread() - Awoken: waiting for XML thread..' )
244
245                                r = os.wait()
246                                ret = r[1]
247                                if ret != 0:
248                                        debug_msg( 7, self.printTime() + ' - mainthread() - Done waiting: ERROR! xmlthread() exited with status %d' %(ret) )
249                                        if DEBUG_LEVEL>=7: sys.exit( 1 )
250                                else:
251
252                                        debug_msg( 7, self.printTime() + ' - mainthread() - Done waiting: xmlthread() finished succesfully' )
253
254        def processXML( self ):
255                "Process XML"
256
257                myXMLGatherer = GangliaXMLGatherer( 'localhost', 8651 ) 
258
259                myParser = make_parser()   
260                myHandler = GangliaXMLHandler()
261                myParser.setContentHandler( myHandler )
262
263                myParser.parse( myXMLGatherer.getFileObject() )
264
265class GangliaConfigParser:
266
267        sources = [ ]
268
269        def __init__( self, config ):
270                self.config = config
271                self.parseValues()
272
273        def parseValues(self):
274                "Parse certain values from gmetad.conf"
275
276                readcfg = open( self.config, 'r' )
277
278                for line in readcfg.readlines():
279
280                        if line.count( '"' ) > 1:
281
282                                if line.find( 'data_source' ) != -1 and line[0] != '#':
283
284                                        source = { }
285                                        source['name'] = line.split( '"' )[1]
286                                        source_words = line.split( '"' )[2].split( ' ' )
287
288                                        for word in source_words:
289
290                                                valid_interval = 1
291
292                                                for letter in word:
293                                                        if letter not in string.digits:
294                                                                valid_interval = 0
295
296                                                if valid_interval and len(word) > 0:
297                                                        source['interval'] = word
298                                                        debug_msg( 9, 'polling interval for %s = %s' %(source['name'], source['interval'] ) )
299               
300                # No interval found, use Ganglia's default     
301                if not source.has_key( 'interval' ):
302                        source['interval'] = 15
303                        debug_msg( 9, 'polling interval for %s defaulted to 15' %(source['name']) )
304
305                self.sources.append( source )
306
307        def getInterval( self, source_name ):
308                for source in self.sources:
309                        if source['name'] == source_name:
310                                return source['interval']
311                return None
312
313class RRDHandler:
314
315        def __init__( self, cluster ):
316                self.cluster = cluster
317                self.gmetad_conf = GangliaConfigParser( GMETAD_CONF )
318
319        def makeTimeSerial( self ):
320
321                # YYYYMMDDhhmmss: 20050321143411
322                #mytime = time.strftime( "%Y%m%d%H%M%S" )
323
324                # Seconds since epoch
325                mytime = int( time.time() )
326
327                return mytime
328
329        def makeRrdPath( self, host, metric=None, timeserial=None ):
330
331                if not timeserial:     
332                        rrd_dir = '%s/%s/%s' %( check_dir(ARCHIVE_PATH), self.cluster, host )
333                else:
334                        rrd_dir = '%s/%s/%s/%s' %( check_dir(ARCHIVE_PATH), self.cluster, host, timeserial )
335                if metric:
336                        rrd_file = '%s/%s.rrd' %( rrd_dir, metric['name'] )
337                else:
338                        rrd_file = None
339
340                return rrd_dir, rrd_file
341
342        def getLastRrdTimeSerial( self, host ):
343
344                rrd_dir, rrd_file = self.makeRrdPath( host )
345
346                newest_timeserial = 0
347
348                if os.path.exists( rrd_dir ):
349                        for root, dirs, files in os.walk( rrd_dir ):
350
351                                for dir in dirs:
352
353                                        valid_dir = 1
354
355                                        for letter in dir:
356                                                if letter not in string.digits:
357                                                        valid_dir = 0
358
359                                        if valid_dir:
360                                                timeserial = dir
361                                                if timeserial > newest_timeserial:
362                                                        newest_timeserial = timeserial
363
364                if newest_timeserial:
365                        return newest_timeserial
366                else:
367                        return 0
368
369        def checkNewRrdPeriod( self, host, current_timeserial ):
370
371                last_timeserial = int( self.getLastRrdTimeSerial( host ) )
372                debug_msg( 9, 'last timeserial of %s is %s' %( host, last_timeserial ) )
373
374                if not last_timeserial:
375                        serial = current_timeserial
376                else:
377
378                        archive_secs = ARCHIVE_HOURS_PER_RRD * (60 * 60)
379
380                        if (current_timeserial - last_timeserial) >= archive_secs:
381                                serial = current_timeserial
382                        else:
383                                serial = last_timeserial
384
385                return serial
386
387        def createCheck( self, host, metric, timeserial ):
388                "Check if an .rrd allready exists for this metric, create if not"
389
390                debug_msg( 9, 'rrdcreate: using timeserial %s for %s/%s' %( timeserial, host, metric['name'] ) )
391
392                rrd_dir, rrd_file = self.makeRrdPath( host, metric, timeserial )
393
394                if not os.path.exists( rrd_dir ):
395                        os.makedirs( rrd_dir )
396                        debug_msg( 9, 'created dir %s' %( str(rrd_dir) ) )
397
398                if not os.path.exists( rrd_file ):
399
400                        interval = self.gmetad_conf.getInterval( self.cluster )
401                        heartbeat = 8 * int(interval)
402
403                        param_step1 = '--step'
404                        param_step2 = str( interval )
405
406                        param_start1 = '--start'
407                        param_start2 = str( int( metric['time'] ) - 1 )
408
409                        param_ds = 'DS:sum:GAUGE:%d:U:U' %heartbeat
410                        param_rra = 'RRA:AVERAGE:0.5:1:%s' %(ARCHIVE_HOURS_PER_RRD * 240)
411
412                        rrdtool.create( str(rrd_file), param_step1, param_step2, param_start1, param_start2, param_ds, param_rra )
413
414                        debug_msg( 9, 'created rrd %s' %( str(rrd_file) ) )
415
416        def update( self, host, metric, timeserial ):
417
418                debug_msg( 9, 'rrdupdate: using timeserial %s for %s/%s' %( timeserial, host, metric['name'] ) )
419
420                rrd_dir, rrd_file = self.makeRrdPath( host, metric, timeserial )
421
422                timestamp = metric['time']
423                val = metric['val']
424
425                update_string = '%s:%s' %(timestamp, val)
426
427                try:
428                        rrdtool.update( str(rrd_file), str(update_string) )
429                except rrdtool.error, detail:
430                        debug_msg( 0, 'EXCEPTION! While trying to update rrd:' )
431                        debug_msg( 0, '\trrd %s with %s' %( str(rrd_file), update_string ) )
432                        debug_msg( 0, str(detail) )
433                        sys.exit( 1 )
434               
435                debug_msg( 9, 'updated rrd %s with %s' %( str(rrd_file), update_string ) )
436
437def main():
438        "Program startup"
439
440        myProcessor = GangliaXMLProcessor()
441
442        if DAEMONIZE:
443                myProcessor.daemon()
444        else:
445                myProcessor.run()
446
447def check_dir( directory ):
448        "Check if directory is a proper directory. I.e.: Does _not_ end with a '/'"
449
450        if directory[-1] == '/':
451                directory = directory[:-1]
452
453        return directory
454
455def debug_msg( level, msg ):
456
457        if (DEBUG_LEVEL >= level):
458                sys.stderr.write( msg + '\n' )
459
460# Let's go
461if __name__ == '__main__':
462        main()
Note: See TracBrowser for help on using the repository browser.