danyadjan/db_kyc_project
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Wrap energy decr in transaction
Some checks are pending
CI/CD Pipeline / lint-sast (push) Waiting to run
Details
CI/CD Pipeline / deploy (push) Blocked by required conditions
Details

Browse Source
This commit is contained in:
Michail Kostochka 2025-06-09 13:56:18 +03:00
parent 42ab9eb8f4
commit 2b0322d962
1 changed files with 12 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
batcher/app/src/domain/click/repos/pg.py
View File

@ -18,7 +18,7 @@ async def store(conn: Connection, click: Click) -> int:
async def delete_user_info(conn: Connection, user_id: int): async def delete_user_info(conn: Connection, user_id: int):
async with conn.transaction(): async with conn.transaction():
await conn.execute('DELETE FROM clicks WHERE user_id=$1', user_id) await conn.execute('DELETE FROM clicks WHERE user_id=$1', user_id)
await conn.execute('DELTE FROM users WHERE id=$1', user_id) await conn.execute('DELETE FROM users WHERE id=$1', user_id)
async def get_period_sum(conn: Connection, user_id: int, period: int) -> decimal.Decimal: async def get_period_sum(conn: Connection, user_id: int, period: int) -> decimal.Decimal:
@ -65,6 +65,7 @@ async def get_energy(conn: Connection, user_id: int) -> int:
async def decr_energy(conn: Connection, user_id: int, amount: int) -> Tuple[int, int]: async def decr_energy(conn: Connection, user_id: int, amount: int) -> Tuple[int, int]:
with conn.transaction('serializable'):
new_energy, spent = await conn.fetchrow(''' new_energy, spent = await conn.fetchrow('''
WITH energy_cte AS ( WITH energy_cte AS (
SELECT energy, (CASE WHEN energy < $2 THEN energy ELSE $2 END) AS delta FROM users WHERE id=$1 SELECT energy, (CASE WHEN energy < $2 THEN energy ELSE $2 END) AS delta FROM users WHERE id=$1