From 4211fd8525bfe5e45c323a9389df1ec5f54aab8d Mon Sep 17 00:00:00 2001 From: Itamar Turner-Trauring Date: Mon, 27 Mar 2023 13:41:30 -0400 Subject: [PATCH] Revert to old code. --- src/allmydata/storage/http_client.py | 46 ++++++++++++++++------------ 1 file changed, 27 insertions(+), 19 deletions(-) diff --git a/src/allmydata/storage/http_client.py b/src/allmydata/storage/http_client.py index a61f94708..44ba64363 100644 --- a/src/allmydata/storage/http_client.py +++ b/src/allmydata/storage/http_client.py @@ -355,7 +355,6 @@ class StorageClient(object): ) return headers - @inlineCallbacks def request( self, method, @@ -378,26 +377,35 @@ class StorageClient(object): Default timeout is 60 seconds. """ - with start_action( - action_type="allmydata:storage:http-client:request", - method=method, - url=str(url), - ) as ctx: - headers = self._get_headers(headers) + headers = self._get_headers(headers) - # Add secrets: - for secret, value in [ - (Secrets.LEASE_RENEW, lease_renew_secret), - (Secrets.LEASE_CANCEL, lease_cancel_secret), - (Secrets.UPLOAD, upload_secret), - (Secrets.WRITE_ENABLER, write_enabler_secret), - ]: - if value is None: - continue - headers.addRawHeader( - "X-Tahoe-Authorization", - b"%s %s" % (secret.value.encode("ascii"), b64encode(value).strip()), + # Add secrets: + for secret, value in [ + (Secrets.LEASE_RENEW, lease_renew_secret), + (Secrets.LEASE_CANCEL, lease_cancel_secret), + (Secrets.UPLOAD, upload_secret), + (Secrets.WRITE_ENABLER, write_enabler_secret), + ]: + if value is None: + continue + headers.addRawHeader( + "X-Tahoe-Authorization", + b"%s %s" % (secret.value.encode("ascii"), b64encode(value).strip()), + ) + + # Note we can accept CBOR: + headers.addRawHeader("Accept", CBOR_MIME_TYPE) + + # If there's a request message, serialize it and set the Content-Type + # header: + if message_to_serialize is not None: + if "data" in kwargs: + raise TypeError( + "Can't use both `message_to_serialize` and `data` " + "as keyword arguments at the same time" ) + kwargs["data"] = dumps(message_to_serialize) + headers.addRawHeader("Content-Type", CBOR_MIME_TYPE) return self._treq.request( method, url, headers=headers, timeout=timeout, **kwargs