IntenseFinance/nosqlbench
3
0
Fork 0
mirror of https://github.com/nosqlbench/nosqlbench.git synced 2025-01-26 15:36:33 -06:00
Code Issues Packages Projects Releases Wiki Activity

implement bulk op

Browse Source
This commit is contained in:
Jonathan Shook 2024-03-21 19:04:39 -05:00
parent 2e8a435f21
commit 27a02dfe28
1 changed files with 16 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
adapter-milvus/src/main/java/io/nosqlbench/adapter/milvus/opdispensers/MilvusBulkInsertOpDispenser.java
View File

@ -23,6 +23,8 @@ import io.nosqlbench.adapter.milvus.ops.MilvusBaseOp;
import io.nosqlbench.adapter.milvus.ops.MilvusBulkInsertOp; import io.nosqlbench.adapter.milvus.ops.MilvusBulkInsertOp;
import io.nosqlbench.adapters.api.templating.ParsedOp; import io.nosqlbench.adapters.api.templating.ParsedOp;
import java.util.List;
import java.util.Map;
import java.util.function.LongFunction; import java.util.function.LongFunction;
public class MilvusBulkInsertOpDispenser extends MilvusBaseOpDispenser<BulkInsertParam> { public class MilvusBulkInsertOpDispenser extends MilvusBaseOpDispenser<BulkInsertParam> {
@ -41,17 +43,20 @@ public class MilvusBulkInsertOpDispenser extends MilvusBaseOpDispenser<BulkInser
) { ) {
LongFunction<BulkInsertParam.Builder> ebF = LongFunction<BulkInsertParam.Builder> ebF =
l -> BulkInsertParam.newBuilder().withCollectionName(targetF.apply(l)); l -> BulkInsertParam.newBuilder().withCollectionName(targetF.apply(l));
// Add enhancement functions here
throw new RuntimeException("implement me");
ebF = op.enhanceFuncOptionally(ebF, "options", Map.class,
(BulkInsertParam.Builder builder, Map map) -> {
// And remove test function map.forEach((k, v) -> {
// BulkInsertParam.Builder test = ebF.apply(0); builder.withOption(k.toString(), v.toString());
});
// final LongFunction<BulkInsertParam.Builder> lastF = ebF; return builder;
// final LongFunction<BulkInsertParam> collectionParamF = l -> lastF.apply(l).build(); }
// return null; );
ebF = op.enhanceFuncOptionally(ebF, "files", List.class, BulkInsertParam.Builder::withFiles);
ebF = op.enhanceFuncOptionally(ebF, List.of("partition_name", "partition"), String.class,
BulkInsertParam.Builder::withPartitionName);
LongFunction<BulkInsertParam.Builder> finalEbF = ebF;
return l -> finalEbF.apply(l).build();
} }
@Override @Override