ExternalVendorCode/tahoe-lafs
1
0
Fork 0
mirror of https://github.com/tahoe-lafs/tahoe-lafs.git synced 2025-04-19 00:06:57 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Run in thread pool

Browse Source
This commit is contained in:
Itamar Turner-Trauring 2023-10-16 11:31:48 -04:00
parent daec717903
commit 6e93b12cb5
1 changed files with 15 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
src/allmydata/mutable/publish.py
View File

@ -708,7 +708,8 @@ class Publish(object):
writer.put_salt(salt)
def _encode_segment(self, segnum):
@async_to_deferred
async def _encode_segment(self, segnum):
"""
I encrypt and encode the segment segnum.
"""
@ -730,11 +731,16 @@ class Publish(object):
salt = os.urandom(16)
key = hashutil.ssk_readkey_data_hash(salt, self.readkey)
key = await defer_to_thread(hashutil.ssk_readkey_data_hash, salt, self.readkey)
self._status.set_status("Encrypting")
encryptor = aes.create_encryptor(key)
crypttext = aes.encrypt_data(encryptor, data)
assert len(crypttext) == len(data)
def encrypt():
encryptor = aes.create_encryptor(key)
crypttext = aes.encrypt_data(encryptor, data)
assert len(crypttext) == len(data)
return crypttext
crypttext = await defer_to_thread(encrypt)
now = time.time()
self._status.accumulate_encrypt_time(now - started)
@ -755,14 +761,11 @@ class Publish(object):
piece = piece + b"\x00"*(piece_size - len(piece)) # padding
crypttext_pieces[i] = piece
assert len(piece) == piece_size
d = fec.encode(crypttext_pieces)
def _done_encoding(res):
elapsed = time.time() - started
self._status.accumulate_encode_time(elapsed)
return (res, salt)
d.addCallback(_done_encoding)
return d
res = await fec.encode(crypttext_pieces)
elapsed = time.time() - started
self._status.accumulate_encode_time(elapsed)
return (res, salt)
@async_to_deferred
async def _push_segment(self, encoded_and_salt, segnum):