Changed XContentType from MediaType (#912)

opensearch-project · Jul 26, 2023 · e40420b · e40420b
1 parent ec7c4a6
commit e40420b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/src/main/java/org/opensearch/sdk/sample/helloworld/rest/RestHelloAction.java b/src/main/java/org/opensearch/sdk/sample/helloworld/rest/RestHelloAction.java
@@ -11,6 +11,7 @@
 
 import org.opensearch.OpenSearchParseException;
 import org.opensearch.core.common.bytes.BytesReference;
+import org.opensearch.core.xcontent.MediaType;
 import org.opensearch.core.xcontent.XContentBuilder;
 import org.opensearch.common.xcontent.XContentType;
 import org.opensearch.common.xcontent.json.JsonXContent;
@@ -100,11 +101,11 @@ public List<NamedRoute> routes() {
     private Function<RestRequest, RestResponse> handlePostRequest = (request) -> {
         if (request.hasContent()) {
             String adjective = "";
-            XContentType contentType = request.getXContentType();
-            if (contentType == null) {
+            MediaType mediaType = request.getMediaType();
+            if (mediaType == null) {
                 // Plain text
                 adjective = request.content().utf8ToString();
-            } else if (contentType.equals(XContentType.JSON)) {
+            } else if (mediaType.equals(XContentType.JSON)) {
                 try {
                     adjective = request.contentParser().mapStrings().get("adjective");
                 } catch (IOException | OpenSearchParseException e) {

diff --git a/src/test/java/org/opensearch/sdk/rest/TestSDKRestRequest.java b/src/test/java/org/opensearch/sdk/rest/TestSDKRestRequest.java
@@ -56,7 +56,7 @@ public void testSDKRestRequestMethods() throws IOException {
         assertEquals(expectedPath, sdkRestRequest.path());
         assertEquals(expectedParams, sdkRestRequest.params());
         assertEquals(expectedHeaders, sdkRestRequest.getHeaders());
-        assertEquals(exptectedXContentType, sdkRestRequest.getXContentType());
+        assertEquals(exptectedXContentType, sdkRestRequest.getMediaType());
         assertEquals(expectedContent, sdkRestRequest.content());
 
         Map<String, String> source = sdkRestRequest.contentParser().mapStrings();