fix error code (#2120)

Signed-off-by: Jing Zhang <[email protected]>
opensearch-project · Mar 12, 2024 · a8f983b · a8f983b
1 parent 0b07a99
commit a8f983b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/plugin/src/main/java/org/opensearch/ml/rest/RestMLExecuteAction.java b/plugin/src/main/java/org/opensearch/ml/rest/RestMLExecuteAction.java
@@ -32,6 +32,7 @@
 import org.opensearch.ml.repackage.com.google.common.annotations.VisibleForTesting;
 import org.opensearch.ml.repackage.com.google.common.collect.ImmutableList;
 import org.opensearch.ml.settings.MLFeatureEnabledSetting;
+import org.opensearch.ml.utils.error.ErrorMessage;
 import org.opensearch.ml.utils.error.ErrorMessageFactory;
 import org.opensearch.rest.BaseRestHandler;
 import org.opensearch.rest.BytesRestResponse;
@@ -130,7 +131,8 @@ private void sendResponse(RestChannel channel, MLExecuteTaskResponse response) t
     }
 
     private void reportError(final RestChannel channel, final Exception e, final RestStatus status) {
-        channel.sendResponse(new BytesRestResponse(status, ErrorMessageFactory.createErrorMessage(e, status.getStatus()).toString()));
+        ErrorMessage errorMessage = ErrorMessageFactory.createErrorMessage(e, status.getStatus());
+        channel.sendResponse(new BytesRestResponse(RestStatus.fromCode(errorMessage.getStatus()), errorMessage.toString()));
     }
 
     private boolean isClientError(Exception e) {

diff --git a/plugin/src/main/java/org/opensearch/ml/utils/error/ErrorMessage.java b/plugin/src/main/java/org/opensearch/ml/utils/error/ErrorMessage.java
@@ -20,6 +20,7 @@ public class ErrorMessage {
 
     protected Throwable exception;
 
+    @Getter
     private final int status;
 
     @Getter
-Original file line number
+Diff line change
@@ Expand Up / @@ -20,6 +20,7 @@ public class ErrorMessage { @@
         protected Throwable exception;
+        @Getter
         private final int status;
         @Getter
@@ Expand Down @@