Merge pull request #3153 from derekpierre/middleware-timeouts

Stop gap measure for limiting operational timeouts
pull/3157/head
Derek Pierre 2023-06-22 13:22:37 -04:00 committed by GitHub
commit 43b5792a3c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 8 additions and 5 deletions

View File

@ -0,0 +1 @@
Increase default timeout for ``/reencrypt`` requests to limit timeouts when multiple retrieval kits are included in a single request.

View File

@ -47,7 +47,9 @@ class ThresholdDecryptionClient(ThresholdAccessControlClient):
node_or_sprout.mature()
response = (
self._learner.network_middleware.get_encrypted_decryption_share(
node_or_sprout, bytes(encrypted_request)
ursula=node_or_sprout,
decryption_request_bytes=bytes(encrypted_request),
timeout=timeout,
)
)
if response.status_code == 200:

View File

@ -253,23 +253,23 @@ class RestMiddleware:
)
return response
def reencrypt(self, ursula: 'Ursula', reencryption_request_bytes: bytes):
def reencrypt(self, ursula: "Ursula", reencryption_request_bytes: bytes, timeout=8):
response = self.client.post(
node_or_sprout=ursula,
path=f"reencrypt",
data=reencryption_request_bytes,
timeout=2
timeout=timeout,
)
return response
def get_encrypted_decryption_share(
self, ursula: "Ursula", decryption_request_bytes: bytes
self, ursula: "Ursula", decryption_request_bytes: bytes, timeout=8
):
response = self.client.post(
node_or_sprout=ursula,
path=f"decrypt",
data=decryption_request_bytes,
timeout=8
timeout=timeout,
)
return response