summaryrefslogtreecommitdiffstats
path: root/testing/web-platform/tests/lifecycle/resources/beacon.py
blob: 09915ffbcf045ad65d69d564b72736dc5a20365c (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"")