| 1 | import collections |
| 2 | try: |
| 3 | import pdm.perf |
| 4 | except: |
| 5 | pdm = None |
| 6 | try: |
| 7 | import time |
| 8 | clock_thread = time.CLOCK_THREAD_CPUTIME_ID |
| 9 | except: |
| 10 | clock_thread = None |
| 11 | |
| 12 | reqstat = {} |
| 13 | |
| 14 | if pdm: |
| 15 | statistics = pdm.perf.staticdir() |
| 16 | statistics["req"] = pdm.perf.valueattr(reqstat) |
| 17 | requests = pdm.perf.eventobj() |
| 18 | |
| 19 | class reqstart(pdm.perf.startevent): |
| 20 | def __init__(self, env): |
| 21 | super().__init__() |
| 22 | self.method = env.get("REQUEST_METHOD") |
| 23 | self.uri = env.get("REQUEST_URI") |
| 24 | self.host = env.get("HTTP_HOST") |
| 25 | self.script_uri = env.get("SCRIPT_NAME") |
| 26 | self.script_path = env.get("SCRIPT_FILENAME") |
| 27 | self.pathinfo = env.get("PATH_INFO") |
| 28 | self.querystring = env.get("QUERY_STRING") |
| 29 | self.remoteaddr = env.get("REMOTE_ADDR") |
| 30 | self.remoteport = env.get("REMOTE_PORT") |
| 31 | self.scheme = env.get("wsgi.url_scheme") |
| 32 | if clock_thread is not None: |
| 33 | self.icpu = time.clock_gettime(clock_thread) |
| 34 | |
| 35 | class reqfinish(pdm.perf.finishevent): |
| 36 | def __init__(self, start, aborted, status): |
| 37 | super().__init__(start, aborted) |
| 38 | self.status = status |
| 39 | self.cputime = 0 |
| 40 | if clock_thread is not None: |
| 41 | self.cputime = time.clock_gettime(clock_thread) - start.icpu |
| 42 | |
| 43 | class request(object): |
| 44 | def __init__(self, env): |
| 45 | self.resp = None |
| 46 | if pdm: |
| 47 | self.startev = reqstart(env) |
| 48 | requests.notify(self.startev) |
| 49 | |
| 50 | def response(self, resp): |
| 51 | self.resp = resp |
| 52 | |
| 53 | def finish(self, aborted): |
| 54 | key = None |
| 55 | status = None |
| 56 | try: |
| 57 | if len(self.resp) > 0: |
| 58 | status = self.resp[0] |
| 59 | if isinstance(status, collections.ByteString): |
| 60 | status = status.decode("latin-1") |
| 61 | else: |
| 62 | status = str(status) |
| 63 | p = status.find(" ") |
| 64 | if p < 0: |
| 65 | key = status |
| 66 | else: |
| 67 | key = status[:p] |
| 68 | except: |
| 69 | pass |
| 70 | reqstat[key] = reqstat.setdefault(key, 0) + 1 |
| 71 | if pdm: |
| 72 | requests.notify(reqfinish(self.startev, aborted, status)) |
| 73 | |
| 74 | def __enter__(self): |
| 75 | return self |
| 76 | |
| 77 | def __exit__(self, *excinfo): |
| 78 | self.finish(bool(excinfo[0])) |
| 79 | return False |