X-Git-Url: http://www.dolda2000.com/gitweb/?a=blobdiff_plain;f=python%2Fashd%2Fperf.py;fp=python%2Fashd%2Fperf.py;h=d3d3cb776c7a06ae47ea621c8041c430a3ad7ba8;hb=d5ee5cdea79ab756f4a7908c19b8232d57a1a3a8;hp=0000000000000000000000000000000000000000;hpb=2824afabef34b9bd0aa41644c7da1858d1532da8;p=ashd.git

diff --git a/python/ashd/perf.py b/python/ashd/perf.py
new file mode 100644
index 0000000..d3d3cb7
--- /dev/null
+++ b/python/ashd/perf.py
@@ -0,0 +1,57 @@
+try:
+    import pdm.perf
+except:
+    pdm = None
+
+reqstat = {}
+
+if pdm:
+    statistics = pdm.perf.staticdir()
+    statistics["req"] = pdm.perf.valueattr(reqstat)
+    requests = pdm.perf.eventobj()
+
+    class reqstart(pdm.perf.startevent):
+        def __init__(self, env):
+            super(reqstart, self).__init__()
+            self.method = env.get("REQUEST_METHOD")
+            self.uri = env.get("REQUEST_URI")
+            self.host = env.get("HTTP_HOST")
+
+    class reqfinish(pdm.perf.finishevent):
+        def __init__(self, start, aborted, status):
+            super(reqfinish, self).__init__(start, aborted)
+            self.status = status
+
+class request(object):
+    def __init__(self, env):
+        self.resp = None
+        if pdm:
+            self.startev = reqstart(env)
+            requests.notify(self.startev)
+
+    def response(self, resp):
+        self.resp = resp
+
+    def finish(self, aborted):
+        key = None
+        status = None
+        try:
+            if len(self.resp) > 0:
+                status = self.resp[0]
+                p = status.find(" ")
+                if p < 0:
+                    key = status
+                else:
+                    key = status[:p]
+        except:
+            pass
+        reqstat[key] = reqstat.setdefault(key, 0) + 1
+        if pdm:
+            requests.notify(reqfinish(self.startev, aborted, status))
+
+    def __enter__(self):
+        return self
+    
+    def __exit__(self, *excinfo):
+        self.finish(bool(excinfo[0]))
+        return False