chore: make server streaming transformer safer

Change-Id: I0fed7ea426a4b775a7fe5e661ed8e67f1faa53eb
googleapis · Nov 5, 2024 · d6b9b1f · d6b9b1f
1 parent 6685aa3
commit d6b9b1f
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/...main/java/com/google/cloud/bigtable/data/v2/stub/TransformingServerStreamingCallable.java b/...main/java/com/google/cloud/bigtable/data/v2/stub/TransformingServerStreamingCallable.java
@@ -46,24 +46,24 @@ public void call(
 
     inner.call(
         innerReq,
-        new ResponseObserver<InnerRespT>() {
+        new SafeResponseObserver<InnerRespT>(outerObserver) {
           @Override
-          public void onStart(StreamController streamController) {
+          public void onStartImpl(StreamController streamController) {
             outerObserver.onStart(streamController);
           }
 
           @Override
-          public void onResponse(InnerRespT innerResp) {
+          public void onResponseImpl(InnerRespT innerResp) {
             outerObserver.onResponse(responseTransformer.apply(innerResp));
           }
 
           @Override
-          public void onError(Throwable throwable) {
+          public void onErrorImpl(Throwable throwable) {
             outerObserver.onError(throwable);
           }
 
           @Override
-          public void onComplete() {
+          public void onCompleteImpl() {
             outerObserver.onComplete();
           }
         },