feat: add data.exists method

Author: bevzzz

src/it/java/io/weaviate/integration/DataITest.java

@@ -45,8 +45,10 @@ public void testCreateGetDelete() throws IOException {
         .returnProperties("name")
         .returnMetadata(Metadata.ID, Metadata.VECTOR));
 
+    Assertions.assertThat(artists.data.exists(id))
+        .as("object exists after insert").isTrue();
     Assertions.assertThat(object)
-        .as("object exists after insert").get()
+        .as("object has correct properties").get()
         .satisfies(obj -> {
           Assertions.assertThat(obj.metadata().uuid())
               .as("object id").isEqualTo(id);
@@ -60,8 +62,8 @@ public void testCreateGetDelete() throws IOException {
         });
 
     artists.data.delete(id);
-    object = artists.query.byId(id);
-    Assertions.assertThat(object).isEmpty().as("object not exists after deletion");
+    Assertions.assertThat(artists.data.exists(id))
+        .as("object not exists after deletion").isFalse();
   }
 
   @Test
@@ -175,4 +177,12 @@ public void testReferences_AddReplaceDelete() throws IOException {
         .asInstanceOf(InstanceOfAssertFactories.list(WeaviateObject.class))
         .isEmpty();
   }
+
+  @Test
+  public void testReplace() {
+  }
+
+  @Test
+  public void testUpdate() {
+  }
 }

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

@@ -20,8 +20,8 @@ public CollectionHandle(
       CollectionDescriptor<T> collectionDescriptor) {
 
     this.config = new WeaviateConfigClient(collectionDescriptor, restTransport, grpcTransport);
-    this.data = new WeaviateDataClient<>(collectionDescriptor, restTransport);
     this.query = new WeaviateQueryClient<>(collectionDescriptor, grpcTransport);
+    this.data = new WeaviateDataClient<>(collectionDescriptor, restTransport, this.query);
     this.aggregate = new WeaviateAggregateClient(collectionDescriptor, grpcTransport);
   }
 }

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

@@ -20,8 +20,8 @@ public CollectionHandleAsync(
       CollectionDescriptor<T> collectionDescriptor) {
 
     this.config = new WeaviateConfigClientAsync(collectionDescriptor, restTransport, grpcTransport);
-    this.data = new WeaviateDataClientAsync<>(collectionDescriptor, restTransport);
     this.query = new WeaviateQueryClientAsync<>(collectionDescriptor, grpcTransport);
+    this.data = new WeaviateDataClientAsync<>(collectionDescriptor, restTransport, this.query);
     this.aggregate = new WeaviateAggregateClientAsync(collectionDescriptor, grpcTransport);
   }
 }

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

@@ -5,6 +5,7 @@
 
 import io.weaviate.client6.v1.api.collections.ObjectMetadata;
 import io.weaviate.client6.v1.api.collections.WeaviateObject;
+import io.weaviate.client6.v1.api.collections.query.WeaviateQueryClient;
 import io.weaviate.client6.v1.internal.ObjectBuilder;
 import io.weaviate.client6.v1.internal.orm.CollectionDescriptor;
 import io.weaviate.client6.v1.internal.rest.RestTransport;
@@ -13,9 +14,13 @@ public class WeaviateDataClient<T> {
   private final RestTransport restTransport;
   private final CollectionDescriptor<T> collectionDescriptor;
 
-  public WeaviateDataClient(CollectionDescriptor<T> collectionDescriptor, RestTransport restTransport) {
+  private final WeaviateQueryClient<T> query;
+
+  public WeaviateDataClient(CollectionDescriptor<T> collectionDescriptor, RestTransport restTransport,
+      WeaviateQueryClient<T> query) {
     this.restTransport = restTransport;
     this.collectionDescriptor = collectionDescriptor;
+    this.query = query;
   }
 
   public WeaviateObject<T, Object, ObjectMetadata> insert(T properties) throws IOException {
@@ -32,6 +37,10 @@ public WeaviateObject<T, Object, ObjectMetadata> insert(InsertObjectRequest<T> r
     return this.restTransport.performRequest(request, InsertObjectRequest.endpoint(collectionDescriptor));
   }
 
+  public boolean exists(String uuid) throws IOException {
+    return this.query.byId(uuid).isPresent();
+  }
+
   public void delete(String uuid) throws IOException {
     this.restTransport.performRequest(new DeleteObjectRequest(collectionDescriptor.name(), uuid),
         DeleteObjectRequest._ENDPOINT);

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

@@ -2,11 +2,13 @@
 
 import java.io.IOException;
 import java.util.Collection;
+import java.util.Optional;
 import java.util.concurrent.CompletableFuture;
 import java.util.function.Function;
 
 import io.weaviate.client6.v1.api.collections.ObjectMetadata;
 import io.weaviate.client6.v1.api.collections.WeaviateObject;
+import io.weaviate.client6.v1.api.collections.query.WeaviateQueryClientAsync;
 import io.weaviate.client6.v1.internal.ObjectBuilder;
 import io.weaviate.client6.v1.internal.orm.CollectionDescriptor;
 import io.weaviate.client6.v1.internal.rest.RestTransport;
@@ -15,9 +17,13 @@ public class WeaviateDataClientAsync<T> {
   private final RestTransport restTransport;
   private final CollectionDescriptor<T> collectionDescriptor;
 
-  public WeaviateDataClientAsync(CollectionDescriptor<T> collectionDescriptor, RestTransport restTransport) {
+  private final WeaviateQueryClientAsync<T> query;
+
+  public WeaviateDataClientAsync(CollectionDescriptor<T> collectionDescriptor, RestTransport restTransport,
+      WeaviateQueryClientAsync<T> query) {
     this.restTransport = restTransport;
     this.collectionDescriptor = collectionDescriptor;
+    this.query = query;
   }
 
   public CompletableFuture<WeaviateObject<T, Object, ObjectMetadata>> insert(T properties) throws IOException {
@@ -35,6 +41,10 @@ public CompletableFuture<WeaviateObject<T, Object, ObjectMetadata>> insert(Inser
     return this.restTransport.performRequestAsync(request, InsertObjectRequest.endpoint(collectionDescriptor));
   }
 
+  public CompletableFuture<Boolean> exists(String uuid) {
+    return this.query.byId(uuid).thenApply(Optional::isPresent);
+  }
+
   public CompletableFuture<Void> delete(String uuid) {
     return this.restTransport.performRequestAsync(new DeleteObjectRequest(collectionDescriptor.name(), uuid),
         DeleteObjectRequest._ENDPOINT);