1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
|
import json
import os.path
import time
from wptserve.pipes import template
def main(request, response):
response.headers.set(b"Content-Type", b"text/html")
response.headers.set(b"Cache-Control", b"no-store")
uuid = request.GET[b"uuid"]
# The lock is needed because the server receives concurrent requests to the
# same `uuid` e.g. due to `race-network-and-fetch-handler`.
with request.server.stash.lock:
request_count = request.server.stash.take(uuid)
if request_count is None:
request_count = {"prefetch": 0, "nonPrefetch": 0}
if b"check" in request.GET:
response.content = json.dumps(request_count)
return
prefetch = request.headers.get(
"Sec-Purpose", b"").decode("utf-8").startswith("prefetch")
request_count["prefetch" if prefetch else "nonPrefetch"] += 1
request.server.stash.put(uuid, request_count)
# This delay is after the `stash.put` above so that the request is counted
# upon received (i.e. we don't have to wait for the delay completion).
if b"delay" in request.GET:
time.sleep(float(request.GET[b"delay"]) / 1E3)
if b"location" in request.GET:
response.status = 302
response.headers.set(b"Location", request.GET[b"location"])
return
response.content = template(
request,
open(os.path.join(os.path.dirname(__file__), "executor.sub.html"), "rb").read())
|
