11246:: addressing the review comments.

Author: vimanikag

core/src/main/java/io/grpc/internal/ServerImpl.java

@@ -56,6 +56,7 @@
 import io.grpc.ServerServiceDefinition;
 import io.grpc.ServerTransportFilter;
 import io.grpc.Status;
+import io.grpc.StatusException;
 import io.grpc.StatusRuntimeException;
 import io.perfmark.Link;
 import io.perfmark.PerfMark;
@@ -813,10 +814,14 @@ private void internalClose(Throwable t) {
       // TODO(ejona86): this is not thread-safe :)
       String description = "Application error processing RPC";
       Status statusToPropagate = Status.UNKNOWN.withDescription(description).withCause(t);
+      Status extractedStatus = null;
       if (t instanceof StatusRuntimeException) {
-        if (((StatusRuntimeException) t).getStatus().getCode() == Status.Code.RESOURCE_EXHAUSTED) {
-          statusToPropagate = ((StatusRuntimeException) t).getStatus().withCause(t);
-        }
+        extractedStatus = ((StatusRuntimeException) t).getStatus();
+      }  else if (t instanceof StatusException) {
+        extractedStatus = ((StatusException) t).getStatus();
+      }
+      if (extractedStatus != null && extractedStatus.getCode() == Status.Code.RESOURCE_EXHAUSTED) {
+        statusToPropagate = extractedStatus.withCause(t);
       }
       stream.close(statusToPropagate, new Metadata());
     }