report error to sentry in stream interceptor

skit-ai · Aug 3, 2021 · c999984 · c999984
1 parent 0805d50
commit c999984
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/surveillance/sentry.go b/surveillance/sentry.go
@@ -233,6 +233,12 @@ func (wrapper *Sentry) StreamServerInterceptor() grpc.StreamServerInterceptor {
 
 		wrapped := sentryWrapper.WrapServerStream(stream)
 		wrapped.WrappedContext = ctx
-		return handler(srv, wrapped)
+		err := handler(srv, wrapped)
+
+		if opts.ReportOn(err) {
+			hub.CaptureException(err)
+		}
+
+		return err
 	}
 }