Source Code for Module Products.ZenUtils.DaemonStats

  1  ########################################################################### 
  2  # 
  3  # This program is part of Zenoss Core, an open source monitoring platform. 
  4  # Copyright (C) 2007, Zenoss Inc. 
  5  # 
  6  # This program is free software; you can redistribute it and/or modify it 
  7  # under the terms of the GNU General Public License version 2 as published by 
  8  # the Free Software Foundation. 
  9  # 
 10  # For complete information please visit: http://www.zenoss.com/oss/ 
 11  # 
 12  ########################################################################### 
 13  import Globals 
 14  from Products.ZenUtils.Utils import zenPath 
 15   
 16  # FIXME dependency from Utils to Thresholds 
 17  from Products.ZenRRD.Thresholds import Thresholds 
 18   
 19  import rrdtool 
 20  import os 
 21  import time 
 22   
 23  import logging 
 24  log = logging.getLogger("zen.DaemonStats") 
 25   
 26 -def fullname(partial): 
 27      return zenPath('perf', partial + '.rrd') 
 28   
 29 -class DaemonStats: 
 30      "Utility for a daemon to write out internal performance statistics" 
 31   
 32      name = "" 
 33      monitor = "" 
 34      rrdCreateCommand = "" 
 35   
 36 -    def __init__(self): 
 37          self.thresholds = Thresholds() 
 38   
 39   
 40 -    def config(self, name, monitor, thresholds, rrdCreateCommand = None): 
 41          """Initialize the object.  We could do this in __init__, but 
 42          that would delay creation to after configuration time, which 
 43          may run asynchronously with collection or heartbeats.  By 
 44          deferring initialization, this object implements the Null 
 45          Object pattern until the application is ready to start writing 
 46          real statistics. 
 47          """ 
 48          self.name = name 
 49          self.monitor = monitor 
 50          if not rrdCreateCommand: 
 51              from Products.ZenModel.PerformanceConf import PerformanceConf 
 52              rrdCreateCommand = PerformanceConf.defaultRRDCreateCommand 
 53          if type(rrdCreateCommand) != type(''): 
 54              self.createCommand = rrdCreateCommand 
 55          else: 
 56              self.createCommand = rrdCreateCommand.split('\n') 
 57          self.thresholds = Thresholds() 
 58          self.thresholds.updateList(thresholds) 
 59   
 60   
 61 -    def rrdFile(self, type, cycleTime, name, minVal = 'U', maxVal = 'U'): 
 62          """Create an RRD file if it does not exist. 
 63          Returns the basename of the rrdFile, suitable for checking thresholds. 
 64          """ 
 65          if not self.name: return None 
 66          base = os.path.join('Daemons', self.name) 
 67          directory = zenPath('perf', base) 
 68          if not os.path.exists(directory): 
 69              os.makedirs(directory) 
 70          base = os.path.join(base, '%s_%s' % (self.monitor, name)) 
 71          fileName = fullname(base) 
 72          if not os.path.exists(fileName): 
 73              rrdtool.create(fileName, 
 74                             'DS:ds0:%s:%s:%s:%s' % (type, 
 75                                                     cycleTime * 3, 
 76                                                     minVal, 
 77                                                     maxVal), 
 78                             *self.createCommand) 
 79          return base 
 80   
 81   
 82 -    def counter(self, name, cycleTime, value): 
 83          "Write a counter value, return threshold events" 
 84          fileName = self.rrdFile('DERIVE', cycleTime, name, 0) 
 85          if fileName: 
 86              full = fullname(fileName) 
 87              try: 
 88                  rrdtool.update(full, 'N:%s' % int(value)) 
 89                  startStop, names, values = \ 
 90                      rrdtool.fetch(full, 'AVERAGE', 
 91                          '-s', 'now-%d' % (cycleTime*2), 
 92                          '-e', 'now') 
 93                  value = values[0][0] 
 94                  if value is not None: 
 95                      return self.thresholds.check(fileName, time.time(), value) 
 96              except rrdtool.error, err: 
 97                  log.error('rrdtool reported error %s %s', err, full) 
 98          return [] 
 99   
100   
101 -    def gauge(self, name, cycleTime, value): 
102          "Write a gauge value, return threshold events" 
103          fileName = self.rrdFile('GAUGE', cycleTime, name) 
104          if fileName: 
105              full = fullname(fileName) 
106              try: 
107                  rrdtool.update(full, 'N:%s' % value) 
108              except rrdtool.error, err: 
109                  log.error('rrdtool reported error %s %s', err, full) 
110          if value is not None: 
111              return self.thresholds.check(fileName, time.time(), value) 
112          return [] 
113