Changeset 33


Ignore:
Timestamp:
04/04/05 11:11:07 (17 years ago)
Author:
bastiaans
Message:

daemon/togad.py:

  • Rewrote classes to be reusable
  • Seperated storing from gathering
  • Setup RRD creation/updating to use in-memory metrics
File:
1 edited

Legend:

Unmodified
Added
Removed
  • trunk/daemon/togad.py

    r32 r33  
    2020# <=7  = daemon threading - NOTE: Daemon will 'halt on all errors' from this level
    2121#
    22 DEBUG_LEVEL = 8
     22DEBUG_LEVEL = 9
    2323
    2424# Where is the gmetad.conf located
     
    3737#
    3838ARCHIVE_HOURS_PER_RRD = 12
    39 
    40 # Interval at which to grab&store XML
    41 #
    42 GRAB_INTERVAL = 15
    4339
    4440# Wether or not to run as a daemon in background
     
    6359        "Parse Ganglia's XML"
    6460
    65         def __init__( self ):
    66                 self.metrics = [ ]
    67                 self.clusters = { }
     61        clusters = { }
     62        config = None
     63
     64        def __init__( self, config ):
     65                self.config = config
    6866
    6967        def startElement( self, name, attrs ):
     
    8987                        self.time = attrs.get( 'LOCALTIME', "" )
    9088
    91                         self.clusters[ self.clusterName ] = RRDHandler( self.clusterName )
     89                        if not self.clusters.has_key( self.clusterName ):
     90
     91                                self.clusters[ self.clusterName ] = RRDHandler( self.clusterName )
    9292
    9393                        debug_msg( 10, ' |-Cluster found: %s' %( self.clusterName ) )
     
    118118        def storeMetrics( self, hostname, timeserial ):
    119119
    120                 for metric in self.metrics:
    121                         if metric['type'] not in UNSUPPORTED_ARCHIVE_TYPES:
    122 
    123                                 self.rrd.createCheck( hostname, metric, timeserial )   
    124                                 self.rrd.update( hostname, metric, timeserial )
    125                                 debug_msg( 9, 'stored metric %s for %s: %s' %( hostname, metric['name'], metric['val'] ) )
    126                                 #sys.exit(1)
     120                for cluster in self.clusters:
     121
     122                        cluster.storeMetrics()
    127123
    128124class GangliaXMLGatherer:
     
    136132                self.host = host
    137133                self.port = port
    138 
    139         def __del__( self ):
    140                 "Kill the socket before we leave"
    141 
    142                 self.s.close()
    143 
    144         def getFileObject( self ):
    145                 "Connect, and return a file object"
     134                self.connect()
     135
     136        def connect( self ):
     137                "Setup connection to XML source"
    146138
    147139                for res in socket.getaddrinfo( self.host, self.port, socket.AF_UNSPEC, socket.SOCK_STREAM ):
     
    172164                if self.s is None:
    173165
    174                         print 'Could not open socket'
    175                         sys.exit(1)
     166                        debug_msg( 0, 'Could not open socket' )
     167                        sys.exit( 1 )
     168
     169        def disconnect( self ):
     170                "Close socket"
     171
     172                if self.s:
     173                        self.s.close()
     174                        self.s = None
     175
     176        def __del__( self ):
     177                "Kill the socket before we leave"
     178
     179                self.disconnect()
     180
     181        def getFileObject( self ):
     182                "Connect, and return a file object"
     183
     184                if not self.s:
     185                        self.connect()
    176186
    177187                return self.s.makefile( 'r' )
    178188
    179189class GangliaXMLProcessor:
     190
     191        def __init__( self ):
     192                "Setup initial XML connection and handlers"
     193
     194                self.config = GangliaConfigParser( GMETAD_CONF )
     195
     196                self.myXMLGatherer = GangliaXMLGatherer( 'localhost', 8651 )
     197                self.myParser = make_parser()   
     198                self.myHandler = GangliaXMLHandler( self.config )
     199                self.myParser.setContentHandler( self.myHandler )
    180200
    181201        def daemon( self ):
     
    220240
    221241        def printTime( self ):
    222 
    223                 return time.strftime("%a, %d %b %Y %H:%M:%S")
     242                "Print current time in human readable format"
     243
     244                return time.strftime("%a %d %b %Y %H:%M:%S")
     245
     246        def grabXML( self ):
     247
     248                debug_msg( 7, self.printTime() + ' - mainthread() - xmlthread() started' )
     249                pid = os.fork()
     250
     251                if pid == 0:
     252                        # Child - XML Thread
     253                        #
     254                        # Process XML and exit
     255
     256                        debug_msg( 7, self.printTime() + ' - xmlthread()  - Start XML processing..' )
     257                        self.processXML()
     258                        debug_msg( 7, self.printTime() + ' - xmlthread()  - Done processing; exiting.' )
     259                        sys.exit( 0 )
     260
     261                elif pid > 0:
     262                        # Parent - Time/sleep Thread
     263                        #
     264                        # Make sure XML is processed on time and at regular intervals
     265
     266                        debug_msg( 7, self.printTime() + ' - mainthread() - Sleep '+ str( self.config.getInterval() ) +'s: zzzzz..' )
     267                        time.sleep( self.config.getInterval() )
     268                        debug_msg( 7, self.printTime() + ' - mainthread() - Awoken: waiting for XML thread..' )
     269
     270                        r = os.wait()
     271                        ret = r[1]
     272                        if ret != 0:
     273                                debug_msg( 7, self.printTime() + ' - mainthread() - Done waiting: ERROR! xmlthread() exited with status %d' %(ret) )
     274                                if DEBUG_LEVEL>=7: sys.exit( 1 )
     275                        else:
     276
     277                                debug_msg( 7, self.printTime() + ' - mainthread() - Done waiting: xmlthread() finished succesfully' )
    224278
    225279        def run( self ):
    226280                "Main thread"
    227281
    228                 while( 1 ):
    229 
    230                         debug_msg( 7, self.printTime() + ' - mainthread() - xmlthread() started' )
     282                # Daemonized not working yet
     283                if DAEMONIZE:
    231284                        pid = os.fork()
    232285
     286                        # Handle XML grabbing in Child
    233287                        if pid == 0:
    234                                 # Child - XML Thread
    235                                 #
    236                                 # Process XML and exit
    237 
    238                                 debug_msg( 7, self.printTime() + ' - xmlthread()  - Start XML processing..' )
    239                                 self.processXML()
    240                                 debug_msg( 7, self.printTime() + ' - xmlthread()  - Done processing; exiting.' )
    241                                 sys.exit( 0 )
    242 
    243                         elif pid > 0:
    244                                 # Parent - Daemon Thread
    245 
    246                                 debug_msg( 7, self.printTime() + ' - mainthread() - Sleep '+ str(GRAB_INTERVAL) +'s: zzzzz..' )
    247                                 time.sleep( GRAB_INTERVAL )
    248                                 debug_msg( 7, self.printTime() + ' - mainthread() - Awoken: waiting for XML thread..' )
    249 
    250                                 r = os.wait()
    251                                 ret = r[1]
    252                                 if ret != 0:
    253                                         debug_msg( 7, self.printTime() + ' - mainthread() - Done waiting: ERROR! xmlthread() exited with status %d' %(ret) )
    254                                         if DEBUG_LEVEL>=7: sys.exit( 1 )
    255                                 else:
    256 
    257                                         debug_msg( 7, self.printTime() + ' - mainthread() - Done waiting: xmlthread() finished succesfully' )
     288
     289                                while( 1 ):
     290                                        self.grabXML()
     291
     292                        # Do scheduled RRD storing in Parent
     293                        #elif pid > ):
     294
     295                else:
     296                        self.grabXML()
     297                        self.storeMetrics()
     298
     299        def storeMetrics( self ):
     300                "Store metrics retained in memory to disk"
     301
     302                self.myHandler.storeMetrics()
    258303
    259304        def processXML( self ):
    260305                "Process XML"
    261306
    262                 myXMLGatherer = GangliaXMLGatherer( 'localhost', 8651 )
    263 
    264                 myParser = make_parser()   
    265                 myHandler = GangliaXMLHandler()
    266                 myParser.setContentHandler( myHandler )
    267 
    268                 myParser.parse( myXMLGatherer.getFileObject() )
     307                self.myParser.parse( self.myXMLGatherer.getFileObject() )
    269308
    270309class GangliaConfigParser:
    271310
    272         sources = [ ]
     311        sources = { }
    273312
    274313        def __init__( self, config ):
     
    306345                                                        source['interval'] = word
    307346                                                        debug_msg( 9, 'polling interval for %s = %s' %(source['name'], source['interval'] ) )
    308                
    309                 # No interval found, use Ganglia's default     
    310                 if not source.has_key( 'interval' ):
    311 
    312                         source['interval'] = 15
    313                         debug_msg( 9, 'polling interval for %s defaulted to 15' %(source['name']) )
    314 
    315                 self.sources.append( source )
     347       
     348                                        # No interval found, use Ganglia's default     
     349                                        if not source.has_key( 'interval' ):
     350                                                source['interval'] = 15
     351                                                debug_msg( 9, 'polling interval for %s defaulted to 15' %(source['name']) )
     352
     353                                        self.sources.append( source )
    316354
    317355        def getInterval( self, source_name ):
     
    329367        myMetrics = { }
    330368
    331         def __init__( self, cluster ):
     369        def __init__( self, config, cluster ):
    332370                self.cluster = cluster
    333                 self.gmetad_conf = GangliaConfigParser( GMETAD_CONF )
     371                self.config = config
    334372
    335373        def getClusterName( self ):
     
    365403                return update_string
    366404
     405        def storeMetrics( self ):
     406
     407                for hostname, mymetrics in self.myMetrics.items():     
     408
     409                        for metricname, mymetric in mymetrics.items():
     410
     411                                self.rrd.createCheck( hostname, metricname, timeserial )       
     412                                update_okay = self.rrd.update( hostname, metricname, timeserial )
     413
     414                                if not update_okay:
     415
     416                                        del self.myMetrics[ hostname ][ metricname ]
     417                                        debug_msg( 9, 'stored metric %s for %s' %( hostname, metricname ) )
     418                                else:
     419                                        debug_msg( 9, 'metric update failed' )
     420
     421                                sys.exit(1)
     422
    367423        def makeTimeSerial( self ):
    368424                "Generate a time serial. Seconds since epoch"
    369425
    370                 # YYYYMMDDhhmmss: 20050321143411
    371                 #mytime = time.strftime( "%Y%m%d%H%M%S" )
    372 
    373426                # Seconds since epoch
    374427                mytime = int( time.time() )
     
    376429                return mytime
    377430
    378         def makeRrdPath( self, host, metric=None, timeserial=None ):
     431        def makeRrdPath( self, host, metricname=None, timeserial=None ):
    379432                """
    380433                Make a RRD location/path and filename
     
    388441                        rrd_dir = '%s/%s/%s/%s' %( check_dir(ARCHIVE_PATH), self.cluster, host, timeserial )
    389442                if metric:
    390                         rrd_file = '%s/%s.rrd' %( rrd_dir, metric['name'] )
     443                        rrd_file = '%s/%s.rrd' %( rrd_dir, metricname )
    391444                else:
    392445                        rrd_file = None
     
    450503                return serial
    451504
    452         def createCheck( self, host, metric, timeserial ):
     505        def getFirstTime( self, host, metricname ):
     506                "Get the first time of a metric we know of"
     507
     508                first_time = 0
     509
     510                for metric in self.myMetrics[ host ][ metricname ]:
     511
     512                        if not first_time or metric['time'] <= first_time:
     513
     514                                first_time = metric['time']
     515
     516        def createCheck( self, host, metricname, timeserial ):
    453517                "Check if an .rrd allready exists for this metric, create if not"
    454518
    455519                debug_msg( 9, 'rrdcreate: using timeserial %s for %s/%s' %( timeserial, host, metric['name'] ) )
    456520
    457                 rrd_dir, rrd_file = self.makeRrdPath( host, metric, timeserial )
     521                rrd_dir, rrd_file = self.makeRrdPath( host, metricname, timeserial )
    458522
    459523                if not os.path.exists( rrd_dir ):
     
    463527                if not os.path.exists( rrd_file ):
    464528
    465                         interval = self.gmetad_conf.getInterval( self.cluster )
     529                        interval = self.config.getInterval( self.cluster )
    466530                        heartbeat = 8 * int(interval)
    467531
     
    470534
    471535                        param_start1 = '--start'
    472                         param_start2 = str( int( metric['time'] ) - 1 )
     536                        param_start2 = str( int( self.getFirstTime( host, metricname ) ) - 1 )
    473537
    474538                        param_ds = 'DS:sum:GAUGE:%d:U:U' %heartbeat
     
    479543                        debug_msg( 9, 'created rrd %s' %( str(rrd_file) ) )
    480544
    481         def update( self, host, metric, timeserial ):
     545        def update( self, host, metricname, timeserial ):
    482546
    483547                debug_msg( 9, 'rrdupdate: using timeserial %s for %s/%s' %( timeserial, host, metric['name'] ) )
    484548
    485                 rrd_dir, rrd_file = self.makeRrdPath( host, metric, timeserial )
    486 
    487                 timestamp = metric['time']
    488                 val = metric['val']
    489 
    490                 update_string = '%s:%s' %(timestamp, val)
     549                rrd_dir, rrd_file = self.makeRrdPath( host, metricname, timeserial )
     550
     551                #timestamp = metric['time']
     552                #val = metric['val']
     553
     554                #update_string = '%s:%s' %(timestamp, val)
     555                update_string = self.makeUpdateString( host, metricname )
    491556
    492557                try:
     
    500565                        debug_msg( 0, str(detail) )
    501566
    502                         sys.exit( 1 )
     567                        return 1
    503568               
    504569                debug_msg( 9, 'updated rrd %s with %s' %( str(rrd_file), update_string ) )
Note: See TracChangeset for help on using the changeset viewer.