mia/0x0
1
0
Fork 1
mirror of https://git.0x0.st/mia/0x0.git synced 2025-04-19 14:53:02 +02:00
Code Issues Projects Releases Packages Wiki Activity

Coalesce updates to the database during migration

Browse source 
https://git.0x0.st/mia/0x0/pulls/72#issuecomment-226
This commit is contained in:
Emi Simpson 2022-11-28 16:28:22 -05:00
parent 19d989b696
commit 55ee3740b0
No known key found for this signature in database
GPG key ID: 45E9C6E81BD86E7C
1 changed files with 6 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
migrations/versions/939a08e1d6e5_.py
View file

@ -74,16 +74,17 @@ def upgrade():
File.sha256.in_(unexpired_files) File.sha256.in_(unexpired_files)
) )
) )
updates = [] # We coalesce updates to the database here
for file in files: for file in files:
file_path = storage / file.sha256 file_path = storage / file.sha256
stat = os.stat(file_path) stat = os.stat(file_path)
max_age = get_max_lifespan(stat.st_size) # How long the file is allowed to live, in ms max_age = get_max_lifespan(stat.st_size) # How long the file is allowed to live, in ms
file_birth = stat.st_mtime * 1000 # When the file was created, in ms file_birth = stat.st_mtime * 1000 # When the file was created, in ms
op.execute( updates.append({'id': file.id, 'expiration': int(file_birth + max_age)})
sa.update(UpdatedFile)
.where(UpdatedFile.c.id == file.id) # Apply coalesced updates
.values({'expiration': int(file_birth + max_age)}) session.bulk_update_mappings(File, updates)
) session.commit()
def downgrade(): def downgrade():
op.drop_column('file', 'expiration') op.drop_column('file', 'expiration')