Add admin method returning the underlying SDKClient

Signed-off-by: Daniel Widdis <[email protected]>
opensearch-project · Jan 28, 2023 · 7546ee6 · 7546ee6
1 parent 01b02e5
commit 7546ee6
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/src/main/java/org/opensearch/sdk/SDKClient.java b/src/main/java/org/opensearch/sdk/SDKClient.java
@@ -207,6 +207,13 @@ public SDKRestClient(RestHighLevelClient restHighLevelClient) {
             this.restHighLevelClient = restHighLevelClient;
         }
 
+        /**
+         * The admin client that can be used to perform administrative operations.
+         */
+        public SDKRestClient admin() {
+            return this;
+        }
+
         /**
          * A client allowing to perform actions/operations against the cluster.
          */

diff --git a/src/test/java/org/opensearch/sdk/TestSDKClient.java b/src/test/java/org/opensearch/sdk/TestSDKClient.java
@@ -61,6 +61,7 @@ public void testCreateRestClient() throws Exception {
         SDKRestClient restClient = sdkClient.initializeRestClient("localhost", 9200);
         assertInstanceOf(SDKIndicesClient.class, restClient.indices());
         assertInstanceOf(SDKClusterAdminClient.class, restClient.cluster());
+        assertEquals(restClient, restClient.admin());
 
         sdkClient.doCloseHighLevelClient();
     }