Make async gRPC less noisy (#2507)

Author: jyggen

sentry_sdk/integrations/grpc/aio/server.py

@@ -13,7 +13,7 @@
 try:
     import grpc
     from grpc import HandlerCallDetails, RpcMethodHandler
-    from grpc.aio import ServicerContext
+    from grpc.aio import AbortError, ServicerContext
 except ImportError:
     raise DidNotEnable("grpcio is not installed")
 
@@ -52,6 +52,8 @@ async def wrapped(request, context):
                 with hub.start_transaction(transaction=transaction):
                     try:
                         return await handler.unary_unary(request, context)
+                    except AbortError:
+                        raise
                     except Exception as exc:
                         event, hint = event_from_exception(
                             exc,

tests/integrations/grpc/test_grpc_aio.py

@@ -124,6 +124,21 @@ async def test_grpc_server_exception(capture_events, grpc_server):
     assert event["exception"]["values"][0]["mechanism"]["type"] == "grpc"
 
 
+@pytest.mark.asyncio
+async def test_grpc_server_abort(capture_events, grpc_server):
+    events = capture_events()
+
+    async with grpc.aio.insecure_channel("localhost:{}".format(AIO_PORT)) as channel:
+        stub = gRPCTestServiceStub(channel)
+        try:
+            await stub.TestServe(gRPCTestMessage(text="abort"))
+            raise AssertionError()
+        except Exception:
+            pass
+
+    assert len(events) == 1
+
+
 @pytest.mark.asyncio
 async def test_grpc_client_starts_span(
     grpc_server, sentry_init, capture_events_forksafe
@@ -218,6 +233,9 @@ async def TestServe(cls, request, context):  # noqa: N802
         if request.text == "exception":
             raise cls.TestException()
 
+        if request.text == "abort":
+            await context.abort(grpc.StatusCode.ABORTED)
+
         return gRPCTestMessage(text=request.text)
 
     @classmethod