ensure that update runs over unique ids

Author: Reskov

frameworks/Python/aiohttp/app/views.py

@@ -1,11 +1,12 @@
 from operator import attrgetter, itemgetter
 from pathlib import Path
-from random import randint
+from random import randint, sample
 
 import jinja2
 import orjson
 from aiohttp.web import Response
 from sqlalchemy import select
+from sqlalchemy.orm.attributes import flag_modified
 
 from .models import sa_fortunes, sa_worlds, Fortune, World
 
@@ -77,7 +78,7 @@ async def multiple_database_queries_orm(request):
     result = []
     async with request.app['db_session']() as sess:
         for id_ in ids:
-            num = await sess.scalar(READ_SELECT_ORM.filter_by(id=id_))
+            num = await sess.scalar(READ_SELECT_ORM.where(World.id == id_))
             result.append({'id': id_, 'randomNumber': num})
     return json_response(result)
 
@@ -131,30 +132,38 @@ async def updates(request):
     Test 5 ORM
     """
     num_queries = get_num_queries(request)
-    updates = [(randint(1, 10000), randint(1, 10000)) for _ in range(num_queries)]
-    updates.sort()
-    worlds = [{'id': row_id, 'randomNumber': number} for row_id, number in updates]
+
+    ids = sample(range(1, 10000 + 1), num_queries)
+    ids.sort()
+    worlds = []
 
     async with request.app['db_session'].begin() as sess:
-        for id_, number in updates:
-            world = await sess.get(World, id_, populate_existing=True)
-            world.randomnumber = number
+        for row_id in ids:
+            random_number =  randint(1, 10000)
+            world = await sess.get(World, row_id, populate_existing=True)
+            world.randomnumber = random_number
+            # force sqlalchemy to UPDATE entry even if the value has not changed
+            # doesn't make sense in a real application, added only for pass `tfb verify`
+            flag_modified(world, "randomnumber")
+            worlds.append({'id': row_id, 'randomNumber': random_number})
+
     return json_response(worlds)
 
 async def updates_raw(request):
     """
     Test 5 RAW
     """
     num_queries = get_num_queries(request)
-    updates = [(randint(1, 10000), randint(1, 10000)) for _ in range(num_queries)]
-    updates.sort()
+    ids = sample(range(1, 10000 + 1), num_queries)
+    ids.sort()
+    updates = [(row_id, randint(1, 10000)) for row_id in ids]
     worlds = [{'id': row_id, 'randomNumber': number} for row_id, number in updates]
 
     async with request.app['pg'].acquire() as conn:
         stmt = await conn.prepare(READ_ROW_SQL)
-        for id_, _ in updates:
+        for row_id in ids:
             # the result of this is the int previous random number which we don't actually use
-            await stmt.fetchval(id_)
+            await stmt.fetchval(row_id)
         await conn.executemany(WRITE_ROW_SQL, updates)
 
     return json_response(worlds)

frameworks/Python/aiohttp/requirements.txt

@@ -2,7 +2,6 @@ aiohttp==3.11.16
 asyncpg==0.30.0
 gunicorn==23.0.0
 jinja2==3.1.6
-psycopg2==2.9.10
 SQLAlchemy==2.0.39
 orjson==3.10.16
 uvloop==0.21.0