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