#!/usr/bin/env

import datetime, threading, thread

class CheckPerf:
    writelock = threading.Lock()
    def __init__(self):
        self.begintime = datetime.datetime.now()
        self.endtime = None
        self.mintime = 0.0
        self.maxtime = 0.0
        self.successcount = 0
        self.failedcount = 0
        self.totaltime = 0.0
        self.reportmsg = ''
        self.prechktime = None
        
    def startchk(self):
        self.begintime = datetime.datetime.now()
        self.endtime = None
        self.mintime = 99999999.0
        self.maxtime = 0.0
        self.checkcount = 0
        self.totaltime = 0.0
        self.reportmsg = ''
        self.prechktime = self.begintime
    
    def setchkpoint(self, success=True):
        self.endtime = datetime.datetime.now()
        delta = self.endtime - self.prechktime
        t = delta.seconds + delta.microseconds / 1000000.0
        if t < self.mintime: self.mintime = t
        if t > self.maxtime: self.maxtime = t
        if success:
            self.successcount += 1
        else:
            self.failedcount += 1
        self.totaltime += t
        self.prechktime = self.endtime
        
    def _output_file(self, out_file):
        handler = open(out_file, 'a+b')
        handler.write("thrd[%d]" % thread.get_ident())
        handler.write(' - ' + self.reportmsg + '\n')
        handler.close()
        
    def _reportToFile(self, out_file):
        CheckPerf.writelock.acquire()
        try:
            self._output_file(out_file)
            print "generate report file " , out_file
        except Exception, ex:
            print "generator report file ", out_file , " Error ", ex
        CheckPerf.writelock.release()
        
    def report_msg(self):
        totalcheckcount = self.successcount + self.failedcount
        if totalcheckcount > 2:
            avg = (self.totaltime - self.mintime - self.maxtime) / (totalcheckcount - 2)
        elif totalcheckcount == 0:
            rpt = "No check point has been set"
            return rpt
        else:
            avg = self.totaltime / totalcheckcount
        rpt = 'execute[%d]count cost(s) [%.4f]max[%.4f]min[%.4f]avg[%.4f] success[%d] failed[%d]' \
            % (totalcheckcount, self.totaltime, self.maxtime, self.mintime, avg, \
            self.successcount, self.failedcount)
        return rpt
        
    def report(self, out_file=None):
        self.reportmsg = self.report_msg()
        if out_file:
            self._reportToFile(out_file)
        return self.reportmsg