Skip to content
This repository was archived by the owner on May 8, 2026. It is now read-only.

Commit 4baa72e

Browse files
author
Patrick Wrobel
committed
fix: use StatusProto.fromThrowable for ApiException in test-proxy
This should fix the test proxy from prefixing error messages with "io.grpc.StatusRuntimeException: "
1 parent e0b2d6c commit 4baa72e

1 file changed

Lines changed: 8 additions & 58 deletions

File tree

test-proxy/src/main/java/com/google/cloud/bigtable/testproxy/CbtTestProxy.java

Lines changed: 8 additions & 58 deletions
Original file line numberDiff line numberDiff line change
@@ -284,21 +284,13 @@ public void mutateRow(
284284
return;
285285
}
286286

287-
// TODO(developer): evaluate if we want to manually unpack the proto into a model, instead of
288-
// using fromProto. Same for the other methods.
289287
RowMutation mutation = RowMutation.fromProto(request.getRequest());
290288
try {
291289
// This response is empty.
292290
client.dataClient().mutateRow(mutation);
293291
} catch (ApiException e) {
294292
responseObserver.onNext(
295-
MutateRowResult.newBuilder()
296-
.setStatus(
297-
com.google.rpc.Status.newBuilder()
298-
.setCode(e.getStatusCode().getCode().ordinal())
299-
.setMessage(e.getMessage())
300-
.build())
301-
.build());
293+
MutateRowResult.newBuilder().setStatus(StatusProto.fromThrowable(e)).build());
302294
responseObserver.onCompleted();
303295
return;
304296
} catch (StatusRuntimeException e) {
@@ -351,13 +343,7 @@ public void bulkMutateRows(
351343
return;
352344
} catch (ApiException e) {
353345
responseObserver.onNext(
354-
MutateRowsResult.newBuilder()
355-
.setStatus(
356-
com.google.rpc.Status.newBuilder()
357-
.setCode(e.getStatusCode().getCode().ordinal())
358-
.setMessage(e.getMessage())
359-
.build())
360-
.build());
346+
MutateRowsResult.newBuilder().setStatus(StatusProto.fromThrowable(e)).build());
361347
responseObserver.onCompleted();
362348
return;
363349
} catch (StatusRuntimeException e) {
@@ -410,13 +396,7 @@ public void readRow(ReadRowRequest request, StreamObserver<RowResult> responseOb
410396
}
411397
} catch (ApiException e) {
412398
responseObserver.onNext(
413-
RowResult.newBuilder()
414-
.setStatus(
415-
com.google.rpc.Status.newBuilder()
416-
.setCode(e.getStatusCode().getCode().ordinal())
417-
.setMessage(e.getMessage())
418-
.build())
419-
.build());
399+
RowResult.newBuilder().setStatus(StatusProto.fromThrowable(e)).build());
420400
responseObserver.onCompleted();
421401
return;
422402
} catch (StatusRuntimeException e) {
@@ -461,13 +441,7 @@ public void readRows(ReadRowsRequest request, StreamObserver<RowsResult> respons
461441
resultBuilder.setStatus(com.google.rpc.Status.getDefaultInstance()).build());
462442
} catch (ApiException e) {
463443
responseObserver.onNext(
464-
RowsResult.newBuilder()
465-
.setStatus(
466-
com.google.rpc.Status.newBuilder()
467-
.setCode(e.getStatusCode().getCode().ordinal())
468-
.setMessage(e.getMessage())
469-
.build())
470-
.build());
444+
RowsResult.newBuilder().setStatus(StatusProto.fromThrowable(e)).build());
471445
responseObserver.onCompleted();
472446
return;
473447
} catch (StatusRuntimeException e) {
@@ -585,13 +559,7 @@ public void sampleRowKeys(
585559
keyOffsets = client.dataClient().sampleRowKeys(tableId);
586560
} catch (ApiException e) {
587561
responseObserver.onNext(
588-
SampleRowKeysResult.newBuilder()
589-
.setStatus(
590-
com.google.rpc.Status.newBuilder()
591-
.setCode(e.getStatusCode().getCode().ordinal())
592-
.setMessage(e.getMessage())
593-
.build())
594-
.build());
562+
SampleRowKeysResult.newBuilder().setStatus(StatusProto.fromThrowable(e)).build());
595563
responseObserver.onCompleted();
596564
return;
597565
} catch (StatusRuntimeException e) {
@@ -631,13 +599,7 @@ public void checkAndMutateRow(
631599
matched = client.dataClient().checkAndMutateRow(mutation);
632600
} catch (ApiException e) {
633601
responseObserver.onNext(
634-
CheckAndMutateRowResult.newBuilder()
635-
.setStatus(
636-
com.google.rpc.Status.newBuilder()
637-
.setCode(e.getStatusCode().getCode().ordinal())
638-
.setMessage(e.getMessage())
639-
.build())
640-
.build());
602+
CheckAndMutateRowResult.newBuilder().setStatus(StatusProto.fromThrowable(e)).build());
641603
responseObserver.onCompleted();
642604
return;
643605
} catch (StatusRuntimeException e) {
@@ -682,13 +644,7 @@ public void readModifyWriteRow(
682644
}
683645
} catch (ApiException e) {
684646
responseObserver.onNext(
685-
RowResult.newBuilder()
686-
.setStatus(
687-
com.google.rpc.Status.newBuilder()
688-
.setCode(e.getStatusCode().getCode().ordinal())
689-
.setMessage(e.getMessage())
690-
.build())
691-
.build());
647+
RowResult.newBuilder().setStatus(StatusProto.fromThrowable(e)).build());
692648
responseObserver.onCompleted();
693649
return;
694650
} catch (StatusRuntimeException e) {
@@ -745,13 +701,7 @@ public void executeQuery(
745701
return;
746702
} catch (ApiException e) {
747703
responseObserver.onNext(
748-
ExecuteQueryResult.newBuilder()
749-
.setStatus(
750-
com.google.rpc.Status.newBuilder()
751-
.setCode(e.getStatusCode().getCode().ordinal())
752-
.setMessage(e.getMessage())
753-
.build())
754-
.build());
704+
ExecuteQueryResult.newBuilder().setStatus(StatusProto.fromThrowable(e)).build());
755705
responseObserver.onCompleted();
756706
return;
757707
} catch (StatusRuntimeException e) {

0 commit comments

Comments
 (0)