blob: 7d0fa36019ef58c8c7764767f267628cd05bc285 (
plain)
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
|
def main(request, response):
# `token` should be a unique UUID request parameter for the duration of this
# request. It will get stored in the server stash and will be used later in
# a query request.
# `query` should be a request parameter indicating the request would like
# to know how many times the server has seen the request (with the
# same token).
token = request.GET.first(b"token", None)
is_query = request.GET.first(b"query", None) is not None
with request.server.stash.lock:
value = request.server.stash.take(token)
count = 0
if value is not None:
count = int(value)
if is_query:
request.server.stash.put(token, count)
else:
count += 1
request.server.stash.put(token, count)
headers = []
if is_query:
headers = [(b"Count", count)]
return (200, headers, b"")
|
