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 or (at your 
  8  # option) any later version as published by 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 not isinstance(rrdCreateCommand, basestring): 
 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 derive(self, name, cycleTime, value): 
 83          "Write a DERIVE value, return threshold events" 
 84          return self.counter(name, cycleTime, value) 
 85   
 86 -    def counter(self, name, cycleTime, value): 
 87          "Write a DERIVE(! NOT COUNTER!) value, return threshold events" 
 88          fileName = self.rrdFile('DERIVE', cycleTime, name, 0) 
 89          if fileName: 
 90              full = fullname(fileName) 
 91              try: 
 92                  rrdtool.update(full, 'N:%s' % int(value)) 
 93                  startStop, names, values = \ 
 94                      rrdtool.fetch(full, 'AVERAGE', 
 95                          '-s', 'now-%d' % (cycleTime*2), 
 96                          '-e', 'now') 
 97                  value = values[0][0] 
 98                  if value is not None: 
 99                      return self.thresholds.check(fileName, time.time(), value) 
100              except rrdtool.error, err: 
101                  log.error('rrdtool reported error %s %s', err, full) 
102          return [] 
103   
104   
105 -    def gauge(self, name, cycleTime, value): 
106          "Write a gauge value, return threshold events" 
107          fileName = self.rrdFile('GAUGE', cycleTime, name) 
108          if fileName: 
109              full = fullname(fileName) 
110              try: 
111                  rrdtool.update(full, 'N:%s' % value) 
112              except rrdtool.error, err: 
113                  log.error('rrdtool reported error %s %s', err, full) 
114          if value is not None: 
115              return self.thresholds.check(fileName, time.time(), value) 
116          return [] 
117