|
3 | 3 | package com.azure.spring.data.cosmos.exception; |
4 | 4 |
|
5 | 5 | import com.azure.cosmos.CosmosException; |
6 | | -import com.azure.spring.data.cosmos.Constants; |
7 | 6 | import com.azure.spring.data.cosmos.common.CosmosUtils; |
8 | 7 | import com.azure.spring.data.cosmos.core.ResponseDiagnosticsProcessor; |
| 8 | +import org.springframework.http.HttpStatus; |
9 | 9 | import org.springframework.util.ObjectUtils; |
10 | 10 | import reactor.core.Exceptions; |
11 | 11 | import reactor.core.publisher.Mono; |
@@ -43,66 +43,52 @@ public static <T> Mono<T> exceptionHandler(String message, Throwable throwable, |
43 | 43 | CosmosException cosmosException = (CosmosException) unwrappedThrowable; |
44 | 44 | CosmosUtils.fillAndProcessCosmosExceptionDiagnostics(responseDiagnosticsProcessor, cosmosException); |
45 | 45 |
|
46 | | - switch (cosmosException.getStatusCode()) { |
47 | | - case Constants.CosmosExceptionStatusCodes.BADREQUEST: |
48 | | - cosmosAccessException = new CosmosBadRequestException(message, cosmosException); |
49 | | - break; |
50 | | - case Constants.CosmosExceptionStatusCodes.CONFLICT: |
51 | | - cosmosAccessException = new CosmosConflictException(message, cosmosException); |
52 | | - break; |
53 | | - case Constants.CosmosExceptionStatusCodes.FORBIDDEN: |
54 | | - cosmosAccessException = new CosmosForbiddenException(message, cosmosException); |
55 | | - break; |
56 | | - case Constants.CosmosExceptionStatusCodes.GONE: |
57 | | - if (cosmosException.getSubStatusCode() == Constants.CosmosExceptionSubStatusCodes.NAME_CACHE_IS_STALE) { |
58 | | - cosmosAccessException = new CosmosInvalidPartitionException(message, cosmosException); |
59 | | - } else if (cosmosException.getSubStatusCode() == Constants.CosmosExceptionSubStatusCodes.COMPLETING_PARTITION_MIGRATION) { |
60 | | - cosmosAccessException = new CosmosPartitionIsMigratingException(message, cosmosException); |
61 | | - } else if (cosmosException.getSubStatusCode() == Constants.CosmosExceptionSubStatusCodes.PARTITION_KEY_RANGE_GONE) { |
62 | | - cosmosAccessException = new CosmosPartitionKeyRangeGoneException(message, cosmosException); |
63 | | - } else if (cosmosException.getSubStatusCode() == Constants.CosmosExceptionSubStatusCodes.COMPLETING_SPLIT_OR_MERGE) { |
64 | | - cosmosAccessException = new CosmosPartitionKeyRangeIsSplittingException(message, cosmosException); |
65 | | - } else { |
66 | | - cosmosAccessException = new CosmosGoneException(message, cosmosException); |
67 | | - } |
68 | | - break; |
69 | | - case Constants.CosmosExceptionStatusCodes.INTERNAL_SERVER_ERROR: |
70 | | - cosmosAccessException = new CosmosInternalServerErrorException(message, cosmosException); |
71 | | - break; |
72 | | - case Constants.CosmosExceptionStatusCodes.METHOD_NOT_ALLOWED: |
73 | | - cosmosAccessException = new CosmosMethodNotAllowedException(message, cosmosException); |
74 | | - break; |
75 | | - case Constants.CosmosExceptionStatusCodes.NOTFOUND: |
76 | | - cosmosAccessException = new CosmosNotFoundException(message, cosmosException); |
77 | | - break; |
78 | | - case Constants.CosmosExceptionStatusCodes.REQUEST_TIMEOUT: |
79 | | - if (((CosmosException) unwrappedThrowable).getSubStatusCode() == Constants.CosmosExceptionSubStatusCodes.CLIENT_OPERATION_TIMEOUT) { |
80 | | - cosmosAccessException = new CosmosOperationCancelledException(message, cosmosException); |
81 | | - } else { |
82 | | - cosmosAccessException = new CosmosRequestTimeoutException(message, cosmosException); |
83 | | - } |
84 | | - break; |
85 | | - case Constants.CosmosExceptionStatusCodes.PRECONDITION_FAILED: |
86 | | - cosmosAccessException = new CosmosPreconditionFailedException(message, cosmosException); |
87 | | - break; |
88 | | - case Constants.CosmosExceptionStatusCodes.REQUEST_ENTITY_TOO_LARGE: |
89 | | - cosmosAccessException = new CosmosRequestEntityTooLargeException(message, cosmosException); |
90 | | - break; |
91 | | - case Constants.CosmosExceptionStatusCodes.TOO_MANY_REQUESTS: |
92 | | - cosmosAccessException = new CosmosRequestRateTooLargeException(message, cosmosException); |
93 | | - break; |
94 | | - case Constants.CosmosExceptionStatusCodes.RETRY_WITH: |
95 | | - cosmosAccessException = new CosmosRetryWithException(message, cosmosException); |
96 | | - break; |
97 | | - case Constants.CosmosExceptionStatusCodes.SERVICE_UNAVAILABLE: |
98 | | - cosmosAccessException = new CosmosServiceUnavailableException(message, cosmosException); |
99 | | - break; |
100 | | - case Constants.CosmosExceptionStatusCodes.UNAUTHORIZED: |
101 | | - cosmosAccessException = new CosmosUnauthorizedException(message, cosmosException); |
102 | | - break; |
103 | | - default: |
104 | | - cosmosAccessException = new CosmosAccessException(message, cosmosException); |
105 | | - break; |
| 46 | + int statusCode = cosmosException.getStatusCode(); |
| 47 | + int subStatusCode = ((CosmosException) unwrappedThrowable).getSubStatusCode(); |
| 48 | + if (statusCode == HttpStatus.BAD_REQUEST.value()) { |
| 49 | + cosmosAccessException = new CosmosBadRequestException(message, cosmosException); |
| 50 | + } else if (statusCode == HttpStatus.CONFLICT.value()) { |
| 51 | + cosmosAccessException = new CosmosConflictException(message, cosmosException); |
| 52 | + } else if (statusCode == HttpStatus.FORBIDDEN.value()) { |
| 53 | + cosmosAccessException = new CosmosForbiddenException(message, cosmosException); |
| 54 | + } else if (statusCode == HttpStatus.GONE.value()) { |
| 55 | + if (subStatusCode == HttpConstants.CosmosExceptionSubStatusCodes.NAME_CACHE_IS_STALE) { |
| 56 | + cosmosAccessException = new CosmosInvalidPartitionException(message, cosmosException); |
| 57 | + } else if (subStatusCode == HttpConstants.CosmosExceptionSubStatusCodes.COMPLETING_PARTITION_MIGRATION) { |
| 58 | + cosmosAccessException = new CosmosPartitionIsMigratingException(message, cosmosException); |
| 59 | + } else if (subStatusCode == HttpConstants.CosmosExceptionSubStatusCodes.PARTITION_KEY_RANGE_GONE) { |
| 60 | + cosmosAccessException = new CosmosPartitionKeyRangeGoneException(message, cosmosException); |
| 61 | + } else if (subStatusCode == HttpConstants.CosmosExceptionSubStatusCodes.COMPLETING_SPLIT_OR_MERGE) { |
| 62 | + cosmosAccessException = new CosmosPartitionKeyRangeIsSplittingException(message, cosmosException); |
| 63 | + } else { |
| 64 | + cosmosAccessException = new CosmosGoneException(message, cosmosException); |
| 65 | + } |
| 66 | + } else if (statusCode == HttpStatus.INTERNAL_SERVER_ERROR.value()) { |
| 67 | + cosmosAccessException = new CosmosInternalServerErrorException(message, cosmosException); |
| 68 | + } else if (statusCode == HttpStatus.METHOD_NOT_ALLOWED.value()) { |
| 69 | + cosmosAccessException = new CosmosMethodNotAllowedException(message, cosmosException); |
| 70 | + } else if (statusCode == HttpStatus.NOT_FOUND.value()) { |
| 71 | + cosmosAccessException = new CosmosNotFoundException(message, cosmosException); |
| 72 | + } else if (statusCode == HttpStatus.REQUEST_TIMEOUT.value()) { |
| 73 | + if (subStatusCode == HttpConstants.CosmosExceptionSubStatusCodes.CLIENT_OPERATION_TIMEOUT) { |
| 74 | + cosmosAccessException = new CosmosOperationCancelledException(message, cosmosException); |
| 75 | + } else { |
| 76 | + cosmosAccessException = new CosmosRequestTimeoutException(message, cosmosException); |
| 77 | + } |
| 78 | + } else if (statusCode == HttpStatus.PRECONDITION_FAILED.value()) { |
| 79 | + cosmosAccessException = new CosmosPreconditionFailedException(message, cosmosException); |
| 80 | + } else if (statusCode == HttpStatus.PAYLOAD_TOO_LARGE.value()) { |
| 81 | + cosmosAccessException = new CosmosRequestEntityTooLargeException(message, cosmosException); |
| 82 | + } else if (statusCode == HttpStatus.TOO_MANY_REQUESTS.value()) { |
| 83 | + cosmosAccessException = new CosmosRequestRateTooLargeException(message, cosmosException); |
| 84 | + } else if (statusCode == HttpConstants.CosmosExceptionStatusCodes.RETRY_WITH) { |
| 85 | + cosmosAccessException = new CosmosRetryWithException(message, cosmosException); |
| 86 | + } else if (statusCode == HttpStatus.SERVICE_UNAVAILABLE.value()) { |
| 87 | + cosmosAccessException = new CosmosServiceUnavailableException(message, cosmosException); |
| 88 | + } else if (statusCode == HttpStatus.UNAUTHORIZED.value()) { |
| 89 | + cosmosAccessException = new CosmosUnauthorizedException(message, cosmosException); |
| 90 | + } else { |
| 91 | + cosmosAccessException = new CosmosAccessException(message, cosmosException); |
106 | 92 | } |
107 | 93 | } else { |
108 | 94 | cosmosAccessException = new CosmosAccessException(message, unwrappedThrowable); |
|
0 commit comments