fix: Status.RESOURCE_EXHAUSTED wasn't handled

timemates · Jul 26, 2023 · b42dc46 · b42dc46
1 parent eb450b6
commit b42dc46
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/grpc-engine/src/main/kotlin/io/timemates/api/grpc/GrpcTimeMatesRequestsEngine.kt b/grpc-engine/src/main/kotlin/io/timemates/api/grpc/GrpcTimeMatesRequestsEngine.kt
@@ -33,6 +33,7 @@ import io.timemates.sdk.common.exceptions.InternalServerError
 import io.timemates.sdk.common.exceptions.InvalidArgumentException
 import io.timemates.sdk.common.exceptions.NotFoundException
 import io.timemates.sdk.common.exceptions.PermissionDeniedException
+import io.timemates.sdk.common.exceptions.TooManyRequestsException
 import io.timemates.sdk.common.exceptions.UnauthorizedException
 import io.timemates.sdk.common.exceptions.UnavailableException
 import io.timemates.sdk.common.exceptions.UnsupportedException
@@ -383,6 +384,9 @@ public class GrpcTimeMatesRequestsEngine(
         val message = exception.message ?: NO_MESSAGE
 
         when (status) {
+            Status.RESOURCE_EXHAUSTED ->
+                TooManyRequestsException("Too many requests.", exception)
+
             Status.INVALID_ARGUMENT, Status.FAILED_PRECONDITION ->
                 InvalidArgumentException(message, exception)
 

diff --git a/sdk/src/commonMain/kotlin/io/timemates/sdk/common/exceptions/TooManyRequestsException.kt b/sdk/src/commonMain/kotlin/io/timemates/sdk/common/exceptions/TooManyRequestsException.kt
@@ -9,4 +9,5 @@ package io.timemates.sdk.common.exceptions
  */
 public data class TooManyRequestsException(
     override val message: String,
+    override val cause: Throwable?,
 ) : TimeMatesException(message)