IntenseFinance/nosqlbench
3
0
Fork 0
mirror of https://github.com/nosqlbench/nosqlbench.git synced 2024-12-24 08:00:00 -06:00
Code Issues Packages Projects Releases Wiki Activity

add key in pojo for vector result

Browse Source
This commit is contained in:
Jonathan Shook 2024-02-08 17:38:00 -06:00
parent 8f4aa5cec5
commit ac8ca0be37
1 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
adapter-aws-opensearch/src/main/java/io/nosqlbench/adapter/opensearch/dispensers/KnnSearchOpDispenser.java
View File

@ -65,9 +65,12 @@ public class KnnSearchOpDispenser extends BaseOpenSearchOpDispenser {
public static class Doc { public static class Doc {
private float[] values; private float[] values;
private String key;
public Doc() {} public Doc() {}
public Doc(float[] values) { public Doc(float[] values, String key) {
this.values = values; this.values = values;
this.key = key;
} }
public float[] getValues() { public float[] getValues() {
return values; return values;
@ -79,6 +82,14 @@ public class KnnSearchOpDispenser extends BaseOpenSearchOpDispenser {
public String toString() { public String toString() {
return "{" + "values=" + values + "}"; return "{" + "values=" + values + "}";
} }
public String getKey() {
return key;
}
public void setKey(String key) {
this.key = key;
}
} }
} }