try-catch

Signed-off-by: Cassandra Coyle <[email protected]>

Author: cicoyle

client/src/main/java/com/microsoft/durabletask/DurableTaskGrpcWorker.java

@@ -136,7 +136,17 @@ public void startAndBlock() {
                                     .setCustomStatus(StringValue.of(taskOrchestratorResult.getCustomStatus()))
                                     .build();
 
-                            this.sidecarClient.completeOrchestratorTask(response);
+                            try {
+                                this.sidecarClient.completeOrchestratorTask(response);
+                            } catch (StatusRuntimeException e) {
+                                if (e.getStatus().getCode() == Status.Code.UNAVAILABLE) {
+                                    logger.log(Level.WARNING, "The sidecar at address {0} is unavailable while completing the orchestrator task.", this.getSidecarAddress());
+                                } else if (e.getStatus().getCode() == Status.Code.CANCELLED) {
+                                    logger.log(Level.WARNING, "Durable Task worker has disconnected from {0} while completing the orchestrator task.", this.getSidecarAddress());
+                                } else {
+                                    logger.log(Level.WARNING, "Unexpected failure completing the orchestrator task at {0}.", this.getSidecarAddress());
+                                }
+                            }
                         });
                     } else if (requestType == RequestCase.ACTIVITYREQUEST) {
                         ActivityRequest activityRequest = workItem.getActivityRequest();
@@ -169,7 +179,17 @@ public void startAndBlock() {
                                 responseBuilder.setFailureDetails(failureDetails);
                             }
 
-                            this.sidecarClient.completeActivityTask(responseBuilder.build());
+                            try {
+                                this.sidecarClient.completeActivityTask(responseBuilder.build());
+                            } catch (StatusRuntimeException e) {
+                                if (e.getStatus().getCode() == Status.Code.UNAVAILABLE) {
+                                    logger.log(Level.WARNING, "The sidecar at address {0} is unavailable while completing the activity task.", this.getSidecarAddress());
+                                } else if (e.getStatus().getCode() == Status.Code.CANCELLED) {
+                                    logger.log(Level.WARNING, "Durable Task worker has disconnected from {0} while completing the activity task.", this.getSidecarAddress());
+                                } else {
+                                    logger.log(Level.WARNING, "Unexpected failure completing the activity task at {0}.", this.getSidecarAddress());
+                                }
+                            }
                         });
                     } else {
                         logger.log(Level.WARNING, "Received and dropped an unknown '{0}' work-item from the sidecar.", requestType);