Matrix-Mirrors/anonymousland-synapse
1
0
Fork 0
mirror of https://git.anonymousland.org/anonymousland/synapse.git synced 2024-10-01 11:49:51 -04:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #639 from matrix-org/erikj/as_user_update_batch

Browse Source 
Update users table in a batched manner
This commit is contained in:
Erik Johnston 2016-03-10 15:38:00 +00:00
commit 703826886c
1 changed files with 13 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
synapse/storage/schema/delta/30/as_users.py
View File

@ -52,12 +52,17 @@ def run_upgrade(cur, database_engine, config, *args, **kwargs):
" service (IDs %s and %s); assigning arbitrarily to %s" % " service (IDs %s and %s); assigning arbitrarily to %s" %
(user_id, owned[user_id], appservice.id, owned[user_id]) (user_id, owned[user_id], appservice.id, owned[user_id])
) )
owned[user_id] = appservice.id owned.setdefault(appservice.id, []).append(user_id)
for user_id, as_id in owned.items(): for as_id, user_ids in owned.items():
n = 100
user_chunks = (user_ids[i:i + 100] for i in xrange(0, len(user_ids), n))
for chunk in user_chunks:
cur.execute( cur.execute(
database_engine.convert_param_style( database_engine.convert_param_style(
"UPDATE users SET appservice_id = ? WHERE name = ?" "UPDATE users SET appservice_id = ? WHERE name IN (%s)" % (
), ",".join("?" for _ in chunk),
(as_id, user_id) )
),
[as_id] + chunk
) )