Skip to content

Commit 4c20c39

Browse files
committed
more docs
1 parent 7db3d93 commit 4c20c39

File tree

4 files changed

+114
-2
lines changed

4 files changed

+114
-2
lines changed
Lines changed: 50 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,54 @@
11
package com.dtsx.astra.sdk.documentation;
22

3+
import com.dtsx.astra.sdk.AstraDB;
4+
import com.dtsx.astra.sdk.AstraDBCollection;
5+
import com.fasterxml.jackson.annotation.JsonProperty;
6+
import io.stargate.sdk.json.domain.JsonResult;
7+
import io.stargate.sdk.json.domain.odm.Result;
8+
import io.stargate.sdk.json.domain.odm.ResultMapper;
9+
10+
import java.util.Optional;
11+
312
public class FindById {
13+
public static void main(String[] args) {
14+
15+
// Accessing existing DB
16+
AstraDB db = new AstraDB("<token>", "<api_endpoint>");
17+
18+
// Access existing collection
19+
AstraDBCollection collection = db.collection("collection_vector1");
20+
21+
// (1) Find a Json Result from its id
22+
Optional<JsonResult> res = collection.findById("doc1");
23+
res.ifPresent(jsonResult -> System.out.println(jsonResult.getSimilarity()));
24+
25+
// (2) find by id with Result Mapper
26+
Optional<Result<MyBean>> res2 = collection.findById("doc1", new ResultMapper<MyBean>() {
27+
@Override
28+
public Result<MyBean> map(JsonResult record) {
29+
MyBean bean = new MyBean(
30+
(String) record.getData().get("product_name"),
31+
(Double) record.getData().get("product_price"));
32+
return new Result<>(record, bean);
33+
}
34+
});
35+
36+
// (3) find by id with class Mapping
37+
Optional<Result<MyBean>> res3 = collection.findById("doc1", MyBean.class);
38+
39+
// (4) Test document existence
40+
boolean exists = collection.isDocumentExists("doc1");
41+
42+
}
43+
44+
public static class MyBean {
45+
@JsonProperty("product_name") String name;
46+
@JsonProperty("product_price") Double price;
47+
public MyBean(String name, Double price) {
48+
this.name = name;
49+
this.price = price;
50+
}
51+
// getters and setters
52+
}
53+
454
}
Lines changed: 51 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,55 @@
11
package com.dtsx.astra.sdk.documentation;
22

3+
import com.dtsx.astra.sdk.AstraDB;
4+
import com.dtsx.astra.sdk.AstraDBCollection;
5+
import com.fasterxml.jackson.annotation.JsonProperty;
6+
import io.stargate.sdk.json.domain.JsonResult;
7+
import io.stargate.sdk.json.domain.odm.Result;
8+
import io.stargate.sdk.json.domain.odm.ResultMapper;
9+
10+
import java.util.Optional;
11+
312
public class FindByVector {
13+
public static void main(String[] args) {
14+
15+
// Accessing existing DB
16+
AstraDB db = new AstraDB("<token>", "<api_endpoint>");
17+
18+
// Access existing collection
19+
AstraDBCollection collection = db.collection("collection_vector1");
20+
21+
// (1) Find a Json Result from its vector
22+
collection
23+
.findOneByVector(new float[]{1f, 0f, 1f, 1f, 1f, 1f, 0f, 0f, 0f, 0f, 0f, 0f, 0f, 0f})
24+
.ifPresent(jsonResult -> System.out.println(jsonResult.getSimilarity()));
25+
26+
// (2) find by id with Result Mapper
27+
Optional<Result<MyBean>> res2 = collection
28+
.findOneByVector(new float[]{1f, 0f, 1f, 1f, 1f, 1f, 0f, 0f, 0f, 0f, 0f, 0f, 0f, 0f}, new ResultMapper<MyBean>() {
29+
@Override
30+
public Result<MyBean> map(JsonResult record) {
31+
MyBean bean = new MyBean(
32+
(String) record.getData().get("product_name"),
33+
(Double) record.getData().get("product_price"));
34+
return new Result<>(record, bean);
35+
}
36+
});
37+
38+
// (3) find by id with class Mapping
39+
Optional<Result<MyBean>> res3 = collection
40+
.findOneByVector(new float[]{1f, 0f, 1f, 1f, 1f, 1f, 0f, 0f, 0f, 0f, 0f, 0f, 0f, 0f},
41+
MyBean.class);
42+
43+
}
44+
45+
public static class MyBean {
46+
@JsonProperty("product_name") String name;
47+
@JsonProperty("product_price") Double price;
48+
public MyBean(String name, Double price) {
49+
this.name = name;
50+
this.price = price;
51+
}
52+
// getters and setters
53+
}
54+
455
}
Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,15 @@
11
package com.dtsx.astra.sdk.documentation;
22

3+
import com.dtsx.astra.sdk.AstraDB;
4+
import com.dtsx.astra.sdk.AstraDBCollection;
5+
36
public class FindOne {
7+
public static void main(String[] args) {
8+
// Accessing existing DB
9+
AstraDB db = new AstraDB("<token>", "<api_endpoint>");
10+
11+
// Access existing collection
12+
AstraDBCollection collection = db.collection("collection_vector1");
13+
14+
}
415
}

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

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
import com.dtsx.astra.sdk.AstraDB;
44
import com.dtsx.astra.sdk.AstraDBCollection;
55
import io.stargate.sdk.json.domain.JsonDocument;
6-
import io.stargate.sdk.json.exception.ApiException;
6+
import io.stargate.sdk.json.exception.JsonApiException;
77

88
import java.util.Map;
99

@@ -60,7 +60,7 @@ public static void main(String[] args) {
6060
// (5) You cannot insert a document with an existing id
6161
try {
6262
collection.insertOne(new JsonDocument("doc4"));
63-
} catch(ApiException e) {
63+
} catch(JsonApiException e) {
6464
System.out.println("Expected ERROR: " + e.getMessage());
6565
}
6666

0 commit comments

Comments
 (0)