feat: extend cluster/nodes to async client

Author: bevzzz

src/main/java/io/weaviate/client6/v1/api/WeaviateClientAsync.java

@@ -6,6 +6,7 @@
 
 import io.weaviate.client6.v1.api.alias.WeaviateAliasClientAsync;
 import io.weaviate.client6.v1.api.backup.WeaviateBackupClientAsync;
+import io.weaviate.client6.v1.api.cluster.WeaviateClusterClientAsync;
 import io.weaviate.client6.v1.api.collections.WeaviateCollectionsClient;
 import io.weaviate.client6.v1.api.collections.WeaviateCollectionsClientAsync;
 import io.weaviate.client6.v1.api.rbac.groups.WeaviateGroupsClientAsync;
@@ -52,6 +53,12 @@ public class WeaviateClientAsync implements AutoCloseable {
    */
   public final WeaviateUsersClientAsync users;
 
+  /**
+   * Client for {@code /nodes} and {@code /replication} endpoints
+   * for managing replication and sharding.
+   */
+  public final WeaviateClusterClientAsync cluster;
+
   /**
    * This constructor is blocking if {@link Authentication} configured,
    * as the client will need to do the initial token exchange.
@@ -110,6 +117,7 @@ public WeaviateClientAsync(Config config) {
     this.roles = new WeaviateRolesClientAsync(restTransport);
     this.groups = new WeaviateGroupsClientAsync(restTransport);
     this.users = new WeaviateUsersClientAsync(restTransport);
+    this.cluster = new WeaviateClusterClientAsync(restTransport);
     this.collections = new WeaviateCollectionsClientAsync(restTransport, grpcTransport);
   }
 

src/main/java/io/weaviate/client6/v1/api/cluster/WeaviateClusterClientAsync.java

@@ -0,0 +1,59 @@
+package io.weaviate.client6.v1.api.cluster;
+
+import java.io.IOException;
+import java.util.List;
+import java.util.Optional;
+import java.util.concurrent.CompletableFuture;
+import java.util.function.Function;
+
+import io.weaviate.client6.v1.internal.ObjectBuilder;
+import io.weaviate.client6.v1.internal.rest.RestTransport;
+
+public class WeaviateClusterClientAsync {
+  private final RestTransport restTransport;
+
+  public WeaviateClusterClientAsync(RestTransport restTransport) {
+    this.restTransport = restTransport;
+  }
+
+  /**
+   * Query sharding state of a collection.
+   *
+   * @param collection Collection name.
+   */
+  public CompletableFuture<Optional<ShardingState>> shardingState(String collection) throws IOException {
+    return this.restTransport.performRequestAsync(ListShardsRequest.of(collection), ListShardsRequest._ENDPOINT);
+  }
+
+  /**
+   * Query sharding state of a collection.
+   *
+   * @param collection Collection name.
+   * @param fn         Lambda expression for optional parameters.
+   */
+  public CompletableFuture<Optional<ShardingState>> shardingState(String collection,
+      Function<ListShardsRequest.Builder, ObjectBuilder<ListShardsRequest>> fn)
+      throws IOException {
+    return this.restTransport.performRequestAsync(ListShardsRequest.of(collection, fn), ListShardsRequest._ENDPOINT);
+  }
+
+  /**
+   * Get the status of all nodes in the cluster.
+   *
+   * @param fn Lambda expression for optional parameters.
+   */
+  public CompletableFuture<List<Node>> listNodes()
+      throws IOException {
+    return this.restTransport.performRequestAsync(ListNodesRequest.of(), ListNodesRequest._ENDPOINT);
+  }
+
+  /**
+   * Get the status of all nodes in the cluster.
+   *
+   * @param fn Lambda expression for optional parameters.
+   */
+  public CompletableFuture<List<Node>> listNodes(Function<ListNodesRequest.Builder, ObjectBuilder<ListNodesRequest>> fn)
+      throws IOException {
+    return this.restTransport.performRequestAsync(ListNodesRequest.of(fn), ListNodesRequest._ENDPOINT);
+  }
+}