Additional APIs implemented

nb-adapters/adapter-qdrant/pom.xml

@@ -35,16 +35,16 @@
         <dependency>
             <groupId>io.grpc</groupId>
             <artifactId>grpc-protobuf</artifactId>
-            <version>1.63.0</version>
+            <!-- <version>1.63.0</version> -->
             <!-- If Trying to match https://github.com/qdrant/java-client/blob/v1.9.1/build.gradle#L80 -->
-            <!-- <version>1.59.0</version> -->
+            <version>1.59.0</version>
         </dependency>
         <dependency>
             <groupId>com.google.protobuf</groupId>
             <artifactId>protobuf-java-util</artifactId>
-            <version>3.25.3</version>
+            <!-- <version>3.25.3</version> -->
             <!-- If Trying to match https://github.com/qdrant/java-client/blob/v1.9.1/build.gradle#L81 -->
-            <!-- <version>3.24.0</version> -->
+            <version>3.24.0</version>
         </dependency>
         <dependency>
             <groupId>com.google.guava</groupId>

nb-adapters/adapter-qdrant/src/main/java/io/nosqlbench/adapter/qdrant/QdrantOpMapper.java

@@ -63,6 +63,12 @@ public OpDispenser<? extends QdrantBaseOp<?>> apply(ParsedOp op) {
             case search_points -> new QdrantSearchPointsOpDispenser(adapter, op, typeAndTarget.targetFunction);
             case upsert_points -> new QdrantUpsertPointsOpDispenser(adapter, op, typeAndTarget.targetFunction);
             case count_points -> new QdrantCountPointsOpDispenser(adapter, op, typeAndTarget.targetFunction);
+            case list_collections -> new QdrantListCollectionsOpDispenser(adapter, op, typeAndTarget.targetFunction);
+            case collection_info -> new QdrantCollectionInfoOpDispenser(adapter, op, typeAndTarget.targetFunction);
+            case collection_exists -> new QdrantCollectionExistsOpDispenser(adapter, op, typeAndTarget.targetFunction);
+            case list_collection_aliases ->
+                new QdrantListCollectionAliasesOpDispenser(adapter, op, typeAndTarget.targetFunction);
+            case list_snapshots -> new QdrantListSnapshotsOpDispenser(adapter, op, typeAndTarget.targetFunction);
 //            default -> throw new RuntimeException("Unrecognized op type '" + typeAndTarget.enumId.name() + "' while " +
 //                "mapping parsed op " + op);
         };

nb-adapters/adapter-qdrant/src/main/java/io/nosqlbench/adapter/qdrant/opdispensers/QdrantCollectionExistsOpDispenser.java

@@ -0,0 +1,51 @@
+/*
+ * Copyright (c) 2020-2024 nosqlbench
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package io.nosqlbench.adapter.qdrant.opdispensers;
+
+import io.nosqlbench.adapter.qdrant.QdrantDriverAdapter;
+import io.nosqlbench.adapter.qdrant.ops.QdrantBaseOp;
+import io.nosqlbench.adapter.qdrant.ops.QdrantCollectionExistsOp;
+import io.nosqlbench.adapters.api.templating.ParsedOp;
+import io.qdrant.client.QdrantClient;
+import io.qdrant.client.grpc.Collections.CollectionExistsRequest;
+
+import java.util.function.LongFunction;
+
+public class QdrantCollectionExistsOpDispenser extends QdrantBaseOpDispenser<CollectionExistsRequest> {
+    public QdrantCollectionExistsOpDispenser(QdrantDriverAdapter adapter, ParsedOp op, LongFunction<String> targetFunction) {
+        super(adapter, op, targetFunction);
+    }
+
+    @Override
+    public LongFunction<CollectionExistsRequest> getParamFunc(
+        LongFunction<QdrantClient> clientF, ParsedOp op, LongFunction<String> targetF) {
+        LongFunction<CollectionExistsRequest.Builder> ebF =
+            l -> CollectionExistsRequest.newBuilder().setCollectionName(targetF.apply(l));
+
+        final LongFunction<CollectionExistsRequest.Builder> lastF = ebF;
+        return l -> lastF.apply(l).build();
+    }
+
+    @Override
+    public LongFunction<QdrantBaseOp<CollectionExistsRequest>> createOpFunc(
+        LongFunction<CollectionExistsRequest> paramF,
+        LongFunction<QdrantClient> clientF,
+        ParsedOp op,
+        LongFunction<String> targetF) {
+        return l -> new QdrantCollectionExistsOp(clientF.apply(l), paramF.apply(l));
+    }
+}

nb-adapters/adapter-qdrant/src/main/java/io/nosqlbench/adapter/qdrant/opdispensers/QdrantCollectionInfoOpDispenser.java

@@ -0,0 +1,46 @@
+/*
+ * Copyright (c) 2020-2024 nosqlbench
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package io.nosqlbench.adapter.qdrant.opdispensers;
+
+import io.nosqlbench.adapter.qdrant.QdrantDriverAdapter;
+import io.nosqlbench.adapter.qdrant.ops.QdrantBaseOp;
+import io.nosqlbench.adapter.qdrant.ops.QdrantCollectionInfoOp;
+import io.nosqlbench.adapters.api.templating.ParsedOp;
+import io.qdrant.client.QdrantClient;
+
+import java.util.function.LongFunction;
+
+public class QdrantCollectionInfoOpDispenser extends QdrantBaseOpDispenser<String> {
+    public QdrantCollectionInfoOpDispenser(QdrantDriverAdapter adapter, ParsedOp op, LongFunction<String> targetFunction) {
+        super(adapter, op, targetFunction);
+    }
+
+    @Override
+    public LongFunction<String> getParamFunc(
+        LongFunction<QdrantClient> clientF, ParsedOp op, LongFunction<String> targetF) {
+        return l -> targetF.apply(l);
+    }
+
+    @Override
+    public LongFunction<QdrantBaseOp<String>> createOpFunc(
+        LongFunction<String> paramF,
+        LongFunction<QdrantClient> clientF,
+        ParsedOp op,
+        LongFunction<String> targetF) {
+        return l -> new QdrantCollectionInfoOp(clientF.apply(l), paramF.apply(l));
+    }
+}

nb-adapters/adapter-qdrant/src/main/java/io/nosqlbench/adapter/qdrant/opdispensers/QdrantCountPointsOpDispenser.java

@@ -37,9 +37,13 @@ public LongFunction<CountPoints> getParamFunc(
         LongFunction<CountPoints.Builder> ebF =
             l -> CountPoints.newBuilder().setCollectionName(targetF.apply(l));
 
+        ebF = op.enhanceFuncOptionally(ebF, "exact", Boolean.class, CountPoints.Builder::setExact);
+
         LongFunction<Filter.Builder> filterBuilder = getFilterFromOp(op);
-        final LongFunction<CountPoints.Builder> filterF = ebF;
-        ebF = l -> filterF.apply(l).setFilter(filterBuilder.apply(l));
+        if (filterBuilder != null) {
+            final LongFunction<CountPoints.Builder> filterF = ebF;
+            ebF = l -> filterF.apply(l).setFilter(filterBuilder.apply(l));
+        }
 
         final LongFunction<CountPoints.Builder> lastF = ebF;
         return l -> lastF.apply(l).build();

nb-adapters/adapter-qdrant/src/main/java/io/nosqlbench/adapter/qdrant/opdispensers/QdrantListCollectionAliasesOpDispenser.java

@@ -0,0 +1,50 @@
+/*
+ * Copyright (c) 2020-2024 nosqlbench
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package io.nosqlbench.adapter.qdrant.opdispensers;
+
+import io.nosqlbench.adapter.qdrant.QdrantDriverAdapter;
+import io.nosqlbench.adapter.qdrant.ops.QdrantBaseOp;
+import io.nosqlbench.adapter.qdrant.ops.QdrantListCollectionAliasesOp;
+import io.nosqlbench.adapters.api.templating.ParsedOp;
+import io.qdrant.client.QdrantClient;
+import io.qdrant.client.grpc.Collections.ListCollectionAliasesRequest;
+
+import java.util.function.LongFunction;
+
+public class QdrantListCollectionAliasesOpDispenser extends QdrantBaseOpDispenser<ListCollectionAliasesRequest> {
+    public QdrantListCollectionAliasesOpDispenser(QdrantDriverAdapter adapter, ParsedOp op,
+                                                  LongFunction<String> targetFunction) {
+        super(adapter, op, targetFunction);
+    }
+
+    @Override
+    public LongFunction<ListCollectionAliasesRequest> getParamFunc(LongFunction<QdrantClient> clientF, ParsedOp op,
+                                                                   LongFunction<String> targetF) {
+        LongFunction<ListCollectionAliasesRequest.Builder> ebF =
+            l -> ListCollectionAliasesRequest.newBuilder().setCollectionName(targetF.apply(l));
+
+        final LongFunction<ListCollectionAliasesRequest.Builder> lastF = ebF;
+        return l -> lastF.apply(l).build();
+    }
+
+    @Override
+    public LongFunction<QdrantBaseOp<ListCollectionAliasesRequest>> createOpFunc(
+        LongFunction<ListCollectionAliasesRequest> paramF,
+        LongFunction<QdrantClient> clientF, ParsedOp op, LongFunction<String> targetF) {
+        return l -> new QdrantListCollectionAliasesOp(clientF.apply(l), paramF.apply(l));
+    }
+}

nb-adapters/adapter-qdrant/src/main/java/io/nosqlbench/adapter/qdrant/opdispensers/QdrantListCollectionsOpDispenser.java

@@ -0,0 +1,45 @@
+/*
+ * Copyright (c) 2020-2024 nosqlbench
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package io.nosqlbench.adapter.qdrant.opdispensers;
+
+import io.nosqlbench.adapter.qdrant.QdrantDriverAdapter;
+import io.nosqlbench.adapter.qdrant.ops.QdrantBaseOp;
+import io.nosqlbench.adapter.qdrant.ops.QdrantListCollectionsOp;
+import io.nosqlbench.adapters.api.templating.ParsedOp;
+import io.qdrant.client.QdrantClient;
+
+import java.util.function.LongFunction;
+
+public class QdrantListCollectionsOpDispenser extends QdrantBaseOpDispenser<Object> {
+    public QdrantListCollectionsOpDispenser(QdrantDriverAdapter adapter,
+                                            ParsedOp op, LongFunction<String> targetFunction) {
+        super(adapter, op, targetFunction);
+    }
+
+    @Override
+    public LongFunction<Object> getParamFunc(LongFunction<QdrantClient> clientF, ParsedOp op, LongFunction<String> targetF) {
+        return l -> "";
+    }
+
+    @Override
+    public LongFunction<QdrantBaseOp<Object>> createOpFunc(
+        LongFunction<Object> paramF,
+        LongFunction<QdrantClient> clientF,
+        ParsedOp op, LongFunction<String> targetF) {
+            return l -> new QdrantListCollectionsOp(clientF.apply(l), null);
+    }
+}

nb-adapters/adapter-qdrant/src/main/java/io/nosqlbench/adapter/qdrant/opdispensers/QdrantListSnapshotsOpDispenser.java

@@ -0,0 +1,49 @@
+/*
+ * Copyright (c) 2020-2024 nosqlbench
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package io.nosqlbench.adapter.qdrant.opdispensers;
+
+import io.nosqlbench.adapter.qdrant.QdrantDriverAdapter;
+import io.nosqlbench.adapter.qdrant.ops.QdrantBaseOp;
+import io.nosqlbench.adapter.qdrant.ops.QdrantListSnapshotsOp;
+import io.nosqlbench.adapters.api.templating.ParsedOp;
+import io.qdrant.client.QdrantClient;
+import io.qdrant.client.grpc.SnapshotsService.ListSnapshotsRequest;
+
+import java.util.function.LongFunction;
+
+public class QdrantListSnapshotsOpDispenser extends QdrantBaseOpDispenser<ListSnapshotsRequest> {
+    public QdrantListSnapshotsOpDispenser(QdrantDriverAdapter adapter, ParsedOp op, LongFunction<String> targetFunction) {
+        super(adapter, op, targetFunction);
+    }
+
+    @Override
+    public LongFunction<ListSnapshotsRequest> getParamFunc(LongFunction<QdrantClient> clientF,
+                                                           ParsedOp op, LongFunction<String> targetF) {
+        LongFunction<ListSnapshotsRequest.Builder> ebF =
+            l -> ListSnapshotsRequest.newBuilder().setCollectionName(targetF.apply(l));
+
+        final LongFunction<ListSnapshotsRequest.Builder> lastF = ebF;
+        return l -> lastF.apply(l).build();
+    }
+
+    @Override
+    public LongFunction<QdrantBaseOp<ListSnapshotsRequest>> createOpFunc(LongFunction<ListSnapshotsRequest> paramF,
+                                                                         LongFunction<QdrantClient> clientF,
+                                                                         ParsedOp op, LongFunction<String> targetF) {
+        return l -> new QdrantListSnapshotsOp(clientF.apply(l), paramF.apply(l));
+    }
+}

nb-adapters/adapter-qdrant/src/main/java/io/nosqlbench/adapter/qdrant/opdispensers/QdrantSearchPointsOpDispenser.java

@@ -114,8 +114,10 @@ public LongFunction<SearchPoints> getParamFunc(
         }
 
         LongFunction<Filter.Builder> filterBuilder = getFilterFromOp(op);
-        final LongFunction<SearchPoints.Builder> filterF = ebF;
-        ebF = l -> filterF.apply(l).setFilter(filterBuilder.apply(l));
+        if (filterBuilder != null) {
+            final LongFunction<SearchPoints.Builder> filterF = ebF;
+            ebF = l -> filterF.apply(l).setFilter(filterBuilder.apply(l));
+        }
 
         final LongFunction<SearchPoints.Builder> lastF = ebF;
         return l -> lastF.apply(l).build();

nb-adapters/adapter-qdrant/src/main/java/io/nosqlbench/adapter/qdrant/ops/QdrantCollectionExistsOp.java

@@ -0,0 +1,40 @@
+/*
+ * Copyright (c) 2020-2024 nosqlbench
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package io.nosqlbench.adapter.qdrant.ops;
+
+import io.qdrant.client.QdrantClient;
+import io.qdrant.client.grpc.Collections.CollectionExistsRequest;
+
+import java.time.Duration;
+
+public class QdrantCollectionExistsOp extends QdrantBaseOp<CollectionExistsRequest> {
+    public QdrantCollectionExistsOp(QdrantClient client, CollectionExistsRequest request) {
+        super(client, request);
+    }
+
+    @Override
+    public Object applyOp(long value) {
+        Boolean response;
+        try {
+            response = client.collectionExistsAsync(request.getCollectionName(), Duration.ofSeconds(600)).get();
+            logger.info("Collection {} exists: {}", request.getCollectionName(), response);
+        } catch (Exception e) {
+            throw new RuntimeException(e);
+        }
+        return response;
+    }
+}

nb-adapters/adapter-qdrant/src/main/java/io/nosqlbench/adapter/qdrant/ops/QdrantCollectionInfoOp.java

@@ -0,0 +1,40 @@
+/*
+ * Copyright (c) 2020-2024 nosqlbench
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package io.nosqlbench.adapter.qdrant.ops;
+
+import io.qdrant.client.QdrantClient;
+import io.qdrant.client.grpc.Collections.CollectionInfo;
+
+import java.time.Duration;
+
+public class QdrantCollectionInfoOp extends QdrantBaseOp<String> {
+    public QdrantCollectionInfoOp(QdrantClient client, String request) {
+        super(client, request);
+    }
+
+    @Override
+    public Object applyOp(long value) {
+        CollectionInfo response;
+        try {
+            response = client.getCollectionInfoAsync(request, Duration.ofSeconds(600)).get();
+            logger.debug("Collection info: {}", response.toString());
+        } catch (Exception e) {
+            throw new RuntimeException(e);
+        }
+        return response;
+    }
+}