aljaz/nutshell
1
0
Fork 0
mirror of https://github.com/aljazceru/nutshell.git synced 2025-12-20 18:44:20 +01:00
Code Issues Packages Projects Releases Wiki Activity

Batch update Y columns in proofs_used and proofs_pending tables (#475)

Browse Source 
* mint: migration 16 batch update

* migrate: only if data exists
This commit is contained in:
callebtc
2024-03-16 10:56:51 +01:00
committed by GitHub
parent 7ecf14f87b
commit e7b1e0c0ed
1 changed files with 46 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

53
cashu/mint/migrations.py
View File

@@ -555,15 +555,54 @@ async def m016_recompute_Y_with_new_h2c(db: Database):
)
proofs_pending = [Proof(**r) for r in rows]
# overwrite the old Y columns with the new Y
# Prepare data for batch update
proofs_used_data = [(proof.Y, proof.secret) for proof in proofs_used]
proofs_pending_data = [(proof.Y, proof.secret) for proof in proofs_pending]
# Perform batch update in a single transaction
async with db.connect() as conn:
for proof in proofs_used:
if len(proofs_used_data):
# For proofs_used
await conn.execute(
f"UPDATE {table_with_schema(db, 'proofs_used')} SET Y = '{proof.Y}'"
f" WHERE secret = '{proof.secret}'"
"CREATE TABLE IF NOT EXISTS tmp_proofs_used (Y TEXT, secret TEXT)"
)
values_placeholder = ", ".join(
f"('{y}', '{secret}')" for y, secret in proofs_used_data
)
for proof in proofs_pending:
await conn.execute(
f"UPDATE {table_with_schema(db, 'proofs_pending')} SET Y = '{proof.Y}'"
f" WHERE secret = '{proof.secret}'"
f"INSERT INTO tmp_proofs_used (Y, secret) VALUES {values_placeholder}",
)
await conn.execute(
f"""
UPDATE {table_with_schema(db, 'proofs_used')}
SET Y = tmp_proofs_used.Y
FROM tmp_proofs_used
WHERE {table_with_schema(db, 'proofs_used')}.secret = tmp_proofs_used.secret
"""
)
if len(proofs_pending_data):
# For proofs_pending
await conn.execute(
"CREATE TABLE IF NOT EXISTS tmp_proofs_pending (Y TEXT, secret TEXT)"
)
values_placeholder = ", ".join(
f"('{y}', '{secret}')" for y, secret in proofs_pending_data
)
await conn.execute(
f"INSERT INTO tmp_proofs_used (Y, secret) VALUES {values_placeholder}",
)
await conn.execute(
f"""
UPDATE {table_with_schema(db, 'proofs_pending')}
SET Y = tmp_proofs_pending.Y
FROM tmp_proofs_pending
WHERE {table_with_schema(db, 'proofs_pending')}.secret = tmp_proofs_pending.secret
"""
)
async with db.connect() as conn:
if len(proofs_used_data):
await conn.execute("DROP TABLE tmp_proofs_used")
if len(proofs_pending_data):
await conn.execute("DROP TABLE tmp_proofs_pending")