@@ -53,12 +53,11 @@ def load_fortunes_template():
5353app .on_start += setup_db
5454app .on_stop += shutdown_db
5555
56- def get_num_queries (request ):
56+ def get_num_queries (queries ):
5757 try :
58- value = request .query .get ('queries' )
59- if value is None :
58+ if queries is None :
6059 return 1
61- query_count = int (value [ 0 ] )
60+ query_count = int (queries )
6261 except (KeyError , IndexError , ValueError ):
6362 return 1
6463 return min (max (query_count , 1 ), 500 )
@@ -78,9 +77,9 @@ async def single_db_query_test(request):
7877 number = await cursor .fetchone ()
7978 return bs .json ({'id' : row_id , 'randomNumber' : number [1 ]})
8079
81- @bs .get ('/queries' )
82- async def multiple_db_queries_test (request ):
83- num_queries = get_num_queries (request )
80+ @bs .get ('/queries/{queries} ' )
81+ async def multiple_db_queries_test (request , queries ):
82+ num_queries = get_num_queries (queries )
8483 row_ids = random .sample (range (1 , 10000 ), num_queries )
8584 worlds = []
8685 async with db_pool .connection () as db_conn :
@@ -102,9 +101,9 @@ async def fortunes_test(request):
102101 data = fortune_template .render (fortunes = fortunes )
103102 return bs .html (data )
104103
105- @bs .get ('/updates' )
106- async def db_updates_test (request ):
107- num_queries = get_num_queries (request )
104+ @bs .get ('/updates/{queries} ' )
105+ async def db_updates_test (request , queries ):
106+ num_queries = get_num_queries (queries )
108107 updates = list (zip (
109108 random .sample (range (1 , 10000 ), num_queries ),
110109 sorted (random .sample (range (1 , 10000 ), num_queries ))
@@ -123,6 +122,7 @@ async def plaintext_test(request):
123122 return bs .Response (200 , content = bs .Content (b"text/plain" , b'Hello, World!' ))
124123
125124if platform .python_implementation () == 'PyPy' :
125+ import logging
126126 from socketify import ASGI
127127 workers = int (multiprocessing .cpu_count ())
128128 if os .environ .get ('TRAVIS' ) == 'true' :
0 commit comments