python/ashd/perf.py

   1 try:
   2     import pdm.perf
   3 except:
   4     pdm = None
   5
   6 reqstat = {}
   7
   8 if pdm:
   9     statistics = pdm.perf.staticdir()
  10     statistics["req"] = pdm.perf.valueattr(reqstat)
  11     requests = pdm.perf.eventobj()
  12
  13     class reqstart(pdm.perf.startevent):
  14         def __init__(self, env):
  15             super(reqstart, self).__init__()
  16             self.method = env.get("REQUEST_METHOD")
  17             self.uri = env.get("REQUEST_URI")
  18             self.host = env.get("HTTP_HOST")
  19             self.remoteaddr = env.get("REMOTE_ADDR")
  20             self.remoteport = env.get("REMOTE_PORT")
  21
  22     class reqfinish(pdm.perf.finishevent):
  23         def __init__(self, start, aborted, status):
  24             super(reqfinish, self).__init__(start, aborted)
  25             self.status = status
  26
  27 class request(object):
  28     def __init__(self, env):
  29         self.resp = None
  30         if pdm:
  31             self.startev = reqstart(env)
  32             requests.notify(self.startev)
  33
  34     def response(self, resp):
  35         self.resp = resp
  36
  37     def finish(self, aborted):
  38         key = None
  39         status = None
  40         try:
  41             if len(self.resp) > 0:
  42                 status = self.resp[0]
  43                 p = status.find(" ")
  44                 if p < 0:
  45                     key = status
  46                 else:
  47                     key = status[:p]
  48         except:
  49             pass
  50         reqstat[key] = reqstat.setdefault(key, 0) + 1
  51         if pdm:
  52             requests.notify(reqfinish(self.startev, aborted, status))
  53
  54     def __enter__(self):
  55         return self
  56
  57     def __exit__(self, *excinfo):
  58         self.finish(bool(excinfo[0]))
  59         return False