diff options
author | Jürg Billeter <j@bitron.ch> | 2020-06-25 17:46:29 +0200 |
---|---|---|
committer | Jürg Billeter <j@bitron.ch> | 2020-06-25 18:12:09 +0200 |
commit | 771c332530f161da64a696a2757ca460145730de (patch) | |
tree | bb0824219d675a1eeb7ca48ff42f821250d720fa /src/buildstream | |
parent | 43126fddfaa3a122ecba42f1bd94ae69468293cc (diff) | |
download | buildstream-771c332530f161da64a696a2757ca460145730de.tar.gz |
casserver.py: Fix error handling for requests forwarded to casd
Diffstat (limited to 'src/buildstream')
-rw-r--r-- | src/buildstream/_cas/casserver.py | 25 |
1 files changed, 20 insertions, 5 deletions
diff --git a/src/buildstream/_cas/casserver.py b/src/buildstream/_cas/casserver.py index dd822d53b..71d7d9071 100644 --- a/src/buildstream/_cas/casserver.py +++ b/src/buildstream/_cas/casserver.py @@ -230,13 +230,19 @@ class _ByteStreamServicer(bytestream_pb2_grpc.ByteStreamServicer): def Read(self, request, context): self.logger.debug("Reading %s", request.resource_name) - return self.bytestream.Read(request) + try: + return self.bytestream.Read(request) + except grpc.RpcError as err: + context.abort(err.code(), err.details()) def Write(self, request_iterator, context): # Note that we can't easily give more information because the # data is stuck in an iterator that will be consumed if read. self.logger.debug("Writing data") - return self.bytestream.Write(request_iterator) + try: + return self.bytestream.Write(request_iterator) + except grpc.RpcError as err: + context.abort(err.code(), err.details()) class _ContentAddressableStorageServicer(remote_execution_pb2_grpc.ContentAddressableStorageServicer): @@ -248,15 +254,24 @@ class _ContentAddressableStorageServicer(remote_execution_pb2_grpc.ContentAddres def FindMissingBlobs(self, request, context): self.logger.info("Finding '%s'", request.blob_digests) - return self.cas.FindMissingBlobs(request) + try: + return self.cas.FindMissingBlobs(request) + except grpc.RpcError as err: + context.abort(err.code(), err.details()) def BatchReadBlobs(self, request, context): self.logger.info("Reading '%s'", request.digests) - return self.cas.BatchReadBlobs(request) + try: + return self.cas.BatchReadBlobs(request) + except grpc.RpcError as err: + context.abort(err.code(), err.details()) def BatchUpdateBlobs(self, request, context): self.logger.info("Updating: '%s'", [request.digest for request in request.requests]) - return self.cas.BatchUpdateBlobs(request) + try: + return self.cas.BatchUpdateBlobs(request) + except grpc.RpcError as err: + context.abort(err.code(), err.details()) class _CapabilitiesServicer(remote_execution_pb2_grpc.CapabilitiesServicer): |