feat: add sync features to async client

Author: bevzzz

src/main/java/io/weaviate/client6/v1/api/collections/WeaviateCollectionsClientAsync.java

@@ -1,5 +1,7 @@
 package io.weaviate.client6.v1.api.collections;
 
+import java.io.IOException;
+import java.util.List;
 import java.util.Map;
 import java.util.Optional;
 import java.util.concurrent.CompletableFuture;
@@ -42,7 +44,24 @@ public CompletableFuture<Optional<WeaviateCollection>> getConfig(String name) {
     return this.restTransport.performRequestAsync(new GetConfigRequest(name), GetConfigRequest._ENDPOINT);
   }
 
+  public CompletableFuture<List<WeaviateCollection>> list() {
+    return this.restTransport.performRequestAsync(new ListCollectionRequest(), ListCollectionRequest._ENDPOINT);
+  }
+
   public CompletableFuture<Void> delete(String name) {
     return this.restTransport.performRequestAsync(new DeleteCollectionRequest(name), DeleteCollectionRequest._ENDPOINT);
   }
+
+  public CompletableFuture<Void> deleteAll() throws IOException {
+    return list().thenCompose(collections -> {
+      var futures = collections.stream()
+          .map(collection -> delete(collection.name()))
+          .toArray(CompletableFuture[]::new);
+      return CompletableFuture.allOf(futures);
+    });
+  }
+
+  public CompletableFuture<Boolean> exists(String name) {
+    return getConfig(name).thenApply(Optional::isPresent);
+  }
 }

src/main/java/io/weaviate/client6/v1/api/collections/data/WeaviateDataClientAsync.java

@@ -1,6 +1,7 @@
 package io.weaviate.client6.v1.api.collections.data;
 
 import java.io.IOException;
+import java.util.Collection;
 import java.util.concurrent.CompletableFuture;
 import java.util.function.Function;
 
@@ -38,4 +39,50 @@ public CompletableFuture<Void> delete(String uuid) {
     return this.restTransport.performRequestAsync(new DeleteObjectRequest(collectionDescriptor.name(), uuid),
         DeleteObjectRequest._ENDPOINT);
   }
+
+  public CompletableFuture<Void> referenceAdd(String fromUuid, String fromProperty, Reference reference) {
+    return forEachAsync(reference.uuids(), uuid -> {
+      var singleRef = new Reference(reference.collection(), (String) uuid);
+      return this.restTransport.performRequestAsync(new ReferenceAddRequest(fromUuid, fromProperty, singleRef),
+          ReferenceAddRequest.endpoint(collectionDescriptor));
+    });
+  }
+
+  public CompletableFuture<Void> referenceDelete(String fromUuid, String fromProperty, Reference reference) {
+    return forEachAsync(reference.uuids(), uuid -> {
+      var singleRef = new Reference(reference.collection(), (String) uuid);
+      return this.restTransport.performRequestAsync(new ReferenceDeleteRequest(fromUuid, fromProperty, singleRef),
+          ReferenceDeleteRequest.endpoint(collectionDescriptor));
+    });
+  }
+
+  public CompletableFuture<Void> referenceReplace(String fromUuid, String fromProperty, Reference reference) {
+    return forEachAsync(reference.uuids(), uuid -> {
+      var singleRef = new Reference(reference.collection(), (String) uuid);
+      return this.restTransport.performRequestAsync(new ReferenceReplaceRequest(fromUuid, fromProperty, singleRef),
+          ReferenceReplaceRequest.endpoint(collectionDescriptor));
+    });
+  }
+
+  /**
+   * Spawn execution {@code fn} for each of the {@code elements} and return a
+   * flattened {@link CompletableFuture#allOf}.
+   *
+   * <p>
+   * Usage:
+   *
+   * <pre>{@code
+   *  // With elements immediately available
+   *  forEachAsync(myElements, element -> doNetworkIo(element));
+   *
+   *  // Chain to another CompletableFuture
+   *  fetch(request).thenCompose(elements -> forEachAsync(...));
+   * }</pre>
+   */
+  private static <T> CompletableFuture<Void> forEachAsync(Collection<T> elements,
+      Function<T, CompletableFuture<?>> fn) {
+    var futures = elements.stream().map(el -> fn.apply(el))
+        .toArray(CompletableFuture[]::new);
+    return CompletableFuture.allOf(futures);
+  }
 }