source: branches/1.1/jobarchived/jobarchived.py @ 929

Last change on this file since 929 was 929, checked in by ramonb, 11 years ago
  • revert
  • Property svn:executable set to *
  • Property svn:keywords set to Id
File size: 62.2 KB
RevLine 
[3]1#!/usr/bin/env python
[225]2#
3# This file is part of Jobmonarch
4#
[771]5# Copyright (C) 2006-2013  Ramon Bastiaans
[225]6#
7# Jobmonarch is free software; you can redistribute it and/or modify
8# it under the terms of the GNU General Public License as published by
9# the Free Software Foundation; either version 2 of the License, or
10# (at your option) any later version.
11#
12# Jobmonarch is distributed in the hope that it will be useful,
13# but WITHOUT ANY WARRANTY; without even the implied warranty of
14# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15# GNU General Public License for more details.
16#
17# You should have received a copy of the GNU General Public License
18# along with this program; if not, write to the Free Software
19# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
20#
[230]21# SVN $Id: jobarchived.py 929 2013-07-31 14:27:18Z ramonb $
22#
[3]23
[466]24import getopt, syslog, ConfigParser, sys
[3]25
[913]26VERSION='__VERSION__'
[284]27
[466]28def usage( ver ):
[284]29
[770]30    print 'jobarchived %s' %VERSION
[284]31
[770]32    if ver:
33        return 0
[284]34
[770]35    print
36    print 'Purpose:'
37    print '  The Job Archive Daemon (jobarchived) stores batch job information in a SQL database'
38    print '  and node statistics in a RRD archive'
39    print
40    print 'Usage:    jobarchived [OPTIONS]'
41    print
42    print '  -c, --config=FILE    The configuration file to use (default: /etc/jobarchived.conf)'
43    print '  -p, --pidfile=FILE    Use pid file to store the process id'
44    print '  -h, --help        Print help and exit'
45    print '  -v, --version        Print version and exit'
46    print
[435]47
[214]48def processArgs( args ):
[6]49
[858]50    SHORT_L   = 'p:hvc:'
[773]51    LONG_L    = [ 'help', 'config=', 'pidfile=', 'version' ]
[169]52
[773]53    config_filename = '/etc/jobarchived.conf'
[169]54
[770]55    global PIDFILE
[435]56
[858]57    PIDFILE   = None
[435]58
[773]59    try:
[169]60
[773]61        opts, args = getopt.getopt( args, SHORT_L, LONG_L )
[9]62
[773]63    except getopt.error, detail:
[60]64
[773]65        print detail
66        sys.exit(1)
[9]67
[773]68    for opt, value in opts:
[60]69
[773]70        if opt in [ '--config', '-c' ]:
[13]71
[773]72            config_filename = value
[198]73
[770]74        if opt in [ '--pidfile', '-p' ]:
[435]75
[770]76            PIDFILE         = value
[435]77
[770]78        if opt in [ '--help', '-h' ]:
[435]79
[770]80            usage( False )
81            sys.exit( 0 )
[435]82
[770]83        if opt in [ '--version', '-v' ]:
[60]84
[770]85            usage( True )
86            sys.exit( 0 )
[22]87
[770]88    try:
89        return loadConfig( config_filename )
[13]90
[770]91    except ConfigParser.NoOptionError, detail:
[214]92
[770]93        print detail
94        sys.exit( 1 )
[214]95
96def loadConfig( filename ):
97
[770]98    def getlist( cfg_string ):
[214]99
[770]100        my_list = [ ]
[214]101
[770]102        for item_txt in cfg_string.split( ',' ):
[214]103
[770]104            sep_char = None
[214]105
[770]106            item_txt = item_txt.strip()
[214]107
[770]108            for s_char in [ "'", '"' ]:
[214]109
[770]110                if item_txt.find( s_char ) != -1:
[214]111
[770]112                    if item_txt.count( s_char ) != 2:
[214]113
[770]114                        print 'Missing quote: %s' %item_txt
115                        sys.exit( 1 )
[214]116
[770]117                    else:
[214]118
[770]119                        sep_char = s_char
120                        break
[214]121
[770]122            if sep_char:
[214]123
[770]124                item_txt = item_txt.split( sep_char )[1]
[214]125
[770]126            my_list.append( item_txt )
[214]127
[770]128        return my_list
[214]129
[770]130    cfg = ConfigParser.ConfigParser()
[214]131
[770]132    cfg.read( filename )
[214]133
[770]134    global DEBUG_LEVEL, USE_SYSLOG, SYSLOG_LEVEL, SYSLOG_FACILITY, GMETAD_CONF, ARCHIVE_XMLSOURCE
135    global ARCHIVE_DATASOURCES, ARCHIVE_PATH, ARCHIVE_HOURS_PER_RRD, ARCHIVE_EXCLUDE_METRICS
[774]136    global JOB_SQL_DBASE, DAEMONIZE, RRDTOOL, JOB_TIMEOUT, MODRRDTOOL, JOB_SQL_PASSWORD, JOB_SQL_USER
[214]137
[858]138    ARCHIVE_PATH           = cfg.get( 'DEFAULT', 'ARCHIVE_PATH' )
[214]139
[858]140    ARCHIVE_HOURS_PER_RRD  = cfg.getint( 'DEFAULT', 'ARCHIVE_HOURS_PER_RRD' )
[214]141
[858]142    DEBUG_LEVEL            = cfg.getint( 'DEFAULT', 'DEBUG_LEVEL' )
[214]143
[858]144    USE_SYSLOG             = cfg.getboolean( 'DEFAULT', 'USE_SYSLOG' )
[214]145
[858]146    SYSLOG_LEVEL           = cfg.getint( 'DEFAULT', 'SYSLOG_LEVEL' )
[214]147
[858]148    MODRRDTOOL             = False
[375]149
[770]150    try:
151        global rrdtool
152        import rrdtool
[214]153
[770]154        MODRRDTOOL        = True
[375]155
[770]156    except ImportError:
[375]157
[770]158        MODRRDTOOL        = False
[375]159
[770]160        print "ERROR: py-rrdtool import FAILED: failing back to DEPRECATED use of rrdtool binary. This will slow down jobarchived significantly!"
[375]161
[770]162        RRDTOOL            = cfg.get( 'DEFAULT', 'RRDTOOL' )
[455]163
[770]164    try:
[375]165
[770]166        SYSLOG_FACILITY    = eval( 'syslog.LOG_' + cfg.get( 'DEFAULT', 'SYSLOG_FACILITY' ) )
[214]167
[770]168    except AttributeError, detail:
[214]169
[770]170        print 'Unknown syslog facility'
171        sys.exit( 1 )
[214]172
[858]173    GMETAD_CONF             = cfg.get( 'DEFAULT', 'GMETAD_CONF' )
[214]174
[858]175    ARCHIVE_XMLSOURCE       = cfg.get( 'DEFAULT', 'ARCHIVE_XMLSOURCE' )
[214]176
[858]177    ARCHIVE_DATASOURCES     = getlist( cfg.get( 'DEFAULT', 'ARCHIVE_DATASOURCES' ) )
[214]178
[858]179    ARCHIVE_EXCLUDE_METRICS = getlist( cfg.get( 'DEFAULT', 'ARCHIVE_EXCLUDE_METRICS' ) )
[214]180
[858]181    JOB_SQL_DBASE           = cfg.get( 'DEFAULT', 'JOB_SQL_DBASE' )
182    JOB_SQL_USER            = cfg.get( 'DEFAULT', 'JOB_SQL_USER' )
[774]183    JOB_SQL_PASSWORD        = cfg.get( 'DEFAULT', 'JOB_SQL_PASSWORD' )
[214]184
[858]185    JOB_TIMEOUT             = cfg.getint( 'DEFAULT', 'JOB_TIMEOUT' )
[295]186
[858]187    DAEMONIZE               = cfg.getboolean( 'DEFAULT', 'DAEMONIZE' )
[214]188
[770]189    return True
[214]190
[17]191# What XML data types not to store
[13]192#
[17]193UNSUPPORTED_ARCHIVE_TYPES = [ 'string' ]
[9]194
[47]195# Maximum time (in seconds) a parsethread may run
196#
197PARSE_TIMEOUT = 60
198
199# Maximum time (in seconds) a storethread may run
200#
201STORE_TIMEOUT = 360
202
[8]203"""
[224]204The Job Archiving Daemon
[8]205"""
206
[214]207from types import *
208
209import xml.sax, xml.sax.handler, socket, string, os, os.path, time, thread, threading, random, re
210
[486]211try:
[773]212    import psycopg2
[486]213
214except ImportError, details:
215
[773]216    print "FATAL ERROR: psycopg2 python module not found"
[770]217    sys.exit( 1 )
[486]218
[379]219class InitVars:
220        Vars = {}
221       
222        def __init__(self, **key_arg):
223                for (key, value) in key_arg.items():
224                        if value:
225                                self.Vars[key] = value
226                        else:   
227                                self.Vars[key] = None
228                               
229        def __call__(self, *key):
230                key = "%s" % key
231                return self.Vars[key]
232               
233        def __getitem__(self, key):
234                return self.Vars[key]
235               
236        def __repr__(self):
237                return repr(self.Vars)
238               
239        def keys(self):
240                barf =  map(None, self.Vars.keys())
241                return barf
242               
243        def values(self):
244                barf =  map(None, self.Vars.values())
245                return barf
246               
247        def has_key(self, key):
248                if self.Vars.has_key(key):
249                        return 1
250                else:   
251                        return 0
252                       
253class DBError(Exception):
254        def __init__(self, msg=''):
255                self.msg = msg
256                Exception.__init__(self, msg)
257        def __repr__(self):
258                return self.msg
259        __str__ = __repr__
260
261#
262# Class to connect to a database
263# and return the queury in a list or dictionairy.
264#
265class DB:
[773]266    def __init__(self, db_vars):
[379]267
[773]268        self.dict = db_vars
[379]269
[773]270        if self.dict.has_key('User'):
271            self.user = self.dict['User']
272        else:
273            self.user = 'postgres'
[379]274
[773]275        if self.dict.has_key('Host'):
276            self.host = self.dict['Host']
277        else:
278            self.host = 'localhost'
[379]279
[773]280        if self.dict.has_key('Password'):
281            self.passwd = self.dict['Password']
282        else:
283            self.passwd = ''
[379]284
[773]285        if self.dict.has_key('DataBaseName'):
286            self.db = self.dict['DataBaseName']
287        else:
288            self.db = 'jobarchive'
[379]289
[773]290        # connect_string = 'host:port:database:user:password:
291        dsn = "host='%s' dbname='%s' user='%s' password='%s'" %(self.host, self.db, self.user, self.passwd)
[379]292
[773]293        try:
294            self.SQL = psycopg2.connect(dsn)
295        except psycopg2.Error, details:
296            str = "%s" %details
297            raise DBError(str)
[379]298
[773]299    def __repr__(self):
300        return repr(self.result)
[379]301
[773]302    def __nonzero__(self):
303        return not(self.result == None)
[379]304
[773]305    def __len__(self):
306        return len(self.result)
[379]307
[773]308    def __getitem__(self,i):
309        return self.result[i]
[379]310
[773]311    def __getslice__(self,i,j):
312        return self.result[i:j]
[379]313
[773]314    def Get(self, q_str):
315        c = self.SQL.cursor()
316        try:
317            c.execute(q_str)
318            result = c.fetchall()
319        except psycopg2.Error, details:
320            c.close()
321            str = "%s" %details
322            raise DBError(str)
[379]323
[773]324        c.close()
325        return result
[379]326
[773]327    def Set(self, q_str):
328        c = self.SQL.cursor()
329        try:
330            c.execute(q_str)
[379]331
[773]332        except psycopg2.Error, details:
333            c.close()
334            str = "%s" %details
335            raise DBError(str)
[379]336
[773]337        c.close()
[774]338        return True
[379]339
[773]340    def Commit(self):
[379]341
[792]342        return self.SQL.commit()
343
344    def Rollback( self ):
345
346        return self.SQL.rollback()
347
[84]348class DataSQLStore:
349
[770]350    db_vars = None
351    dbc = None
[84]352
[770]353    def __init__( self, hostname, database ):
[84]354
[774]355        global JOB_SQL_USER, JOB_SQL_PASSWORD
356
[770]357        self.db_vars = InitVars(DataBaseName=database,
[774]358                User=JOB_SQL_USER,
[770]359                Host=hostname,
[774]360                Password=JOB_SQL_PASSWORD,
[770]361                Dictionary='true')
[84]362
[770]363        try:
364            self.dbc     = DB(self.db_vars)
365        except DBError, details:
366            debug_msg( 0, 'FATAL ERROR: Unable to connect to database!: ' +str(details) )
367            sys.exit(1)
[84]368
[770]369    def setDatabase(self, statement):
[792]370
[770]371        ret = self.doDatabase('set', statement)
372        return ret
373       
374    def getDatabase(self, statement):
[792]375
[770]376        ret = self.doDatabase('get', statement)
377        return ret
[84]378
[792]379    def doCommit( self ):
380
381        return self.dbc.Commit()
382
383    def doRollback( self ):
384
385        return self.dbc.Rollback()
386
[770]387    def doDatabase(self, type, statement):
[84]388
[770]389        debug_msg( 10, 'doDatabase(): %s: %s' %(type, statement) )
390        try:
391            if type == 'set':
392                result = self.dbc.Set( statement )
393            elif type == 'get':
394                result = self.dbc.Get( statement )
395               
396        except DBError, detail:
397            operation = statement.split(' ')[0]
[792]398            debug_msg( 0, 'ERROR: ' +operation+ ' on database failed while doing ['+statement+'] full msg: '+str(detail) )
399            return False
[84]400
[770]401        debug_msg( 10, 'doDatabase(): result: %s' %(result) )
402        return result
[84]403
[770]404    def getJobNodeId( self, job_id, node_id ):
[191]405
[770]406        id = self.getDatabase( "SELECT job_id,node_id FROM job_nodes WHERE job_id = '%s' AND node_id = '%s'" %(job_id, node_id) )
[792]407        if not id:
408            return False
409
[770]410        if len( id ) > 0:
[191]411
[770]412            if len( id[0] ) > 0 and id[0] != '':
413           
[792]414                return True
[191]415
[792]416        return False
[191]417
[770]418    def getNodeId( self, hostname ):
[84]419
[770]420        id = self.getDatabase( "SELECT node_id FROM nodes WHERE node_hostname = '%s'" %hostname )
[84]421
[770]422        if len( id ) > 0:
[84]423
[770]424            id = id[0][0]
[89]425
[770]426            return id
427        else:
428            return None
[84]429
[770]430    def getNodeIds( self, hostnames ):
[84]431
[770]432        ids = [ ]
[84]433
[770]434        for node in hostnames:
[84]435
[770]436            id = self.getNodeId( node )
[84]437
[770]438            if id:
439                ids.append( id )
[84]440
[770]441        return ids
[84]442
[770]443    def getJobId( self, jobid ):
[84]444
[770]445        id = self.getDatabase( "SELECT job_id FROM jobs WHERE job_id = '%s'" %jobid )
[84]446
[770]447        if id:
448            id = id[0][0]
[84]449
[770]450            return id
451        else:
452            return None
[84]453
[770]454    def addJob( self, job_id, jobattrs ):
[84]455
[770]456        if not self.getJobId( job_id ):
[84]457
[792]458            return self.mutateJob( 'insert', job_id, jobattrs ) 
[770]459        else:
[792]460            return self.mutateJob( 'update', job_id, jobattrs )
[84]461
[770]462    def mutateJob( self, action, job_id, jobattrs ):
[84]463
[782]464        job_values     = [ 'name', 'queue', 'owner', 'requested_time', 'requested_memory', 'ppn', 'status', 'start_timestamp', 'stop_timestamp' ]
[84]465
[782]466        insert_col_str = 'job_id'
467        insert_val_str = "'%s'" %job_id
468        update_str     = None
[84]469
[770]470        debug_msg( 10, 'mutateJob(): %s %s' %(action,job_id))
[84]471
[770]472        ids = [ ]
[99]473
[770]474        for valname, value in jobattrs.items():
[84]475
[770]476            if valname in job_values and value != '':
[84]477
[770]478                column_name = 'job_' + valname
[84]479
[770]480                if action == 'insert':
[84]481
[770]482                    if not insert_col_str:
483                        insert_col_str = column_name
484                    else:
485                        insert_col_str = insert_col_str + ',' + column_name
[84]486
[770]487                    if not insert_val_str:
488                        insert_val_str = value
489                    else:
490                        insert_val_str = insert_val_str + ",'%s'" %value
[84]491
[770]492                elif action == 'update':
493                   
494                    if not update_str:
495                        update_str = "%s='%s'" %(column_name, value)
496                    else:
497                        update_str = update_str + ",%s='%s'" %(column_name, value)
[84]498
[770]499            elif valname == 'nodes' and value:
[84]500
[770]501                node_valid = 1
[190]502
[770]503                if len(value) == 1:
504               
505                    if jobattrs['status'] == 'Q':
[190]506
[770]507                        node_valid = 0
[190]508
[770]509                    else:
[190]510
[770]511                        node_valid = 0
[190]512
[770]513                        for node_char in str(value[0]):
[190]514
[770]515                            if string.find( string.digits, node_char ) != -1 and not node_valid:
[190]516
[770]517                                node_valid = 1
[84]518
[770]519                if node_valid:
[191]520
[770]521                    ids = self.addNodes( value, jobattrs['domain'] )
[191]522
[770]523        if action == 'insert':
[84]524
[792]525            db_ret = self.setDatabase( "INSERT INTO jobs ( %s ) VALUES ( %s )" %( insert_col_str, insert_val_str ) )
[86]526
[770]527        elif action == 'update':
[84]528
[792]529            db_ret = self.setDatabase( "UPDATE jobs SET %s WHERE job_id='%s'" %(update_str, job_id) )
[84]530
[770]531        if len( ids ) > 0:
532            self.addJobNodes( job_id, ids )
[190]533
[792]534        return db_ret
535
[770]536    def addNodes( self, hostnames, domain ):
[84]537
[770]538        ids = [ ]
[98]539
[770]540        for node in hostnames:
[84]541
[770]542            node    = '%s.%s' %( node, domain )
543            id    = self.getNodeId( node )
544   
545            if not id:
546                self.setDatabase( "INSERT INTO nodes ( node_hostname ) VALUES ( '%s' )" %node )
547                id = self.getNodeId( node )
[84]548
[770]549            ids.append( id )
[98]550
[770]551        return ids
[98]552
[770]553    def addJobNodes( self, jobid, nodes ):
[86]554
[770]555        for node in nodes:
[86]556
[770]557            if not self.getJobNodeId( jobid, node ):
[191]558
[770]559                self.addJobNode( jobid, node )
[191]560
[770]561    def addJobNode( self, jobid, nodeid ):
[84]562
[774]563        self.setDatabase( "INSERT INTO job_nodes (job_id,node_id) VALUES ( '%s',%s )" %(jobid, nodeid) )
[84]564
[770]565    def storeJobInfo( self, jobid, jobattrs ):
[84]566
[792]567        return self.addJob( jobid, jobattrs )
[84]568
[857]569    def checkTimedoutJobs( self ):
[295]570
[857]571        debug_msg( 1, 'Housekeeping: checking database for timed out jobs..' )
572
[770]573        # Locate all jobs in the database that are not set to finished
574        #
575        q = "SELECT * from jobs WHERE job_status != 'F'"
[295]576
[770]577        r = self.getDatabase( q )
[295]578
[770]579        if len( r ) == 0:
[295]580
[770]581            return None
[295]582
[857]583        timeoutjobs  = [ ]
[295]584
[857]585        jobtimeout_sec = JOB_TIMEOUT * (60 * 60)
586        cur_time       = time.time()
[295]587
[770]588        for row in r:
[295]589
[857]590            job_id              = row[0]
591            job_requested_time  = row[4]
592            job_status          = row[7]
593            job_start_timestamp = row[8]
[295]594
[770]595            # If it was set to queued and we didn't see it started
596            # there's not point in keeping it around
597            #
[857]598            if job_status == 'R' and job_start_timestamp:
[295]599
[770]600                start_timestamp = int( job_start_timestamp )
[295]601
[770]602                # If it was set to running longer than JOB_TIMEOUT
603                # close the job: it probably finished while we were not running
604                #
605                if ( cur_time - start_timestamp ) > jobtimeout_sec:
[295]606
[770]607                    if job_requested_time:
[295]608
[770]609                        rtime_epoch    = reqtime2epoch( job_requested_time )
610                    else:
611                        rtime_epoch    = None
612                   
613                    timeoutjobs.append( (job_id, job_start_timestamp, rtime_epoch) )
[295]614
[857]615        debug_msg( 1, 'Housekeeping: Found ' + str( len( timeoutjobs ) ) + ' timed out jobs in database: closing entries' )
[295]616
[857]617        ret_jobids_clean = [ ]
[295]618
[770]619        # Close these jobs in the database
620        # update the stop_timestamp to: start_timestamp + requested wallclock
621        # and set state: finished
622        #
623        for j in timeoutjobs:
[295]624
[770]625            ( i, s, r )        = j
[295]626
[770]627            if r:
628                new_end_timestamp    = int( s ) + r
[295]629
[857]630                q = "UPDATE jobs SET job_status='F',job_stop_timestamp = '" + str( new_end_timestamp ) + "' WHERE job_id = '" + str(i) + "'"
631                self.setDatabase( q )
632            else:
633
634                # Requested walltime unknown: cannot guess end time: sorry delete them
635                q = "DELETE FROM jobs WHERE job_id = '" + str( i ) + "'"
636                self.setDatabase( q )
637
638            ret_jobids_clean.append( i )
639
640        debug_msg( 1, 'Housekeeping: done.' )
641
642        return ret_jobids_clean
643
644    def checkStaleJobs( self ):
645
646        debug_msg( 1, 'Housekeeping: checking database for stale jobs..' )
647
648        # Locate all jobs in the database that are not set to finished
649        #
650        q = "SELECT * from jobs WHERE job_status != 'F'"
651
652        r = self.getDatabase( q )
653
654        if len( r ) == 0:
655
656            return None
657
658        cleanjobs      = [ ]
659
660        cur_time       = time.time()
661
662        for row in r:
663
664            job_id              = row[0]
665            job_requested_time  = row[4]
666            job_status          = row[7]
667            job_start_timestamp = row[8]
668
669            # If it was set to queued and we didn't see it started
670            # there's not point in keeping it around
671            #
672            if job_status == 'Q' or not job_start_timestamp:
673
674                cleanjobs.append( job_id )
675
676        debug_msg( 1, 'Housekeeping: Found ' + str( len( cleanjobs ) ) + ' stale jobs in database: deleting entries' )
677
678        # Purge these from database
679        #
680        for j in cleanjobs:
681
682            q = "DELETE FROM jobs WHERE job_id = '" + str( j ) + "'"
[770]683            self.setDatabase( q )
[295]684
[857]685        debug_msg( 1, 'Housekeeping: done.' )
686
687        return cleanjobs
688
[37]689class RRDMutator:
[770]690    """A class for performing RRD mutations"""
[37]691
[770]692    binary = None
[37]693
[770]694    def __init__( self, binary=None ):
695        """Set alternate binary if supplied"""
[37]696
[770]697        if binary:
698            self.binary = binary
[37]699
[770]700    def create( self, filename, args ):
701        """Create a new rrd with args"""
[63]702
[770]703        global MODRRDTOOL
[37]704
[770]705        if MODRRDTOOL:
706            return self.perform( 'create', filename, args )
707        else:
708            return self.perform( 'create', '"' + filename + '"', args )
[375]709
[770]710    def update( self, filename, args ):
711        """Update a rrd with args"""
[63]712
[770]713        global MODRRDTOOL
[37]714
[770]715        if MODRRDTOOL:
716            return self.perform( 'update', filename, args )
717        else:
718            return self.perform( 'update', '"' + filename + '"', args )
[375]719
[770]720    def grabLastUpdate( self, filename ):
721        """Determine the last update time of filename rrd"""
[42]722
[770]723        global MODRRDTOOL
[375]724
[770]725        last_update = 0
[42]726
[770]727        # Use the py-rrdtool module if it's available on this system
728        #
729        if MODRRDTOOL:
[53]730
[770]731            debug_msg( 8, 'rrdtool.info( ' + filename + ' )' )
[42]732
[770]733            rrd_header     = { }
[292]734
[770]735            try:
736                rrd_header    = rrdtool.info( filename )
737            except rrdtool.error, msg:
738                debug_msg( 8, str( msg ) )
[375]739
[770]740            if rrd_header.has_key( 'last_update' ):
741                return last_update
742            else:
743                return 0
[375]744
[770]745        # For backwards compatiblity: use the rrdtool binary if py-rrdtool is unavailable
746        # DEPRECATED (slow!)
747        #
748        else:
749            debug_msg( 8, self.binary + ' info ' + filename )
[42]750
[770]751            my_pipe        = os.popen( self.binary + ' info "' + filename + '"' )
[375]752
[770]753            for line in my_pipe.readlines():
[375]754
[770]755                if line.find( 'last_update') != -1:
[375]756
[770]757                    last_update = line.split( ' = ' )[1]
[375]758
[770]759            if my_pipe:
[375]760
[770]761                my_pipe.close()
[375]762
[770]763            if last_update:
764                return last_update
765            else:
766                return 0
[375]767
768
[770]769    def perform( self, action, filename, args ):
770        """Perform action on rrd filename with args"""
[37]771
[770]772        global MODRRDTOOL
[375]773
[770]774        arg_string = None
[37]775
[770]776        if type( args ) is not ListType:
777            debug_msg( 8, 'Arguments needs to be of type List' )
778            return 1
[40]779
[770]780        for arg in args:
[37]781
[770]782            if not arg_string:
[37]783
[770]784                arg_string = arg
785            else:
786                arg_string = arg_string + ' ' + arg
[37]787
[770]788        if MODRRDTOOL:
[37]789
[770]790            debug_msg( 8, 'rrdtool.' + action + "( " + filename + ' ' + arg_string + ")" )
[292]791
[770]792            try:
793                debug_msg( 8, "filename '" + str(filename) + "' type "+ str(type(filename)) + " args " + str( args ) )
[37]794
[770]795                if action == 'create':
[146]796
[770]797                    rrdtool.create( str( filename ), *args )
[37]798
[770]799                elif action == 'update':
[37]800
[770]801                    rrdtool.update( str( filename ), *args )
[365]802
[770]803            except rrdtool.error, msg:
[365]804
[770]805                error_msg = str( msg )
806                debug_msg( 8, error_msg )
807                return 1
[375]808
[770]809        else:
[375]810
[770]811            debug_msg( 8, self.binary + ' ' + action + ' ' + filename + ' ' + arg_string  )
[375]812
[770]813            cmd     = os.popen( self.binary + ' ' + action + ' ' + filename + ' ' + arg_string )
814            lines   = cmd.readlines()
[375]815
[770]816            cmd.close()
[375]817
[770]818            for line in lines:
[375]819
[770]820                if line.find( 'ERROR' ) != -1:
[375]821
[770]822                    error_msg = string.join( line.split( ' ' )[1:] )
823                    debug_msg( 8, error_msg )
824                    return 1
[375]825
[770]826        return 0
[37]827
[78]828class XMLProcessor:
[770]829    """Skeleton class for XML processor's"""
[78]830
[770]831    def run( self ):
832        """Do main processing of XML here"""
[78]833
[770]834        pass
[78]835
[782]836class JobXMLProcessor( XMLProcessor ):
[770]837    """Main class for processing XML and acting with it"""
[78]838
[770]839    def __init__( self, XMLSource, DataStore ):
840        """Setup initial XML connection and handlers"""
[78]841
[782]842        self.myXMLSource  = XMLSource
843        self.myXMLHandler = JobXMLHandler( DataStore )
844        self.myXMLError   = XMLErrorHandler()
[78]845
[782]846        self.config       = GangliaConfigParser( GMETAD_CONF )
[287]847
[782]848        self.kill_thread  = False
849
850    def killThread( self ):
851
852        self.kill_thread  = True
853
[770]854    def run( self ):
855        """Main XML processing"""
[78]856
[782]857        debug_msg( 1, 'job_xml_thread(): started.' )
[87]858
[770]859        while( 1 ):
[78]860
[782]861            debug_msg( 1, 'job_xml_thread(): Retrieving XML data..' )
[176]862
[770]863            my_data    = self.myXMLSource.getData()
[287]864
[782]865            debug_msg( 1, 'job_xml_thread(): Done retrieving: data size %d' %len(my_data) )
[469]866
[770]867            if my_data:
[782]868                debug_msg( 1, 'job_xml_thread(): Parsing XML..' )
[469]869
[770]870                xml.sax.parseString( my_data, self.myXMLHandler, self.myXMLError )
[469]871
[926]872                if self.myXMLError.isFatal():
873
874                    sys.exit( 1 )
875
[782]876                debug_msg( 1, 'job_xml_thread(): Done parsing.' )
[774]877            else:
[782]878                debug_msg( 1, 'job_xml_thread(): Got no data.' )
[774]879
[782]880            if self.kill_thread:
881
882                debug_msg( 1, 'job_xml_thread(): killed.' )
883                return None
[770]884               
[782]885            debug_msg( 1, 'job_xml_thread(): Sleeping.. (%ss)' %(str( self.config.getLowestInterval() ) ) )
[770]886            time.sleep( self.config.getLowestInterval() )
[78]887
[782]888class JobXMLHandler( xml.sax.handler.ContentHandler ):
889    """Parse Job's jobinfo XML from our plugin"""
[63]890
[770]891    def __init__( self, datastore ):
[84]892
[782]893        self.ds              = datastore
894        self.jobs_processed  = [ ]
895        self.jobs_to_store   = [ ]
896        self.jobAttrs        = { }
897        self.jobAttrsSaved   = { }
898
[857]899        self.iteration       = 0
900
901        self.ds.checkTimedoutJobs()
902        self.ds.checkStaleJobs()
903
[774]904        debug_msg( 1, "XML: Handler created" )
[84]905
[770]906    def startDocument( self ):
[183]907
[782]908        self.jobs_processed = [ ]
909        self.heartbeat      = 0
910        self.elementct      = 0
[857]911        self.iteration      = self.iteration + 1
[782]912
[857]913        if self.iteration > 20:
[183]914
[857]915            timedout_jobs = self.ds.checkTimedoutJobs()
916            self.iteration = 0
917
[925]918            if timedout_jobs != None:
[857]919
[925]920                for j in timedout_jobs:
[857]921
[925]922                    del self.jobAttrs[ j ]
923                    del self.jobAttrsSaved[ j ]
924
[857]925        debug_msg( 1, "XML: Start document: iteration %s" %str(self.iteration) )
926
[770]927    def startElement( self, name, attrs ):
928        """
929        This XML will be all gmetric XML
930        so there will be no specific start/end element
931        just one XML statement with all info
932        """
[782]933
[770]934        jobinfo = { }
[63]935
[770]936        self.elementct    += 1
[365]937
[770]938        if name == 'CLUSTER':
[63]939
[770]940            self.clustername = str( attrs.get( 'NAME', "" ) )
[199]941
[770]942        elif name == 'METRIC' and self.clustername in ARCHIVE_DATASOURCES:
[199]943
[770]944            metricname = str( attrs.get( 'NAME', "" ) )
[63]945
[772]946            if metricname == 'zplugin_monarch_heartbeat':
[782]947
[770]948                self.heartbeat = str( attrs.get( 'VAL', "" ) )
[63]949
[772]950            elif metricname.find( 'zplugin_monarch_job' ) != -1:
[63]951
[782]952                job_id  = metricname.split( 'zplugin_monarch_job_' )[1].split( '_' )[1]
953                val     = str( attrs.get( 'VAL', "" ) )
[63]954
[770]955                valinfo = val.split( ' ' )
[63]956
[770]957                for myval in valinfo:
[63]958
[770]959                    if len( myval.split( '=' ) ) > 1:
[63]960
[782]961                        valname = myval.split( '=' )[0]
962                        value   = myval.split( '=' )[1]
[70]963
[770]964                        if valname == 'nodes':
[782]965
[770]966                            value = value.split( ';' )
[72]967
[770]968                        jobinfo[ valname ] = value
[84]969
[782]970                self.jobAttrs[ job_id ] = jobinfo
[84]971
[782]972                self.jobs_processed.append( job_id )
[770]973                   
974    def endDocument( self ):
975        """When all metrics have gone, check if any jobs have finished"""
[72]976
[783]977        jobs_finished = [ ]
978
[782]979        debug_msg( 1, "XML: Processed "+str(self.elementct)+ " elements - found "+str(len(self.jobs_processed))+" jobs" )
[365]980
[782]981        if self.heartbeat == 0:
982            return None
[74]983
[782]984        for jobid, jobinfo in self.jobAttrs.items():
[102]985
[782]986            if jobinfo['reported'] != self.heartbeat:
[74]987
[782]988                if (jobinfo['status'] != 'R'):
989                    debug_msg( 1, 'job %s report time %s does not match current heartbeat %s : ignoring job' %(jobid, jobinfo['reported'], self.heartbeat ) )
990                    del self.jobAttrs[ jobid ]
[96]991
[782]992                    if jobid in self.jobs_to_store:
993                        del self.jobs_to_store[ jobid ]
994
995                    continue
996
997                elif jobid not in self.jobs_processed:
[783]998
[782]999                    # Was running previous heartbeat but not anymore: must be finished
[770]1000                    self.jobAttrs[ jobid ]['status'] = 'F'
1001                    self.jobAttrs[ jobid ]['stop_timestamp'] = str( self.heartbeat )
[782]1002                    debug_msg( 1, 'job %s appears to have finished' %jobid )
[96]1003
[783]1004                    jobs_finished.append( jobid )
1005
[770]1006                    if not jobid in self.jobs_to_store:
1007                        self.jobs_to_store.append( jobid )
[74]1008
[782]1009                    continue
[87]1010
[782]1011            elif self.jobAttrsSaved.has_key( jobid ):
[84]1012
[783]1013                # This should pretty much never happen, but hey let's be careful
1014                # Perhaps if someone altered their job while in queue with qalter
1015
[782]1016                if self.jobinfoChanged( jobid, jobinfo ):
[184]1017
[782]1018                    self.jobAttrs[ jobid ]['stop_timestamp'] = ''
1019                    self.jobAttrs[ jobid ]                   = self.setJobAttrs( self.jobAttrs[ jobid ], jobinfo )
[184]1020
[782]1021                    if not jobid in self.jobs_to_store:
[87]1022
[782]1023                        self.jobs_to_store.append( jobid )
[84]1024
[782]1025                    debug_msg( 10, 'jobinfo for job %s has changed' %jobid )
1026            else:
[929]1027                debug_msg( 1, 'new job %s' %jobid )
[782]1028
1029                if not jobid in self.jobs_to_store:
1030
1031                    self.jobs_to_store.append( jobid )
1032
1033        debug_msg( 1, 'job_xml_thread(): Found %s updated jobs.' %len(self.jobs_to_store) )
1034
[792]1035        failed_store = [ ]
1036        succes_store = [ ]
1037
[782]1038        if len( self.jobs_to_store ) > 0:
1039
1040            debug_msg( 1, 'job_xml_thread(): Storing jobs to database..' )
1041
[792]1042            for n in range( 0, len(self.jobs_to_store ) ):
[782]1043
[792]1044                if len( self.jobs_to_store ) == 0:
1045                    break
1046
[782]1047                jobid = self.jobs_to_store.pop( 0 )
1048
[792]1049                db_ok = self.ds.storeJobInfo( jobid, self.jobAttrs[ jobid ] )
[782]1050
[792]1051                if not db_ok:
1052
1053                    self.ds.doRollback()
1054                    failed_store.append( jobid )
1055                    continue
1056
1057                self.ds.doCommit()
1058                succes_store.append( jobid )
1059
[783]1060                if not jobid in jobs_finished:
[782]1061
[783]1062                    self.jobAttrsSaved[ jobid ] = self.jobAttrs[ jobid ]
1063
1064                elif self.jobAttrsSaved.has_key( jobid ):
1065
1066                    del self.jobAttrsSaved[ jobid ]
1067
[782]1068                if self.jobAttrs[ jobid ]['status'] == 'F':
1069
1070                    del self.jobAttrs[ jobid ]
1071
[792]1072            result_str = 'succesfully stored: %s jobs' %str(len(succes_store))
[782]1073
[792]1074            if len( failed_store ) > 0:
1075                result_str = result_str + ' - failed to store: %s jobs - deferred to next interval' %str(len(failed_store))
1076
1077            debug_msg( 1, 'job_xml_thread(): Done storing. %s' %result_str )
1078
[782]1079        else:
1080            debug_msg( 1, 'job_xml_thread(): No jobs to store.' )
1081
1082        self.jobs_processed = [ ]
1083
[783]1084        # TODO: once in while check database AND self.jobAttrsSaved for stale jobs
1085
[770]1086    def setJobAttrs( self, old, new ):
1087        """
1088        Set new job attributes in old, but not lose existing fields
1089        if old attributes doesn't have those
1090        """
[82]1091
[770]1092        for valname, value in new.items():
1093            old[ valname ] = value
[82]1094
[770]1095        return old
1096       
[82]1097
[782]1098    def jobinfoChanged( self, jobid, jobinfo ):
[770]1099        """
1100        Check if jobinfo has changed from jobattrs[jobid]
1101        if it's report time is bigger than previous one
1102        and it is report time is recent (equal to heartbeat)
1103        """
[72]1104
[770]1105        ignore_changes = [ 'reported' ]
[87]1106
[782]1107        if self.jobAttrsSaved.has_key( jobid ):
[73]1108
[770]1109            for valname, value in jobinfo.items():
[73]1110
[770]1111                if valname not in ignore_changes:
[73]1112
[782]1113                    if self.jobAttrsSaved[ jobid ].has_key( valname ):
[73]1114
[782]1115                        if value != self.jobAttrsSaved[ jobid ][ valname ]:
[73]1116
[782]1117                            if jobinfo['reported'] > self.jobAttrsSaved[ jobid ][ 'reported' ]:
1118
1119                                debug_msg( 1, "job %s field '%s' changed since saved from: %s to: %s" %( jobid, valname, value, self.jobAttrsSaved[ jobid ][ valname ] ) )
1120
[770]1121                                return True
[73]1122
[770]1123                    else:
[782]1124                        debug_msg( 1, "job %s did not have field '%s'" %( jobid, valname )  )
[770]1125                        return True
[87]1126
[770]1127        return False
[73]1128
[71]1129class GangliaXMLHandler( xml.sax.handler.ContentHandler ):
[770]1130    """Parse Ganglia's XML"""
[3]1131
[770]1132    def __init__( self, config, datastore ):
1133        """Setup initial variables and gather info on existing rrd archive"""
[63]1134
[857]1135        self.config          = config
1136        self.clusters        = { }
1137        self.ds              = datastore
[324]1138
[857]1139        debug_msg( 1, 'Housekeeping: checking RRD archive (may take a while)..' )
[770]1140        self.gatherClusters()
[857]1141        debug_msg( 1, 'Housekeeping: RRD check complete.' )
[33]1142
[770]1143    def gatherClusters( self ):
1144        """Find all existing clusters in archive dir"""
[44]1145
[770]1146        archive_dir    = check_dir(ARCHIVE_PATH)
[44]1147
[770]1148        hosts        = [ ]
[44]1149
[770]1150        if os.path.exists( archive_dir ):
[44]1151
[770]1152            dirlist    = os.listdir( archive_dir )
[44]1153
[770]1154            for cfgcluster in ARCHIVE_DATASOURCES:
[369]1155
[770]1156                if cfgcluster not in dirlist:
[369]1157
[770]1158                    # Autocreate a directory for this cluster
1159                    # assume it is new
1160                    #
1161                    cluster_dir = '%s/%s' %( check_dir(ARCHIVE_PATH), cfgcluster )
[369]1162
[770]1163                    os.mkdir( cluster_dir )
[369]1164
[770]1165                    dirlist.append( cfgcluster )
[370]1166
[770]1167            for item in dirlist:
[44]1168
[770]1169                clustername = item
[44]1170
[770]1171                if not self.clusters.has_key( clustername ) and clustername in ARCHIVE_DATASOURCES:
[44]1172
[770]1173                    self.clusters[ clustername ] = RRDHandler( self.config, clustername )
[44]1174
[929]1175        debug_msg( 9, "Found "+str(len(self.clusters.keys()))+" clusters" )
[365]1176
[770]1177    def startElement( self, name, attrs ):
1178        """Memorize appropriate data from xml start tags"""
[3]1179
[770]1180        if name == 'GANGLIA_XML':
[32]1181
[782]1182            self.XMLSource      = str( attrs.get( 'SOURCE',  "" ) )
1183            self.gangliaVersion = str( attrs.get( 'VERSION', "" ) )
[32]1184
[770]1185            debug_msg( 10, 'Found XML data: source %s version %s' %( self.XMLSource, self.gangliaVersion ) )
[6]1186
[770]1187        elif name == 'GRID':
[32]1188
[770]1189            self.gridName    = str( attrs.get( 'NAME', "" ) )
[858]1190            self.time        = str( attrs.get( 'LOCALTIME', "" ) )
[32]1191
[770]1192            debug_msg( 10, '`-Grid found: %s' %( self.gridName ) )
[6]1193
[770]1194        elif name == 'CLUSTER':
[32]1195
[782]1196            self.clusterName = str( attrs.get( 'NAME',      "" ) )
[770]1197            self.time        = str( attrs.get( 'LOCALTIME', "" ) )
[32]1198
[770]1199            if not self.clusters.has_key( self.clusterName ) and self.clusterName in ARCHIVE_DATASOURCES:
[32]1200
[770]1201                self.clusters[ self.clusterName ] = RRDHandler( self.config, self.clusterName )
[33]1202
[929]1203                debug_msg( 10, ' |-Cluster found: %s' %( self.clusterName ) )
[6]1204
[770]1205        elif name == 'HOST' and self.clusterName in ARCHIVE_DATASOURCES:     
[32]1206
[782]1207            self.hostName     = str( attrs.get( 'NAME',     "" ) )
1208            self.hostIp       = str( attrs.get( 'IP',       "" ) )
1209            self.hostReported = str( attrs.get( 'REPORTED', "" ) )
[32]1210
[770]1211            debug_msg( 10, ' | |-Host found: %s - ip %s reported %s' %( self.hostName, self.hostIp, self.hostReported ) )
[6]1212
[770]1213        elif name == 'METRIC' and self.clusterName in ARCHIVE_DATASOURCES:
[6]1214
[770]1215            type = str( attrs.get( 'TYPE', "" ) )
1216           
1217            exclude_metric = False
1218           
1219            for ex_metricstr in ARCHIVE_EXCLUDE_METRICS:
[6]1220
[770]1221                orig_name = str( attrs.get( 'NAME', "" ) )
[3]1222
[770]1223                if string.lower( orig_name ) == string.lower( ex_metricstr ):
1224               
1225                    exclude_metric = True
[198]1226
[770]1227                elif re.match( ex_metricstr, orig_name ):
[198]1228
[770]1229                    exclude_metric = True
[198]1230
[770]1231            if type not in UNSUPPORTED_ARCHIVE_TYPES and not exclude_metric:
[198]1232
[782]1233                myMetric         = { }
1234                myMetric['name'] = str( attrs.get( 'NAME', "" ) )
1235                myMetric['val']  = str( attrs.get( 'VAL',  "" ) )
1236                myMetric['time'] = self.hostReported
[3]1237
[770]1238                self.clusters[ self.clusterName ].memMetric( self.hostName, myMetric )
[3]1239
[770]1240                debug_msg( 11, ' | | |-metric: %s:%s' %( myMetric['name'], myMetric['val'] ) )
[6]1241
[770]1242    def storeMetrics( self ):
1243        """Store metrics of each cluster rrd handler"""
[9]1244
[770]1245        for clustername, rrdh in self.clusters.items():
[16]1246
[770]1247            ret = rrdh.storeMetrics()
[9]1248
[770]1249            if ret:
1250                debug_msg( 9, 'An error occured while storing metrics for cluster %s' %clustername )
1251                return 1
[38]1252
[770]1253        return 0
[38]1254
[71]1255class XMLErrorHandler( xml.sax.handler.ErrorHandler ):
1256
[926]1257    def __init__( self ):
1258
1259        self.me_fatal = False
1260
[770]1261    def error( self, exception ):
1262        """Recoverable error"""
[71]1263
[770]1264        debug_msg( 0, 'Recoverable XML error ' + str( exception ) + ' ignored.' )
[71]1265
[770]1266    def fatalError( self, exception ):
1267        """Non-recoverable error"""
[71]1268
[770]1269        exception_str = str( exception )
[71]1270
[770]1271        # Ignore 'no element found' errors
1272        if exception_str.find( 'no element found' ) != -1:
[926]1273
[770]1274            debug_msg( 0, 'No XML data found: Socket not (re)connected or datasource not available.' )
1275            return 0
[71]1276
[926]1277        self.me_fatal = True
1278
[770]1279        debug_msg( 0, 'FATAL ERROR: Non-recoverable XML error ' + str( exception ) )
[71]1280
[926]1281    def isFatal( self ):
1282
1283        return self.me_fatal
1284
[770]1285    def warning( self, exception ):
1286        """Warning"""
[71]1287
[770]1288        debug_msg( 0, 'Warning ' + str( exception ) )
[71]1289
[78]1290class XMLGatherer:
[770]1291    """Setup a connection and file object to Ganglia's XML"""
[3]1292
[858]1293    s           = None
1294    fd          = None
[770]1295    data        = None
1296    slot        = None
[8]1297
[770]1298    # Time since the last update
1299    #
1300    LAST_UPDATE    = 0
[287]1301
[770]1302    # Minimum interval between updates
1303    #
1304    MIN_UPDATE_INT    = 10
[287]1305
[770]1306    # Is a update occuring now
1307    #
1308    update_now    = False
[287]1309
[770]1310    def __init__( self, host, port ):
1311        """Store host and port for connection"""
[8]1312
[770]1313        self.host    = host
1314        self.port    = port
[782]1315        self.slot    = threading.Lock()
[3]1316
[770]1317        self.retrieveData()
[287]1318
[770]1319    def retrieveData( self ):
1320        """Setup connection to XML source"""
[8]1321
[858]1322        self.update_now = True
[287]1323
[770]1324        self.slot.acquire()
[293]1325
[858]1326        self.data       = None
[469]1327
[770]1328        for res in socket.getaddrinfo( self.host, self.port, socket.AF_UNSPEC, socket.SOCK_STREAM ):
[32]1329
[770]1330            af, socktype, proto, canonname, sa = res
[32]1331
[770]1332            try:
[32]1333
[770]1334                self.s = socket.socket( af, socktype, proto )
[32]1335
[770]1336            except ( socket.error, socket.gaierror, socket.herror, socket.timeout ), msg:
[32]1337
[770]1338                self.s = None
1339                continue
[32]1340
[773]1341            try:
[32]1342
[770]1343                self.s.connect( sa )
[32]1344
[773]1345            except ( socket.error, socket.gaierror, socket.herror, socket.timeout ), msg:
[32]1346
[770]1347                self.disconnect()
1348                continue
[32]1349
[773]1350            break
[3]1351
[770]1352        if self.s is None:
[32]1353
[770]1354            debug_msg( 0, 'FATAL ERROR: Could not open socket or unable to connect to datasource!' )
1355            self.update_now    = False
1356            #sys.exit( 1 )
[5]1357
[770]1358        else:
1359            #self.s.send( '\n' )
[287]1360
[770]1361            my_fp            = self.s.makefile( 'r' )
[773]1362            my_data          = my_fp.readlines()
1363            my_data          = string.join( my_data, '' )
[287]1364
[770]1365            self.data        = my_data
[287]1366
[773]1367            self.LAST_UPDATE = time.time()
[287]1368
[770]1369        self.slot.release()
[293]1370
[770]1371        self.update_now    = False
[287]1372
[770]1373    def disconnect( self ):
1374        """Close socket"""
[33]1375
[770]1376        if self.s:
1377            #self.s.shutdown( 2 )
1378            self.s.close()
1379            self.s = None
[33]1380
[770]1381    def __del__( self ):
1382        """Kill the socket before we leave"""
[33]1383
[770]1384        self.disconnect()
[33]1385
[770]1386    def reGetData( self ):
1387        """Reconnect"""
[33]1388
[770]1389        while self.update_now:
[287]1390
[770]1391            # Must be another update in progress:
1392            # Wait until the update is complete
1393            #
1394            time.sleep( 1 )
[287]1395
[770]1396        if self.s:
1397            self.disconnect()
[33]1398
[770]1399        self.retrieveData()
[5]1400
[770]1401    def getData( self ):
[287]1402
[770]1403        """Return the XML data"""
[287]1404
[770]1405        # If more than MIN_UPDATE_INT seconds passed since last data update
1406        # update the XML first before returning it
1407        #
[287]1408
[770]1409        cur_time    = time.time()
[287]1410
[770]1411        if ( cur_time - self.LAST_UPDATE ) > self.MIN_UPDATE_INT:
[287]1412
[770]1413            self.reGetData()
[287]1414
[770]1415        while self.update_now:
[287]1416
[770]1417            # Must be another update in progress:
1418            # Wait until the update is complete
1419            #
1420            time.sleep( 1 )
1421           
1422        return self.data
[287]1423
[770]1424    def makeFileDescriptor( self ):
1425        """Make file descriptor that points to our socket connection"""
[70]1426
[770]1427        self.reconnect()
[70]1428
[770]1429        if self.s:
1430            self.fd = self.s.makefile( 'r' )
[70]1431
[770]1432    def getFileObject( self ):
1433        """Connect, and return a file object"""
[70]1434
[770]1435        self.makeFileDescriptor()
[78]1436
[770]1437        if self.fd:
1438            return self.fd
[70]1439
[78]1440class GangliaXMLProcessor( XMLProcessor ):
[770]1441    """Main class for processing XML and acting with it"""
[5]1442
[770]1443    def __init__( self, XMLSource, DataStore ):
1444        """Setup initial XML connection and handlers"""
[33]1445
[782]1446        self.config       = GangliaConfigParser( GMETAD_CONF )
1447        self.myXMLSource  = XMLSource
1448        self.ds           = DataStore
1449        self.myXMLHandler = GangliaXMLHandler( self.config, self.ds )
1450        self.myXMLError   = XMLErrorHandler()
[33]1451
[770]1452    def run( self ):
1453        """Main XML processing; start a xml and storethread"""
[8]1454
[782]1455        xml_thread   = threading.Thread( None, self.processXML,   'xmlthread' )
[770]1456        store_thread = threading.Thread( None, self.storeMetrics, 'storethread' )
[22]1457
[770]1458        while( 1 ):
[36]1459
[770]1460            if not xml_thread.isAlive():
1461                # Gather XML at the same interval as gmetad
[36]1462
[770]1463                # threaded call to: self.processXML()
1464                #
1465                try:
1466                    xml_thread = threading.Thread( None, self.processXML, 'xml_thread' )
1467                    xml_thread.start()
1468                except thread.error, msg:
1469                    debug_msg( 0, 'ERROR: Unable to start xml_thread!: '+str(msg))
1470                    #return 1
[36]1471
[770]1472            if not store_thread.isAlive():
1473                # Store metrics every .. sec
[36]1474
[770]1475                # threaded call to: self.storeMetrics()
1476                #
1477                try:
1478                    store_thread = threading.Thread( None, self.storeMetrics, 'store_thread' )
1479                    store_thread.start()
1480                except thread.error, msg:
1481                    debug_msg( 0, 'ERROR: Unable to start store_thread!: '+str(msg))
1482                    #return 1
1483       
1484            # Just sleep a sec here, to prevent daemon from going mad. We're all threads here anyway
1485            time.sleep( 1 )   
[36]1486
[770]1487    def storeMetrics( self ):
1488        """Store metrics retained in memory to disk"""
[22]1489
[770]1490        global DEBUG_LEVEL
[365]1491
[770]1492        # Store metrics somewhere between every 360 and 640 seconds
1493        #
[774]1494        if DEBUG_LEVEL >= 1:
1495            STORE_INTERVAL = 60
[770]1496        else:
[857]1497            STORE_INTERVAL = random.randint( 300, 600 )
[22]1498
[770]1499        try:
1500            store_metric_thread = threading.Thread( None, self.storeThread, 'store_metric_thread' )
1501            store_metric_thread.start()
1502        except thread.error, msg:
1503            debug_msg( 0, 'ERROR: Unable to start ganglia_store_thread()!: '+str(msg) )
1504            return 1
[36]1505
[770]1506        debug_msg( 1, 'ganglia_store_thread(): started.' )
[169]1507
[770]1508        debug_msg( 1, 'ganglia_store_thread(): Sleeping.. (%ss)' %STORE_INTERVAL )
1509        time.sleep( STORE_INTERVAL )
1510        debug_msg( 1, 'ganglia_store_thread(): Done sleeping.' )
[36]1511
[770]1512        if store_metric_thread.isAlive():
[36]1513
[782]1514            debug_msg( 1, 'ganglia_store_thread(): storemetricthread() still running, waiting to finish..' )
[770]1515            store_metric_thread.join( STORE_TIMEOUT ) # Maximum time is for storing thread to finish
[36]1516
[782]1517            if store_metric_thread.isAlive():
1518
1519                debug_msg( 1, 'ganglia_store_thread(): Done waiting: storemetricthread() still running :( now what?' )
1520            else:
1521                debug_msg( 1, 'ganglia_store_thread(): Done waiting: storemetricthread() has finished' )
1522
[770]1523        debug_msg( 1, 'ganglia_store_thread(): finished.' )
[36]1524
[770]1525        return 0
[36]1526
[770]1527    def storeThread( self ):
1528        """Actual metric storing thread"""
[39]1529
[770]1530        debug_msg( 1, 'ganglia_store_metric_thread(): started.' )
1531        debug_msg( 1, 'ganglia_store_metric_thread(): Storing data..' )
[782]1532
[770]1533        ret = self.myXMLHandler.storeMetrics()
1534        if ret > 0:
1535            debug_msg( 0, 'ganglia_store_metric_thread(): UNKNOWN ERROR %s while storing Metrics!' %str(ret) )
[782]1536
[770]1537        debug_msg( 1, 'ganglia_store_metric_thread(): Done storing.' )
1538        debug_msg( 1, 'ganglia_store_metric_thread(): finished.' )
1539       
1540        return 0
[39]1541
[770]1542    def processXML( self ):
1543        """Process XML"""
[8]1544
[770]1545        try:
1546            parsethread = threading.Thread( None, self.parseThread, 'parsethread' )
1547            parsethread.start()
1548        except thread.error, msg:
1549            debug_msg( 0, 'ERROR: Unable to start ganglia_xml_thread()!: ' + str(msg) )
1550            return 1
[8]1551
[770]1552        debug_msg( 1, 'ganglia_xml_thread(): started.' )
[36]1553
[770]1554        debug_msg( 1, 'ganglia_xml_thread(): Sleeping.. (%ss)' %self.config.getLowestInterval() )
1555        time.sleep( float( self.config.getLowestInterval() ) )   
1556        debug_msg( 1, 'ganglia_xml_thread(): Done sleeping.' )
[36]1557
[770]1558        if parsethread.isAlive():
[36]1559
[782]1560            debug_msg( 1, 'ganglia_xml_thread(): parsethread() still running, waiting (%ss) to finish..' %PARSE_TIMEOUT )
[770]1561            parsethread.join( PARSE_TIMEOUT ) # Maximum time for XML thread to finish
[36]1562
[782]1563            if parsethread.isAlive():
1564                debug_msg( 1, 'ganglia_xml_thread(): Done waiting: parsethread() still running :( now what?' )
1565            else:
1566                debug_msg( 1, 'ganglia_xml_thread(): Done waiting: parsethread() finished' )
1567
[770]1568        debug_msg( 1, 'ganglia_xml_thread(): finished.' )
[36]1569
[770]1570        return 0
[36]1571
[770]1572    def parseThread( self ):
1573        """Actual parsing thread"""
[39]1574
[770]1575        debug_msg( 1, 'ganglia_parse_thread(): started.' )
1576        debug_msg( 1, 'ganglia_parse_thread(): Retrieving XML data..' )
1577       
1578        my_data    = self.myXMLSource.getData()
[176]1579
[782]1580        debug_msg( 1, 'ganglia_parse_thread(): Done retrieving: data size %d' %len(my_data) )
[293]1581
[770]1582        if my_data:
1583            debug_msg( 1, 'ganglia_parse_thread(): Parsing XML..' )
1584            xml.sax.parseString( my_data, self.myXMLHandler, self.myXMLError )
1585            debug_msg( 1, 'ganglia_parse_thread(): Done parsing.' )
[176]1586
[770]1587        debug_msg( 1, 'ganglia_parse_thread(): finished.' )
[39]1588
[770]1589        return 0
[39]1590
[9]1591class GangliaConfigParser:
1592
[770]1593    sources = [ ]
[9]1594
[770]1595    def __init__( self, config ):
1596        """Parse some stuff from our gmetad's config, such as polling interval"""
[32]1597
[770]1598        self.config = config
1599        self.parseValues()
[9]1600
[770]1601    def parseValues( self ):
1602        """Parse certain values from gmetad.conf"""
[9]1603
[770]1604        readcfg = open( self.config, 'r' )
[9]1605
[770]1606        for line in readcfg.readlines():
[9]1607
[770]1608            if line.count( '"' ) > 1:
[9]1609
[770]1610                if line.find( 'data_source' ) != -1 and line[0] != '#':
[9]1611
[855]1612                    source                = { }
1613                    source['name']        = line.split( '"' )[1]
1614                    source_value_words    = line.split( '"' )[2].split( ' ' )
[9]1615
[855]1616                    check_interval        = source_value_words[0]
[9]1617
[855]1618                    try:
[9]1619
[855]1620                        source['interval'] = int( check_interval )
1621                        debug_msg( 9, 'polling interval for %s = %s' %(source['name'], str( source['interval'] ) ) )
1622                    except ValueError:
[32]1623
[770]1624                        source['interval'] = 15
1625                        debug_msg( 9, 'polling interval for %s defaulted to 15' %(source['name']) )
[32]1626
[770]1627                    self.sources.append( source )
[9]1628
[855]1629        readcfg.close()
1630
1631    def clusterExists( self, source_name ):
1632
1633        for source in self.sources:
1634
1635            if source['name'] == source_name:
1636
1637                return True
1638
1639        return False
1640
[770]1641    def getInterval( self, source_name ):
1642        """Return interval for source_name"""
[32]1643
[770]1644        for source in self.sources:
[32]1645
[770]1646            if source['name'] == source_name:
[32]1647
[770]1648                return source['interval']
[32]1649
[770]1650        return None
[9]1651
[770]1652    def getLowestInterval( self ):
1653        """Return the lowest interval of all clusters"""
[34]1654
[770]1655        lowest_interval = 0
[34]1656
[770]1657        for source in self.sources:
[34]1658
[770]1659            if not lowest_interval or source['interval'] <= lowest_interval:
[34]1660
[770]1661                lowest_interval = source['interval']
[34]1662
[770]1663        # Return 15 when nothing is found, so that the daemon won't go insane with 0 sec delays
1664        if lowest_interval:
1665            return lowest_interval
1666        else:
1667            return 15
[34]1668
[9]1669class RRDHandler:
[770]1670    """Class for handling RRD activity"""
[9]1671
[929]1672    myMetrics   = { }
1673    lastStored  = { }
1674    timeserials = { }
1675    slot = None
[32]1676
[770]1677    def __init__( self, config, cluster ):
1678        """Setup initial variables"""
[78]1679
[770]1680        global MODRRDTOOL
[455]1681
[858]1682        self.block   = 0
1683        self.cluster = cluster
1684        self.config  = config
[770]1685        self.slot    = threading.Lock()
[292]1686
[770]1687        if MODRRDTOOL:
[455]1688
[770]1689            self.rrdm    = RRDMutator()
1690        else:
1691            self.rrdm    = RRDMutator( RRDTOOL )
[455]1692
[770]1693        global DEBUG_LEVEL
[9]1694
[770]1695        if DEBUG_LEVEL <= 2:
1696            self.gatherLastUpdates()
[365]1697
[770]1698    def gatherLastUpdates( self ):
1699        """Populate the lastStored list, containing timestamps of all last updates"""
[42]1700
[770]1701        cluster_dir = '%s/%s' %( check_dir(ARCHIVE_PATH), self.cluster )
[42]1702
[770]1703        hosts = [ ]
[42]1704
[770]1705        if os.path.exists( cluster_dir ):
[42]1706
[770]1707            dirlist = os.listdir( cluster_dir )
[42]1708
[770]1709            for dir in dirlist:
[42]1710
[770]1711                hosts.append( dir )
[42]1712
[770]1713        for host in hosts:
[42]1714
[770]1715            host_dir    = cluster_dir + '/' + host
[858]1716            dirlist     = os.listdir( host_dir )
[47]1717
[770]1718            for dir in dirlist:
[47]1719
[770]1720                if not self.timeserials.has_key( host ):
[47]1721
[770]1722                    self.timeserials[ host ] = [ ]
[47]1723
[770]1724                self.timeserials[ host ].append( dir )
[47]1725
[770]1726            last_serial = self.getLastRrdTimeSerial( host )
[292]1727
[770]1728            if last_serial:
[42]1729
[770]1730                metric_dir = cluster_dir + '/' + host + '/' + last_serial
[292]1731
[770]1732                if os.path.exists( metric_dir ):
[42]1733
[770]1734                    dirlist = os.listdir( metric_dir )
[42]1735
[770]1736                    for file in dirlist:
[42]1737
[770]1738                        metricname = file.split( '.rrd' )[0]
[42]1739
[770]1740                        if not self.lastStored.has_key( host ):
[42]1741
[770]1742                            self.lastStored[ host ] = { }
[42]1743
[770]1744                        self.lastStored[ host ][ metricname ] = self.rrdm.grabLastUpdate( metric_dir + '/' + file )
[42]1745
[770]1746    def getClusterName( self ):
1747        """Return clustername"""
[63]1748
[770]1749        return self.cluster
[32]1750
[770]1751    def memMetric( self, host, metric ):
1752        """Store metric from host in memory"""
[32]1753
[770]1754        # <ATOMIC>
1755        #
1756        self.slot.acquire()
1757       
1758        if self.myMetrics.has_key( host ):
[32]1759
[770]1760            if self.myMetrics[ host ].has_key( metric['name'] ):
[32]1761
[770]1762                for mymetric in self.myMetrics[ host ][ metric['name'] ]:
[32]1763
[770]1764                    if mymetric['time'] == metric['time']:
[32]1765
[770]1766                        # Allready have this metric, abort
1767                        self.slot.release()
1768                        return 1
1769            else:
1770                self.myMetrics[ host ][ metric['name'] ] = [ ]
1771        else:
[858]1772            self.myMetrics[ host ]                   = { }
1773            self.myMetrics[ host ][ metric['name'] ] = [ ]
[32]1774
[770]1775        # Push new metric onto stack
1776        # atomic code; only 1 thread at a time may access the stack
[63]1777
[770]1778        self.myMetrics[ host ][ metric['name'] ].append( metric )
[32]1779
[770]1780        self.slot.release()
1781        #
1782        # </ATOMIC>
[40]1783
[770]1784    def makeUpdateList( self, host, metriclist ):
1785        """
1786        Make a list of update values for rrdupdate
1787        but only those that we didn't store before
1788        """
[37]1789
[770]1790        update_list    = [ ]
1791        metric        = None
[37]1792
[770]1793        while len( metriclist ) > 0:
[37]1794
[770]1795            metric = metriclist.pop( 0 )
[37]1796
[770]1797            if self.checkStoreMetric( host, metric ):
[292]1798
[770]1799                u_val    = str( metric['time'] ) + ':' + str( metric['val'] )
1800                #update_list.append( str('%s:%s') %( metric['time'], metric['val'] ) )
1801                update_list.append( u_val )
[40]1802
[770]1803        return update_list
[37]1804
[770]1805    def checkStoreMetric( self, host, metric ):
1806        """Check if supplied metric if newer than last one stored"""
[40]1807
[770]1808        if self.lastStored.has_key( host ):
[40]1809
[770]1810            if self.lastStored[ host ].has_key( metric['name'] ):
[40]1811
[770]1812                if metric['time'] <= self.lastStored[ host ][ metric['name'] ]:
[40]1813
[770]1814                    # This is old
1815                    return 0
[40]1816
[770]1817        return 1
[50]1818
[770]1819    def memLastUpdate( self, host, metricname, metriclist ):
1820        """
1821        Memorize the time of the latest metric from metriclist
1822        but only if it wasn't allready memorized
1823        """
[50]1824
[770]1825        if not self.lastStored.has_key( host ):
1826            self.lastStored[ host ] = { }
[54]1827
[770]1828        last_update_time = 0
[50]1829
[770]1830        for metric in metriclist:
[50]1831
[770]1832            if metric['name'] == metricname:
[50]1833
[770]1834                if metric['time'] > last_update_time:
[50]1835
[770]1836                    last_update_time = metric['time']
[40]1837
[770]1838        if self.lastStored[ host ].has_key( metricname ):
1839           
1840            if last_update_time <= self.lastStored[ host ][ metricname ]:
1841                return 1
[40]1842
[770]1843        self.lastStored[ host ][ metricname ] = last_update_time
[52]1844
[770]1845    def storeMetrics( self ):
1846        """
1847        Store all metrics from memory to disk
1848        and do it to the RRD's in appropriate timeperiod directory
1849        """
[33]1850
[770]1851        debug_msg( 5, "Entering storeMetrics()")
[365]1852
[782]1853        count_values  = 0
1854        count_metrics = 0
[770]1855        count_bits    = 0
[365]1856
[770]1857        for hostname, mymetrics in self.myMetrics.items():   
[33]1858
[770]1859            for metricname, mymetric in mymetrics.items():
[33]1860
[770]1861                count_metrics += 1
[365]1862
[770]1863                for dmetric in mymetric:
[365]1864
[770]1865                    count_values += 1
[365]1866
[782]1867                    count_bits   += len( dmetric['time'] )
1868                    count_bits   += len( dmetric['val'] )
[365]1869
[770]1870        count_bytes    = count_bits / 8
[365]1871
[770]1872        debug_msg( 5, "size of cluster '" + self.cluster + "': " + 
1873            str( len( self.myMetrics.keys() ) ) + " hosts " + 
1874            str( count_metrics ) + " metrics " + str( count_values ) + " values " +
1875            str( count_bits ) + " bits " + str( count_bytes ) + " bytes " )
[365]1876
[770]1877        for hostname, mymetrics in self.myMetrics.items():   
[365]1878
[770]1879            for metricname, mymetric in mymetrics.items():
[365]1880
[770]1881                metrics_to_store = [ ]
[53]1882
[770]1883                # Pop metrics from stack for storing until none is left
1884                # atomic code: only 1 thread at a time may access myMetrics
[63]1885
[770]1886                # <ATOMIC>
1887                #
1888                self.slot.acquire() 
[33]1889
[770]1890                while len( self.myMetrics[ hostname ][ metricname ] ) > 0:
[53]1891
[770]1892                    if len( self.myMetrics[ hostname ][ metricname ] ) > 0:
[53]1893
[770]1894                        try:
1895                            metrics_to_store.append( self.myMetrics[ hostname ][ metricname ].pop( 0 ) )
1896                        except IndexError, msg:
[176]1897
[770]1898                            # Somehow sometimes myMetrics[ hostname ][ metricname ]
1899                            # is still len 0 when the statement is executed.
1900                            # Just ignore indexerror's..
1901                            pass
[176]1902
[770]1903                self.slot.release()
1904                #
1905                # </ATOMIC>
[53]1906
[770]1907                # Create a mapping table, each metric to the period where it should be stored
1908                #
1909                metric_serial_table = self.determineSerials( hostname, metricname, metrics_to_store )
[33]1910
[770]1911                update_rets = [ ]
[50]1912
[770]1913                for period, pmetric in metric_serial_table.items():
[47]1914
[770]1915                    create_ret = self.createCheck( hostname, metricname, period )   
[47]1916
[770]1917                    update_ret = self.update( hostname, metricname, period, pmetric )
[47]1918
[770]1919                    if update_ret == 0:
[47]1920
[770]1921                        debug_msg( 9, 'stored metric %s for %s' %( hostname, metricname ) )
1922                    else:
1923                        debug_msg( 9, 'metric update failed' )
[47]1924
[770]1925                    update_rets.append( create_ret )
1926                    update_rets.append( update_ret )
[47]1927
[770]1928                # Lets ignore errors here for now, we need to make sure last update time
1929                # is correct!
1930                #
1931                #if not (1) in update_rets:
[50]1932
[770]1933                self.memLastUpdate( hostname, metricname, metrics_to_store )
[50]1934
[770]1935        debug_msg( 5, "Leaving storeMetrics()")
[365]1936
[770]1937    def makeTimeSerial( self ):
1938        """Generate a time serial. Seconds since epoch"""
[17]1939
[770]1940        # Seconds since epoch
1941        mytime = int( time.time() )
[17]1942
[770]1943        return mytime
[17]1944
[770]1945    def makeRrdPath( self, host, metricname, timeserial ):
1946        """Make a RRD location/path and filename"""
[17]1947
[858]1948        rrd_dir  = '%s/%s/%s/%s'    %( check_dir(ARCHIVE_PATH), self.cluster, host, timeserial )
1949        rrd_file = '%s/%s.rrd'    %( rrd_dir, metricname )
[17]1950
[770]1951        return rrd_dir, rrd_file
[17]1952
[770]1953    def getLastRrdTimeSerial( self, host ):
1954        """Find the last timeserial (directory) for this host"""
[17]1955
[770]1956        newest_timeserial = 0
[19]1957
[770]1958        for dir in self.timeserials[ host ]:
[32]1959
[770]1960            valid_dir = 1
[17]1961
[770]1962            for letter in dir:
1963                if letter not in string.digits:
1964                    valid_dir = 0
[17]1965
[770]1966            if valid_dir:
1967                timeserial = dir
1968                if timeserial > newest_timeserial:
1969                    newest_timeserial = timeserial
[17]1970
[770]1971        if newest_timeserial:
1972            return newest_timeserial
1973        else:
1974            return 0
[17]1975
[770]1976    def determinePeriod( self, host, check_serial ):
1977        """Determine to which period (directory) this time(serial) belongs"""
[47]1978
[770]1979        period_serial = 0
[47]1980
[770]1981        if self.timeserials.has_key( host ):
[47]1982
[770]1983            for serial in self.timeserials[ host ]:
[47]1984
[770]1985                if check_serial >= serial and period_serial < serial:
[47]1986
[770]1987                    period_serial = serial
[56]1988
[770]1989        return period_serial
[47]1990
[770]1991    def determineSerials( self, host, metricname, metriclist ):
1992        """
1993        Determine the correct serial and corresponding rrd to store
1994        for a list of metrics
1995        """
[47]1996
[770]1997        metric_serial_table = { }
[47]1998
[770]1999        for metric in metriclist:
[47]2000
[770]2001            if metric['name'] == metricname:
[47]2002
[858]2003                period       = self.determinePeriod( host, metric['time'] )   
[47]2004
[858]2005                archive_secs = ARCHIVE_HOURS_PER_RRD * (60 * 60)
[47]2006
[770]2007                if (int( metric['time'] ) - int( period ) ) > archive_secs:
[47]2008
[770]2009                    # This one should get it's own new period
2010                    period = metric['time']
[57]2011
[770]2012                    if not self.timeserials.has_key( host ):
2013                        self.timeserials[ host ] = [ ]
[57]2014
[770]2015                    self.timeserials[ host ].append( period )
[47]2016
[770]2017                if not metric_serial_table.has_key( period ):
[47]2018
[770]2019                    metric_serial_table[ period ] = [ ]
[47]2020
[770]2021                metric_serial_table[ period ].append( metric )
[47]2022
[770]2023        return metric_serial_table
[47]2024
[770]2025    def createCheck( self, host, metricname, timeserial ):
2026        """Check if an rrd allready exists for this metric, create if not"""
[9]2027
[770]2028        debug_msg( 9, 'rrdcreate: using timeserial %s for %s/%s' %( timeserial, host, metricname ) )
2029       
2030        rrd_dir, rrd_file = self.makeRrdPath( host, metricname, timeserial )
[17]2031
[770]2032        if not os.path.exists( rrd_dir ):
[58]2033
[770]2034            try:
2035                os.makedirs( rrd_dir )
[58]2036
[770]2037            except os.OSError, msg:
[58]2038
[770]2039                if msg.find( 'File exists' ) != -1:
[58]2040
[770]2041                    # Ignore exists errors
2042                    pass
[58]2043
[770]2044                else:
[58]2045
[770]2046                    print msg
2047                    return
[58]2048
[770]2049            debug_msg( 9, 'created dir %s' %( str(rrd_dir) ) )
[9]2050
[770]2051        if not os.path.exists( rrd_file ):
[9]2052
[855]2053            interval     = self.config.getInterval( self.cluster )
[770]2054            heartbeat    = 8 * int( interval )
[9]2055
[858]2056            params       = [ ]
[12]2057
[770]2058            params.append( '--step' )
2059            params.append( str( interval ) )
[12]2060
[770]2061            params.append( '--start' )
2062            params.append( str( int( timeserial ) - 1 ) )
[12]2063
[770]2064            params.append( 'DS:sum:GAUGE:%d:U:U' %heartbeat )
2065            params.append( 'RRA:AVERAGE:0.5:1:%s' %(ARCHIVE_HOURS_PER_RRD * 240) )
[13]2066
[770]2067            self.rrdm.create( str(rrd_file), params )
[37]2068
[770]2069            debug_msg( 9, 'created rrd %s' %( str(rrd_file) ) )
[14]2070
[770]2071    def update( self, host, metricname, timeserial, metriclist ):
2072        """
2073        Update rrd file for host with metricname
2074        in directory timeserial with metriclist
2075        """
[9]2076
[770]2077        debug_msg( 9, 'rrdupdate: using timeserial %s for %s/%s' %( timeserial, host, metricname ) )
[9]2078
[858]2079        rrd_dir, rrd_file = self.makeRrdPath( host, metricname, timeserial )
[18]2080
[858]2081        update_list       = self.makeUpdateList( host, metriclist )
[15]2082
[770]2083        if len( update_list ) > 0:
2084            ret = self.rrdm.update( str(rrd_file), update_list )
[32]2085
[770]2086            if ret:
2087                return 1
2088       
2089            debug_msg( 9, 'updated rrd %s with %s' %( str(rrd_file), string.join( update_list ) ) )
[15]2090
[770]2091        return 0
[36]2092
[169]2093def daemon():
[770]2094    """daemonized threading"""
[8]2095
[770]2096    # Fork the first child
2097    #
2098    pid = os.fork()
[169]2099
[770]2100    if pid > 0:
[169]2101
[770]2102        sys.exit(0)  # end parent
[169]2103
[770]2104    # creates a session and sets the process group ID
2105    #
2106    os.setsid()
[169]2107
[770]2108    # Fork the second child
2109    #
2110    pid = os.fork()
[169]2111
[770]2112    if pid > 0:
[169]2113
[770]2114        sys.exit(0)  # end parent
[169]2115
[770]2116    write_pidfile()
[435]2117
[770]2118    # Go to the root directory and set the umask
2119    #
2120    os.chdir('/')
2121    os.umask(0)
[169]2122
[770]2123    sys.stdin.close()
2124    sys.stdout.close()
2125    sys.stderr.close()
[169]2126
[770]2127    os.open('/dev/null', os.O_RDWR)
2128    os.dup2(0, 1)
2129    os.dup2(0, 2)
[169]2130
[770]2131    run()
[169]2132
2133def run():
[770]2134    """Threading start"""
[169]2135
[855]2136    global ARCHIVE_DATASOURCES
2137
[782]2138    config             = GangliaConfigParser( GMETAD_CONF )
[855]2139
2140    for ds in ARCHIVE_DATASOURCES:
2141
2142        if not config.clusterExists( ds ):
2143
2144            print "FATAL ERROR: Data source with name '%s' not found in %s" %( ds, GMETAD_CONF )
2145            sys.exit( 1 )
2146
[782]2147    s_timeout          = int( config.getLowestInterval() - 1 )
[469]2148
[770]2149    socket.setdefaulttimeout( s_timeout )
[469]2150
[770]2151    myXMLSource        = XMLGatherer( ARCHIVE_XMLSOURCE.split( ':' )[0], ARCHIVE_XMLSOURCE.split( ':' )[1] )
2152    myDataStore        = DataSQLStore( JOB_SQL_DBASE.split( '/' )[0], JOB_SQL_DBASE.split( '/' )[1] )
[8]2153
[782]2154    myJobProcessor     = JobXMLProcessor( myXMLSource, myDataStore )
2155    myGangliaProcessor = GangliaXMLProcessor( myXMLSource, myDataStore )
[287]2156
[770]2157    try:
[782]2158        job_xml_thread     = threading.Thread( None, myJobProcessor.run, 'job_proc_thread' )
2159        ganglia_xml_thread = threading.Thread( None, myGangliaProcessor.run, 'ganglia_proc_thread' )
[22]2160
[782]2161        job_xml_thread.start()
[770]2162        ganglia_xml_thread.start()
2163       
2164    except thread.error, msg:
2165        debug_msg( 0, 'FATAL ERROR: Unable to start main threads!: '+ str(msg) )
2166        syslog.closelog()
2167        sys.exit(1)
2168       
2169    debug_msg( 0, 'main threading started.' )
[78]2170
[169]2171def main():
[770]2172    """Program startup"""
[169]2173
[770]2174    global DAEMONIZE, USE_SYSLOG
[375]2175
[770]2176    if not processArgs( sys.argv[1:] ):
2177        sys.exit( 1 )
[214]2178
[770]2179    if( DAEMONIZE and USE_SYSLOG ):
2180        syslog.openlog( 'jobarchived', syslog.LOG_NOWAIT, SYSLOG_FACILITY )
[169]2181
[770]2182    if DAEMONIZE:
2183        daemon()
2184    else:
2185        run()
[169]2186
2187#
[81]2188# Global functions
[169]2189#
[81]2190
[9]2191def check_dir( directory ):
[770]2192    """Check if directory is a proper directory. I.e.: Does _not_ end with a '/'"""
[9]2193
[770]2194    if directory[-1] == '/':
2195        directory = directory[:-1]
[9]2196
[770]2197    return directory
[9]2198
[295]2199def reqtime2epoch( rtime ):
2200
[770]2201    (hours, minutes, seconds )    = rtime.split( ':' )
[295]2202
[770]2203    etime    = int(seconds)
2204    etime    = etime + ( int(minutes) * 60 )
2205    etime    = etime + ( int(hours) * 60 * 60 )
[295]2206
[770]2207    return etime
[295]2208
[12]2209def debug_msg( level, msg ):
[770]2210    """Only print msg if correct levels"""
[12]2211
[770]2212    if (not DAEMONIZE and DEBUG_LEVEL >= level):
2213        sys.stderr.write( printTime() + ' - ' + msg + '\n' )
2214   
2215    if (DAEMONIZE and USE_SYSLOG and SYSLOG_LEVEL >= level):
2216        syslog.syslog( msg )
[12]2217
[46]2218def printTime( ):
[770]2219    """Print current time in human readable format"""
[46]2220
[770]2221    return time.strftime("%a %d %b %Y %H:%M:%S")
[46]2222
[435]2223def write_pidfile():
2224
[770]2225    # Write pidfile if PIDFILE exists
2226    if PIDFILE:
[435]2227
[770]2228        pid     = os.getpid()
[435]2229
[770]2230        pidfile = open(PIDFILE, 'w')
[435]2231
[770]2232        pidfile.write( str( pid ) )
2233        pidfile.close()
[435]2234
[63]2235# Ooohh, someone started me! Let's go..
[469]2236#
[9]2237if __name__ == '__main__':
[770]2238    main()
Note: See TracBrowser for help on using the repository browser.