Add Samples

Author: clun

astra-db-client/src/test/java/com/dtsx/astra/sdk/documentation/FindById.java

@@ -0,0 +1,4 @@
+package com.dtsx.astra.sdk.documentation;
+
+public class FindById {
+}

astra-db-client/src/test/java/com/dtsx/astra/sdk/documentation/FindByVector.java

@@ -0,0 +1,4 @@
+package com.dtsx.astra.sdk.documentation;
+
+public class FindByVector {
+}

astra-db-client/src/test/java/com/dtsx/astra/sdk/documentation/FindOne.java

@@ -0,0 +1,4 @@
+package com.dtsx.astra.sdk.documentation;
+
+public class FindOne {
+}

astra-db-client/src/test/java/com/dtsx/astra/sdk/documentation/InsertMany.java

@@ -0,0 +1,32 @@
+package com.dtsx.astra.sdk.documentation;
+
+import com.dtsx.astra.sdk.AstraDB;
+import com.dtsx.astra.sdk.AstraDBCollection;
+import io.stargate.sdk.json.domain.JsonDocument;
+import io.stargate.sdk.json.domain.SimilarityMetric;
+
+import java.util.List;
+
+public class InsertMany {
+    public static void main(String[] args) {
+
+// Given an active db and a collection with a vector field (see CreateCollection.java)
+AstraDB db = new AstraDB("<token>", "<api_endpoint>");
+
+// Create collection if not exists
+AstraDBCollection collection = db
+        .createCollection("collection_vector1",14);
+
+// Insert documents, ids are generated here
+List<String> identifiers = collection.insertMany(List.of(
+  new JsonDocument()
+    .vector(new float[]{1f, 0f, 1f, 1f, .5f, 1f, 0f, 0.3f, 0f, 0f, 0f, 0f, 0f, 0f})
+    .put("product_name", "Yet another product")
+    .put("product_price", 99.99),
+  new JsonDocument()
+    .vector(new float[]{1f, 0f, 1f, 1f, .5f, 1f, 0f, 0.3f, 0f, 0f, 0f, 0f, 0f, 0f})
+    .put("product_name", "product3")
+    .put("product_price", 99.99)));
+
+}
+}

astra-db-client/src/test/java/com/dtsx/astra/sdk/documentation/InsertOne.java

@@ -16,11 +16,13 @@ public static void main(String[] args) {
 /*
  * Given a collection with vector (dimension 14)
  * Can be created with:
- * AstraDBCollection collection = db.createCollection("collection_vector1", 14);
+ * AstraDBCollection collection = db
+ *   .createCollection("collection_vector1", 14);
  */
 AstraDBCollection collection = db.collection("collection_vector1");
 // Adding this list to allow the test to be re-runnable
 collection.deleteAll();
+
 //  (1) You can insert records with key/value.
 collection.insertOne(new JsonDocument()
  .id("doc1") // uuid is generated if not explicitely set
@@ -45,7 +47,7 @@ public static void main(String[] args) {
  .data(Map.of("product_name", "HealthyFresh - Chicken raw dog food"))
 );
 
-// (4) All hybrid combination are possible (key/value, json, map)
+// (4) Any combination is possible (key/value, json, map)
 collection.insertOne(new JsonDocument()
  .id("doc4")
  .vector(new float[]{1f, 1f, 1f, 1f, 1f, 0f, 0f, 0f, 0f, 0f, 0f, 0f, 0f, 0f})
@@ -63,12 +65,7 @@ public static void main(String[] args) {
 }
 
 /*
-* (5) Insertion rules
-*
-* - all attributes are nullable
-* - id is generated if needed, the id is returned by insertOne()
-* - Add any property you want, but needs to use [A-Za-z_-.] and not starting by '$'
-* - It is schema less, values can be any simple type
+* (6) If not provided id is generated and returned
 */
 String generatedId = collection.insertOne(
         new JsonDocument().put("demo", 1));

astra-db-client/src/test/java/com/dtsx/astra/sdk/documentation/QuickStart.java

@@ -27,10 +27,8 @@ public static void main(String[] args) {
 demoCollection.insertOne(new JsonDocument()
  .id("doc2")
  .vector(new float[]{1f, 1f, 1f, 1f, 1f, 0f, 0f, 0f, 0f, 0f, 0f, 0f, 0f, 0f})
- .data("{"
-  + "   \"product_name\": \"HealthyFresh - Chicken raw dog food\", "
-  + "  \"product_price\": 9.99"
-  + "}"));
+ .put("product_name", "HealthyFresh - Chicken raw dog food")
+ .put("product_price", 9.99));
 demoCollection.insertOne(new JsonDocument()
  .id("doc3")
  .vector(new float[]{1f, 1f, 1f, 1f, 1f, 0f, 0f, 0f, 0f, 0f, 0f, 0f, 0f, 0f})