source: trunk/Monitoring/Analyze/analyze @ 879

#!/usr/bin/env python
# -*- coding: ISO-8859-1 -*-

##################################
#  @program        smon
#  @description    simulation monitor
#  @copyright      Copyright “(c)2009 Centre National de la Recherche Scientifique CNRS. 
#                             All Rights Reserved”
#  @svn_file       $Id: analyzer 2545 2013-02-01 09:58:10Z jripsl $
#  @version        $Rev: 2545 $
#  @lastrevision   $Date: 2013-02-01 10:58:10 +0100 (Fri, 01 Feb 2013) $
#  @license        CeCILL (http://dods.ipsl.jussieu.fr/jripsl/smon/LICENSE)
##################################

import sys
import signal
import traceback
import smtplib
from email.mime.text import MIMEText
import time
import datetime


# line below is to include "smon" package in the search path
sys.path.append("/home/jripsl/snapshot/Monitoring")

import smon.repo_io as repo_io
import smon.types as types



class CheckList():
        max_time_between_msg=20 # unit => seconds

        @classmethod
        def msg_timeout(cls,message):

                # get current epoch
                current_epoch=time.time()

                # get msg epoch
                msg_time=time.strptime(str(message.crea_date), "%Y-%m-%d %H:%M:%S.%f")
                msg_epoch=time.mktime(msg_time)

                diff =  current_epoch - msg_epoch

                # debug
                #print "cur=%i,ms=%s"%(current_epoch,message.crea_date)

                # debug
                #print "diff=%i"%int(diff)

                if diff>cls.max_time_between_msg:

                        return True
                else:
                        return False

        @classmethod
        def C0001(cls):
                """
                description
                  check heartbeat (currently, heartbeat is implemented using simulation progress messages).
                  if progress messages suddently stops, it is likely that the simulation was Killed or
                  a Segfault occurs. In such case, we inform the other components (failover, prodiguer GUI..) by
                  changing the simulation status 
                """


                for simulation in repo_io.get_running_simulations():

                        print "hhh"

                        print "checking heartbeat ('%s')"%simulation.name

                        try:
                                message=repo_io.retrieve_last_message(simulation)

                                # debug
                                #print "found"

                        except types.MessageNotFoundException, e:
                                # when we are here, it mean we are in the interval when a new simulation have just been inserted but the corresponding message have not been inserted yet

                                print "no message found for simulation ('%s')"%simulation.name

                                continue


                        if cls.msg_timeout(message):

                                simulation.status="error"

                                repo_io.update_simulation_status(simulation)

                                print "heartbeat NOK - simulation status set to 'error' (%s)"%simulation.name


                        else:
                                print "heartbeat OK (%s)"%simulation.name

class Analyzer():

        @classmethod
        def start(cls):
                repo_io.init() # open DB connection

                Analyzer.main()

        @classmethod
        def stop(cls):
                repo_io.free() # close DB connection

        @classmethod
        def main(self):

                """
                # parse args
                parser = argparse.ArgumentParser(prog='analyzer')
                parser.add_argument('-v', dest='verbose',required=False,action='store_true')
                args = parser.parse_args()

                # check
                if not os.path.exists(SMON.smon_home):
                        sys.exit(1)

                SMON.init_singleton()
                """

                print ' [*] Analyzer running. To exit press CTRL+C'

                while True:

                        print "checking simulations heartbeats"

                        CheckList.C0001()


                        time.sleep(3)

                """
                SMON.free_singleton()
                """

def signal_handler(signal, frame):
                print 'You pressed Ctrl+C!'

                Analyzer.stop()

                sys.exit(0)

if __name__ == '__main__':

        signal.signal(signal.SIGINT, signal_handler)

        try:
                Analyzer.start()

                sys.exit(0)

        except Exception, e:

                traceback.print_exc()

                sys.exit(1)